Update pipeline example and test to include currying

sopherapps · Feb 9, 2023 · 66ff540 · 66ff540
1 parent 6fc14a7
commit 66ff540
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git a/funml/pipeline.py b/funml/pipeline.py
@@ -22,8 +22,11 @@ def execute(*args: Any, **kwargs: Any) -> ExecutionExpression:
         ```python
         import funml as ml
 
-        output = ml.val(90) >> (lambda x: x**2) >> (lambda v: v/90) >> ml.execute()
-        # prints 90
+        to_power_of = ml.val(lambda power, v: v**power)
+        divided_by = ml.val(lambda divisor, v: v / divisor)
+
+        output = ml.val(90) >> to_power_of(3) >> divided_by(90) >> divided_by(3) >> ml.execute()
+        # prints 2700
         ```
     """
     return ExecutionExpression(*args, **kwargs)
diff --git a/tests/test_pipeline.py b/tests/test_pipeline.py
@@ -8,10 +8,14 @@
 
 def test_execute():
     """execute terminates pipeline"""
+    to_power_of = val(lambda power, v: v**power)
+    divided_by = val(lambda divisor, v: v / divisor)
+    with_suffix = val(lambda suffix, v: f"{v}{suffix}")
+
     test_data = [
-        (val(90) >> (lambda x: x**2) >> (lambda v: v / 90), 90),
+        (val(90) >> to_power_of(3) >> divided_by(90) >> divided_by(3), 2700),
         (
-            val("hey") >> (lambda x: f"{x} you") >> (lambda g: f"{g}, John"),
+            val("hey") >> with_suffix(" you") >> with_suffix(f", John"),
             "hey you, John",
         ),
     ]