From 17123ac7dd5e203cc8b70532adead378bbc8b661 Mon Sep 17 00:00:00 2001 From: Luke LB Date: Mon, 4 Sep 2023 21:46:18 +0100 Subject: [PATCH] fixed test failing for log2 and log10 (due to floating point inprecision in the graph) --- tests/logprob/test_transforms.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/tests/logprob/test_transforms.py b/tests/logprob/test_transforms.py index aaaab0bb6c7..79b8ac0d6b3 100644 --- a/tests/logprob/test_transforms.py +++ b/tests/logprob/test_transforms.py @@ -1125,7 +1125,13 @@ def test_special_log_exp_transforms(transform): logp_test = logp(transformed_rv, vv) logp_ref = logp(ref_transformed_rv, vv) - assert equal_computations([logp_test], [logp_ref]) + if transform in ["log2", "log10"]: + # in the cases of log2 and log10 floating point inprecision causes failure + # from equal_computations so evaluate logp and check all close instead + vv_test = np.array(0.25) + np.testing.assert_allclose(logp_ref.eval({vv: vv_test}), logp_test.eval({vv: vv_test})) + else: + assert equal_computations([logp_test], [logp_ref]) @pytest.mark.parametrize(