Skip to content

Commit b97b512

Browse files
committed
Add test
1 parent b92c34a commit b97b512

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

Lib/test/test_capi/test_opt.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1286,5 +1286,15 @@ def testfunc(n):
12861286
self.assertEqual(res, 32 * 32)
12871287
self.assertIsNone(ex)
12881288

1289+
def test_return_generator(self):
1290+
def gen():
1291+
yield None
1292+
def testfunc(n):
1293+
for i in range(n):
1294+
gen()
1295+
res, ex = self._run_with_optimizer(testfunc, 20)
1296+
self.assertIsNotNone(ex)
1297+
self.assertIn("_RETURN_GENERATOR", get_opnames(ex))
1298+
12891299
if __name__ == "__main__":
12901300
unittest.main()

0 commit comments

Comments
 (0)