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)