We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
None
1 parent d4e5db1 commit 9792dffCopy full SHA for 9792dff
tests/model/test_core.py
@@ -899,12 +899,14 @@ def test_none_coords_autonumbering(self):
899
# TODO: Either allow dims without coords everywhere or nowhere
900
with pm.Model() as m:
901
m.add_coord(name="a", values=None, length=3)
902
- m.add_coord(name="b", values=range(5))
903
- x = pm.Normal("x", dims=("a", "b"))
+ m.add_coord(name="b", values=range(-5, 0))
+ m.add_coord(name="c", values=None, length=7)
904
+ x = pm.Normal("x", dims=("a", "b", "c"))
905
prior = pm.sample_prior_predictive(draws=2).prior
- assert prior["x"].shape == (1, 2, 3, 5)
906
+ assert prior["x"].shape == (1, 2, 3, 5, 7)
907
assert list(prior.coords["a"].values) == list(range(3))
- assert list(prior.coords["b"].values) == list(range(5))
908
+ assert list(prior.coords["b"].values) == list(range(-5, 0))
909
+ assert list(prior.coords["c"].values) == list(range(7))
910
911
def test_set_data_indirect_resize_without_coords(self):
912
with pm.Model() as pmodel:
0 commit comments