commit e9da81c62bd489e7c0347e896bf6d108f8a10f97
author Brian Curtin <brian.curtin@gmail.com> Sat Apr 03 00:59:32 2010 +0000
committer Brian Curtin <brian.curtin@gmail.com> Sat Apr 03 00:59:32 2010 +0000
tree 7894e1978092f1f73d4198a1232e43a22b68020b
parent bea424af98468a7fc5c505c949218e22862d2b40

Fix assertRaises usage on reflection functions which should raise
NotImplementedError on Windows XP and below.

Lib/test/test_winreg.py

diff --git a/Lib/test/test_winreg.py b/Lib/test/test_winreg.py
index c1a3d18..753be74 100644
--- a/Lib/test/test_winreg.py
+++ b/Lib/test/test_winreg.py
@@ -242,11 +242,14 @@
             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)