def test_b(self): """Test add_value adds the right events.""" timeseries = SparseTimeseriesStub() timeseries.add_value(datetime(2011, 4, 8), 10.0) timeseries.add_value(datetime(2011, 4, 9), 20.0) timeseries.add_value(datetime(2011, 4, 10), 30.0) expected_events = [(datetime(2011, 4, 8), 10.0), (datetime(2011, 4, 9), 20.0), (datetime(2011, 4, 10), 30.0)] self.assertEqual(expected_events, list(timeseries.events()))
def test_h(self): """Test events returns a subset of the events. The subset starts after the first event and does not include the later events. """ timeseries = SparseTimeseriesStub() timeseries.add_value(datetime(2011, 4, 8), 10.0) timeseries.add_value(datetime(2011, 4, 9), 20.0) timeseries.add_value(datetime(2011, 4, 10), 30.0) timeseries.add_value(datetime(2011, 4, 11), 40.0) start_date, end_date = datetime(2011, 4, 9), datetime(2011, 4, 11) events = list(timeseries.events(start_date, end_date)) self.assertEqual(2, len(events)) self.assertEqual((datetime(2011, 4, 9), 20.0), events[0]) self.assertEqual((datetime(2011, 4, 10), 30.0), events[1])
def test_e(self): """Test events returns a subset of the events. The subset includes all events. """ timeseries = SparseTimeseriesStub() timeseries.add_value(datetime(2011, 4, 8), 10.0) timeseries.add_value(datetime(2011, 4, 9), 20.0) timeseries.add_value(datetime(2011, 4, 10), 30.0) timeseries.add_value(datetime(2011, 4, 11), 40.0) start_date, end_date = datetime(2011, 4, 8), datetime(2011, 4, 12) events = list(timeseries.events(start_date, end_date)) self.assertEqual(4, len(events)) self.assertEqual((datetime(2011, 4, 8), 10.0), events[0]) self.assertEqual((datetime(2011, 4, 9), 20.0), events[1]) self.assertEqual((datetime(2011, 4, 10), 30.0), events[2]) self.assertEqual((datetime(2011, 4, 11), 40.0), events[3])
def test_c(self): """Test add_value can only add events on consecutive days.""" timeseries = SparseTimeseriesStub() timeseries.add_value(datetime(2011, 4, 8), 10.0) self.assertRaises(AssertionError, timeseries.add_value, \ datetime(2011, 4, 10), 30.0)