Python code coverage for Lib/unittest/test/test_assertions.py

#	count	content
1	n/a	import datetime
2	n/a	import warnings
3	n/a	import weakref
4	n/a	import unittest
5	n/a	from itertools import product
6	n/a
7	n/a
8	n/a	class Test_Assertions(unittest.TestCase):
9	n/a	def test_AlmostEqual(self):
10	n/a	self.assertAlmostEqual(1.00000001, 1.0)
11	n/a	self.assertNotAlmostEqual(1.0000001, 1.0)
12	n/a	self.assertRaises(self.failureException,
13	n/a	self.assertAlmostEqual, 1.0000001, 1.0)
14	n/a	self.assertRaises(self.failureException,
15	n/a	self.assertNotAlmostEqual, 1.00000001, 1.0)
16	n/a
17	n/a	self.assertAlmostEqual(1.1, 1.0, places=0)
18	n/a	self.assertRaises(self.failureException,
19	n/a	self.assertAlmostEqual, 1.1, 1.0, places=1)
20	n/a
21	n/a	self.assertAlmostEqual(0, .1+.1j, places=0)
22	n/a	self.assertNotAlmostEqual(0, .1+.1j, places=1)
23	n/a	self.assertRaises(self.failureException,
24	n/a	self.assertAlmostEqual, 0, .1+.1j, places=1)
25	n/a	self.assertRaises(self.failureException,
26	n/a	self.assertNotAlmostEqual, 0, .1+.1j, places=0)
27	n/a
28	n/a	self.assertAlmostEqual(float('inf'), float('inf'))
29	n/a	self.assertRaises(self.failureException, self.assertNotAlmostEqual,
30	n/a	float('inf'), float('inf'))
31	n/a
32	n/a	def test_AmostEqualWithDelta(self):
33	n/a	self.assertAlmostEqual(1.1, 1.0, delta=0.5)
34	n/a	self.assertAlmostEqual(1.0, 1.1, delta=0.5)
35	n/a	self.assertNotAlmostEqual(1.1, 1.0, delta=0.05)
36	n/a	self.assertNotAlmostEqual(1.0, 1.1, delta=0.05)
37	n/a
38	n/a	self.assertAlmostEqual(1.0, 1.0, delta=0.5)
39	n/a	self.assertRaises(self.failureException, self.assertNotAlmostEqual,
40	n/a	1.0, 1.0, delta=0.5)
41	n/a
42	n/a	self.assertRaises(self.failureException, self.assertAlmostEqual,
43	n/a	1.1, 1.0, delta=0.05)
44	n/a	self.assertRaises(self.failureException, self.assertNotAlmostEqual,
45	n/a	1.1, 1.0, delta=0.5)
46	n/a
47	n/a	self.assertRaises(TypeError, self.assertAlmostEqual,
48	n/a	1.1, 1.0, places=2, delta=2)
49	n/a	self.assertRaises(TypeError, self.assertNotAlmostEqual,
50	n/a	1.1, 1.0, places=2, delta=2)
51	n/a
52	n/a	first = datetime.datetime.now()
53	n/a	second = first + datetime.timedelta(seconds=10)
54	n/a	self.assertAlmostEqual(first, second,
55	n/a	delta=datetime.timedelta(seconds=20))
56	n/a	self.assertNotAlmostEqual(first, second,
57	n/a	delta=datetime.timedelta(seconds=5))
58	n/a
59	n/a	def test_assertRaises(self):
60	n/a	def _raise(e):
61	n/a	raise e
62	n/a	self.assertRaises(KeyError, _raise, KeyError)
63	n/a	self.assertRaises(KeyError, _raise, KeyError("key"))
64	n/a	try:
65	n/a	self.assertRaises(KeyError, lambda: None)
66	n/a	except self.failureException as e:
67	n/a	self.assertIn("KeyError not raised", str(e))
68	n/a	else:
69	n/a	self.fail("assertRaises() didn't fail")
70	n/a	try:
71	n/a	self.assertRaises(KeyError, _raise, ValueError)
72	n/a	except ValueError:
73	n/a	pass
74	n/a	else:
75	n/a	self.fail("assertRaises() didn't let exception pass through")
76	n/a	with self.assertRaises(KeyError) as cm:
77	n/a	try:
78	n/a	raise KeyError
79	n/a	except Exception as e:
80	n/a	exc = e
81	n/a	raise
82	n/a	self.assertIs(cm.exception, exc)
83	n/a
84	n/a	with self.assertRaises(KeyError):
85	n/a	raise KeyError("key")
86	n/a	try:
87	n/a	with self.assertRaises(KeyError):
88	n/a	pass
89	n/a	except self.failureException as e:
90	n/a	self.assertIn("KeyError not raised", str(e))
91	n/a	else:
92	n/a	self.fail("assertRaises() didn't fail")
93	n/a	try:
94	n/a	with self.assertRaises(KeyError):
95	n/a	raise ValueError
96	n/a	except ValueError:
97	n/a	pass
98	n/a	else:
99	n/a	self.fail("assertRaises() didn't let exception pass through")
100	n/a
101	n/a	def test_assertRaises_frames_survival(self):
102	n/a	# Issue #9815: assertRaises should avoid keeping local variables
103	n/a	# in a traceback alive.
104	n/a	class A:
105	n/a	pass
106	n/a	wr = None
107	n/a
108	n/a	class Foo(unittest.TestCase):
109	n/a
110	n/a	def foo(self):
111	n/a	nonlocal wr
112	n/a	a = A()
113	n/a	wr = weakref.ref(a)
114	n/a	try:
115	n/a	raise IOError
116	n/a	except IOError:
117	n/a	raise ValueError
118	n/a
119	n/a	def test_functional(self):
120	n/a	self.assertRaises(ValueError, self.foo)
121	n/a
122	n/a	def test_with(self):
123	n/a	with self.assertRaises(ValueError):
124	n/a	self.foo()
125	n/a
126	n/a	Foo("test_functional").run()
127	n/a	self.assertIsNone(wr())
128	n/a	Foo("test_with").run()
129	n/a	self.assertIsNone(wr())
130	n/a
131	n/a	def testAssertNotRegex(self):
132	n/a	self.assertNotRegex('Ala ma kota', r'r+')
133	n/a	try:
134	n/a	self.assertNotRegex('Ala ma kota', r'k.t', 'Message')
135	n/a	except self.failureException as e:
136	n/a	self.assertIn('Message', e.args[0])
137	n/a	else:
138	n/a	self.fail('assertNotRegex should have failed.')
139	n/a
140	n/a
141	n/a	class TestLongMessage(unittest.TestCase):
142	n/a	"""Test that the individual asserts honour longMessage.
143	n/a	This actually tests all the message behaviour for
144	n/a	asserts that use longMessage."""
145	n/a
146	n/a	def setUp(self):
147	n/a	class TestableTestFalse(unittest.TestCase):
148	n/a	longMessage = False
149	n/a	failureException = self.failureException
150	n/a
151	n/a	def testTest(self):
152	n/a	pass
153	n/a
154	n/a	class TestableTestTrue(unittest.TestCase):
155	n/a	longMessage = True
156	n/a	failureException = self.failureException
157	n/a
158	n/a	def testTest(self):
159	n/a	pass
160	n/a
161	n/a	self.testableTrue = TestableTestTrue('testTest')
162	n/a	self.testableFalse = TestableTestFalse('testTest')
163	n/a
164	n/a	def testDefault(self):
165	n/a	self.assertTrue(unittest.TestCase.longMessage)
166	n/a
167	n/a	def test_formatMsg(self):
168	n/a	self.assertEqual(self.testableFalse._formatMessage(None, "foo"), "foo")
169	n/a	self.assertEqual(self.testableFalse._formatMessage("foo", "bar"), "foo")
170	n/a
171	n/a	self.assertEqual(self.testableTrue._formatMessage(None, "foo"), "foo")
172	n/a	self.assertEqual(self.testableTrue._formatMessage("foo", "bar"), "bar : foo")
173	n/a
174	n/a	# This blows up if _formatMessage uses string concatenation
175	n/a	self.testableTrue._formatMessage(object(), 'foo')
176	n/a
177	n/a	def test_formatMessage_unicode_error(self):
178	n/a	one = ''.join(chr(i) for i in range(255))
179	n/a	# this used to cause a UnicodeDecodeError constructing msg
180	n/a	self.testableTrue._formatMessage(one, '\uFFFD')
181	n/a
182	n/a	def assertMessages(self, methodName, args, errors):
183	n/a	"""
184	n/a	Check that methodName(*args) raises the correct error messages.
185	n/a	errors should be a list of 4 regex that match the error when:
186	n/a	1) longMessage = False and no msg passed;
187	n/a	2) longMessage = False and msg passed;
188	n/a	3) longMessage = True and no msg passed;
189	n/a	4) longMessage = True and msg passed;
190	n/a	"""
191	n/a	def getMethod(i):
192	n/a	useTestableFalse = i < 2
193	n/a	if useTestableFalse:
194	n/a	test = self.testableFalse
195	n/a	else:
196	n/a	test = self.testableTrue
197	n/a	return getattr(test, methodName)
198	n/a
199	n/a	for i, expected_regex in enumerate(errors):
200	n/a	testMethod = getMethod(i)
201	n/a	kwargs = {}
202	n/a	withMsg = i % 2
203	n/a	if withMsg:
204	n/a	kwargs = {"msg": "oops"}
205	n/a
206	n/a	with self.assertRaisesRegex(self.failureException,
207	n/a	expected_regex=expected_regex):
208	n/a	testMethod(args, *kwargs)
209	n/a
210	n/a	def testAssertTrue(self):
211	n/a	self.assertMessages('assertTrue', (False,),
212	n/a	["^False is not true$", "^oops$", "^False is not true$",
213	n/a	"^False is not true : oops$"])
214	n/a
215	n/a	def testAssertFalse(self):
216	n/a	self.assertMessages('assertFalse', (True,),
217	n/a	["^True is not false$", "^oops$", "^True is not false$",
218	n/a	"^True is not false : oops$"])
219	n/a
220	n/a	def testNotEqual(self):
221	n/a	self.assertMessages('assertNotEqual', (1, 1),
222	n/a	["^1 == 1$", "^oops$", "^1 == 1$",
223	n/a	"^1 == 1 : oops$"])
224	n/a
225	n/a	def testAlmostEqual(self):
226	n/a	self.assertMessages('assertAlmostEqual', (1, 2),
227	n/a	["^1 != 2 within 7 places$", "^oops$",
228	n/a	"^1 != 2 within 7 places$", "^1 != 2 within 7 places : oops$"])
229	n/a
230	n/a	def testNotAlmostEqual(self):
231	n/a	self.assertMessages('assertNotAlmostEqual', (1, 1),
232	n/a	["^1 == 1 within 7 places$", "^oops$",
233	n/a	"^1 == 1 within 7 places$", "^1 == 1 within 7 places : oops$"])
234	n/a
235	n/a	def test_baseAssertEqual(self):
236	n/a	self.assertMessages('_baseAssertEqual', (1, 2),
237	n/a	["^1 != 2$", "^oops$", "^1 != 2$", "^1 != 2 : oops$"])
238	n/a
239	n/a	def testAssertSequenceEqual(self):
240	n/a	# Error messages are multiline so not testing on full message
241	n/a	# assertTupleEqual and assertListEqual delegate to this method
242	n/a	self.assertMessages('assertSequenceEqual', ([], [None]),
243	n/a	[r"\+ \[None\]$", "^oops$", r"\+ \[None\]$",
244	n/a	r"\+ \[None\] : oops$"])
245	n/a
246	n/a	def testAssertSetEqual(self):
247	n/a	self.assertMessages('assertSetEqual', (set(), set([None])),
248	n/a	["None$", "^oops$", "None$",
249	n/a	"None : oops$"])
250	n/a
251	n/a	def testAssertIn(self):
252	n/a	self.assertMessages('assertIn', (None, []),
253	n/a	[r'^None not found in \[\]$', "^oops$",
254	n/a	r'^None not found in \[\]$',
255	n/a	r'^None not found in \[\] : oops$'])
256	n/a
257	n/a	def testAssertNotIn(self):
258	n/a	self.assertMessages('assertNotIn', (None, [None]),
259	n/a	[r'^None unexpectedly found in \[None\]$', "^oops$",
260	n/a	r'^None unexpectedly found in \[None\]$',
261	n/a	r'^None unexpectedly found in \[None\] : oops$'])
262	n/a
263	n/a	def testAssertDictEqual(self):
264	n/a	self.assertMessages('assertDictEqual', ({}, {'key': 'value'}),
265	n/a	[r"\+ \{'key': 'value'\}$", "^oops$",
266	n/a	r"\+ \{'key': 'value'\}$",
267	n/a	r"\+ \{'key': 'value'\} : oops$"])
268	n/a
269	n/a	def testAssertDictContainsSubset(self):
270	n/a	with warnings.catch_warnings():
271	n/a	warnings.simplefilter("ignore", DeprecationWarning)
272	n/a
273	n/a	self.assertMessages('assertDictContainsSubset', ({'key': 'value'}, {}),
274	n/a	["^Missing: 'key'$", "^oops$",
275	n/a	"^Missing: 'key'$",
276	n/a	"^Missing: 'key' : oops$"])
277	n/a
278	n/a	def testAssertMultiLineEqual(self):
279	n/a	self.assertMessages('assertMultiLineEqual', ("", "foo"),
280	n/a	[r"\+ foo$", "^oops$",
281	n/a	r"\+ foo$",
282	n/a	r"\+ foo : oops$"])
283	n/a
284	n/a	def testAssertLess(self):
285	n/a	self.assertMessages('assertLess', (2, 1),
286	n/a	["^2 not less than 1$", "^oops$",
287	n/a	"^2 not less than 1$", "^2 not less than 1 : oops$"])
288	n/a
289	n/a	def testAssertLessEqual(self):
290	n/a	self.assertMessages('assertLessEqual', (2, 1),
291	n/a	["^2 not less than or equal to 1$", "^oops$",
292	n/a	"^2 not less than or equal to 1$",
293	n/a	"^2 not less than or equal to 1 : oops$"])
294	n/a
295	n/a	def testAssertGreater(self):
296	n/a	self.assertMessages('assertGreater', (1, 2),
297	n/a	["^1 not greater than 2$", "^oops$",
298	n/a	"^1 not greater than 2$",
299	n/a	"^1 not greater than 2 : oops$"])
300	n/a
301	n/a	def testAssertGreaterEqual(self):
302	n/a	self.assertMessages('assertGreaterEqual', (1, 2),
303	n/a	["^1 not greater than or equal to 2$", "^oops$",
304	n/a	"^1 not greater than or equal to 2$",
305	n/a	"^1 not greater than or equal to 2 : oops$"])
306	n/a
307	n/a	def testAssertIsNone(self):
308	n/a	self.assertMessages('assertIsNone', ('not None',),
309	n/a	["^'not None' is not None$", "^oops$",
310	n/a	"^'not None' is not None$",
311	n/a	"^'not None' is not None : oops$"])
312	n/a
313	n/a	def testAssertIsNotNone(self):
314	n/a	self.assertMessages('assertIsNotNone', (None,),
315	n/a	["^unexpectedly None$", "^oops$",
316	n/a	"^unexpectedly None$",
317	n/a	"^unexpectedly None : oops$"])
318	n/a
319	n/a	def testAssertIs(self):
320	n/a	self.assertMessages('assertIs', (None, 'foo'),
321	n/a	["^None is not 'foo'$", "^oops$",
322	n/a	"^None is not 'foo'$",
323	n/a	"^None is not 'foo' : oops$"])
324	n/a
325	n/a	def testAssertIsNot(self):
326	n/a	self.assertMessages('assertIsNot', (None, None),
327	n/a	["^unexpectedly identical: None$", "^oops$",
328	n/a	"^unexpectedly identical: None$",
329	n/a	"^unexpectedly identical: None : oops$"])
330	n/a
331	n/a	def testAssertRegex(self):
332	n/a	self.assertMessages('assertRegex', ('foo', 'bar'),
333	n/a	["^Regex didn't match:",
334	n/a	"^oops$",
335	n/a	"^Regex didn't match:",
336	n/a	"^Regex didn't match: (.*) : oops$"])
337	n/a
338	n/a	def testAssertNotRegex(self):
339	n/a	self.assertMessages('assertNotRegex', ('foo', 'foo'),
340	n/a	["^Regex matched:",
341	n/a	"^oops$",
342	n/a	"^Regex matched:",
343	n/a	"^Regex matched: (.*) : oops$"])
344	n/a
345	n/a
346	n/a	def assertMessagesCM(self, methodName, args, func, errors):
347	n/a	"""
348	n/a	Check that the correct error messages are raised while executing:
349	n/a	with method(*args):
350	n/a	func()
351	n/a	errors should be a list of 4 regex that match the error when:
352	n/a	1) longMessage = False and no msg passed;
353	n/a	2) longMessage = False and msg passed;
354	n/a	3) longMessage = True and no msg passed;
355	n/a	4) longMessage = True and msg passed;
356	n/a	"""
357	n/a	p = product((self.testableFalse, self.testableTrue),
358	n/a	({}, {"msg": "oops"}))
359	n/a	for (cls, kwargs), err in zip(p, errors):
360	n/a	method = getattr(cls, methodName)
361	n/a	with self.assertRaisesRegex(cls.failureException, err):
362	n/a	with method(args, *kwargs) as cm:
363	n/a	func()
364	n/a
365	n/a	def testAssertRaises(self):
366	n/a	self.assertMessagesCM('assertRaises', (TypeError,), lambda: None,
367	n/a	['^TypeError not raised$', '^oops$',
368	n/a	'^TypeError not raised$',
369	n/a	'^TypeError not raised : oops$'])
370	n/a
371	n/a	def testAssertRaisesRegex(self):
372	n/a	# test error not raised
373	n/a	self.assertMessagesCM('assertRaisesRegex', (TypeError, 'unused regex'),
374	n/a	lambda: None,
375	n/a	['^TypeError not raised$', '^oops$',
376	n/a	'^TypeError not raised$',
377	n/a	'^TypeError not raised : oops$'])
378	n/a	# test error raised but with wrong message
379	n/a	def raise_wrong_message():
380	n/a	raise TypeError('foo')
381	n/a	self.assertMessagesCM('assertRaisesRegex', (TypeError, 'regex'),
382	n/a	raise_wrong_message,
383	n/a	['^"regex" does not match "foo"$', '^oops$',
384	n/a	'^"regex" does not match "foo"$',
385	n/a	'^"regex" does not match "foo" : oops$'])
386	n/a
387	n/a	def testAssertWarns(self):
388	n/a	self.assertMessagesCM('assertWarns', (UserWarning,), lambda: None,
389	n/a	['^UserWarning not triggered$', '^oops$',
390	n/a	'^UserWarning not triggered$',
391	n/a	'^UserWarning not triggered : oops$'])
392	n/a
393	n/a	def testAssertWarnsRegex(self):
394	n/a	# test error not raised
395	n/a	self.assertMessagesCM('assertWarnsRegex', (UserWarning, 'unused regex'),
396	n/a	lambda: None,
397	n/a	['^UserWarning not triggered$', '^oops$',
398	n/a	'^UserWarning not triggered$',
399	n/a	'^UserWarning not triggered : oops$'])
400	n/a	# test warning raised but with wrong message
401	n/a	def raise_wrong_message():
402	n/a	warnings.warn('foo')
403	n/a	self.assertMessagesCM('assertWarnsRegex', (UserWarning, 'regex'),
404	n/a	raise_wrong_message,
405	n/a	['^"regex" does not match "foo"$', '^oops$',
406	n/a	'^"regex" does not match "foo"$',
407	n/a	'^"regex" does not match "foo" : oops$'])
408	n/a
409	n/a
410	n/a	if __name__ == "__main__":
411	n/a	unittest.main()