diff --git a/tests/unit_tests/test_version_strings.py b/tests/unit_tests/test_version_strings.py index 6c4f935b..f71e59b8 100644 --- a/tests/unit_tests/test_version_strings.py +++ b/tests/unit_tests/test_version_strings.py @@ -59,15 +59,13 @@ def _git(command: str) -> str: def temp_dir(): """Return the path of a temporary directory (set as working dir).""" working_dir = os.getcwd() - try: - with tempfile.TemporaryDirectory() as tmpdir: + with tempfile.TemporaryDirectory() as tmpdir: + try: os.chdir(tmpdir) yield tmpdir + finally: # Change back to working dir before closing context manager or Windows will error os.chdir(working_dir) - finally: - # Ensure changed to working dir - os.chdir(working_dir) @pytest.fixture