run test in temp dir

Author: oesteban

nipype/interfaces/base/core.py

@@ -33,7 +33,7 @@
 
 from ... import config, logging, LooseVersion
 from ...utils.provenance import write_provenance
-from ...utils.misc import trim, str2bool
+from ...utils.misc import trim, str2bool, rgetcwd
 from ...utils.filemanip import (FileNotFoundError, split_filename, read_stream,
                                 which, get_dependencies, canonicalize_env as
                                 _canonicalize_env)
@@ -438,14 +438,16 @@ def _duecredit_cite(self):
             r['path'] = self.__module__
             due.cite(**r)
 
-    def run(self, **inputs):
+    def run(self, cwd=None, **inputs):
         """Execute this interface.
 
         This interface will not raise an exception if runtime.returncode is
         non-zero.
 
         Parameters
         ----------
+
+        cwd : specify a folder where the interface should be run
         inputs : allows the interface settings to be updated
 
         Returns
@@ -455,6 +457,13 @@ def run(self, **inputs):
         """
         from ...utils.profiler import ResourceMonitor
 
+        # Tear-up: get current and prev directories
+        syscwd = rgetcwd(error=False)  # Recover when wd does not exist
+        if cwd is None:
+            cwd = syscwd
+
+        os.chdir(cwd)  # Change to the interface wd
+
         enable_rm = config.resource_monitor and self.resource_monitor
         force_raise = not getattr(self.inputs, 'ignore_exception', False)
         self.inputs.trait_set(**inputs)
@@ -471,7 +480,8 @@ def run(self, **inputs):
             env['DISPLAY'] = config.get_display()
 
         runtime = Bunch(
-            cwd=os.getcwd(),
+            cwd=cwd,
+            prevcwd=syscwd,
             returncode=None,
             duration=None,
             environ=env,
@@ -556,6 +566,7 @@ def run(self, **inputs):
                         'rss_GiB': (vals[:, 2] / 1024).tolist(),
                         'vms_GiB': (vals[:, 3] / 1024).tolist(),
                     }
+            os.chdir(syscwd)
 
         return results
 

nipype/pipeline/engine/nodes.py

@@ -449,26 +449,13 @@ def run(self, updatehash=False):
         savepkl(op.join(outdir, '_node.pklz'), self)
         savepkl(op.join(outdir, '_inputs.pklz'), self.inputs.get_traitsfree())
 
-        try:
-            cwd = os.getcwd()
-        except OSError:
-            # Changing back to cwd is probably not necessary
-            # but this makes sure there's somewhere to change to.
-            cwd = op.split(outdir)[0]
-            logger.warning(
-                'Current folder "%s" does not exist, changing to "%s" instead.',
-                os.getenv('PWD', 'unknown'), cwd)
-
-        os.chdir(outdir)
         try:
             result = self._run_interface(execute=True)
         except Exception:
             logger.warning('[Node] Error on "%s" (%s)', self.fullname, outdir)
             # Tear-up after error
             os.remove(hashfile_unfinished)
             raise
-        finally:  # Ensure we come back to the original CWD
-            os.chdir(cwd)
 
         # Tear-up after success
         shutil.move(hashfile_unfinished, hashfile)
@@ -586,17 +573,18 @@ def _run_command(self, execute, copyfiles=True):
                 logger.info("[Node] Cached - collecting precomputed outputs")
                 return result
 
+        outdir = self.output_dir()
         # Run command: either execute is true or load_results failed.
-        runtime = Bunch(
-            returncode=1,
-            environ=dict(os.environ),
-            hostname=socket.gethostname())
         result = InterfaceResult(
             interface=self._interface.__class__,
-            runtime=runtime,
+            runtime=Bunch(
+                cwd=outdir,
+                returncode=1,
+                environ=dict(os.environ),
+                hostname=socket.gethostname()
+            ),
             inputs=self._interface.inputs.get_traitsfree())
 
-        outdir = self.output_dir()
         if copyfiles:
             self._originputs = deepcopy(self._interface.inputs)
             self._copyfiles_to_wd(execute=execute)
@@ -618,7 +606,7 @@ def _run_command(self, execute, copyfiles=True):
             message += ', a CommandLine Interface with command:\n{}'.format(cmd)
         logger.info(message)
         try:
-            result = self._interface.run()
+            result = self._interface.run(cwd=outdir)
         except Exception as msg:
             result.runtime.stderr = '%s\n\n%s'.format(
                 getattr(result.runtime, 'stderr', ''), msg)

nipype/utils/tests/test_provenance.py

@@ -11,9 +11,10 @@
 from nipype.utils.provenance import ProvStore, safe_encode
 
 
-def test_provenance():
-    ps = ProvStore()
+def test_provenance(tmpdir):
     from nipype.interfaces.base import CommandLine
+    tmpdir.chdir()
+    ps = ProvStore()
     results = CommandLine('echo hello').run()
     ps.add_results(results)
     provn = ps.g.get_provn()