forked from cs327e-spring-2014/xml-tests
-
Notifications
You must be signed in to change notification settings - Fork 0
/
mw24859-TestXML.py
169 lines (138 loc) · 4.61 KB
/
mw24859-TestXML.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
#!/usr/bin/env python
# -------------------------------
# projects/xml/XML.py
# Copyright (C) 2014
# Matt Woods, Carvey Leung
# -------------------------------
"""
To test the program:
% python TestXML.py >& TestXML.out
% chmod ugo+x TestXML.py
% TestXML.py >& TestXML.out
"""
import StringIO
import unittest
from xml.etree.ElementTree import Element, fromstring
from XML import xml_read, xml_search, xml_children, xml_print, xml_solve
import XML
class TestXML(unittest.TestCase):
# ----
# read
# ----
def test_xml_read_1 (self) :
s = ''
r = StringIO.StringIO("<THU><Team><ACRush></ACRush><Jelly></Jelly><Cooly></Cooly></Team></THU><Team><Cooly></Cooly></Team>")
xml_read(r)
self.assertTrue(type(s) == str)
def test_xml_read_2 (self) :
xml = fromstring("<Team></Team>")
r = StringIO.StringIO("<THU><Team><ACRush></ACRush><Jelly></Jelly><Cooly></Cooly></Team></THU><Team><Cooly></Cooly></Team>")
xml_read(r)
self.assertTrue(type(xml) == Element)
def test_xml_read_3 (self) :
p = fromstring("<THU></THU>")
r = StringIO.StringIO("<THU><Team><ACRush></ACRush><Jelly></Jelly><Cooly></Cooly></Team></THU><Team><Cooly></Cooly></Team>")
xml_read(r)
self.assertTrue(type(p) == Element)
# ------
# search
# ------
def test_xml_search_1 (self) :
XML.index = 1
l = []
xml = fromstring("<xml><a></a></xml>")
p = fromstring("<a></a>")
d = ["a"]
xml_search(xml, p, d, l)
self.assertTrue(l[0] == 2)
self.assertTrue(len(l) == 1)
def test_xml_search_2 (self) :
XML.index = 1
l = []
xml = fromstring("<xml><a><dog></dog></a></xml>")
p = fromstring("<a></a>")
d = ["a"]
xml_search(xml, p, d, l)
self.assertTrue(l[0] == 2)
self.assertTrue(len(l) == 1)
def test_xml_search_3 (self) :
XML.index = 1
l = []
xml = fromstring("<xml><a></a><dog></dog></xml>")
p = fromstring("<dog></dog>")
d = ["dog"]
xml_search(xml, p, d, l)
self.assertTrue(l[0] == 3)
self.assertTrue(len(l) == 1)
# --------
# children
# --------
def test_xml_children_1 (self) :
index = 1
l = []
length = 1
i = fromstring("<xml><a></a></xml>")
p = fromstring("<xml><a></a></xml>")
d = ["xml","a"]
xml_children(i, p, index, length, d, l)
self.assertTrue(l[0] == 1)
self.assertTrue(len(l) == 1)
def test_xml_children_1 (self) :
index = 1
l = []
length = 1
i = fromstring("<xml><a></a></xml>")
p = fromstring("<xml><a></a></xml>")
d = ["xml","a"]
xml_children(i, p, index, length, d, l)
self.assertTrue(l[0] == 1)
self.assertTrue(len(l) == 1)
def test_xml_children_1 (self) :
index = 1
l = []
length = 1
i = fromstring("<xml><a></a></xml>")
p = fromstring("<xml><a></a></xml>")
d = ["xml","a"]
xml_children(i, p, index, length, d, l)
self.assertTrue(l[0] == 1)
self.assertTrue(len(l) == 1)
# -----
# print
# -----
def test_xml_print_1 (self) :
w = StringIO.StringIO()
l = [3, 4, 5]
xml_print(l, w)
self.assertTrue(w.getvalue() == "3\n3\n4\n5")
def test_xml_print_2 (self) :
w = StringIO.StringIO()
l = [7, 20, 398]
xml_print(l, w)
self.assertTrue(w.getvalue() == "3\n7\n20\n398")
def test_xml_print_3 (self) :
w = StringIO.StringIO()
l = [9, 90, 178, 999]
xml_print(l, w)
self.assertTrue(w.getvalue() == "4\n9\n90\n178\n999")
# -----
# solve
# -----
def test_xml_solve_1 (self) :
r = StringIO.StringIO("<xml><a></a></xml><RUN></RUN>")
w = StringIO.StringIO()
xml_solve(r, w)
self.assertTrue(w.getvalue() == "0")
def test_xml_solve_2 (self) :
r = StringIO.StringIO("<THU><Team><ACRush></ACRush><Jelly></Jelly><Cooly></Cooly></Team><JiaJia><Team><Ahyangyi></Ahyangyi><Dragon></Dragon><Cooly><Amber></Amber></Cooly></Team></JiaJia></THU><Team><Cooly></Cooly></Team>")
w = StringIO.StringIO()
xml_solve(r, w)
self.assertTrue(w.getvalue() == "2\n2\n7")
def test_xml_solve_3 (self) :
r = StringIO.StringIO("<THU><Team><ACRush></ACRush><Jelly></Jelly><Cooly></Cooly></Team></THU><Team><Cooly></Cooly></Team>")
w = StringIO.StringIO()
xml_solve(r, w)
self.assertTrue(w.getvalue() == "1\n2")
print "TestXML.py"
unittest.main()
print "Done"