Skip to content

Commit 57092d9

Browse files
committed
make all side param comparisons almost equal to account for slight numpy differences
1 parent c21c6c6 commit 57092d9

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

tests/test_bayesian_optimization.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -626,14 +626,16 @@ def area_of_triangle(sides):
626626
# Test that key properties match
627627
assert len(optimizer.space) == len(new_optimizer.space)
628628
assert optimizer.max["target"] == new_optimizer.max["target"]
629-
assert optimizer.max["params"] == new_optimizer.max["params"]
629+
np.testing.assert_array_almost_equal(
630+
optimizer.max["params"]["sides"], new_optimizer.max["params"]["sides"], decimal=10
631+
)
630632

631633
# Test that all historical data matches
632634
for i in range(len(optimizer.space)):
633-
np.testing.assert_array_almost_equal(optimizer.space.params[i], new_optimizer.space.params[i])
635+
np.testing.assert_array_almost_equal(optimizer.space.params[i], new_optimizer.space.params[i], decimal=10)
634636
assert optimizer.space.target[i] == new_optimizer.space.target[i]
635637
np.testing.assert_array_almost_equal(
636-
optimizer.res[i]["params"]["sides"], new_optimizer.res[i]["params"]["sides"]
638+
optimizer.res[i]["params"]["sides"], new_optimizer.res[i]["params"]["sides"], decimal=10
637639
)
638640
assert optimizer.res[i]["target"] == new_optimizer.res[i]["target"]
639641

0 commit comments

Comments
 (0)