def test_get(self): "Tests fields retrieval" [d, m, mrec, dlist, dates, ts, mts] = self.data assert(isinstance(mts['f0'], TimeSeries)) assert_equal(mts['f0']._dates, dates) assert_equal(mts['f0']._data, d) assert_equal(mts['f0']._mask, m) # assert(isinstance(mts[0], MultiTimeSeries)) assert_equal(mts._data[0], mrec[0]) # We can't use assert_equal here, as it tries to convert the tuple into a singleton assert(mts[0]._data.view(N.ndarray) == mrec[0]) assert_equal(mts._dates[0], dates[0]) assert_equal(mts[0]._dates, dates[0]) # assert(isinstance(mts['2007-01'], MultiTimeSeries)) assert(mts['2007-01']._data == mrec[0]) assert_equal(mts['2007-01']._dates, dates[0]) # assert(isinstance(mts.f0, TimeSeries)) assert_equal(mts.f0, time_series(d, dates=dates, mask=m)) assert_equal(mts.f1, time_series(d[::-1], dates=dates, mask=m[::-1])) assert((mts._fieldmask == N.core.records.fromarrays([m, m[::-1]])).all()) assert_equal(mts._mask, N.r_[[m,m[::-1]]].all(0)) assert_equal(mts.f0[1], mts[1].f0) # assert(isinstance(mts[:2], MultiTimeSeries)) assert_equal(mts[:2]._data.f0, mrec[:2].f0) assert_equal(mts[:2]._data.f1, mrec[:2].f1) assert_equal(mts[:2]._dates, dates[:2])
def setup(self): "Generic setup" d = N.arange(5) m = MA.make_mask([1,0,0,1,1]) base_d = N.r_[d,d[::-1]].reshape(2,-1).T base_m = N.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) ts = time_series(mrec,dates) mts = MultiTimeSeries(mrec,dates) self.data = [d, m, mrec, dlist, dates, ts, mts]