def test_tag_expansion_full(self):
"""End result of tag expansion."""
# check empty
empty = KPA([], [], [], [], [])
self.assert_eq_rec([], empty.calc_full_tagdag())
#check sample
expected = [
{'catID':0, 'setID':0, 'tagID':1},
{'catID':0, 'setID':5, 'tagID':1},
{'catID':0, 'setID':0, 'tagID':2},
{'catID':0, 'setID':1, 'tagID':2},
{'catID':0, 'setID':5, 'tagID':2},
{'catID':0, 'setID':0, 'tagID':3},
{'catID':0, 'setID':1, 'tagID':3},
{'catID':0, 'setID':2, 'tagID':3},
{'catID':0, 'setID':5, 'tagID':3},
{'catID':0, 'setID':0, 'tagID':4},
{'catID':0, 'setID':1, 'tagID':4},
{'catID':0, 'setID':2, 'tagID':4},
{'catID':0, 'setID':3, 'tagID':4},
{'catID':0, 'setID':5, 'tagID':4},
{'catID':1, 'setID':0, 'tagID':1},
{'catID':2, 'setID':0, 'tagID':1},
{'catID':2, 'setID':2, 'tagID':1},
]
sample = KPA([], [], [], [], tagdag = self.sample_input_tagdag)
actual = sample.calc_full_tagdag()
self.assertEqual(self.comparableTagdag(expected),
self.comparableTagdag(actual))
def test_tag_expansion_intermediate(self):
"""Checking the intermediate form of expansion."""
# check empty
self.assert_eq_rec({}, KPA.tag_closure([]))
#check sample
expected = {
0:{
0:set([]),
1:set([0,5]),
2:set([0,1,5]),
3:set([0,1,2,5]),
4:set([0,1,2,3,5]),
5:set([])
},
1:{
0:set([]),
1:set([0])
},
2:{
0:set([]),
1:set([0,2]),
2:set([])
}}
actual = KPA.tag_closure(self.sample_input_tagdag)
self.assertEqual(self.comparableClosure(expected),
self.comparableClosure(actual))
