mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-07-30 06:34:15 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

assertRaises as context manager now allows you to access exception as documented

Browse source
This commit is contained in:
Michael Foord 2010-02-07 18:44:12 +00:00
parent a4f46e1292
commit 2bd52dcccb
2 changed files with 8 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

9
Lib/test/test_unittest.py
View file

@ -3059,8 +3059,13 @@ class Test_Assertions(TestCase):
pass pass
else: else:
self.fail("assertRaises() didn't let exception pass through") self.fail("assertRaises() didn't let exception pass through")
with self.assertRaises(KeyError): with self.assertRaises(KeyError) as cm:
raise KeyError try:
raise KeyError
except Exception, e:
raise
self.assertIs(cm.exception, e)
with self.assertRaises(KeyError): with self.assertRaises(KeyError):
raise KeyError("key") raise KeyError("key")
try: try:

2
Lib/unittest/case.py
View file

@ -91,7 +91,7 @@ class _AssertRaisesContext(object):
self.expected_regexp = expected_regexp self.expected_regexp = expected_regexp
def __enter__(self): def __enter__(self):
pass return self
def __exit__(self, exc_type, exc_value, tb): def __exit__(self, exc_type, exc_value, tb):
if exc_type is None: if exc_type is None: