pandas-js/src/js/__tests__/tseries/tools.js

Pandas for JavaScript

'use strict';

var _immutable = require('immutable');

var _immutable2 = _interopRequireDefault(_immutable);

var _index = require('../../core/index');

var _tools = require('../../tseries/tools');

function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }

describe('tools', function () {
  describe('to_datetime', function () {
    it('converts a Series to a datetime Series', function () {
      var ds = new _index.Series(['March 1 2015', 'May 2 2016']);
      expect(ds.dtype.dtype).toEqual('object');

      var dsTs = (0, _tools.to_datetime)(ds);
      expect(dsTs.dtype.dtype).toEqual('datetime');
      expect(dsTs.iloc(0)).toBeInstanceOf(Date);
      expect(dsTs.iloc(0).getMonth() + 1).toEqual(3);
      expect(dsTs.iloc(0).getDay() + 1).toEqual(1);
      expect(dsTs.iloc(0).getYear() + 1900).toEqual(2015);
      expect(dsTs.iloc(1)).toBeInstanceOf(Date);
      expect(dsTs.iloc(1).getMonth() + 1).toEqual(5);
      expect(dsTs.iloc(1).getDay() + 1).toEqual(2);
      expect(dsTs.iloc(1).getYear() + 1900).toEqual(2016);
    });

    it('converts a DataFrame to a datetime Series', function () {
      var df = new _index.DataFrame([{ x: 'March 1 2015', y: 'May 2 2016' }]);

      var dfTs = (0, _tools.to_datetime)(df);
      expect(dfTs.length).toEqual(1);
      expect(dfTs.get('x').dtype.dtype).toEqual('datetime');
      expect(dfTs.get('x').iloc(0)).toBeInstanceOf(Date);
      expect(dfTs.get('x').iloc(0).getMonth() + 1).toEqual(3);
      expect(dfTs.get('x').iloc(0).getDay() + 1).toEqual(1);
      expect(dfTs.get('x').iloc(0).getYear() + 1900).toEqual(2015);
      expect(dfTs.get('y').dtype.dtype).toEqual('datetime');
      expect(dfTs.get('y').iloc(0)).toBeInstanceOf(Date);
      expect(dfTs.get('y').iloc(0).getMonth() + 1).toEqual(5);
      expect(dfTs.get('y').iloc(0).getDay() + 1).toEqual(2);
      expect(dfTs.get('y').iloc(0).getYear() + 1900).toEqual(2016);
    });

    it('converts a List to a datetime List', function () {
      var testList = _immutable2.default.List.of('March 1 2015', 'May 2 2016');

      var listTs = (0, _tools.to_datetime)(testList);

      expect(listTs.get(0)).toBeInstanceOf(Date);
      expect(listTs.get(0).getMonth() + 1).toEqual(3);
      expect(listTs.get(0).getDay() + 1).toEqual(1);
      expect(listTs.get(0).getYear() + 1900).toEqual(2015);
      expect(listTs.get(1)).toBeInstanceOf(Date);
      expect(listTs.get(1).getMonth() + 1).toEqual(5);
      expect(listTs.get(1).getDay() + 1).toEqual(2);
      expect(listTs.get(1).getYear() + 1900).toEqual(2016);
    });

    it('converts an Array to a datetime List', function () {
      var testArray = ['March 1 2015', 'May 2 2016'];

      var arrayTs = (0, _tools.to_datetime)(testArray);

      expect(arrayTs[0]).toBeInstanceOf(Date);
      expect(arrayTs[0].getMonth() + 1).toEqual(3);
      expect(arrayTs[0].getDay() + 1).toEqual(1);
      expect(arrayTs[0].getYear() + 1900).toEqual(2015);
      expect(arrayTs[1]).toBeInstanceOf(Date);
      expect(arrayTs[1].getMonth() + 1).toEqual(5);
      expect(arrayTs[1].getDay() + 1).toEqual(2);
      expect(arrayTs[1].getYear() + 1900).toEqual(2016);
    });

    it('allows for setting an index as a datetime', function () {
      var ds = new _index.Series([1, 2, 3, 4], { index: ['January 2016', 'February 2016', 'March 2016', 'April 2016'] });
      ds.index = (0, _tools.to_datetime)(ds.index);

      expect(ds.index.get(0).getMonth() + 1).toEqual(1);
      expect(ds.index.get(3).getMonth() + 1).toEqual(4);
    });
  });
}); /**
     * to_datetime
     *
     * Description:
     * Primary author(s):
     * Secondary author(s):
     *
     * Notes:
     *
     * January 11, 2017
     * StratoDem Analytics, LLC
     */

//# sourceMappingURL=tools.js.map