diff options
Diffstat (limited to 'coverage/test_helpers.py')
-rw-r--r-- | coverage/test_helpers.py | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/coverage/test_helpers.py b/coverage/test_helpers.py index 574a38d..50cc329 100644 --- a/coverage/test_helpers.py +++ b/coverage/test_helpers.py @@ -53,16 +53,20 @@ def change_dir(new_dir): """ old_dir = os.getcwd() os.chdir(new_dir) - yield os.getcwd() - os.chdir(old_dir) + try: + yield os.getcwd() + finally: + os.chdir(old_dir) @contextlib.contextmanager def saved_sys_path(): """Save sys.path, and restore it later.""" old_syspath = sys.path[:] - yield - sys.path = old_syspath + try: + yield + finally: + sys.path = old_syspath def setup_with_context_manager(testcase, cm): |