summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--fable.py9
1 files changed, 9 insertions, 0 deletions
diff --git a/fable.py b/fable.py
index 18ac72d..0b47d4e 100644
--- a/fable.py
+++ b/fable.py
@@ -1,4 +1,7 @@
+import os
import sys
+import tempfile
+
class Context:
@@ -20,6 +23,8 @@ class Run:
def __init__(self):
self._name = None
self._context = None
+ self._srcdir = None
+ self._tmpdir = None
def get_context(self):
return self._context
@@ -27,9 +32,13 @@ class Run:
def start(self, name):
self._context = Context()
self._name = name
+ self._tmpdir = tempfile.mkdtemp()
+ self._srcdir = os.getcwd()
+ os.chdir(self._tmpdir)
def end(self):
sys.stdout.write('OK: {}\n'.format(self._name))
+ os.chdir(self._srcdir)
def assertEqual(a, b):