Skip to content

Commit b7200d7

Browse files
committed
create a failing test
1 parent f91dd1c commit b7200d7

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

tests/distributions/test_multivariate.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1714,6 +1714,14 @@ class TestDirichlet(BaseTestDistributionRandom):
17141714
"check_rv_size",
17151715
]
17161716

1717+
@pytensor.config.change_flags(warn_float64="raise", floatX="float32")
1718+
def test_dirichlet_float32(self):
1719+
"""https://github.com/pymc-devs/pymc/issues/6779
1720+
"""
1721+
with pm.Model() as model:
1722+
c = pm.floatX([1, 1, 1])
1723+
pm.Dirichlet("a", c)
1724+
model.point_logps()
17171725

17181726
class TestMultinomial(BaseTestDistributionRandom):
17191727
pymc_dist = pm.Multinomial
@@ -2126,3 +2134,4 @@ def test_posdef_symmetric(matrix, result):
21262134
"""
21272135
data = np.array(matrix, dtype=pytensor.config.floatX)
21282136
assert posdef(data) == result
2137+

0 commit comments

Comments
 (0)