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