Add DM test case based on BetaBinomial.

pymc-devs · Dec 16, 2019 · 46ceefb · 46ceefb
1 parent f41450d
commit 46ceefb
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pymc3/tests/test_distributions_random.py b/pymc3/tests/test_distributions_random.py
@@ -405,6 +405,11 @@ class TestBetaBinomial(BaseTestCases.BaseTestCase):
     params = {'n': 5, 'alpha': 1., 'beta': 1.}
 
 
+class TestDirichletMultinomial(BaseTestCases.BaseTestCase):
+    distribution = pm.DirichletMultinomial
+    params = {'n': [5], 'alpha': [[1., 1., 1., 1.]]}
+
+
 class TestBernoulli(BaseTestCases.BaseTestCase):
     distribution = pm.Bernoulli
     params = {'p': 0.5}