correct patching ?

IPython/utils/path.py

@@ -198,12 +198,11 b' def get_home_dir(require_writable=False) -> str:'
         # expanduser failed, use the registry to get the 'My Documents' folder.
         try:
             import winreg as wreg
-            key = wreg.OpenKey(
+            with wreg.OpenKey(
                 wreg.HKEY_CURRENT_USER,
                 r"Software\Microsoft\Windows\CurrentVersion\Explorer\Shell Folders"
-            )
-            homedir = wreg.QueryValueEx(key,'Personal')[0]
-            key.Close()
+            ) as key:
+                homedir = wreg.QueryValueEx(key,'Personal')[0]
         except:
             pass
 

IPython/utils/tests/test_path.py

@@ -171,8 +171,12 b' def test_get_home_dir_8():'
         env.pop(key, None)
 
     class key:
+        def __enter__(self):
+            pass
         def Close(self):
             pass
+        def __exit__(*args, **kwargs):
+            pass
 
     with patch.object(wreg, 'OpenKey', return_value=key()), \
          patch.object(wreg, 'QueryValueEx', return_value=[abspath(HOME_TEST_DIR)]):