mend

AmpersandTV · Jul 13, 2020 · 0b99b8d · 0b99b8d
1 parent af8d690
commit 0b99b8d
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/tests/test_cmp.py b/tests/test_cmp.py
@@ -152,4 +152,5 @@ def test_seasonality_cmp_sampling(N: int = 200, off_param=1):
         check_metrics_for_sampling(trace_, simulation)
         betas_np = np.concatenate([betas_intercept, betas_hour, betas_week])
         beta_pred = trace_.posterior["beta_s"].values[0].mean(0)
-        assert np.allclose(beta_pred, betas_np)
+        beta_mape = abs(beta_pred - betas_np) / betas_np
+        assert beta_mape.mean() < 0.2
diff --git a/tests/test_sampling_seasonality.py b/tests/test_sampling_seasonality.py
@@ -82,4 +82,9 @@ def test_seasonality_sampling(N: int = 200, off_param=1):
         check_metrics_for_sampling(trace_, simulation)
         betas_np = np.concatenate([betas_intercept, betas_hour, betas_week])
         beta_pred = trace_.posterior["beta_s"].values[0].mean(0)
-        assert np.allclose(beta_pred, betas_np)
+        beta_mape = abs(beta_pred - betas_np) / betas_np
+        assert beta_mape.mean() < 0.1
+
+
+#%%
+test_seasonality_sampling()
diff --git a/tests/test_ztp.py b/tests/test_ztp.py
@@ -132,4 +132,5 @@ def test_seasonality_ztp_sampling(N: int = 200, off_param=1):
         check_metrics_for_sampling(trace_, simulation)
         betas_np = np.concatenate([betas_intercept, betas_hour, betas_week])
         beta_pred = trace_.posterior["beta_s"].values[0].mean(0)
-        assert np.allclose(beta_pred, betas_np)
+        beta_mape = abs(beta_pred - betas_np) / betas_np
+        assert beta_mape.mean() < 0.1