add tests

Author: skoudoro

nipype/interfaces/dipy/tests/test_base.py

@@ -2,7 +2,8 @@
 from collections import namedtuple
 from ...base import traits, TraitedSpec, BaseInterfaceInputSpec
 from ..base import (convert_to_traits_type, create_interface_specs,
-                    dipy_to_nipype_interface, DipyBaseInterface, no_dipy)
+                    dipy_to_nipype_interface, DipyBaseInterface, no_dipy,
+                    get_dipy_workflows)
 
 
 def test_convert_to_traits_type():
@@ -136,6 +137,16 @@ def run(self, in_files, param1=1, out_dir='', out_ref='out1.txt'):
         new_specs().run()
 
 
+@pytest.mark.skipif(no_dipy(), reason="DIPY is not installed")
+def test_get_dipy_workflows():
+    from dipy.workflows import align
+
+    l_wkflw = get_dipy_workflows(align)
+    for name, obj in l_wkflw:
+        assert name.endswith('Flow')
+        assert issubclass(align.Workflow)
+
+
 if __name__ == "__main__":
     test_convert_to_traits_type()
     test_create_interface_specs()