Skip to content

Commit 37f3ebf

Browse files
committed
fix one more test
1 parent c8a3c95 commit 37f3ebf

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

pymc/tests/test_model.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1269,8 +1269,12 @@ def test_interval_missing_observations():
12691269
)
12701270
assert np.all(np.var(pp_trace["theta1"], 0) > 0.0)
12711271
assert np.all(np.var(pp_trace["theta2"], 0) > 0.0)
1272-
assert np.mean(pp_trace["theta1"][:, ~obs1.mask] - pp_trace["theta1_observed"]) == 0.0
1273-
assert np.mean(pp_trace["theta2"][:, ~obs2.mask] - pp_trace["theta2_observed"]) == 0.0
1272+
assert np.isclose(
1273+
np.mean(pp_trace["theta1"][:, ~obs1.mask] - pp_trace["theta1_observed"]), 0
1274+
)
1275+
assert np.isclose(
1276+
np.mean(pp_trace["theta2"][:, ~obs2.mask] - pp_trace["theta2_observed"]), 0
1277+
)
12741278

12751279

12761280
def test_double_counting():

0 commit comments

Comments
 (0)