1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
import { Frame, range } from '../src/index';
import { data, header } from './support/people';
import { expect } from 'chai';
const frame = new Frame(data, header);
describe('columns.ts', () => {
describe('Get a single column', () => {
it('should return a column of the frame using column name', () => {
const dataToExpect = [
'Arthur',
'Betty',
'Victor',
'Dodger',
'Rayan',
'Skitley',
'Victoria',
'Tiger',
'Killjoy'
];
const names = frame.column('Name');
expect(names.flatten()).to.deep.equal(dataToExpect);
expect(names.columns).to.deep.equal(['Name']);
});
});
describe('Get multiple columns', () => {
it('should return multiple columns', () => {
const dataToExpect = [
[['Arthur'], ['Male']],
[['Betty'], ['Female']],
[['Victor'], ['Male']],
[['Dodger'], ['Male']],
[['Rayan'], ['Male']],
[['Skitley'], ['Female']],
[['Victoria'], ['Female']],
[['Tiger'], ['Male']],
[['Killjoy'], ['Female']]
];
const headersToExpect = ['Name', 'Gender'];
const namesAndGenders = frame.column([0, 2]);
expect(namesAndGenders.rowdata).to.deep.equal(dataToExpect);
expect(namesAndGenders.columns).to.deep.equal(headersToExpect);
});
});
describe('Get multiple columns over a range', () => {
it('should return multiple columns over a range', () => {
const dataToExpect = [
[['Arthur'], ['Male']],
[['Betty'], ['Female']],
[['Victor'], ['Male']],
[['Dodger'], ['Male']],
[['Rayan'], ['Male']],
[['Skitley'], ['Female']],
[['Victoria'], ['Female']],
[['Tiger'], ['Male']],
[['Killjoy'], ['Female']]
];
const headersToExpect = ['Name', 'Gender'];
const namesAndGenders = frame.column(range(0, 2, 2));
expect(namesAndGenders.rowdata).to.deep.equal(dataToExpect);
expect(namesAndGenders.columns).to.deep.equal(headersToExpect);
});
});
});
|
