diff options
Diffstat (limited to 'tests/test_command.py')
| -rw-r--r-- | tests/test_command.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/tests/test_command.py b/tests/test_command.py index 5ec3567..0937930 100644 --- a/tests/test_command.py +++ b/tests/test_command.py @@ -1221,14 +1221,16 @@ class CommandLineTest(TestBase): mock.call( os.path.abspath(os.path.join(path, "__init__.py")), "w" ), - mock.call().close(), + mock.call().__enter__(), + mock.call().__exit__(None, None, None), mock.call( os.path.abspath( os.path.join(path, "versions", "__init__.py") ), "w", ), - mock.call().close(), + mock.call().__enter__(), + mock.call().__exit__(None, None, None), ], ) |