naive test

Author: TonyBagnall

aeon/forecasting/tests/test_naive.py

@@ -69,13 +69,14 @@ def test_predict():
     """Test different input for private predict."""
     forecaster = NaiveForecaster(strategy="mean")
     y = np.array([1, 2, 3, 4, 5, 6, 7, 8])
-    x = forecaster._predict(y)
-    assert isinstance(x, float)
+    pred = forecaster._predict(y)
+    np.testing.assert_allclose(pred, 4.5)
+    assert isinstance(pred, float)
     forecaster = NaiveForecaster(strategy="seasonal_last", seasonal_period=2)
     forecaster._fit(y)
     assert forecaster.forecast_ == 7.0
     forecaster = NaiveForecaster(strategy="FOOBAR")
     with pytest.raises(ValueError, match="Unknown strategy"):
-        x = forecaster._fit(y)
+        forecaster._fit(y)
     with pytest.raises(ValueError, match="Unknown strategy"):
-        x = forecaster._predict(y)
+        forecaster._predict(y)