def test_adjustendpoints(self):
"Tests adjust_endpoints"
(series, data, dates) = self.d
dseries = adjust_endpoints(series, series.dates[0], series.dates[-1])
assert_equal(dseries, series)
dseries = adjust_endpoints(series, series.dates[3], series.dates[-3])
assert_equal(dseries, series[3:-2])
dseries = adjust_endpoints(series, end_date=Date('D', string='2007-01-31'))
assert_equal(dseries.size, 31)
assert_equal(dseries._mask, N.r_[series._mask, [1]*16])
dseries = adjust_endpoints(series, end_date=Date('D', string='2007-01-06'))
assert_equal(dseries.size, 6)
assert_equal(dseries, series[:6])
dseries = adjust_endpoints(series,
start_date=Date('D', string='2007-01-06'),
end_date=Date('D', string='2007-01-31'))
assert_equal(dseries.size, 26)
assert_equal(dseries._mask, N.r_[series._mask[5:], [1]*16])
def _test_read_append_tser(self):
"test reading of appended time series"
result = ts.adjust_endpoints(data['tser']['float32'],
start_date=data['tser']['float32'].start_date,
end_date=appendTSer.end_date)
result[appendTSer.start_date:appendTSer.end_date+1] = appendTSer
ser = self.db.read('$appendTSer')
assert_array_equal(result, ser)
def _test_read_range_tser(self):
"test reading a time series over specified ranges"
src = data['tser']['float32']
s1 = src.start_date+2
s2 = src.start_date-2
e1 = src.end_date+2
e2 = src.end_date-2
dateList = [(s1, e1),
(s1, e2),
(s2, e1),
(s2, e2)]
for s, e in dateList:
res = ts.adjust_endpoints(src, start_date=s, end_date=e)
ser = self.db.read('$tser_float32', start_date=s, end_date=e)
assert_array_equal(res, ser)
