mirror of
https://github.com/RustPython/RustPython.git
synced 2026-06-02 19:39:49 +09:00
Update test_generators.py from 3.14.3
This commit is contained in:
39
Lib/test/test_generators.py
vendored
39
Lib/test/test_generators.py
vendored
@@ -136,6 +136,18 @@ class FinalizationTest(unittest.TestCase):
|
||||
self.assertEqual(len(resurrected), 1)
|
||||
self.assertIsInstance(resurrected[0].gi_code, types.CodeType)
|
||||
|
||||
def test_exhausted_generator_frame_cycle(self):
|
||||
def g():
|
||||
yield
|
||||
|
||||
generator = g()
|
||||
frame = generator.gi_frame
|
||||
self.assertIsNone(frame.f_back)
|
||||
next(generator)
|
||||
self.assertIsNone(frame.f_back)
|
||||
next(generator, None)
|
||||
self.assertIsNone(frame.f_back)
|
||||
|
||||
|
||||
class GeneratorTest(unittest.TestCase):
|
||||
|
||||
@@ -292,6 +304,33 @@ class GeneratorTest(unittest.TestCase):
|
||||
|
||||
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 = [
|
||||
|
||||
Reference in New Issue
Block a user