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