mirrors
/
cpython
mirror of https://github.com/python/cpython.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[3.11] gh-102088 Optimize iter_index itertools recipe (GH-102360) (GH-102363)

This commit is contained in:
Miss Islington (bot) 2023-03-01 19:52:37 -08:00 committed by GitHub
parent 90ec292ab8
commit 3effccee48
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
Doc/library/itertools.rst
View File

@ -875,9 +875,12 @@ which incur interpreter overhead.
except AttributeError: except AttributeError:
# Slow path for general iterables # Slow path for general iterables
it = islice(iterable, start, None) it = islice(iterable, start, None)
for i, element in enumerate(it, start): i = start - 1
if element is value or element == value: try:
yield i while True:
yield (i := i + operator.indexOf(it, value) + 1)
except ValueError:
pass
else: else:
# Fast path for sequences # Fast path for sequences
i = start - 1 i = start - 1

3
Lib/test/test_operator.py
View File

@ -208,6 +208,9 @@ def test_indexOf(self):
nan = float("nan") nan = float("nan")
self.assertEqual(operator.indexOf([nan, nan, 21], nan), 0) self.assertEqual(operator.indexOf([nan, nan, 21], nan), 0)
self.assertEqual(operator.indexOf([{}, 1, {}, 2], {}), 0) self.assertEqual(operator.indexOf([{}, 1, {}, 2], {}), 0)
it = iter('leave the iterator at exactly the position after the match')
self.assertEqual(operator.indexOf(it, 'a'), 2)
self.assertEqual(next(it), 'v')
def test_invert(self): def test_invert(self):
operator = self.module operator = self.module