mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-10-13 18:33:34 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

GH-98363: Fix exception handling in batched() (GH-98523)

Browse source
This commit is contained in:
Raymond Hettinger 2022-10-21 12:31:52 -05:00 • committed by GitHub
parent ec1f6f5f13
commit a5ff80c8bc
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 36 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

15
Lib/test/test_itertools.py
View file

@ -2012,6 +2012,20 @@ class E:
def __next__(self):
3 // 0
class E2:
'Test propagation of exceptions after two iterations'
def __init__(self, seqn):
self.seqn = seqn
self.i = 0
def __iter__(self):
return self
def __next__(self):
if self.i == 2:
raise ZeroDivisionError
v = self.seqn[self.i]
self.i += 1
return v
class S:
'Test immediate stop'
def __init__(self, seqn):
@ -2050,6 +2064,7 @@ class TestVariousIteratorArgs(unittest.TestCase):
self.assertRaises(TypeError, batched, X(s), 2)
self.assertRaises(TypeError, batched, N(s), 2)
self.assertRaises(ZeroDivisionError, list, batched(E(s), 2))
self.assertRaises(ZeroDivisionError, list, batched(E2(s), 4))
def test_chain(self):
for s in ("123", "", range(1000), ('do', 1.2), range(2000,2200,5)):