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_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()))
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()))
