def test_create_keras_timeseriesgenerator_raise_error_on_neg_lookahead(): """Check create_keras_timeseriesgenerator raises an error on negative lookahead""" X = np.array([[0, 1]]) y = X.copy() with pytest.raises(ValueError): create_keras_timeseriesgenerator(X, y, batch_size=2, lookback_window=2, lookahead=-1)
def test_create_keras_timeseriesgenerator_raise_error_on_neg_lookahead( self): # Check that create_keras_timeseriesgenerator raises an error on negative # lookahead X = np.array([[0, 1]]) with self.assertRaises(ValueError): create_keras_timeseriesgenerator(X, batch_size=2, lookback_window=2, lookahead=-1)
def test_create_keras_timeseriesgenerator_lb3_loah2_bs2(): """ Check right output is generated from create_keras_timeseriesgenerator We use lookback_window 2 to get some more interesting batches with lookahead 1 """ X = np.array([[0, 1], [2, 3], [4, 5], [6, 7], [8, 9]]) y = X.copy() gen = create_keras_timeseriesgenerator(X, y, batch_size=2, lookback_window=2, lookahead=2) batch_1 = gen[0] batch_2 = gen[1] batch_1_x = batch_1[0].tolist() batch_1_y = batch_1[1].tolist() batch_2_x = batch_2[0].tolist() batch_2_y = batch_2[1].tolist() assert [[[0, 1], [2, 3]], [[2, 3], [4, 5]]] == batch_1_x assert [[6, 7], [8, 9]] == batch_1_y assert [] == batch_2_x # No more elements left assert [] == batch_2_y
def test_create_keras_timeseriesgenerator_lb3_loah2_bs2(self): # Check that right output is generated from create_keras_timeseriesgenerator # We use lookback_window 2 to get some more interesting batches with lookahead 1 X = np.array([[0, 1], [2, 3], [4, 5], [6, 7], [8, 9]]) gen = create_keras_timeseriesgenerator(X, batch_size=2, lookback_window=2, lookahead=2) batch_1 = gen[0] batch_2 = gen[1] batch_1_x = batch_1[0].tolist() batch_1_y = batch_1[1].tolist() batch_2_x = batch_2[0].tolist() batch_2_y = batch_2[1].tolist() self.assertEquals([[[0, 1], [2, 3]], [[2, 3], [4, 5]]], batch_1_x) self.assertEqual([[6, 7], [8, 9]], batch_1_y) self.assertEqual([], batch_2_x) # No more elements left self.assertEqual([], batch_2_y)
def test_create_keras_timeseriesgenerator_lb3_loah0_bs2(self): # Check that right output is generated from create_keras_timeseriesgenerator X = np.array([[0, 1], [2, 3], [4, 5], [6, 7], [8, 9]]) gen = create_keras_timeseriesgenerator(X, batch_size=2, lookback_window=3, lookahead=0) batch_1 = gen[0] batch_2 = gen[1] batch_1_x = batch_1[0].tolist() batch_1_y = batch_1[1].tolist() batch_2_x = batch_2[0].tolist() batch_2_y = batch_2[1].tolist() self.assertEquals([[[0, 1], [2, 3], [4, 5]], [[2, 3], [4, 5], [6, 7]]], batch_1_x) self.assertEqual([[4, 5], [6, 7]], batch_1_y) self.assertEqual([[[4, 5], [6, 7], [8, 9]]], batch_2_x) self.assertEqual([[8, 9]], batch_2_y)
def test_create_keras_timeseriesgenerator_lb3_loah0_bs2(): """Check that right output is generated from create_keras_timeseriesgenerator""" X = np.array([[0, 1], [2, 3], [4, 5], [6, 7], [8, 9]]) y = X.copy() gen = create_keras_timeseriesgenerator(X, y, batch_size=2, lookback_window=3, lookahead=0) batch_1 = gen[0] batch_2 = gen[1] batch_1_x = batch_1[0].tolist() batch_1_y = batch_1[1].tolist() batch_2_x = batch_2[0].tolist() batch_2_y = batch_2[1].tolist() assert [[[0, 1], [2, 3], [4, 5]], [[2, 3], [4, 5], [6, 7]]] == batch_1_x assert [[4, 5], [6, 7]] == batch_1_y assert [[[4, 5], [6, 7], [8, 9]]] == batch_2_x assert [[8, 9]] == batch_2_y