def test_zero_fill_time_series_date(self):
result = zh.zero_fill_time_series(
self.zarkov_data['a']['a1'], 'date',
datetime(2010, 5, 1), datetime(2011, 9, 1))
self.assertEqual(len(result), 489)
days_ms = 24 * 3600 * 1000
for p1, p2 in zip(result, result[1:]):
delta = p2[0]-p1[0]
assert delta == days_ms
def test_zero_fill_time_series_month(self):
result = zh.zero_fill_time_series(
self.zarkov_data['a']['a1'], 'month',
datetime(2010, 5, 1), datetime(2011, 9, 1))
self.assertEqual(result[0][1], 0)
self.assertEqual(result[-1][1], 0)
self.assertEqual(len(result), 17)
self.assertEqual(result[1][1], 1000)
self.assertEqual(result[-3][1], 1000)
days_ms = 24 * 3600 * 1000
min_delta = 28 * days_ms
max_delta= 31 * days_ms
for p1, p2 in zip(result, result[1:]):
delta = p2[0]-p1[0]
assert min_delta <= delta <= max_delta, delta