def postprocess(result, function): """ Clean up types for caller """ if function == 'classes_of_article': result = [util.from_wikipedia_class(c) for c in result] elif function == 'hypernyms_of_article': # flatten DBpedia hypernyms to a set types = [] for dbpedia_classes in result.values(): types += dbpedia_classes result = set(types) if 'thing' in result: # every entity has type owl:Thing # remove it form the output because it is not informative result.remove('thing') # replace dashes with spaces return [s.replace('-', ' ').lower().strip() for s in result]
def test_from_wikipedia_class(self): self.assertEqual('french-commune', util.from_wikipedia_class('wikipedia-french-commune'))
def get_hypernyms(wiki_class): wiki_class = from_wikipedia_class(wiki_class) return ontology.classes_above_infobox(wiki_class)
def test_from_wikipedia_class(self): self.assertEqual('french-commune', util.from_wikipedia_class('wikipedia-french-commune'))