Skip to content

Commit 1526261

Browse files
committed
Fixes in test caas
1 parent a942b6e commit 1526261

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

tests/test_model.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1062,7 +1062,7 @@ def test_deterministic():
10621062
assert model["y"] == y
10631063

10641064

1065-
def test_potential_with_dims():
1065+
def test_determinsitic_with_dims():
10661066
"""
10671067
Test to check the passing of dims to the potential
10681068
"""
@@ -1072,6 +1072,16 @@ def test_potential_with_dims():
10721072
assert model.named_vars_to_dims == {"y": ("observed",)}
10731073

10741074

1075+
def test_potential_with_dims():
1076+
"""
1077+
Test to check the passing of dims to the potential
1078+
"""
1079+
with pm.Model(coords={"observed": range(10)}) as model:
1080+
x = pm.Normal("x", 0, 1)
1081+
y = pm.Potential("y", x**2, dims=("observed",))
1082+
assert model.named_vars_to_dims == {"y": ("observed",)}
1083+
1084+
10751085
def test_empty_model_representation():
10761086
assert pm.Model().str_repr() == ""
10771087

0 commit comments

Comments
 (0)