add Values tests for composel, remove done TODO.

Technologicat · Technologicat · commit 41c7238f2fe8 · 2021-05-28T02:06:40.000+03:00
diff --git a/unpythonic/seq.py b/unpythonic/seq.py
@@ -99,7 +99,6 @@ def lazy_begin0(*bodys):
         body()
     return out
 
-# TODO: test multiple-return-values support in all function composition utilities (`curry`, `compose` family, `pipe` family)
 # TODO: expand tests of `continuations` to cases with named return values
 # TODO: update code examples in docs
 
diff --git a/unpythonic/tests/test_fun.py b/unpythonic/tests/test_fun.py
@@ -311,6 +311,17 @@ def mymul_typed(y: int):
         test[inc2_then_double(3) == 10]
         test[double_then_inc2(3) == 8]
 
+    with testset("compose with multiple-return-values, named return values"):
+        f = lambda x, y: Values(2 * x, 3 * y)
+        g = lambda x, y: Values(x + 2, y + 3)
+        f_then_g = composel(f, g)
+        test[f_then_g(1, 2) == Values(4, 9)]
+
+        f = lambda x, y: Values(x=2 * x, y=3 * y)
+        g = lambda x, y: Values(x=x + 2, y=y + 3)
+        f_then_g = composel(f, g)
+        test[f_then_g(1, 2) == Values(x=4, y=9)]
+
     with testset("curry in compose chain"):
         def f1(a, b):
             return Values(2 * a, 3 * b)
