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)