def test_fromtextfile(self): "Tests reading from a text file." fcontent = """# 'Dates', 'One (S)','Two (I)','Three (F)','Four (M)','Five (-)','Six (C)' '2007-01', 'strings',1,1.0,'mixed column',,1 '2007-02', 'with embedded "double quotes"',2,2.0,1.0,,1 '2007-03', 'strings',3,3.0E5,3,,1 '2007-05','strings',4,-1e-10,,,1 """ import os from datetime import datetime fname = 'tmp%s' % datetime.now().strftime("%y%m%d%H%M%S%s") f = open(fname, 'w') f.write(fcontent) f.close() mrectxt = fromtextfile(fname,delimitor=',',varnames='ABCDEFG', dates_column=0) os.unlink(fname) # dlist = ['2007-%02i' % i for i in (1,2,3,5)] assert(isinstance(mrectxt, MultiTimeSeries)) assert_equal(mrectxt._dates, date_array(dlist,'M')) assert_equal(mrectxt.dtype.names, ['B','C','D','E','F','G']) assert_equal(mrectxt.G, [1,1,1,1]) assert_equal(mrectxt.F._mask, [1,1,1,1]) assert_equal(mrectxt.D, [1,2,3.e+5,-1e-10])
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]