diff --git a/tests/__init__.py b/tests/__init__.py index 349b248a3be45c245f82dd78d3be3cbd513a1fb4..d8647089277169add9aead66a38bd911615448e3 100644 --- a/tests/__init__.py +++ b/tests/__init__.py @@ -36,7 +36,8 @@ logging.getLogger().setLevel(logging.WARNING) @contextlib.contextmanager def mockFSLDIR(): - oldval = fslplatform.fsldir + oldfsldir = fslplatform.fsldir + oldfsldevdir = fslplatform.fsldevdir try: with tempdir() as td: @@ -44,13 +45,15 @@ def mockFSLDIR(): bindir = op.join(fsldir, 'bin') os.makedirs(bindir) fslplatform.fsldir = fsldir + fslplatform.fsldevdir = None path = op.pathsep.join((bindir, os.environ['PATH'])) with mock.patch.dict(os.environ, {'PATH': path}): yield fsldir finally: - fslplatform.fsldir = oldval + fslplatform.fsldir = oldfsldir + fslplatform.fsldevdir = oldfsldevdir def touch(fname):