datamodel/src/fields/categorical/index.spec.js

Relational algebra compliant in-memory tabular data store

/* global describe, it, beforeEach */
/* eslint-disable no-unused-expressions */

import { expect } from 'chai';
import PartialField from '../partial-field';
import { DimensionSubtype } from '../../enums';
import CategoricalParser from '../parsers/categorical-parser';
import Categorical from './index';
import DataModel from '../../index';

describe('Categorical', () => {
    const schema = {
        name: 'Origin',
        type: 'dimension',
        subtype: DimensionSubtype.CATEGORICAL
    };
    const data = ['India', 'US', 'Canada', 'India', 'US'];
    let catParser;
    let partField;
    let rowDiffset;
    let catField;

    beforeEach(() => {
        catParser = new CategoricalParser();
        partField = new PartialField(schema.name, data, schema, catParser);
        rowDiffset = '0-4';
        catField = new Categorical(partField, rowDiffset);
    });

    describe('#subtype', () => {
        it('should return the correct subtype', () => {
            expect(catField.subtype()).to.equal(schema.subtype);
        });
    });

    describe('#calculateDataDomain', () => {
        it('should return the field domain', () => {
            const expected = ['India', 'US', 'Canada'];
            expect(catField.calculateDataDomain()).to.eql(expected);
        });

        it('should parse null data value as default invalid type', () => {
            const data1 = ['India', 'US', 'Canada', 'India', 'US', null, 'US', null];
            catParser = new CategoricalParser();
            partField = new PartialField(schema.name, data1, schema, catParser);
            rowDiffset = '0-7';
            catField = new Categorical(partField, rowDiffset);
            const expected = ['India', 'US', 'Canada', DataModel.InvalidAwareTypes.NULL];
            expect(catField.calculateDataDomain()).to.eql(expected);
        });
    });
});