Update metrics test to handle zero values.

facebook · May 4, 2024 · fb07236 · fb07236
1 parent 03d5bfa
commit fb07236
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/python/prophet/tests/test_diagnostics.py b/python/prophet/tests/test_diagnostics.py
@@ -234,13 +234,13 @@ def test_performance_metrics(self, ts_short, backend):
             metrics=["coverage", "mse"],
         )
         assert set(df_horizon.columns) == {"coverage", "mse", "horizon"}
-        # Skip MAPE
-        df_cv.loc[0, "y"] = 0.0
+        # Handle zero y and yhat and skip MAPE
+        df_cv["y"] = 0.0
+        df_cv["yhat"] = 0.0
         df_horizon = diagnostics.performance_metrics(
             df_cv,
-            metrics=["coverage", "mape"],
         )
-        assert set(df_horizon.columns) == {"coverage", "horizon"}
+        assert set(df_horizon.columns) == {"coverage", "horizon", "mae", "mdape", "mse", "rmse", "smape"}
         df_horizon = diagnostics.performance_metrics(
             df_cv,
             metrics=["mape"],