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)