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():
    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'