Fix yield inside lambda

Cython/Compiler/ParseTreeTransforms.py

@@ -193,9 +193,8 @@ class PostParse(ScopeTrackingTransform):
         collector = YieldNodeCollector()
         collector.visitchildren(node.result_expr)
         if collector.yields or isinstance(node.result_expr, ExprNodes.YieldExprNode):
-            body = ExprNodes.YieldExprNode(
-                node.result_expr.pos, arg=node.result_expr)
-            body = Nodes.ExprStatNode(node.result_expr.pos, expr=body)
+            body = Nodes.ExprStatNode(
+                node.result_expr.pos, expr=node.result_expr)
         else:
             body = Nodes.ReturnStatNode(
                 node.result_expr.pos, value=node.result_expr)

tests/run/generators.pyx

@@ -291,10 +291,11 @@ def test_inside_lambda():
     >>> obj = test_inside_lambda()()
     >>> next(obj)
     1
-    >>> obj.send('a')
+    >>> next(obj)
     2
-    >>> obj.send('b')
-    ('a', 'b')
+    >>> next(obj)
+    Traceback (most recent call last):
+    StopIteration
     """
     return lambda:((yield 1), (yield 2))
 

tests/run/generators_py.py

@@ -276,10 +276,11 @@ def test_inside_lambda():
     >>> obj = test_inside_lambda()()
     >>> next(obj)
     1
-    >>> obj.send('a')
+    >>> next(obj)
     2
-    >>> obj.send('b')
-    ('a', 'b')
+    >>> next(obj)
+    Traceback (most recent call last):
+    StopIteration
     """
     return lambda:((yield 1), (yield 2))