summaryrefslogtreecommitdiff
path: root/coverage/test_helpers.py
diff options
context:
space:
mode:
Diffstat (limited to 'coverage/test_helpers.py')
-rw-r--r--coverage/test_helpers.py12
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):