def test_set_bounds():
pbounds = {
'p1': (0, 1),
'p3': (0, 3),
'p2': (0, 2),
'p4': (0, 4),
}
space = TargetSpace(target_func, pbounds)
# Ignore unknown keys
space.set_bounds({"other": (7, 8)})
assert all(space.bounds[:, 0] == np.array([0, 0, 0, 0]))
assert all(space.bounds[:, 1] == np.array([1, 2, 3, 4]))
# Update bounds accordingly
space.set_bounds({"p2": (1, 8)})
assert all(space.bounds[:, 0] == np.array([0, 1, 0, 0]))
assert all(space.bounds[:, 1] == np.array([1, 8, 3, 4]))
def test_set_bounds():
pbounds = {'p1': (0, 1), 'p3': (0, 3), 'p2': (0, 2), 'p4': (0, 4)}
ptypes = {'p1': int, 'p2': float, 'p3': int, 'p4': float}
space = TargetSpace(target_func, pbounds, ptypes)
# Ignore unknown keys
space.set_bounds({"other": (7, 8)})
assert all(space.bounds[:, 0] == np.array([0, 0, 0, 0]))
assert all(space.bounds[:, 1] == np.array([1, 2, 3, 4]))
# Update bounds accordingly
space.set_bounds({"p3": (1.1, 8.7)})
print(space.bounds)
assert all(space.bounds[:, 0] == np.array([0, 0, 1, 0]))
assert all(space.bounds[:, 1] == np.array([1, 2, 9, 4]))
ptypes = None
space = TargetSpace(target_func, pbounds, ptypes)
space.set_bounds({"p3": (1.1, 8.7)})
print(space.bounds)
assert all(space.bounds[:, 0] == np.array([0, 0, 1.1, 0]))
assert all(space.bounds[:, 1] == np.array([1, 2, 8.7, 4]))