tests(impact): adjust test to new p-value param

SocialChangeLab · Aug 15, 2024 · 90c1f38 · 90c1f38
1 parent 008c8a4
commit 90c1f38
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/backend-python/media_impact_monitor/impact_estimators/interrupted_time_series_test.py b/backend-python/media_impact_monitor/impact_estimators/interrupted_time_series_test.py
@@ -82,7 +82,7 @@ def test_mean_impact_estimates():
     assert impacts_df.index[0] == -4
     assert impacts_df.index[-1] == 6
     assert len(impacts_df) == 4 + 7
-    assert set(impacts_df.columns) == {"mean", "ci_lower", "ci_upper"}
+    assert set(impacts_df.columns) == {"mean", "ci_lower", "ci_upper", "p_value"}
     for i in range(-4, -1):
         mean = impacts_df.loc[i, "mean"]
         assert -50 <= mean <= 50