def test__merge_annotated_and_original_entities(
parse_original: Dict[Text, Any],
parse_annotated: Dict[Text, Any],
expected_entities: List[Dict[Text, Any]],
):
entities = interactive._merge_annotated_and_original_entities(
parse_annotated, parse_original)
assert entities == expected_entities
def test_entity_annotation_merge_with_original():
parse_original = {
"text":
"Hello there rasa, it's me, paula.",
"entities": [{
"start": 12,
"end": 16,
"entity": "name1",
"value": "rasa",
"extractor": "batman",
}],
"intent": {
"name": "greeting",
"confidence": 0.9
},
}
parse_annotated = {
"text":
"Hello there rasa, it's me, paula.",
"entities": [
{
"start": 12,
"end": 16,
"entity": "name1",
"value": "rasa"
},
{
"start": 26,
"end": 31,
"entity": "name2",
"value": "paula"
},
],
"intent": {
"name": "greeting",
"confidence": 0.9
},
}
entities = interactive._merge_annotated_and_original_entities(
parse_annotated, parse_original)
assert entities == [
{
"start": 12,
"end": 16,
"entity": "name1",
"value": "rasa",
"extractor": "batman",
},
{
"start": 26,
"end": 31,
"entity": "name2",
"value": "paula"
},
]
