Enable pymc3/tests/test_distributions.py::TestBugfixes::test_issue_3051

pymc-devs · May 30, 2021 · 6239763 · 6239763
1 parent 23f321a
commit 6239763
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/pymc3/tests/test_distributions.py b/pymc3/tests/test_distributions.py
@@ -2909,9 +2909,12 @@ class TestBugfixes:
         "dist_cls,kwargs", [(MvNormal, dict(mu=0)), (MvStudentT, dict(mu=0, nu=2))]
     )
     @pytest.mark.parametrize("dims", [1, 2, 4])
-    @pytest.mark.xfail(reason="Distribution not refactored yet")
     def test_issue_3051(self, dims, dist_cls, kwargs):
-        d = dist_cls.dist(**kwargs, cov=np.eye(dims), size=(dims,))
+        mu = np.repeat(kwargs["mu"], dims)
+        if "nu" in kwargs:
+            d = dist_cls.dist(nu=kwargs["nu"], mu=mu, cov=np.eye(dims), size=(20))
+        else:
+            d = dist_cls.dist(mu=mu, cov=np.eye(dims), size=(20))
 
         X = np.random.normal(size=(20, dims))
         actual_t = logpt(d, X)