def test_sorted(self): dates = [ts.Date('D', string='2007-01-%02i' % i) for i in (3, 2, 1)] (a, b) = zip(*[(3., 30), (2., 20), (1., 10), ]) ndtype = [('a', np.float), ('b', np.int)] controldates = date_array(dates, freq='D') controldates.sort_chronologically() series = time_series(zip(*(a, b)), dates, freq='D', dtype=ndtype) assert_equal(series._data.tolist(), [(1., 10), (2., 20), (3., 30)]) assert_equal(series._dates, controldates) # trec = time_records(zip(*(a, b)), dates, freq='D', dtype=ndtype) assert_equal(trec._data.tolist(), [(1., 10), (2., 20), (3., 30)]) assert_equal(trec._dates, controldates) assert_equal(trec['a'], [1., 2., 3.]) assert_equal(trec.a, [1., 2., 3.]) # trec = fromrecords(zip(a, b), dates, names=('a', 'b')) assert_equal(trec._data.tolist(), [(1., 10), (2., 20), (3., 30)]) assert_equal(trec._dates, controldates) assert_equal(trec['a'], [1., 2., 3.]) assert_equal(trec.a, [1., 2., 3.]) # trec = fromarrays([a, b], dates, names=('a', 'b')) assert_equal(trec._data.tolist(), [(1., 10), (2., 20), (3., 30)]) assert_equal(trec._dates, controldates) assert_equal(trec['a'], [1., 2., 3.]) assert_equal(trec.a, [1., 2., 3.])
def test_sorted(self): dates = [ts.Date('D', string='2007-01-%02i' % i) for i in (3, 2, 1)] (a, b) = zip(*[ (3., 30), (2., 20), (1., 10), ]) ndtype = [('a', np.float), ('b', np.int)] controldates = date_array(dates, freq='D') controldates.sort_chronologically() series = time_series(zip(*(a, b)), dates, freq='D', dtype=ndtype) assert_equal(series._data.tolist(), [(1., 10), (2., 20), (3., 30)]) assert_equal(series._dates, controldates) # trec = time_records(zip(*(a, b)), dates, freq='D', dtype=ndtype) assert_equal(trec._data.tolist(), [(1., 10), (2., 20), (3., 30)]) assert_equal(trec._dates, controldates) assert_equal(trec['a'], [1., 2., 3.]) assert_equal(trec.a, [1., 2., 3.]) # trec = fromrecords(zip(a, b), dates, names=('a', 'b')) assert_equal(trec._data.tolist(), [(1., 10), (2., 20), (3., 30)]) assert_equal(trec._dates, controldates) assert_equal(trec['a'], [1., 2., 3.]) assert_equal(trec.a, [1., 2., 3.]) # trec = fromarrays([a, b], dates, names=('a', 'b')) assert_equal(trec._data.tolist(), [(1., 10), (2., 20), (3., 30)]) assert_equal(trec._dates, controldates) assert_equal(trec['a'], [1., 2., 3.]) assert_equal(trec.a, [1., 2., 3.])
def setup(self): a = time_series(np.random.rand(24), start_date=ts.now('M')) b = time_series(np.random.rand(24) * 100, dtype=int, start_date=ts.now('M'),) # c = time_series(["%02i" % _ for _ in np.arange(24)], # start_date=ts.now('M')) c = time_series(np.arange(24), start_date=ts.now('M')) trec = fromarrays([a, b, c], dates=a.dates, names='a,b,c') self.info = (a, b, c, trec)
def setup(self): a = time_series(np.random.rand(24), start_date=ts.now('M')) b = time_series( np.random.rand(24) * 100, dtype=int, start_date=ts.now('M'), ) # c = time_series(["%02i" % _ for _ in np.arange(24)], # start_date=ts.now('M')) c = time_series(np.arange(24), start_date=ts.now('M')) trec = fromarrays([a, b, c], dates=a.dates, names='a,b,c') self.info = (a, b, c, trec)