def test_full_align(self): ra = RecordAligner() records = [Record(t1), Record(t2), Record(t3)] seed, mappings = ra.align(records) self.assertEqual(3, len(mappings)) # all the elements from seed should matched to other 2 trees for tag in ['root', 'a', 'b', 'c']: e = seed[0].xpath('//%s' % tag)[0] expected = [] for record, mapping in mappings.items(): expected.append(mapping[e].tag) self.assertEqual([tag] * 3, expected)
def test_full_align(self): ra = RecordAligner() records = [Record(t1), Record(t2), Record(t3)] seed, mappings = ra.align(records) self.assertEqual(3, len(mappings)) # all the elements from seed should matched to other 2 trees for tag in ['root', 'a', 'b', 'c']: e = seed[0].xpath('//%s' % tag)[0] expected = [] for record, mapping in mappings.iteritems(): expected.append(mapping[e].tag) self.assertEqual([tag] * 3, expected)
def test_align_with_record(self): ra = RecordAligner() seed_record = Record(t4) records = [Record(t1), Record(t2), Record(t3)] seed, mappings = ra.align(records, seed_record) self.assertEqual(4, len(mappings)) # all the elements from seed should matched to other 3 trees for tag in ['root', 'a', 'b', 'c']: root = seed[0].xpath('//%s' % tag)[0] expected = [] for record, mapping in mappings.items(): if seed_record == record: continue expected.append(mapping[root].tag) self.assertEqual([tag] * 3, expected)
def test_align_with_record(self): ra = RecordAligner() seed_record = Record(t4) records = [Record(t1), Record(t2), Record(t3)] seed, mappings = ra.align(records, seed_record) self.assertEqual(4, len(mappings)) # all the elements from seed should matched to other 3 trees for tag in ['root', 'a', 'b', 'c']: root = seed[0].xpath('//%s' % tag)[0] expected = [] for record, mapping in mappings.iteritems(): if seed_record == record: continue expected.append(mapping[root].tag) self.assertEqual([tag] * 3, expected)