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) mrecord = fromarrays(base.T, dtype=[('a',N.float_),('b',N.float_)]) self.data = [d, m, mrecord]
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 = fromarrays(base.T,) self.data = [d, m, mrec]
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]
newdates = __getdates(dates=dates, newdates=newdates, length=nvars, freq=None, start_date=None) return MultiTimeSeries(_datalist, dates=newdates, dtype=mdescr) ################################################################################ if __name__ == "__main__": import numpy as N from maskedarray.testutils import assert_equal if 1: 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] assert isinstance(mts.f0, TimeSeries) if 0: mts[:2] = 5 assert_equal(mts.f0._data, [5, 5, 2, 3, 4]) assert_equal(mts.f1._data, [5, 5, 2, 1, 0]) assert_equal(mts.f0._mask, [0, 0, 0, 1, 1]) assert_equal(mts.f1._mask, [0, 0, 0, 0, 1]) mts.harden_mask()