simplyfying tests, removing some try/except blocks

Author: djarecka

nipype/pipeline/engine/tests/test_engine.py

@@ -156,12 +156,8 @@ def test_expansion():
     pipe5.add_nodes([pipe4])
     pipe6 = pe.Workflow(name="pipe6")
     pipe6.connect([(pipe5, pipe3, [('pipe4.mod5.output1', 'pipe2.mod3.input1')])])
-    error_raised = False
-    try:
-        pipe6._flatgraph = pipe6._create_flat_graph()
-    except:
-        error_raised = True
-    assert not error_raised
+
+    pipe6._flatgraph = pipe6._create_flat_graph()
 
 
 def test_iterable_expansion():
@@ -479,14 +475,9 @@ def func1(in1):
                  nested=False,
                  name='n1')
     n2.inputs.in1 = [[1, [2]], 3, [4, 5]]
-    error_raised = False
-    try:
-        n2.run()
-    except Exception as e:
-        from nipype.pipeline.engine.base import logger
-        logger.info('Exception: %s' % str(e))
-        error_raised = True
-    assert error_raised
+
+    with pytest.raises(Exception): n2.run()
+
 
 
 def test_node_hash(tmpdir):
@@ -518,35 +509,26 @@ def func2(a):
     w1.config['execution'] = {'stop_on_first_crash': 'true',
                               'local_hash_check': 'false',
                               'crashdump_dir': wd}
-    error_raised = False
     # create dummy distributed plugin class
     from nipype.pipeline.plugins.base import DistributedPluginBase
 
     class RaiseError(DistributedPluginBase):
         def _submit_job(self, node, updatehash=False):
             raise Exception('Submit called')
-    try:
+
+    with pytest.raises(Exception) as excinfo:
         w1.run(plugin=RaiseError())
-    except Exception as e:
-        from nipype.pipeline.engine.base import logger
-        logger.info('Exception: %s' % str(e))
-        error_raised = True
-    assert error_raised
+    assert 'Submit called' == str(excinfo.value)        
+
     # rerun to ensure we have outputs
     w1.run(plugin='Linear')
     # set local check
     w1.config['execution'] = {'stop_on_first_crash': 'true',
                               'local_hash_check': 'true',
                               'crashdump_dir': wd}
-    error_raised = False
-    try:
-        w1.run(plugin=RaiseError())
-    except Exception as e:
-        from nipype.pipeline.engine.base import logger
-        logger.info('Exception: %s' % str(e))
-        error_raised = True
-    assert not error_raised
 
+    w1.run(plugin=RaiseError())
+    
 
 def test_old_config(tmpdir):
     wd = str(tmpdir)
@@ -574,14 +556,8 @@ def func2(a):
 
     w1.config['execution']['crashdump_dir'] = wd
     # generate outputs
-    error_raised = False
-    try:
-        w1.run(plugin='Linear')
-    except Exception as e:
-        from nipype.pipeline.engine.base import logger
-        logger.info('Exception: %s' % str(e))
-        error_raised = True
-    assert not error_raised
+
+    w1.run(plugin='Linear')
 
 
 def test_mapnode_json(tmpdir):
@@ -618,13 +594,9 @@ def func1(in1):
     with open(os.path.join(node.output_dir(), 'test.json'), 'wt') as fp:
         fp.write('dummy file')
     w1.config['execution'].update(**{'stop_on_first_rerun': True})
-    error_raised = False
-    try:
-        w1.run()
-    except:
-        error_raised = True
-    assert not error_raised
 
+    w1.run()
+   
 
 def test_parameterize_dirs_false(tmpdir):
     from ....interfaces.utility import IdentityInterface
@@ -643,14 +615,8 @@ def test_parameterize_dirs_false(tmpdir):
     wf.config['execution']['parameterize_dirs'] = False
     wf.connect([(n1, n2, [('output1', 'in1')])])
 
-    error_raised = False
-    try:
-        wf.run()
-    except TypeError as typerr:
-        from nipype.pipeline.engine.base import logger
-        logger.info('Exception: %s' % str(typerr))
-        error_raised = True
-    assert not error_raised
+
+    wf.run()
 
 
 def test_serial_input(tmpdir):
@@ -694,16 +660,8 @@ def func1(in1):
     assert n1.num_subnodes() == 1
 
     # test running the workflow on serial conditions
-    error_raised = False
-    try:
-        w1.run(plugin='MultiProc')
-    except Exception as e:
-        from nipype.pipeline.engine.base import logger
-        logger.info('Exception: %s' % str(e))
-        error_raised = True
-
-    assert not error_raised
-
+    w1.run(plugin='MultiProc')
+    
 
 def test_write_graph_runs(tmpdir):
     os.chdir(str(tmpdir))