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);
    });
  });
});