def main(lines):
named_entities_binary, named_entity_coordinates_binary = detect_named_entities(lines)
named_entities_nonbinary, named_entity_coordinates_nonbinary, named_entity_types_nonbinary = detect_named_entities_and_types(lines)
clustered_named_entities_binary = link_named_entities(named_entities_binary, named_entity_coordinates_binary)
named_entity_types_binary = cherry_pick_entity_class_from_nonbinary(clustered_named_entities_binary, named_entities_nonbinary, named_entity_types_nonbinary)
return clustered_named_entities_binary, named_entity_types_binary
def test_detect_named_entites(self):
named_entities, named_entity_coordinates = detection.detect_named_entities(self.lines)
self.assertEqual(named_entities, ['Einhorn', 'Finkle', 'Finkle', 'Einhorn'])
self.assertEqual(named_entity_coordinates, [('1.1', '1.7'), ('1.12', '1.17'), ('1.24', '1.29'), ('1.34', '1.40')])
