diff options
Diffstat (limited to 'Lib/test')
-rw-r--r-- | Lib/test/test_winreg.py | 13 |
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) |