Fix failing test case

pymc-devs · May 24, 2021 · 3678d71 · 3678d71
1 parent 87607f3
commit 3678d71
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/pymc3/tests/test_model.py b/pymc3/tests/test_model.py
@@ -330,7 +330,6 @@ def test_aesara_switch_broadcast_edge_cases_1(self):
             np.log(0.5) * 10,
         )
 
-    @pytest.mark.xfail(reason="TruncatedNormal not refactored for v4")
     def test_aesara_switch_broadcast_edge_cases_2(self):
         # Known issue 2: https://github.com/pymc-devs/pymc3/issues/4417
         # fmt: off
@@ -343,7 +342,7 @@ def test_aesara_switch_broadcast_edge_cases_2(self):
             mu = pm.Normal("mu", 0, 5)
             obs = pm.TruncatedNormal("obs", mu=mu, sigma=1, lower=-1, upper=2, observed=data)
 
-        npt.assert_allclose(m.dlogp([mu])({"mu": 0}), 2.499424682024436, rtol=1e-5)
+        npt.assert_allclose(m.dlogp([m.rvs_to_values[mu]])({"mu": 0}), 2.499424682024436, rtol=1e-5)
 
 
 @pytest.mark.xfail(reason="DensityDist not refactored for v4")