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]
