Merge pull request #823 from summivox/yield-callable

fix: `yield` expression should be callable
gkz · Feb 1, 2016 · 856ab6b · 856ab6b
2 parents 7816af0 + fade767
commit 856ab6b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/src/ast.ls b/src/ast.ls
@@ -1124,6 +1124,8 @@ class exports.Yield extends Node
 
     show: -> if @op is 'yieldfrom' then 'from' else ''
 
+    ::delegate <[ isCallable ]> -> yes
+
     compile-node: (o) ->
         code = []
 

diff --git a/test/generators.ls b/test/generators.ls
@@ -137,12 +137,16 @@ g5.next!
 g5.next true
 g5.next false
 
-# calling a yield
+# yield expression as argument, as callable
 is-two = -> it == 2
 f6 = ->*
     is-two yield 1
+    ok (yield 1)(2)
+    ok (2 |> yield 1)
 g6 = f6!
 eq 1 g6.next(2).value
+g6.next is-two
+g6.next is-two
 
 # in switch
 f7 = (x) ->*