diff options
-rw-r--r-- | setuptools/tests/test_msvc9compiler.py | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/setuptools/tests/test_msvc9compiler.py b/setuptools/tests/test_msvc9compiler.py index ac5e3914..970f7679 100644 --- a/setuptools/tests/test_msvc9compiler.py +++ b/setuptools/tests/test_msvc9compiler.py @@ -79,11 +79,12 @@ def patch_env(**replacements): os.environ.update(replacements) - yield saved - - for key in replacements: - os.environ.pop(key, None) - os.environ.update(saved) + try: + yield saved + finally: + for key in replacements: + os.environ.pop(key, None) + os.environ.update(saved) class TestMSVC9Compiler(unittest.TestCase): |