Skip to content

Commit 46ceefb

Browse files
committed
Add DM test case based on BetaBinomial.
1 parent f41450d commit 46ceefb

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

pymc3/tests/test_distributions_random.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -405,6 +405,11 @@ class TestBetaBinomial(BaseTestCases.BaseTestCase):
405405
params = {'n': 5, 'alpha': 1., 'beta': 1.}
406406

407407

408+
class TestDirichletMultinomial(BaseTestCases.BaseTestCase):
409+
distribution = pm.DirichletMultinomial
410+
params = {'n': [5], 'alpha': [[1., 1., 1., 1.]]}
411+
412+
408413
class TestBernoulli(BaseTestCases.BaseTestCase):
409414
distribution = pm.Bernoulli
410415
params = {'p': 0.5}

0 commit comments

Comments
 (0)