removing nose from new tests after rebasing

Author: djarecka

nipype/pipeline/engine/tests/test_engine.py

@@ -680,25 +680,23 @@ def func1(in1):
     assert not error_raised
 
 
-def test_write_graph_runs():
-    cwd = os.getcwd()
-    wd = mkdtemp()
-    os.chdir(wd)
+def test_write_graph_runs(tmpdir):
+    os.chdir(str(tmpdir))
 
     for graph in ('orig', 'flat', 'exec', 'hierarchical', 'colored'):
         for simple in (True, False):
             pipe = pe.Workflow(name='pipe')
-            mod1 = pe.Node(interface=TestInterface(), name='mod1')
-            mod2 = pe.Node(interface=TestInterface(), name='mod2')
+            mod1 = pe.Node(interface=EngineTestInterface(), name='mod1')
+            mod2 = pe.Node(interface=EngineTestInterface(), name='mod2')
             pipe.connect([(mod1, mod2, [('output1', 'input1')])])
             try:
                 pipe.write_graph(graph2use=graph, simple_form=simple)
             except Exception:
-                yield assert_true, False, \
-                'Failed to plot {} {} graph'.format(
+                assert False, \
+                    'Failed to plot {} {} graph'.format(
                     'simple' if simple else 'detailed', graph)
 
-            yield assert_true, os.path.exists('graph.dot') or os.path.exists('graph_detailed.dot')
+            assert os.path.exists('graph.dot') or os.path.exists('graph_detailed.dot')
             try:
                 os.remove('graph.dot')
             except OSError:
@@ -708,13 +706,9 @@ def test_write_graph_runs():
             except OSError:
                 pass
 
-    os.chdir(cwd)
-    rmtree(wd)
 
-def test_deep_nested_write_graph_runs():
-    cwd = os.getcwd()
-    wd = mkdtemp()
-    os.chdir(wd)
+def test_deep_nested_write_graph_runs(tmpdir):
+    os.chdir(str(tmpdir))
 
     for graph in ('orig', 'flat', 'exec', 'hierarchical', 'colored'):
         for simple in (True, False):
@@ -724,16 +718,16 @@ def test_deep_nested_write_graph_runs():
                 sub = pe.Workflow(name='pipe_nest_{}'.format(depth))
                 parent.add_nodes([sub])
                 parent = sub
-            mod1 = pe.Node(interface=TestInterface(), name='mod1')
+            mod1 = pe.Node(interface=EngineTestInterface(), name='mod1')
             parent.add_nodes([mod1])
             try:
                 pipe.write_graph(graph2use=graph, simple_form=simple)
             except Exception as e:
-                yield assert_true, False, \
-                'Failed to plot {} {} deep graph: {!s}'.format(
+                assert False, \
+                    'Failed to plot {} {} deep graph: {!s}'.format(
                     'simple' if simple else 'detailed', graph, e)
 
-            yield assert_true, os.path.exists('graph.dot') or os.path.exists('graph_detailed.dot')
+            assert os.path.exists('graph.dot') or os.path.exists('graph_detailed.dot')
             try:
                 os.remove('graph.dot')
             except OSError:
@@ -742,6 +736,3 @@ def test_deep_nested_write_graph_runs():
                 os.remove('graph_detailed.dot')
             except OSError:
                 pass
-
-    os.chdir(cwd)
-    rmtree(wd)