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
Kaydet (Commit) e9da81c6 authored Nis 03, 2010 tarafından Brian Curtin
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 5 deletions

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))
+            with self.assertRaises(NotImplementedError):
-            self.assertRaises(NotImplementedError, EnableReflectionKey(key))
+                DisableReflectionKey(key)
-            self.assertRaises(NotImplementedError, QueryReflectionKey(key))
+            with self.assertRaises(NotImplementedError):
-            self.assertRaises(NotImplementedError,
+                EnableReflectionKey(key)
-                              DeleteKeyEx(HKEY_CURRENT_USER, test_key_name))
+            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)