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): "Generic setup" d = np.arange(5) m = ma.make_mask([1, 0, 0, 1, 1]) base_d = np.r_[d, d[::-1]].reshape(2, -1).T base_m = np.r_[[m, m[::-1]]].T base = ma.array(base_d, mask=base_m) mrec = mr.fromarrays(base.T,) dlist = ['2007-%02i' % (i + 1) for i in d] dates = date_array(dlist) mts = time_series(mrec, dates) rts = time_records(mrec, dates) self.data = [d, m, mrec, dlist, dates, mts, rts]
def setup(self): "Generic setup" d = np.arange(5) m = ma.make_mask([1, 0, 0, 1, 1]) base_d = np.r_[d, d[::-1]].reshape(2, -1).T base_m = np.r_[[m, m[::-1]]].T base = ma.array(base_d, mask=base_m) mrec = mr.fromarrays(base.T, ) dlist = ['2007-%02i' % (i + 1) for i in d] dates = date_array(dlist) mts = time_series(mrec, dates) rts = time_records(mrec, dates) self.data = [d, m, mrec, dlist, dates, mts, rts]
def test_fromrecords(self): "Test from recarray." [d, m, mrec, dlist, dates, mts, rts] = self.data nrec = nr.fromarrays(np.r_[[d, d[::-1]]]) mrecfr = fromrecords(nrec.tolist(), dates=dates) assert_equal(mrecfr.f0, mrec.f0) assert_equal(mrecfr.dtype, mrec.dtype) #.................... altrec = [tuple([d, ] + list(r)) for (d, r) in zip(dlist, nrec)] mrecfr = fromrecords(altrec, names='dates,f0,f1') assert_equal(mrecfr.f0, mrec.f0) assert_equal(mrecfr.dtype, mrec.dtype) #.................... tmp = time_records(rts._series[::-1], dates=rts.dates) mrecfr = fromrecords(tmp) assert_equal(mrecfr.f0, mrec.f0[::-1]) #.................... mrecfr = fromrecords(mrec.data, dates=dates, mask=m) assert_equal(mrecfr.recordmask, m)
def test_fromrecords(self): "Test from recarray." [d, m, mrec, dlist, dates, mts, rts] = self.data nrec = nr.fromarrays(np.r_[[d, d[::-1]]]) mrecfr = fromrecords(nrec.tolist(), dates=dates) assert_equal(mrecfr.f0, mrec.f0) assert_equal(mrecfr.dtype, mrec.dtype) #.................... altrec = [tuple([ d, ] + list(r)) for (d, r) in zip(dlist, nrec)] mrecfr = fromrecords(altrec, names='dates,f0,f1') assert_equal(mrecfr.f0, mrec.f0) assert_equal(mrecfr.dtype, mrec.dtype) #.................... tmp = time_records(rts._series[::-1], dates=rts.dates) mrecfr = fromrecords(tmp) assert_equal(mrecfr.f0, mrec.f0[::-1]) #.................... mrecfr = fromrecords(mrec.data, dates=dates, mask=m) assert_equal(mrecfr.recordmask, m)