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
|
"""Implementations of unittest features from the future."""
import difflib, re, unittest
def _need(method):
"""Do we need to define our own `method` method?"""
return not hasattr(unittest.TestCase, method)
class TestCase(unittest.TestCase):
"""Just like unittest.TestCase, but with assert methods added.
Designed to be compatible with 3.1 unittest. Methods are only defined if
the builtin `unittest` doesn't have them.
"""
if _need('assertIn'):
def assertIn(self, member, container, msg=None):
"""Assert that `member` is in `container`."""
if member not in container:
msg = msg or ('%r not found in %r' % (member, container))
self.fail(msg)
if _need('assertNotIn'):
def assertNotIn(self, member, container, msg=None):
"""Assert that `member` is not in `container`."""
if member in container:
msg = msg or ('%r found in %r' % (member, container))
self.fail(msg)
if _need('assertGreater'):
def assertGreater(self, a, b, msg=None):
"""Assert that `a` is greater than `b`."""
if not a > b:
msg = msg or ('%r not greater than %r' % (a, b))
self.fail(msg)
if _need('assertRaisesRegexp'):
def assertRaisesRegexp(self, excClass, regexp, callobj, *args, **kw):
""" Just like unittest.TestCase.assertRaises,
but checks that the message is right too.
"""
try:
callobj(*args, **kw)
except excClass as exc:
excMsg = str(exc)
if re.search(regexp, excMsg):
# Message provided, and we got the right one: it passes.
return
else:
# Message provided, and it didn't match: fail!
raise self.failureException(
"Right exception, wrong message: "
"%r doesn't match %r" % (excMsg, regexp)
)
# No need to catch other exceptions: They'll fail the test all by
# themselves!
else:
if hasattr(excClass, '__name__'):
excName = excClass.__name__
else:
excName = str(excClass)
raise self.failureException(
"Expected to raise %s, didn't get an exception at all" %
excName
)
if _need('assertSameElements'):
def assertSameElements(self, s1, s2):
"""Assert that the two arguments are equal as sets."""
self.assertEqual(set(s1), set(s2))
if _need('assertRegexpMatches'):
def assertRegexpMatches(self, text, regex, msg=None):
"""Assert that `text` matches `regex`."""
m = re.search(regex, text)
if not m:
msg = msg or ("%r doesn't match %r" % (text, regex))
raise self.failureException(msg)
if _need('assertMultiLineEqual'):
def assertMultiLineEqual(self, first, second, msg=None):
"""Assert that two multi-line strings are equal.
If they aren't, show a nice diff.
"""
# Adapted from Py3.1 unittest.
self.assertTrue(isinstance(first, str),
'First argument is not a string')
self.assertTrue(isinstance(second, str),
'Second argument is not a string')
if first != second:
message = ''.join(difflib.ndiff(first.splitlines(True),
second.splitlines(True)))
if msg:
message += " : " + msg
self.fail("Multi-line strings are unequal:\n" + message)
|
