1 import os
2 import os.path
3 import re
4 import gc
5 import sys
6 import unittest
7
8 try:
9 import urlparse
10 except ImportError:
11 import urllib.parse as urlparse
12
13 try:
14 from urllib import pathname2url
15 except:
16 from urllib.request import pathname2url
17
18 from lxml import etree, html
19
21 l = []
22 for part in re.findall('([0-9]+|[^0-9.]+)', version_string):
23 try:
24 l.append(int(part))
25 except ValueError:
26 l.append(part)
27 return tuple(l)
28
29 IS_PYPY = (getattr(sys, 'implementation', None) == 'pypy' or
30 getattr(sys, 'pypy_version_info', None) is not None)
31
32 IS_PYTHON3 = sys.version_info[0] >= 3
33 IS_PYTHON2 = sys.version_info[0] < 3
34
35 try:
36 from xml.etree import ElementTree
37 except ImportError:
38 try:
39 from elementtree import ElementTree
40 except ImportError:
41 ElementTree = None
42
43 if hasattr(ElementTree, 'VERSION'):
44 ET_VERSION = make_version_tuple(ElementTree.VERSION)
45 else:
46 ET_VERSION = (0,0,0)
47
48 try:
49 from xml.etree import cElementTree
50 except ImportError:
51 try:
52 import cElementTree
53 except ImportError:
54 cElementTree = None
55
56 if hasattr(cElementTree, 'VERSION'):
57 CET_VERSION = make_version_tuple(cElementTree.VERSION)
58 else:
59 CET_VERSION = (0,0,0)
60
62 """Remove test methods that do not work with the current lib version.
63 """
64 find_required_version = version_dict.get
65 def dummy_test_method(self):
66 pass
67 for name in dir(test_class):
68 expected_version = find_required_version(name, (0,0,0))
69 if expected_version > current_version:
70 setattr(test_class, name, dummy_test_method)
71
72 try:
73 import doctest
74
75 doctest.DocFileSuite
76 doctest.DocTestParser
77 doctest.NORMALIZE_WHITESPACE
78 doctest.ELLIPSIS
79 except (ImportError, AttributeError):
80
81 import local_doctest as doctest
82
83 try:
84 sorted
85 except NameError:
87 seq = list(seq)
88 seq.sort(**kwargs)
89 return seq
90 else:
91 locals()['sorted'] = sorted
92
93
94 try:
95 next
96 except NameError:
99 else:
100 locals()['next'] = next
101
102
103 try:
104 import pytest
105 except ImportError:
107 "Using a class because a function would bind into a method when used in classes"
109 - def __call__(self, func, *args): return func
110 else:
111 skipif = pytest.mark.skipif
112
114 module = sys.modules[sys._getframe(frame_depth).f_globals['__name__']]
115 return os.path.normpath(os.path.join(
116 os.path.dirname(getattr(module, '__file__', '')), filename))
117
118 from io import StringIO
119
120 if sys.version_info[0] >= 3:
121
122 from builtins import str as unicode
123 _chr = chr
124 - def _str(s, encoding="UTF-8"):
126 - def _bytes(s, encoding="UTF-8"):
128 from io import BytesIO as _BytesIO
130 if args and isinstance(args[0], str):
131 args = (args[0].encode("UTF-8"),)
132 return _BytesIO(*args)
133
134 doctest_parser = doctest.DocTestParser()
135 _fix_unicode = re.compile(r'(\s+)u(["\'])').sub
136 _fix_exceptions = re.compile(r'(.*except [^(]*),\s*(.*:)').sub
145 else:
146
147 unichr_escape = re.compile(r'\\u[0-9a-fA-F]{4}|\\U[0-9a-fA-F]{8}')
148
149 from __builtin__ import unicode
150 _chr = unichr
151 - def _str(s, encoding="UTF-8"):
156 - def _bytes(s, encoding="UTF-8"):
158 from io import BytesIO
159
160 doctest_parser = doctest.DocTestParser()
161 _fix_traceback = re.compile(r'^(\s*)(?:\w+\.)+(\w*(?:Error|Exception|Invalid):)', re.M).sub
162 _fix_exceptions = re.compile(r'(.*except [^(]*)\s+as\s+(.*:)').sub
163 _fix_bytes = re.compile(r'(\s+)b(["\'])').sub
173
174 try:
175 skipIf = unittest.skipIf
176 except AttributeError:
177 - def skipIf(condition, why,
178 _skip=lambda test_method: None,
179 _keep=lambda test_method: test_method):
180 if condition:
181 return _skip
182 return _keep
183
184
188
189 - def parse(self, text, parser=None):
192
196
197
198 try:
199 unittest.TestCase.assertFalse
200 except AttributeError:
201 assertFalse = unittest.TestCase.failIf
202
203
206 self.xml_data = xml_data
207
208 - def read(self, amount=None):
209 if self.xml_data:
210 if amount:
211 data = self.xml_data[:amount]
212 self.xml_data = self.xml_data[amount:]
213 else:
214 data = self.xml_data
215 self.xml_data = _bytes('')
216 return data
217 return _bytes('')
218
220 - def __init__(self, charlen=100, depth=4, children=5):
221 self.data = BytesIO()
222 self.chars = _bytes('a') * charlen
223 self.children = range(children)
224 self.more = self.iterelements(depth)
225
227 yield _bytes('<root>')
228 depth -= 1
229 if depth > 0:
230 for child in self.children:
231 for element in self.iterelements(depth):
232 yield element
233 yield self.chars
234 else:
235 yield self.chars
236 yield _bytes('</root>')
237
238 - def read(self, amount=None):
256
258 - def __init__(self, charlen=100, depth=4, children=5):
263
265 yield _str('<root>')
266 depth -= 1
267 if depth > 0:
268 for child in self.children:
269 for element in self.iterelements(depth):
270 yield element
271 yield self.chars
272 else:
273 yield self.chars
274 yield _str('</root>')
275
277 _testdir = os.path.dirname(__file__)
278 return os.path.join(_testdir, name)
279
281 return urlparse.urljoin(
282 'file:', pathname2url(path))
283
286
294
301
304
306 tree = etree.parse(BytesIO(xml) if isinstance(xml, bytes) else StringIO(xml))
307 f = BytesIO()
308 tree.write_c14n(f)
309 return f.getvalue()
310
315