diff tests/tmpfs.py @ 673:d6403c21bdea

tests: Improve failure reporting.
author Ludovic Chabant <ludovic@chabant.com>
date Sun, 06 Mar 2016 23:41:35 -0800
parents bab91fcef741
children f987b29d6fab
line wrap: on
line diff
--- a/tests/tmpfs.py	Fri Mar 04 22:37:07 2016 -0800
+++ b/tests/tmpfs.py	Sun Mar 06 23:41:35 2016 -0800
@@ -68,9 +68,10 @@
 
 
 class TempDirScope(object):
-    def __init__(self, fs, open_patches=None):
+    def __init__(self, fs, open_patches=None, keep=False):
         self._fs = fs
         self._open = open
+        self._keep = keep
 
     @property
     def root(self):
@@ -80,5 +81,6 @@
         return self
 
     def __exit__(self, type, value, traceback):
-        shutil.rmtree(self.root)
+        if not self._keep:
+            shutil.rmtree(self.root)