Skip to content

Commit 005edbe

Browse files
committed
Tmp testing with np assert_almost_equal print messages.
1 parent fe5f30b commit 005edbe

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

pymc/testing.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -728,9 +728,11 @@ def check_selfconsistency_discrete_icdf(
728728
computed_value = dist_icdf_fn(
729729
**point, value=np.exp(dist_logcdf_fn(**point, value=value))
730730
)
731-
assert (
732-
expected_value == computed_value
733-
), f"expected_value = {expected_value}, computed_value = {computed_value}, {point}"
731+
npt.assert_almost_equal(
732+
expected_value,
733+
computed_value,
734+
err_msg=f"point: {point}, value: {value}, cdf: {np.exp(dist_logcdf_fn(**point, value=value))}",
735+
)
734736

735737

736738
def assert_moment_is_expected(model, expected, check_finite_logp=True):

0 commit comments

Comments
 (0)