commit 2bd52dcccbd38908ae2a3b3bdceb3b04d6b16c81
author Michael Foord <fuzzyman@voidspace.org.uk> Sun Feb 07 18:44:12 2010 +0000
committer Michael Foord <fuzzyman@voidspace.org.uk> Sun Feb 07 18:44:12 2010 +0000
tree 089ff2404ebb4b86054f3dda2b0396b9c99f8038
parent a4f46e129294c686ef1effdd89c459bd9a624e6d

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

Lib/test/test_unittest.py

diff --git a/Lib/test/test_unittest.py b/Lib/test/test_unittest.py
index ab04382..04a7322 100644
--- a/Lib/test/test_unittest.py
+++ b/Lib/test/test_unittest.py
@@ -3059,8 +3059,13 @@
             pass
         else:
             self.fail("assertRaises() didn't let exception pass through")
-        with self.assertRaises(KeyError):
-            raise KeyError
+        with self.assertRaises(KeyError) as cm:
+            try:
+                raise KeyError
+            except Exception, e:
+                raise
+        self.assertIs(cm.exception, e)
+
         with self.assertRaises(KeyError):
             raise KeyError("key")
         try: