def test_check_in_domain_with_only_bandit_parameters():
p1 = BanditParameter("bandit1",
np.array([[3, 4.0, 0], [3, 4, 0], [4, 4, 1]]))
p2 = BanditParameter("bandit2", np.array([[0, 1], [1, 1], [1.0, 0]]))
one_bandit_space = ParameterSpace([p1])
x_test = np.array([[3.0, 4, 0], [4.0, 3, 0]])
in_domain = one_bandit_space.check_points_in_domain(x_test)
assert np.array_equal(in_domain, np.array([True, False]))
two_bandits_space = ParameterSpace([p1, p2])
x_test = np.array([[3.0, 4, 0, 0, 1], [4.0, 3, 0, 0, 1], [3, 4, 0, 1, 1]])
in_domain = two_bandits_space.check_points_in_domain(x_test)
assert np.array_equal(in_domain, np.array([True, False, True]))
def test_check_in_domain_with_bandit_parameter():
mixed_space_with_bandit = ParameterSpace([
ContinuousParameter("c", 1.0, 5.0),
DiscreteParameter("d", [0, 1, 2]),
CategoricalParameter("cat", OneHotEncoding(["blue", "red"])),
BanditParameter("bandit", np.array([[0, 1], [1, 1], [1.0, 0]])),
])
x_test = np.array([[1.5, 0, 1.0, 0.0, 0, 1], [1.5, 0, 1.0, 0.0, 0.0, 0.0]])
in_domain = mixed_space_with_bandit.check_points_in_domain(x_test)
assert np.array_equal(in_domain, np.array([True, False]))
def test_check_in_domain_with_bandit_parameter():
mixed_space_with_bandit = ParameterSpace([
ContinuousParameter('c', 1.0, 5.0),
DiscreteParameter('d', [0, 1, 2]),
CategoricalParameter('cat', OneHotEncoding(['blue', 'red'])),
BanditParameter('bandit', np.array([[0, 1], [1, 1], [1., 0]]))
])
x_test = np.array([[1.5, 0, 1., 0., 0, 1], [1.5, 0, 1., 0., 0., 0.]])
in_domain = mixed_space_with_bandit.check_points_in_domain(x_test)
assert np.array_equal(in_domain, np.array([True, False]))