def test_infer_num_initial_data_points_too_few_specified(parameters):
with pytest.warns(
UserWarning,
match="num_initial_data_points < number of parameters found"):
assert GPyOpt._infer_num_initial_data_points(3, parameters) == 5
def test_infer_num_initial_data_points(parameters):
assert GPyOpt._infer_num_initial_data_points('infer', parameters) == 5
assert GPyOpt._infer_num_initial_data_points(3, parameters) == 5
assert GPyOpt._infer_num_initial_data_points(7, parameters) == 7
def test_infer_num_initial_data_points(num_initial_data_points, expected,
parameters):
assert GPyOpt._infer_num_initial_data_points(num_initial_data_points,
parameters) == expected
