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]
