Update tests for posterior_to_prior

arviz-devs · Aug 13, 2024 · 6b1f16e · 6b1f16e
1 parent a8d20b7
commit 6b1f16e
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/preliz/ppls/agnostic.py b/preliz/ppls/agnostic.py
@@ -33,7 +33,6 @@ def posterior_to_prior(model, idata, alternative=None):
 
     if alternative is None:
         for var, dist in model_info.items():
-            print(var)
             dist._fit_mle(posterior[var].values)
             new_priors[var] = dist
     else:

diff --git a/preliz/tests/test_posterior_to_prior.py b/preliz/tests/test_posterior_to_prior.py
@@ -1,3 +1,5 @@
+import re
+
 import preliz as pz
 import pymc as pm
 
@@ -12,6 +14,10 @@
 
 def test_p2p():
     pz.posterior_to_prior(model, idata)
-    pz.posterior_to_prior(model, idata, alternative="auto")
+    pattern = r'\s*b = pm.\x1b\[1mGamma\x1b\[0m\("b",\s*alpha\s*=\s*([+-]?\d*\.?\d+),\s*beta\s*=\s*([+-]?\d*\.?\d+)\)'
+    assert bool(re.search(pattern, pz.posterior_to_prior(model, idata, alternative="auto")))
     pz.posterior_to_prior(model, idata, alternative=[pz.LogNormal()])
-    pz.posterior_to_prior(model, idata, alternative={"b": [pz.Gamma(mu=0)]})
+    pattern = r'\s*b = pm.\x1b\[1mGamma\x1b\[0m\("b",\s*mu\s*=\s*([+-]?\d*\.?\d+),\s*sigma\s*=\s*([+-]?\d*\.?\d+)\)'
+    assert bool(
+        re.search(pattern, pz.posterior_to_prior(model, idata, alternative={"b": [pz.Gamma(mu=0)]}))
+    )