tst: adding first test

Author: oesteban

nipype/pipeline/engine/tests/test_utils.py

@@ -16,7 +16,7 @@
 from ....interfaces import base as nib
 from ....interfaces import utility as niu
 from .... import config
-from ..utils import clean_working_directory, write_workflow_prov
+from ..utils import clean_working_directory, write_workflow_prov, load_resultfile
 
 
 class InputSpec(nib.TraitedSpec):
@@ -283,3 +283,17 @@ def test_modify_paths_bug(tmpdir):
     assert outputs.out_dict_path == {out_str: out_path}
     assert outputs.out_dict_str == {out_str: out_str}
     assert outputs.out_list == [out_str] * 2
+
+
+def test_save_load_resultfile(tmpdir):
+    tmpdir.chdir()
+
+    spc = pe.Node(StrPathConfuser(in_str='2'), name='spc')
+    spc.base_dir = tmpdir.mkdir('node').strpath
+    result = spc.run()
+
+    loaded_result = load_resultfile(tmpdir.join('node').join('spc').join('result_spc.pklz'))
+
+    assert result.runtime.dictcopy() == loaded_result.runtime.dictcopy()
+    assert result.inputs == loaded_result.inputs
+    assert result.outputs.get() == loaded_result.outputs.get()

nipype/pipeline/engine/utils.py

@@ -274,40 +274,43 @@ def load_resultfile(results_file, resolve=True):
     """
     Load InterfaceResult file from path.
 
-    Parameter
-    ---------
-    path : base_dir of node
-    name : name of node
+    Parameters
+    ----------
+    results_file : pathlike
+        Path to an existing pickle (``result_<interface name>.pklz``) created with
+        ``save_resultfile``.
+        Raises ``FileNotFoundError`` if ``results_file`` does not exist.
+    resolve : bool
+        Determines whether relative paths will be resolved to absolute (default is ``True``).
 
     Returns
     -------
-    result : InterfaceResult structure
+    result : InterfaceResult
+        A Nipype object containing the runtime, inputs, outputs and other interface information
+        such as a traceback in the case of errors.
 
     """
     results_file = Path(results_file)
-
     if not results_file.exists():
         raise FileNotFoundError(results_file)
 
-    with indirectory(str(results_file.parent)):
-        result = loadpkl(results_file)
-
-        if resolve and result.outputs:
-            try:
-                outputs = result.outputs.get()
-            except TypeError:  # This is a Bunch
-                logger.debug('Outputs object of loaded result %s is a Bunch.', results_file)
-                return result
-
-            logger.debug('Resolving paths in outputs loaded from results file.')
-            for trait_name, old in list(outputs.items()):
-                if isdefined(old):
-                    if result.outputs.trait(trait_name).is_trait_type(OutputMultiPath):
-                        old = result.outputs.trait(trait_name).handler.get_value(
-                            result.outputs, trait_name)
-                    value = resolve_path_traits(result.outputs.trait(trait_name), old,
-                                                results_file.parent)
-                    setattr(result.outputs, trait_name, value)
+    result = loadpkl(results_file)
+    if resolve and result.outputs:
+        try:
+            outputs = result.outputs.get()
+        except TypeError:  # This is a Bunch
+            logger.debug('Outputs object of loaded result %s is a Bunch.', results_file)
+            return result
+
+        logger.debug('Resolving paths in outputs loaded from results file.')
+        for trait_name, old in list(outputs.items()):
+            if isdefined(old):
+                if result.outputs.trait(trait_name).is_trait_type(OutputMultiPath):
+                    old = result.outputs.trait(trait_name).handler.get_value(
+                        result.outputs, trait_name)
+                value = resolve_path_traits(result.outputs.trait(trait_name), old,
+                                            results_file.parent)
+                setattr(result.outputs, trait_name, value)
     return result