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