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