Fix assertRaises usage on reflection functions which should raise

NotImplementedError on Windows XP and below.

Fix assertRaises usage on reflection functions which should raise
NotImplementedError on Windows XP and below.
e9da81c6 · Brian Curtin · bea424af · e9da81c6
Commit e9da81c6 authored Apr 03, 2010 by Brian Curtin
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 5 deletions

Lib/test/test_winreg.py Lib/test/test_winreg.py +8 -5

No files found.
--- 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)