summaryrefslogtreecommitdiff
path: root/Lib/test
diff options
context:
space:
mode:
Diffstat (limited to 'Lib/test')
-rw-r--r--Lib/test/test_winreg.py13
1 files changed, 8 insertions, 5 deletions
diff --git a/Lib/test/test_winreg.py b/Lib/test/test_winreg.py
index c1a3d188b9..753be74604 100644
--- 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)