Fix passing a specific component to a 1D effect

Signed-off-by: martinRenou <[email protected]>

Author: martinRenou

ipygany/ipygany.py

@@ -469,6 +469,9 @@ def _validate_input_impl(self, value):
 
         # Input as a tuple
         if isinstance(value, (tuple, list)):
+            if self.input_dim == 1 and len(value) == 2:
+                return self._validate_input_component(value)
+
             if len(value) != self.input_dim:
                 raise TraitError('input is of dimension {} but expected input dimension is {}'.format(len(value), self.input_dim))
 

tests/test_isocolor.py

@@ -30,12 +30,12 @@ def test_input():
 
     colored_mesh = IsoColor(poly)
 
-    with pytest.raises(TraitError):
-        colored_mesh.input = ('1d', 0)
-
     colored_mesh.input = ('1d', )
     assert colored_mesh.input == (('1d', 'x'), )
 
+    colored_mesh.input = ('3d', 'x')
+    assert colored_mesh.input == ('3d', 'x')
+
     colored_mesh.input = (3.2, )
     assert colored_mesh.input == (3.2, )