def test_read_1 (self) :
r = io.StringIO("<Lion>\n<Pens>\n<6th></6th>\n</Pens>\n</Lion>\n<Pens><6th></6th></Pens>")
a = []
b = []
d = {}
xml_read(r, a, b, d)
self.assertTrue( a == ["<Lion>", "<Pens>" , "<6th>" , "</6th>" , "</Pens>" , "</Lion>"] )
self.assertTrue( b == ["<Pens>" , "<6th>" , "</6th>" , "</Pens>" ] )
self.assertTrue( d == { "<Lion>" : [1] , "<Pens>" : [2] , "<6th>" : [3]} )
def test_read_3 (self) :
r = io.StringIO("<1>\n<2>\n<4></4>\n</2>\n<3>\n<2>\n<4></4>\n<5></5>\n</2>\n</3>\n</1>\n<2><3></3></2>")
a = []
b = []
d = {}
xml_read(r, a, b, d)
self.assertTrue( a == ['<1>', '<2>', '<4>', '</4>', '</2>', '<3>', '<2>', '<4>', '</4>', '<5>', '</5>', '</2>', '</3>', '</1>'] )
self.assertTrue( b == ['<2>', '<3>', '</3>', '</2>'] )
self.assertTrue( d == {'<3>': [4], '<1>': [1], '<4>': [3, 6], '<5>': [7], '<2>': [2, 5]} )
def test_read_2 (self) :
r = io.StringIO("<A>\n<B>\n<C>\n<D></D>\n<E></E>\n</C>\n</B>\n<C>\n<D></D>\n</C>\n</A>\n<C><D></D></C>")
a = []
b = []
d = {}
xml_read(r, a, b, d)
self.assertTrue( a == ['<A>', '<B>', '<C>', '<D>', '</D>', '<E>', '</E>', '</C>', '</B>', '<C>', '<D>', '</D>', '</C>', '</A>'] )
self.assertTrue( b == ['<C>', '<D>', '</D>', '</C>'])
self.assertTrue( d == {'<C>': [3, 6], '<A>': [1], '<D>': [4, 7], '<E>': [5], '<B>': [2]} )
def test_read_2(self):
xml = [ 0, 0 ]
r = StringIO('<THU>\n\t<Team>\n\t\t<ACRush></ACRush>\n\t\t<Jelly></Jelly>\n\t\t<Cooly></Cooly>\n\t</Team>\n\t<JiaJia>\n\t\t<Team>\n\t\t\t<Ahyangyi></Ahyangyi>\n\t\t\t<Dragon></Dragon>\n\t\t\t<Cooly><Amber></Amber></Cooly>\n\t\t</Team>\n\t</JiaJia>\n</THU>\n<Team><Cooly></Cooly></Team>\n\n<Hello><World><Bye></Bye></World><Never><Again></Again><Bye></Bye></Never></Hello>\n<World><Bye></Bye></World>')
valid = xml_read(r, xml)
self.assert_(valid)
self.assert_(xml[0].tag == 'THU')
self.assert_(xml[1].tag == 'Team')
valid = xml_read(r, xml)
self.assert_(valid)
self.assert_(xml[0].tag == 'Hello')
self.assert_(xml[1].tag == 'World')
def test_separate_singles(self):
test_list = []
r = StringIO.StringIO(singles_string)
tree = xml_read(r)
xml_separate(tree, test_list)
self.assert_(test_list[0].tag == 'First')
self.assert_(test_list[1].tag == 'Last')
def test_separate_pretty(self):
test_list = []
r = StringIO.StringIO(pretty_print_string)
tree = xml_read(r)
xml_separate(tree, test_list)
self.assert_(test_list[0].tag == 'This')
self.assert_(test_list[1].tag == 'Query')
def test_separate_default(self):
test_list = []
r = StringIO.StringIO(default_string)
tree = xml_read(r)
xml_separate(tree, test_list)
self.assert_(test_list[0].tag == 'THU')
self.assert_(test_list[1].tag == 'Team')
def test_read_3 (self) :
r = StringIO.StringIO("<e>\n<q>\n<b><i><c></c></i></b>\n</q>\n<b>\n<b><i><c></c></i></b>\n</b>\n</e>\n<b><i><c></c></i></b>")
a = [0, 0]
b = xml_read(r, a)
self.assert_(b == True)
self.assert_(a[0] == "<e>\n<q>\n<b><i><c></c></i></b>\n</q>\n<b>\n<b><i><c></c></i></b>\n</b>\n</e>\n")
self.assert_(a[1] == "<b><i><c></c></i></b>")
def test_read_2 (self) :
r = StringIO.StringIO("<x>\n<f>\n<p></p>\n</f>\n<j>\n<f>\n<p></p>\n</f>\n</j>\n<i>\n<g>\n<f>\n<p></p>\n</f>\n</g>\n</i>\n</x>\n<f><p></p></f>")
a = [0, 0]
b = xml_read(r, a)
self.assert_(b == True)
self.assert_(a[0] == "<x>\n<f>\n<p></p>\n</f>\n<j>\n<f>\n<p></p>\n</f>\n</j>\n<i>\n<g>\n<f>\n<p></p>\n</f>\n</g>\n</i>\n</x>\n")
self.assert_(a[1] == "<f><p></p></f>")
def test_read2(self):
r = StringIO.StringIO(
"<RADIOHEAD>\n\t<DOWN>\n\t\t<UP>\n\t\t\t<NO>\n\t\t\t\t<KARMA></KARMA>\n\t\t\t\t<LET></LET>\n\t\t\t</NO>\n\t\t</UP>\n\t</DOWN>\n\t<FITTER>\n\t\t<POLICE></POLICE>\n\t</FITTER>\n\t<UP>\n\t\t<AIRBAG>\n\t\t\t<HAPPIER></HAPPIER>\n\t\t</AIRBAG>\n\t</UP>\n</RADIOHEAD>\n<UP>\n\t<DOWN>\n\t</DOWN>\n</UP>\n"
)
key, block, path = xml_read(r)
self.assert_(key == "./UP")
self.assert_("".join(path) == "./DOWN")
def test_read_4(self):
r = ["<a><b></b></a>", "<c></c>", "\n", "<a></a>", "<c></c>"]
c = []
b = xml_read(r, c)
self.assert_(b == True)
self.assert_(r == ["<a></a>", "<c></c>"])
self.assert_(c == ["<a><b></b></a>", "<c></c>"])
def test_read_multiple_inputs_whitespace(self):
xml = "<one><five>\n\n\t</five>\t\t\t\t\n</one><two>\n</two>\t\n<three></three>\n\t\n<four></four>"
treeList = xml_read(xml)
self.assertEqual(treeList[0].tag, "one")
self.assertEqual(treeList[1].tag, "two")
self.assertEqual(treeList[2].tag, "three")
self.assertEqual(treeList[3].tag, "four")
def test_solve_3 (self) :
r = StringIO.StringIO("<Cards>\n<Yu>\n<Gi>\n<Oh>\n</Oh>\n</Gi>\n</Yu>" +
"\n</Cards>\n<Yu><Gi><Oh></Oh></Gi></Yu>\n")
a = xml_read(r)
l = xml_solve(a)
self.assert_(len(l) == 1)
self.assert_(l[0] == 2)
def test_solve_2 (self) :
r = StringIO.StringIO("<Thu>\n<Win>\n<Nokia>\n</Nokia>\n</Win>\n" +
"</Thu>\n<Win><Nokia></Nokia></Win>\n")
a = xml_read(r)
l = xml_solve(a)
self.assert_(len(l) == 1)
self.assert_(l[0] == 2)
def test_print_1 (self) :
w = StringIO.StringIO()
r = StringIO.StringIO("<Thu>\n<Win>\n</Win>\n</Thu>\n<Win></Win>\n")
a = xml_read(r)
l = xml_solve(a)
xml_print(w, l)
self.assert_(w.getvalue() == "1\n2\n")
def test_read_1 (self) :
r = StringIO.StringIO("")
a = [0, 0]
b = xml_read(r, a)
self.assert_(b == False)
self.assert_(a[0] == "")
self.assert_(a[1] == "")
def test_read1(self):
r = StringIO.StringIO(
"<THU>\n\t<Team>\n\t\t<ACRush></ACRush>\n\t\t<Jelly></Jelly>\n\t\t<Cooly></Cooly>\n\t</Team>\n\t<JiaJia>\n\t\t<Team>\n\t\t\t<Ahyangyi></Ahyangyi>\n\t\t\t<Dragon></Dragon>\n\t\t\t<Cooly><Amber></Amber></Cooly>\n\t\t</Team>\n\t</JiaJia>\n</THU>\n<Team><Cooly></Cooly></Team>\n"
)
key, block, path = xml_read(r)
self.assert_(key == "./Team")
self.assert_("".join(path) == "./Cooly")
def test_read_1(self):
xml = [ 0, 0 ]
valid = xml_read(StringIO('<THU>\n\t<Team>\n\t\t<ACRush></ACRush>\n\t\t<Jelly></Jelly>\n\t\t<Cooly></Cooly>\n\t</Team>\n\t<JiaJia>\n\t\t<Team>\n\t\t\t<Ahyangyi></Ahyangyi>\n\t\t\t<Dragon></Dragon>\n\t\t\t<Cooly><Amber></Amber></Cooly>\n\t\t</Team>\n\t</JiaJia>\n</THU>\n<Team><Cooly></Cooly></Team>'), xml)
self.assert_(valid)
self.assert_(xml[0].tag == "THU")
self.assert_(xml[1].tag == "Team")
def test_print_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")
w = StringIO.StringIO()
a = xml_read(r)
data = xml_eval(a)
xml_print(w, data)
self.assert_(w.getvalue() == "2\n2\n7\n\n")
def test_print_1(self) :
r = StringIO.StringIO("<THU>\n</THU>\n<THU></THU>\n")
w = StringIO.StringIO()
a = xml_read(r)
data = xml_eval(a)
xml_print(w, data)
self.assert_(w.getvalue() == "1\n1\n\n")
def test_read_2 (self) :
r = StringIO.StringIO("<a><b></b></a>\n<a></a>")
w = StringIO.StringIO()
b = xml_read(r, w)
self.assert_(b.tag == 'XML')
self.assert_(b[0][0].tag == 'b')
self.assert_(b[1].tag == 'a')
def test_read_3(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)
assert type(a[0]) is ET.Element
assert type(a[1]) is ET.Element
assert a[0].tag == "THU"
assert a[1].tag == "Team"
def test_eval_1(self):
r = StringIO.StringIO("<THU>\n</THU>\n<THU></THU>\n")
a = xml_read(r)
assert type(a[0]) is ET.Element
assert type(a[1]) is ET.Element
result = xml_eval(a)
assert(result[0][0] == 1)
def test_read_2 (self) :
try :
r = StringIO.StringIO("<Test><inner></Test></inner>")
a = xml_read(r)
assert False
except Exception, e :
pass
def test_read_2(self):
r = StringIO.StringIO("<THU>\n<Cooley>\n<THU>\n</THU>\n</Cooley>\n</THU><THU><Cooley><THU></THU></Cooley></THU>")
a = xml_read(r)
assert type(a[0]) is ET.Element
assert type(a[1]) is ET.Element
assert a[0].tag == "THU"
assert a[1].tag == "THU"
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_eval3(self):
a = []
r = StringIO.StringIO(
"<RADIOHEAD>\n\t<THE>\n\t\t<MUSIC>\n\t\t\t<OK></OK>\n\t\t</MUSIC>\n\t</THE>\n\t<TOURIST>\n\t\t<THE></THE>\n\t</TOURIST>\n\t<FITTER>\n\t\t<EXIT>\n\t\t\t<WALLS></WALLS>\n\t\t\t<TOURIST></TOURIST>\n\t\t</EXIT>\n\t\t<HOMESICK>\n\t\t\t<DOWN>\n\t\t\t\t<MUSIC></MUSIC>\n\t\t\t</DOWN>\n\t\t</HOMESICK>\n\t</FITTER>\n\t<UP>\n\t\t<CLIMBING></CLIMBING>\n\t</UP>\n</RADIOHEAD>\n<RADIOHEAD>\n\t<THE>\n\t\t<MUSIC>\n\t\t\t<OK></OK>\n\t\t</MUSIC>\n\t</THE>\n\t<TOURIST>\n\t\t<THE></THE>\n\t</TOURIST>\n\t<FITTER>\n\t\t<EXIT>\n\t\t\t<WALLS></WALLS>\n\t\t\t<TOURIST></TOURIST>\n\t\t</EXIT>\n\t\t<HOMESICK>\n\t\t\t<DOWN>\n\t\t\t\t<MUSIC></MUSIC>\n\t\t\t</DOWN>\n\t\t</HOMESICK>\n\t</FITTER>\n\t<UP>\n\t\t<CLIMBING></CLIMBING>\n\t</UP>\n</RADIOHEAD>\n"
)
key, block, path = xml_read(r)
xml_eval(a, key, block, path)
self.assert_(a == [1, 1])
def test_eval2(self):
a = []
r = StringIO.StringIO(
"<RADIOHEAD>\n\t<DOWN>\n\t\t<UP>\n\t\t\t<NO>\n\t\t\t\t<KARMA></KARMA>\n\t\t\t\t<LET></LET>\n\t\t\t</NO>\n\t\t</UP>\n\t</DOWN>\n\t<FITTER>\n\t\t<POLICE></POLICE>\n\t</FITTER>\n\t<UP>\n\t\t<AIRBAG>\n\t\t\t<HAPPIER></HAPPIER>\n\t\t</AIRBAG>\n\t</UP>\n</RADIOHEAD>\n<UP>\n\t<DOWN>\n\t</DOWN>\n</UP>\n"
)
key, block, path = xml_read(r)
xml_eval(a, key, block, path)
self.assert_(a == [0])
def test_eval1(self):
a = []
r = StringIO.StringIO(
"<THU>\n\t<Team>\n\t\t<ACRush></ACRush>\n\t\t<Jelly></Jelly>\n\t\t<Cooly></Cooly>\n\t</Team>\n\t<JiaJia>\n\t\t<Team>\n\t\t\t<Ahyangyi></Ahyangyi>\n\t\t\t<Dragon></Dragon>\n\t\t\t<Cooly><Amber></Amber></Cooly>\n\t\t</Team>\n\t</JiaJia>\n</THU>\n<Team><Cooly></Cooly></Team>\n"
)
key, block, path = xml_read(r)
xml_eval(a, key, block, path)
self.assert_(a == [2, 2, 7])
