diff --git a/preliz/ppls/agnostic.py b/preliz/ppls/agnostic.py index 10e430e9..27b60157 100644 --- 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 index b98c6343..899377ac 100644 --- 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)]})) + )