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
|
# coding: utf-8
"""
unitest generic utils
~~~~~~~~~~~~~~~~~~~~~
Generic utils useable for a markup test.
:copyleft: 2008-2011 by python-creole team, see AUTHORS for more details.
:license: GNU GPL v3 or above, see LICENSE for more details.
"""
import difflib
import textwrap
import unittest
## error output format:
# =1 -> via repr()
# =2 -> raw
VERBOSE = 1
#VERBOSE = 2
def make_diff(block1, block2):
d = difflib.Differ()
block1 = block1.replace("\\n", "\\n\n").split("\n")
block2 = block2.replace("\\n", "\\n\n").split("\n")
diff = d.compare(block1, block2)
result = ["%2s %s\n" % (line, i) for line, i in enumerate(diff)]
return "".join(result)
class MarkupTest(unittest.TestCase):
"""
Special error class: Try to display markup errors in a better way.
"""
def _format_output(self, txt):
txt = txt.split("\\n")
if VERBOSE == 1:
txt = "".join(['%s\\n\n' % i for i in txt])
elif VERBOSE == 2:
txt = "".join(['%s\n' % i for i in txt])
return txt
def assertEqual(self, first, second, msg=""):
if not first == second:
if VERBOSE >= 2:
print("first: %r" % first)
print("second: %r" % second)
#~ first = first.rstrip("\\n")
#~ second = second.rstrip("\\n")
try:
diff = make_diff(first, second)
except AttributeError:
raise self.failureException("%s is not %s" % (repr(first), repr(second)))
if VERBOSE >= 2:
print("diff: %r" % diff)
first = self._format_output(first)
second = self._format_output(second)
msg += (
"\n---[Output:]---\n%s\n"
"---[not equal to:]---\n%s"
"\n---[diff:]---\n%s"
) % (first, second, diff)
raise self.failureException(msg)
def _prepare_text(self, txt):
"""
prepare the multiline, indentation text.
"""
txt = txt.splitlines()
assert txt[0] == "", "First assertion line must be empty! Is: %s" % repr(txt[0])
txt = txt[1:] # Skip the first line
# get the indentation level from the first line
count = False
for count, char in enumerate(txt[0]):
if char != " ":
break
assert count != False, "second line is empty!"
# remove indentation from all lines
txt = [i[count:].rstrip(" ") for i in txt]
#~ txt = re.sub("\n {2,}", "\n", txt)
txt = "\n".join(txt)
# strip *one* newline at the begining...
if txt.startswith("\n"): txt = txt[1:]
# and strip *one* newline at the end of the text
if txt.endswith("\n"): txt = txt[:-1]
#~ print(repr(txt))
#~ print("-"*79)
return txt
def testSelf(self):
"""
Test for self._prepare_text()
"""
out1 = self._prepare_text("""
one line
line two""")
self.assertEqual(out1, "one line\nline two")
out2 = self._prepare_text("""
one line
line two
""")
self.assertEqual(out2, "one line\nline two")
out3 = self._prepare_text("""
one line
line two
""")
self.assertEqual(out3, "one line\n\nline two")
out4 = self._prepare_text("""
one line
line two
""")
self.assertEqual(out4, "one line\n line two\n")
# removing whitespace and the end
self.assertEqual(self._prepare_text("\n 111 \n 222"), "111\n222")
out5 = self._prepare_text("""
one line
line two
dritte Zeile
""")
self.assertEqual(out5, "one line\n line two\ndritte Zeile")
self.assertRaises(
AssertionError, self.assertEqual, "foo", "bar"
)
if __name__ == '__main__':
import doctest
print("DocTest:", doctest.testmod())
unittest.main()
|
