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))