def test_zero_fill_zarkov_month_str(self):
result0 = zh.zero_fill_zarkov_result(
self.zarkov_data, 'month',
datetime(2010, 5, 1), datetime(2011, 9, 1))
result1 = zh.zero_fill_zarkov_result(
self.zarkov_data, 'month',
'2010-5-1', '2011-09-1')
self.assertEqual(result0, result1)
def test_zero_fill_zarkov_month_dt(self):
result = zh.zero_fill_zarkov_result(
self.zarkov_data, 'month',
datetime(2010, 5, 1), datetime(2011, 9, 1))
a_result = result['a']['a1']
b_result = result['b']['b2']
self.assertEqual(a_result[0][1], 0)
self.assertEqual(a_result[-1][1], 0)
self.assertEqual(len(a_result), 17)
self.assertEqual(a_result[1][1], 1000)
self.assertEqual(a_result[-3][1], 1000)
self.assertEqual(b_result[0][1], 0)
self.assertEqual(b_result[-1][1], 0)
self.assertEqual(len(b_result), 17)
self.assertEqual(b_result[1][1], 2000)
self.assertEqual(b_result[-3][1], 2000)
