Context managers wouldn't clean up during exceptions.

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):