Skip to content

Commit f077250

Browse files
refactor: restored missing assert
1 parent a92ba8f commit f077250

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/inference/laplace_approx/test_laplace.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -316,4 +316,8 @@ def test_laplace_scalar_basinhopping(optimizer_method):
316316
assert idata_laplace.fit.mean_vector.shape == (1,)
317317
assert idata_laplace.fit.covariance_matrix.shape == (1, 1)
318318

319+
np.testing.assert_allclose(
320+
idata_laplace.posterior.p.mean(dim=["chain", "draw"]), data.mean(), atol=0.1
321+
)
322+
319323
np.testing.assert_allclose(idata_laplace.fit.mean_vector.values.item(), data.mean(), atol=0.1)

0 commit comments

Comments
 (0)