def test_f(self): """Test missing dates are automatically added as zeros.""" timeserie = TimeseriesStub() today = datetime(2010, 12, 3) tomorrow = datetime(2010, 12, 4) day_after_tomorrow = datetime(2010, 12, 5) timeserie.add_value(today, 20) timeserie.add_value(day_after_tomorrow, 30) events = [event for event in timeserie.events()] expected_events = [(today, 20), (tomorrow, 0), (day_after_tomorrow, 30)] self.assertEqual(expected_events, events)
def test_a(self): timeseries = TimeseriesStub((datetime(2011, 1, 26), 0.0), (datetime(2011, 2, 3), 10.0), (datetime(2011, 2, 28), 0.0)) timeseries_restricted = \ TimeseriesRestrictedStub(timeseries=timeseries, start_date=datetime(2011, 2, 2), end_date=datetime(2011, 2, 5)) expected_timeseries = TimeseriesStub((datetime(2011, 2, 2), 0.0), (datetime(2011, 2, 3), 10.0), (datetime(2011, 2, 4), 0.0)) self.assertEqual(list(expected_timeseries.events()), \ list(timeseries_restricted.events()))