self.assertEqual([1,2], list(i for i in C()))
+ def test_close_clears_frame(self):
+ # gh-142766: Test that closing a generator clears its frame
+ class DetectDelete:
+ def __init__(self):
+ DetectDelete.deleted = False
+
+ def __del__(self):
+ DetectDelete.deleted = True
+
+ def generator(arg):
+ yield
+
+ # Test a freshly created generator (not suspended)
+ g = generator(DetectDelete())
+ g.close()
+ self.assertTrue(DetectDelete.deleted)
+
+ # Test a suspended generator
+ g = generator(DetectDelete())
+ next(g)
+ g.close()
+ self.assertTrue(DetectDelete.deleted)
+
+ # Clear via gi_frame.clear()
+ g = generator(DetectDelete())
+ g.gi_frame.clear()
+ self.assertTrue(DetectDelete.deleted)
class ModifyUnderlyingIterableTest(unittest.TestCase):
iterables = [
if (gen->gi_frame_state == FRAME_CREATED) {
gen->gi_frame_state = FRAME_COMPLETED;
+ gen_clear_frame(gen);
Py_RETURN_NONE;
}
if (FRAME_STATE_FINISHED(gen->gi_frame_state)) {