Skip to content

Commit 1425508

Browse files
committed
Enable pymc3/tests/test_distributions.py::TestBugfixes::test_issue_3051
1 parent e1d9c69 commit 1425508

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

pymc3/tests/test_distributions.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2941,9 +2941,12 @@ class TestBugfixes:
29412941
"dist_cls,kwargs", [(MvNormal, dict(mu=0)), (MvStudentT, dict(mu=0, nu=2))]
29422942
)
29432943
@pytest.mark.parametrize("dims", [1, 2, 4])
2944-
@pytest.mark.xfail(reason="Distribution not refactored yet")
29452944
def test_issue_3051(self, dims, dist_cls, kwargs):
2946-
d = dist_cls.dist(**kwargs, cov=np.eye(dims), size=(dims,))
2945+
mu = np.repeat(kwargs["mu"], dims)
2946+
if "nu" in kwargs:
2947+
d = dist_cls.dist(nu=kwargs["nu"], mu=mu, cov=np.eye(dims), size=(20))
2948+
else:
2949+
d = dist_cls.dist(mu=mu, cov=np.eye(dims), size=(20))
29472950

29482951
X = np.random.normal(size=(20, dims))
29492952
actual_t = logpt(d, X)

0 commit comments

Comments
 (0)