def test_categorical_search_space_ok(): search_space = { "values": ["a", "b", "c"], "probabilities": [0.1, 0.2, 0.7] } search_space = validate_categorical(search_space)
def test_categorical_search_space_bad_values(): search_space = { "values": 3 } with pytest.raises(ValidationError): search_space = validate_categorical(search_space)
def test_categorical_search_space_no_probas(): search_space = {"values": ["a", "b", "c"]} search_space = validate_categorical(search_space) assert "probabilities" in search_space.keys() assert sum(search_space["probabilities"]) == 1
def test_categorical_search_space_wrong_probas(): search_space = { "values": ["a", "b", "c"], "probabilities": [0.5, 0.5, 0.5] } with pytest.raises(ValidationError): search_space = validate_categorical(search_space)
def test_categorical_search_space_not_dict(): search_space = ["a", "b", "c"], with pytest.raises(ValidationError): search_space = validate_categorical(search_space)