GH-141312: Allow only integers to longrangeiter_setstate state (GH-141317)

This fixes an assertion error when the new computed start is not an integer.
This commit is contained in:
Sergey Miryanov 2025-11-14 19:52:01 +05:00 committed by GitHub
parent 49e74210cb
commit 10bec7c1eb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 17 additions and 0 deletions

View file

@ -470,6 +470,16 @@ def test_iterator_setstate(self):
it.__setstate__(2**64 - 7)
self.assertEqual(list(it), [12, 10])
def test_iterator_invalid_setstate(self):
for invalid_value in (1.0, ""):
ranges = (('rangeiter', range(10, 100, 2)),
('longrangeiter', range(10, 2**65, 2)))
for rng_name, rng in ranges:
with self.subTest(invalid_value=invalid_value, range=rng_name):
it = iter(rng)
with self.assertRaises(TypeError):
it.__setstate__(invalid_value)
def test_odd_bug(self):
# This used to raise a "SystemError: NULL result without error"
# because the range validation step was eating the exception