def test_evaluate_entities_cv():
mock_extractors = ["EntityExtractorA", "EntityExtractorB"]
result = align_entity_predictions(EN_entity_result, mock_extractors)
assert result == {
"target_labels": [
"O",
"O",
"O",
"O",
"O",
"O",
"O",
"O",
"food",
"location",
"location",
"datetime",
],
"extractor_labels": {
"EntityExtractorA": [
"O",
"person",
"O",
"O",
"O",
"O",
"O",
"O",
"food",
"O",
"location",
"O",
],
"EntityExtractorB": [
"O",
"O",
"O",
"O",
"O",
"O",
"O",
"O",
"O",
"O",
"movie",
"movie",
],
},
}, "Wrong entity prediction alignment"
def test_evaluate_entities_cv():
mock_extractors = ["A", "B"]
result = align_entity_predictions(EN_targets, EN_predicted,
EN_tokens, mock_extractors)
assert result == {
"target_labels": ["O", "O", "O", "O", "O", "O", "O", "O", "food",
"location", "location", "datetime"],
"extractor_labels": {
"A": ["O", "person", "O", "O", "O", "O", "O", "O", "food",
"O", "location", "O"],
"B": ["O", "O", "O", "O", "O", "O", "O", "O", "O", "O",
"movie", "movie"]
}
}, "Wrong entity prediction alignment"
def test_evaluate_entities_cv_empty_tokens():
mock_extractors = ["EntityExtractorA", "EntityExtractorB"]
result = align_entity_predictions(EN_entity_result_no_tokens,
mock_extractors)
assert result == {
"target_labels": [],
"extractor_labels": {
"EntityExtractorA": [],
"EntityExtractorB": []
},
"confidences": {
"EntityExtractorA": [],
"EntityExtractorB": []
},
}, "Wrong entity prediction alignment"