def test_insertRows_single_row_in_the_middle(self):
model = TimeSeriesModelFixedResolution(
TimeSeriesFixedResolution("2019-07-05T12:00", "2 hours", [-5.0, 7.0], True, False)
)
self.assertTrue(model.insertRows(1, 1))
self.assertEqual(
model.value, TimeSeriesFixedResolution("2019-07-05T12:00", "2 hours", [-5.0, 0.0, 7.0], True, False)
)
self.assertEqual(
model.indexes, np.array(["2019-07-05T12:00", "2019-07-05T14:00", "2019-07-05T16:00"], dtype="datetime64")
)
def test_insertRows_at_the_beginning(self):
model = TimeSeriesModelFixedResolution(
TimeSeriesFixedResolution("2019-07-05T12:00", "2 hours", [-5.0, 7.0], True, False)
)
self.assertTrue(model.insertRows(0, 1))
self.assertEqual(
model.value, TimeSeriesFixedResolution("2019-07-05T12:00", "2 hours", [0.0, -5.0, 7.0], True, False)
)
np.testing.assert_equal(
model.indexes, np.array(["2019-07-05T12:00", "2019-07-05T14:00", "2019-07-05T16:00"], dtype="datetime64")
)
