def test_cannot_remove_all_rows(self):
model = TimeSeriesModelFixedResolution(
TimeSeriesFixedResolution("2019-07-05T12:00", "2 hours", [2.3, -5.0, 7.0], True, False)
)
self.assertTrue(model.removeRows(0, 3))
self.assertEqual(model.value, TimeSeriesFixedResolution("2019-07-05T12:00", "2 hours", [2.3], True, False))
self.assertEqual(model.indexes, np.array(["2019-07-05T12:00"], dtype="datetime64"))
def test_removeRows_from_the_middle(self):
model = TimeSeriesModelFixedResolution(
TimeSeriesFixedResolution("2019-07-05T12:00", "2 hours", [2.3, -5.0, 7.0], True, False)
)
self.assertTrue(model.removeRows(1, 1))
self.assertEqual(model.value, TimeSeriesFixedResolution("2019-07-05T12:00", "2 hours", [2.3, 7.0], True, False))
self.assertEqual(model.indexes, np.array(["2019-07-05T12:00", "2019-07-05T14:00"], dtype="datetime64"))
def test_removing_last_row_fails(self):
model = TimeSeriesModelFixedResolution(
TimeSeriesFixedResolution("2019-07-05T12:00", "2 hours", [2.3],
True, False))
self.assertFalse(model.removeRows(0, 1))
