def test_id_helper_3(self) : r = StringIO.StringIO("<THU>\n<MONKEY>\n<RED>\n</RED>\n</MONKEY>\n</THU>\n<THU><MONKEY></MONKEY></THU>\n") a = xml_read(r) root = a[0] id_map = {root: 1} id_tracker = [2] id_helper(root, id_tracker, id_map) self.assert_(len(id_map.items()) == 3)
def test_id_helper_2(self) : r = StringIO.StringIO("<THU>\n<Team>\n<ACRush></ACRush>\n<Jelly></Jelly>\n<Cooly></Cooly>\n</Team>\n<JiaJia>\n<Team>\n<Ahyangyi></Ahyangyi>\n<Dragon></Dragon>\n<Cooly><Amber></Amber></Cooly>\n</Team>\n</JiaJia>\n</THU>\n<Team><Cooly></Cooly></Team>\n") a = xml_read(r) root = a[0] id_map = {root: 1} id_tracker = [2] id_helper(root, id_tracker, id_map) self.assert_(len(id_map.items()) == 11)
def test_xml_traverse_3(self) : r = StringIO.StringIO("<THU>\n</THU>\n<THU></THU>\n") a = xml_read(r) root = a[0] root_query = a[1] occurrences = [0] id_list = [] query_list = [root_query] element_list = [root] first_match = [None] id_map = {root : 1} id_tracker = [2] id_helper(root, id_tracker, id_map) xml_traverse (element_list, query_list, root_query, occurrences, id_list, first_match, id_map) self.assert_(occurrences[0] == 1) self.assert_(id_list == [1])
def test_xml_traverse_1(self) : r = StringIO.StringIO("<THU>\n<Team>\n<ACRush></ACRush>\n<Jelly></Jelly>\n<Cooly></Cooly>\n</Team>\n<JiaJia>\n<Team>\n<Ahyangyi></Ahyangyi>\n<Dragon></Dragon>\n<Cooly><Amber></Amber></Cooly>\n</Team>\n</JiaJia>\n</THU>\n<Team><Cooly></Cooly></Team>\n") a = xml_read(r) root = a[0] root_query = a[1] occurrences = [0] id_list = [] query_list = [root_query] element_list = [root] first_match = [None] id_map = {root : 1} id_tracker = [2] id_helper(root, id_tracker, id_map) xml_traverse (element_list, query_list, root_query, occurrences, id_list, first_match, id_map) self.assert_(occurrences[0] == 2) self.assert_(id_list == [2, 7])