def setUp(self):
self.example = GraphTaxonomy(
'entity', {
'apple': ['fruit'],
'lemon': ['citrus'],
'orange': ['citrus', 'color'],
'peach': ['fruit', 'color'],
'red': ['color'],
'yellow': ['color'],
'citrus': ['fruit'],
'fruit': ['entity'],
'color': ['entity'],
'entity': []
})
def setUp(self):
self.taxonomy = GraphTaxonomy(
'entity', {
'apple': ['fruit'],
'lemon': ['citrus'],
'orange': ['citrus', 'color'],
'peach': ['fruit', 'color'],
'red': ['color'],
'yellow': ['color'],
'citrus': ['fruit'],
'fruit': ['entity'],
'color': ['entity'],
'entity': []
})
self.wnt = WordnetTaxonomy("entity.n.01")
class TestMetrics(unittest.TestCase):
def setUp(self):
self.taxonomy = GraphTaxonomy(
'entity',
{'apple': ['fruit'],
'lemon': ['citrus'],
'orange': ['citrus', 'color'],
'peach': ['fruit', 'color'],
'red': ['color'],
'yellow': ['color'],
'citrus': ['fruit'],
'fruit': ['entity'],
'color': ['entity'],
'entity': []}
)
def test_num_instances(self):
assert self.taxonomy.num_instances() == 6
def test_flatness(self):
assert flatness(self.taxonomy, 'fruit') == 0.6
assert flatness(self.taxonomy, 'entity') == 2 / 9
def test_wu_palmer(self):
# TODO: check to make sure these numbers are correct for the
# example fruit taxonomy.
print(wu_palmer_similarity(self.taxonomy, "orange", "lemon"))
print(wu_palmer_similarity(self.taxonomy, "orange", "red"))
def test_rosenberg_descendant_sim(self):
# TODO: write tests
pass
class TestMetrics(unittest.TestCase):
def setUp(self):
self.taxonomy = GraphTaxonomy(
'entity', {
'apple': ['fruit'],
'lemon': ['citrus'],
'orange': ['citrus', 'color'],
'peach': ['fruit', 'color'],
'red': ['color'],
'yellow': ['color'],
'citrus': ['fruit'],
'fruit': ['entity'],
'color': ['entity'],
'entity': []
})
self.wnt = WordnetTaxonomy("entity.n.01")
def test_num_instances(self):
assert self.taxonomy.num_instances() == 6
def test_flatness(self):
assert flatness(self.taxonomy, 'fruit') == 5 / 2
assert flatness(self.taxonomy, 'color') == 4 / 1
assert flatness(self.taxonomy, 'entity') == 9 / 4
def test_repetitions(self):
assert repetitions(self.taxonomy, 'fruit') == 3 / 2
assert repetitions(self.taxonomy, 'color') == 3 / 2
assert repetitions(self.taxonomy, 'entity') == 8 / 6
def test_wu_palmer(self):
# TODO: check to make sure these numbers are correct for the
# example fruit taxonomy.
print(wu_palmer_similarity(self.taxonomy, "orange", "lemon", "entity"))
print(wu_palmer_similarity(self.taxonomy, "orange", "red", "entity"))
def test_rosenberg_descendant_sim(self):
rds_geological_formation = rosenberg_descendent_similarity(
self.wnt, "geological_formation.n.01")
assert rds_geological_formation < 0.57 and rds_geological_formation > 0.55
rds_dog = rosenberg_descendent_similarity(self.wnt, "dog.n.01")
assert rds_dog < 0.85 and rds_dog > 0.83
class TestTaxonomy(unittest.TestCase):
def setUp(self):
self.example = GraphTaxonomy(
'entity', {
'apple': ['fruit'],
'lemon': ['citrus'],
'orange': ['citrus', 'color'],
'peach': ['fruit', 'color'],
'red': ['color'],
'yellow': ['color'],
'citrus': ['fruit'],
'fruit': ['entity'],
'color': ['entity'],
'entity': []
})
def test_num_instances(self):
assert self.example.num_instances() == 6
def test_is_instance(self):
assert self.example.is_instance('apple')
assert self.example.is_instance('lemon')
assert self.example.is_instance('orange')
assert self.example.is_instance('peach')
assert self.example.is_instance('red')
assert self.example.is_instance('yellow')
assert not self.example.is_instance('citrus')
assert not self.example.is_instance('fruit')
assert not self.example.is_instance('color')
assert not self.example.is_instance('entity')
def test_is_category(self):
assert not self.example.is_category('apple')
assert not self.example.is_category('lemon')
assert not self.example.is_category('orange')
assert not self.example.is_category('peach')
assert not self.example.is_category('red')
assert not self.example.is_category('yellow')
assert self.example.is_category('citrus')
assert self.example.is_category('fruit')
assert self.example.is_category('color')
assert self.example.is_category('entity')
def test_get_descendant_instances(self):
result = self.example.get_descendant_instances('citrus')
assert result == {'lemon', 'orange'}
result = self.example.get_descendant_instances('color')
assert result == {'orange', 'peach', 'red', 'yellow'}
result = self.example.get_descendant_instances('fruit')
assert result == {'lemon', 'orange', 'apple', 'peach'}
def test_get_descendants(self):
result = self.example.get_descendants('fruit')
assert result == {
'lemon', 'orange', 'apple', 'peach', 'citrus', 'fruit'
}
def test_get_ancestor_categories(self):
result = self.example.get_ancestor_categories('lemon')
assert result == {'citrus', 'fruit', 'entity'}
result = self.example.get_ancestor_categories('peach')
assert result == {'color', 'fruit', 'entity'}
result = self.example.get_ancestor_categories('red')
assert result == {'color', 'entity'}
result = self.example.get_ancestor_categories('orange')
assert result == {'citrus', 'fruit', 'color', 'entity'}
result = self.example.get_ancestor_categories('citrus')
assert result == {'citrus', 'fruit', 'entity'}
def test_lowest_common_ancestor1(self):
result = lowest_common_ancestor(self.example, ['orange', 'lemon'],
'apple')
assert result == (2, 'citrus')
def test_lowest_common_ancestor2(self):
result = lowest_common_ancestor(self.example,
['orange', 'lemon', 'peach'], 'apple')
assert result == (6, 'entity')
def test_lowest_common_ancestor3(self):
result = lowest_common_ancestor(self.example, ['orange', 'peach'],
'lemon')
assert result == (4, 'color')
def test_lowest_common_ancestor4(self):
result = lowest_common_ancestor(self.example, ['orange', 'peach'],
'red')
assert result == (4, 'fruit')