TST: Test interfaces.utility.Merge (vstack)

Author: effigies

nipype/interfaces/tests/test_utility.py

@@ -142,6 +142,29 @@ def test_split():
         shutil.rmtree(tempdir)
 
 
+def test_merge():
+    tempdir = os.path.realpath(mkdtemp())
+    origdir = os.getcwd()
+    os.chdir(tempdir)
+
+    try:
+        node = pe.Node(utility.Merge(3),
+                       name='merge3')
+        node.inputs.in1 = 0
+        node.inputs.in2 = [1, 2]
+        node.inputs.in3 = [3, 4, 5]
+        res = node.run()
+        yield assert_equal, res.outputs.out, [0, 1, 2, 3, 4, 5]
+
+        node = pe.Node(utility.Merge(), name='merge0')
+        node.inputs.in_lists = [0, [1, 2], [3, 4, 5]]
+        res = node.run()
+        yield assert_equal, res.outputs.out, [0, 1, 2, 3, 4, 5]
+    finally:
+        os.chdir(origdir)
+        shutil.rmtree(tempdir)
+
+
 def test_csvReader():
     header = "files,labels,erosion\n"
     lines = ["foo,hello,300.1\n",