def test_g(self):
"""Test insert of a reset as the first event of the quarter."""
dates = [datetime(2010, 9, 30), datetime(2010, 10, 31)]
values = [10.0, 5.0]
expected_dates = [datetime(2010, 9, 30),
datetime(2010, 10, 1),
datetime(2010, 10, 31)]
expected_values = [10.0, 0.0, 5.0]
data = DataForCumulativeGraph(dates, values)
dates, values = data.insert_restart(dates, values, 'quarter')
self.assertEqual(expected_dates, dates)
self.assertEqual(expected_values, values)
def test_h(self):
"""Test insert of a reset as the first event of the month."""
dates = [datetime(2011, 4, 30), datetime(2011, 5, 31)]
values = [10.0, 5.0]
expected_dates = [datetime(2011, 4, 30),
datetime(2011, 5, 1),
datetime(2011, 5, 31)]
expected_values = [10.0, 0.0, 5.0]
data = DataForCumulativeGraph(dates, values)
dates, values = data.insert_restart(dates, values, 'month')
self.assertEqual(expected_dates, dates)
self.assertEqual(expected_values, values)