Alternative method to annoy ARIMA #230

datasciencecampus · Apr 5, 2019 · 3d00743 · 3d00743
1 parent cad6468
commit 3d00743
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/algorithms/test_arima.py b/tests/algorithms/test_arima.py
@@ -51,9 +51,9 @@ def test_linear_sequence(self):
         np_test.assert_almost_equal(actual_prediction, expected_prediction, decimal=4)
 
     def test_flakey_sequence(self):
-        time_series = [1.0, -2.0, 3.0, -4.0, 5.0]
+        time_series = [20.0, -20.0]
         num_predicted_periods = 3
-        expected_prediction = [0, 4.8, np.nan]
+        expected_prediction = [np.nan] * 3
         arima = ARIMAForecast(time_series, num_predicted_periods)
 
         actual_prediction = arima.predict_counts()