def test_disamb_does_not_trigger_when_data_is_missing_in_parse_data():
disambiguator = Disambiguator(
disamb_rule=load_yaml('./tests/disambiguator/test_disambiguator4.yaml')
['disambiguation_policy'])
parse_data = {}
assert disambiguator.should_disambiguate(parse_data) is False
def test_disamb_does_not_trigger_when_intent_is_null():
disambiguator = Disambiguator(
disamb_rule=load_yaml('./tests/disambiguator/test_disambiguator4.yaml')
['disambiguation_policy'])
parse_data = {
"intent": {
"name": None,
"confidence": 0.0
},
"entities": [],
"intent_ranking": []
}
assert disambiguator.should_disambiguate(parse_data) is False
def test_disamb_trigger2():
disambiguator = Disambiguator(
disamb_rule=load_yaml('./tests/disambiguator/test_disambiguator2.yaml')
['disambiguation_policy'])
parse_data = {
"intent_ranking": [{
"name": "intentA",
"confidence": 0.6
}, {
"name": "intentB",
"confidence": 0.2
}]
}
result = disambiguator.should_disambiguate(parse_data)
assert result is False
def test_disamb_and_fallback_trigger_none():
rules = load_yaml('./tests/disambiguator/test_disambiguator9.yaml')
disambiguator = Disambiguator(disamb_rule=rules['disambiguation_policy'],
fallback_rule=rules['fallback_policy'])
parse_data = {
"intent_ranking": [{
"name": "intentA",
"confidence": 0.51
}, {
"name": "intentB",
"confidence": 0.25
}]
}
result = disambiguator.should_fallback(
parse_data) or disambiguator.should_disambiguate(parse_data)
assert result is False
def test_disamb_and_fallback_trigger_both():
rules = load_yaml('./tests/disambiguator/test_disambiguator9.yaml')
disambiguator = Disambiguator(disamb_rule=rules['disambiguation_policy'],
fallback_rule=rules['fallback_policy'])
parse_data = {
"intent_ranking": [{
"name": "intentA",
"confidence": 0.49
}, {
"name": "intentB",
"confidence": 0.4
}]
}
# fallback has precedence and short-circuits disamb in the actual program flow as per rules.py
result = disambiguator.should_fallback(
parse_data) and disambiguator.should_disambiguate(parse_data)
assert result is True