def test_box_discretize_returns_search_space_with_only_points_contained_within_box(
num_samples: int, ) -> None:
box = Box(tf.zeros((3, )), tf.ones((3, )))
dss = box.discretize(num_samples)
samples = dss.sample(num_samples)
assert all(sample in box for sample in samples)
def test_box_discretize_returns_search_space_with_correct_number_of_points(
num_samples: int, ) -> None:
box = Box(tf.zeros((3, )), tf.ones((3, )))
dss = box.discretize(num_samples)
samples = dss.sample(num_samples)
assert len(samples) == num_samples
with pytest.raises(ValueError):
dss.sample(num_samples + 1)
