Python DiscreteSearchSpace.sample Examples

Example #1

def test_discrete_search_space_sampling_raises_when_too_many_samples_are_requested(
    num_samples: int,
) -> None:
    search_space = DiscreteSearchSpace(_points_in_2D_search_space())

    with pytest.raises(ValueError, match="samples"):
        search_space.sample(num_samples)

Example #2

File: test_space.py Project: uri-granta/trieste

def test_discrete_search_space_sampling_raises_when_too_many_samples_are_requested(
    num_samples: int,
) -> None:
    search_space = DiscreteSearchSpace(_points_in_2D_search_space())

    with pytest.raises(tf.errors.InvalidArgumentError):
        search_space.sample(num_samples)

Example #3

def test_discrete_search_space_sampling(num_samples: int) -> None:
    search_space = DiscreteSearchSpace(_points_in_2D_search_space())
    samples = search_space.sample(num_samples)
    _assert_correct_number_of_unique_constrained_samples(
        num_samples, search_space, samples)