Fix assertRaises usage on reflection functions which should raise

NotImplementedError on Windows XP and below.

diff --git a/Lib/test/test_winreg.py b/Lib/test/test_winreg.py
index c1a3d188b97cce2d8a8bc711339a28c4bbfcdfe8..753be746040cca5b29bff3f9e0ae3e99cf402944 100644 (file)
--- a/Lib/test/test_winreg.py
+++ b/Lib/test/test_winreg.py
@@ -242,11 +242,14 @@ class LocalWinregTests(BaseWinregTests):
             key = OpenKey(HKEY_CURRENT_USER, test_key_name)
             self.assertNotEqual(key.handle, 0)
 
-            self.assertRaises(NotImplementedError, DisableReflectionKey(key))
-            self.assertRaises(NotImplementedError, EnableReflectionKey(key))
-            self.assertRaises(NotImplementedError, QueryReflectionKey(key))
-            self.assertRaises(NotImplementedError,
-                              DeleteKeyEx(HKEY_CURRENT_USER, test_key_name))
+            with self.assertRaises(NotImplementedError):
+                DisableReflectionKey(key)
+            with self.assertRaises(NotImplementedError):
+                EnableReflectionKey(key)
+            with self.assertRaises(NotImplementedError):
+                QueryReflectionKey(key)
+            with self.assertRaises(NotImplementedError):
+                DeleteKeyEx(HKEY_CURRENT_USER, test_key_name)
         finally:
             DeleteKey(HKEY_CURRENT_USER, test_key_name)