add tab completion with tests

work still needed for dynamic traits

Author: leej3

nipype/interfaces/base/specs.py

@@ -84,9 +84,11 @@ def __repr__(self):
         """ Return a well-formatted representation of the traits """
         outstr = []
         for name, value in sorted(self.trait_get().items()):
-            outstr.append('%s = %s' % (name, value))
+            if not name == '__all__':
+                outstr.append('%s = %s' % (name, value))
         return '\n{}\n'.format('\n'.join(outstr))
 
+
     def _generate_handlers(self):
         """Find all traits with the 'xor' metadata and attach an event
         handler to them.
@@ -352,7 +354,7 @@ def __deepcopy__(self, memo):
         # clone twice
         dup = self.clone_traits(memo=memo)
         dup.trait_set(**dup_dict)
-        self.__all__ = super(BaseTraitedSpec,self).__all__
+        self.__all__ = self.class_editable_traits()
         return dup
 
 

nipype/interfaces/base/tests/test_specs.py

@@ -11,6 +11,10 @@
 from ....utils.filemanip import split_filename
 from ... import base as nib
 from ...base import traits, Undefined
+from ....interfaces import fsl
+from ...utility.wrappers import Function
+from ....pipeline import Node
+
 
 standard_library.install_aliases()
 
@@ -47,6 +51,23 @@ class spec(nib.TraitedSpec):
     assert infields.__repr__() == '\nfoo = 1\ngoo = 0.0\n'
 
 
+def test_TraitedSpec_tab_completion():
+    bet_nd = Node(fsl.BET(), name = 'bet')
+    bet_interface = fsl.BET()
+    bet_inputs = bet_nd.inputs.class_editable_traits()
+    bet_outputs = bet_nd.outputs.class_editable_traits()
+    
+    # Check __all__ for bet node and interface inputs
+    assert bet_nd.inputs.__all__ == [
+    x for x in bet_inputs if not x == "__all__"]
+    assert bet_interface.inputs.__all__ == [
+    x for x in bet_inputs if not x == "__all__"]
+
+    # Check __all__ for bet node outputs
+    assert bet_nd.outputs.__all__ == [
+    x for x in bet_outputs if not x == "__all__"]
+
+
 @pytest.mark.skip
 def test_TraitedSpec_dynamic():
     from pickle import dumps, loads
@@ -63,6 +84,29 @@ def test_TraitedSpec_dynamic():
         assign_a_again
 
 
+def test_DynamicTraitedSpec_tab_completion():
+    extract_func = lambda list_out: list_out[0]
+
+    # Define interface
+    func_interface = Function(input_names=["list_out"],
+                             output_names=["out_file","another_file"],
+                             function=extract_func)
+    # Check __all__ for interface inputs
+    assert func_interface.inputs.__all__ == func_interface._input_names
+
+    # Define node
+    list_extract = Node(Function(
+        input_names=["list_out"],output_names=["out_file"],
+        function=extract_func), name="list_extract")
+
+    # Check __all__ for node inputs
+    assert list_extract.inputs.__all__ == list_extract._interface._input_names
+    # Check __all__ for node outputs
+    outputs_dict = list_extract._interface._outputs().get().items()
+    outputs = [k for k,v in outputs_dict if not k == "__all__"]
+    assert list_extract.outputs.__all__ == outputs
+
+
 def test_TraitedSpec_logic():
     class spec3(nib.TraitedSpec):
         _xor_inputs = ('foo', 'bar')