Skip to content

Commit 2269bd6

Browse files
committed
add sample to check observed_data
1 parent 89561ec commit 2269bd6

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

tests/test_data.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -547,3 +547,10 @@ def test_scaling_data_works_in_likelihood() -> None:
547547
scaled_target = target / scale
548548
mu = pm.Normal("mu", mu=0, sigma=1)
549549
pm.Normal("x", mu=mu, sigma=1, observed=scaled_target)
550+
551+
idata = pm.sample(10, chains=1, tune=100)
552+
553+
np.testing.assert_allclose(
554+
idata.observed_data["x"].values,
555+
data / scale,
556+
)

0 commit comments

Comments
 (0)