[GR-23231] Fix nested yields with send.

PullRequest: graalpython/1180

Author: steve-s

graalpython/com.oracle.graal.python.test/src/tests/test_generators.py

@@ -235,26 +235,33 @@ def f():
             next(g)
 
 
-    # def test_return_tuple(self):
-    #     def g():
-    #         return (yield 1)
-
-    #     gen = g()
-    #     self.assertEqual(next(gen), 1)
-    #     with self.assertRaises(StopIteration) as cm:
-    #         gen.send((2,))
-    #     self.assertEqual(cm.exception.value, (2,))
-
-    # def test_return_stopiteration(self):
-    #     def g():
-    #         return (yield 1)
-
-    #     gen = g()
-    #     self.assertEqual(next(gen), 1)
-    #     with self.assertRaises(StopIteration) as cm:
-    #         gen.send(StopIteration(2))
-    #     self.assertTrue(isinstance(cm.exception.value, StopIteration))
-    #     self.assertEqual(cm.exception.value.value, 2)
+    def test_return_tuple(self):
+        def g():
+            return (yield 1)
+
+        gen = g()
+        self.assertEqual(next(gen), 1)
+        with self.assertRaises(StopIteration) as cm:
+            gen.send((2,))
+        self.assertEqual(cm.exception.value, (2,))
+
+    def test_return_stopiteration(self):
+        def g():
+            return (yield 1)
+
+        gen = g()
+        self.assertEqual(next(gen), 1)
+        with self.assertRaises(StopIteration) as cm:
+            gen.send(StopIteration(2))
+        self.assertTrue(isinstance(cm.exception.value, StopIteration))
+        self.assertEqual(cm.exception.value.value, 2)
+
+    def test_yield_expr_value_without_send(self):
+        def fn():
+            yield (1,(yield 42))
+        g = fn()
+        self.assertEqual(next(g), 42)
+        self.assertEqual(next(g), (1,None))
 
     def test_generator_caller_frame(self):
         def gen():

graalpython/com.oracle.graal.python.test/src/tests/unittest_tags/test_fstring.txt

@@ -9,35 +9,34 @@
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_ast_line_numbers_multiple_formattedvalues
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_ast_line_numbers_nested
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_backslash_char
+*graalpython.lib-python.3.test.test_fstring.TestCase.test_backslashes_in_string_part
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_call
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_closure
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_comments
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_compile_time_concat
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_compile_time_concat_errors
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_conversions
-*graalpython.lib-python.3.test.test_fstring.TestCase.test_backslashes_in_string_part
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_debug_conversion
-*graalpython.lib-python.3.test.test_fstring.TestCase.test_format_specifier_expressions
-*graalpython.lib-python.3.test.test_fstring.TestCase.test_errors
-*graalpython.lib-python.3.test.test_fstring.TestCase.test_global
-*graalpython.lib-python.3.test.test_fstring.TestCase.test_lambda
-*graalpython.lib-python.3.test.test_fstring.TestCase.test_misformed_unicode_character_name
-*graalpython.lib-python.3.test.test_fstring.TestCase.test_no_backslashes_in_expression_part
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_del
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_dict
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_docstring
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_double_braces
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_empty_format_specifier
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_equal_equal
+*graalpython.lib-python.3.test.test_fstring.TestCase.test_errors
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_expressions_with_triple_quoted_strings
+*graalpython.lib-python.3.test.test_fstring.TestCase.test_format_specifier_expressions
+*graalpython.lib-python.3.test.test_fstring.TestCase.test_global
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_if_conditional
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_invalid_string_prefixes
+*graalpython.lib-python.3.test.test_fstring.TestCase.test_lambda
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_leading_trailing_spaces
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_literal
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_literal_eval
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_locals
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_loop
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_many_expressions
+*graalpython.lib-python.3.test.test_fstring.TestCase.test_misformed_unicode_character_name
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_mismatched_braces
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_mismatched_parens
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_missing_expression
@@ -46,6 +45,7 @@
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_multiple_vars
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_nested_fstrings
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_newlines_in_expressions
+*graalpython.lib-python.3.test.test_fstring.TestCase.test_no_backslashes_in_expression_part
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_no_escapes_for_braces
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_not_equal
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_parens_in_expressions
@@ -54,3 +54,4 @@
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_str_format_differences
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_unterminated_string
 *graalpython.lib-python.3.test.test_fstring.TestCase.test_yield
+*graalpython.lib-python.3.test.test_fstring.TestCase.test_yield_send

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/nodes/generator/YieldNode.java

@@ -68,9 +68,10 @@ public Object execute(VirtualFrame frame) {
                 return specialArgument;
             }
         } else {
+            Object result = right.execute(frame);
             access.setActive(frame, flagSlot, true);
             access.setLastYieldIndex(frame, yieldIndex);
-            throw new YieldException(right.execute(frame));
+            throw new YieldException(result);
         }
     }
 }