Merge pull request #34 from tillahoffmann/args

Add kwargs to placeholder.
spotify · Jun 29, 2018 · 2901619 · 2901619
2 parents ed28c8c + c760fe1
commit 2901619
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/pythonflow/operations.py b/pythonflow/operations.py
@@ -27,8 +27,8 @@ class placeholder(Operation):  # pylint: disable=C0103,R0903
     """
     Placeholder that needs to be given in the context to be evaluated.
     """
-    def __init__(self, name=None):
-        super(placeholder, self).__init__(name=name)
+    def __init__(self, name=None, **kwargs):
+        super(placeholder, self).__init__(name=name, **kwargs)
 
     def _evaluate(self):  # pylint: disable=W0221
         raise ValueError("missing value for placeholder '%s'" % self.name)

diff --git a/tests/test_pythonflow.py b/tests/test_pythonflow.py
@@ -527,3 +527,11 @@ def test_stack_trace():
         raise RuntimeError("did not raise ZeroDivisionError")
     except ZeroDivisionError as ex:
         assert isinstance(ex.__cause__, pf.EvaluationError)
+
+
+def test_placeholder_with_kwargs():
+    with pf.Graph() as graph:
+        a = pf.placeholder(length=2)
+        b, c = a
+
+    assert graph([b, c], {a: [1, 2]}) == (1, 2)