Skip to content

GH-98363: Add itertools.batched() #98364

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
rhettinger merged 24 commits into from
Oct 17, 2022
Merged

GH-98363: Add itertools.batched() #98364

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
24 commits
Select commit Hold shift + click to select a range
057390d
First draft
rhettinger Oct 17, 2022
047c441
.
rhettinger Oct 17, 2022
9061042
Fix NULL test
rhettinger Oct 17, 2022
72a03c6
Improve the docstring and add the regular docs
rhettinger Oct 17, 2022
21a6f67
.
rhettinger Oct 17, 2022
61b83df
Add tests
rhettinger Oct 17, 2022
93bf013
Add entry to toplevel table in the docs
rhettinger Oct 17, 2022
eb6949c
Remove bogus decref
rhettinger Oct 17, 2022
7fd4016
Flatten before joining
rhettinger Oct 17, 2022
8fb8bdf
Add blurb
rhettinger Oct 17, 2022
2b994a3
Add motivating use case and make equivalent code more accurate
rhettinger Oct 17, 2022
fa9c306
Remove TODO comment
rhettinger Oct 17, 2022
c49fb17
Update Lib/test/test_itertools.py
rhettinger Oct 17, 2022
3fbc9c5
Merge branch 'main' into itertools_batched
rhettinger Oct 17, 2022
4c25341
Neaten-up
rhettinger Oct 17, 2022
41acb45
Make the example more self-explanatory
rhettinger Oct 17, 2022
77de77c
Test pure python equivalent
rhettinger Oct 17, 2022
7390670
Test pure python equivalent
rhettinger Oct 17, 2022
c836bc7
Add GC test
rhettinger Oct 17, 2022
c0ca79d
Add "TestVariousIteratorArgs" tests
rhettinger Oct 17, 2022
8238c28
Fix test case I()
rhettinger Oct 17, 2022
b9cbb46
Remove development note
rhettinger Oct 17, 2022
4799e6a
Remove docstring from pure Python equivalent.
rhettinger Oct 17, 2022
bac2f6c
Fix missing argument in the docs table entry.
rhettinger Oct 17, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Fix test case I()
  • Loading branch information
@rhettinger
rhettinger committed Oct 17, 2022
commit 8238c28a71d1bff685a3e0d5d2eb54766ce2f65f
2 changes: 1 addition & 1 deletion Lib/test/test_itertools.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2043,7 +2043,7 @@ def test_batched(self):
s = 'abcde'
r = [['a', 'b'], ['c', 'd'], ['e']]
n = 2
for g in (G, Ig, L, R): # XXX I(s) is failing
for g in (G, I, Ig, L, R):
with self.subTest(g=g):
self.assertEqual(list(batched(g(s), n)), r)
self.assertEqual(list(batched(S(s), 2)), [])
Expand Down
2 changes: 1 addition & 1 deletion Modules/itertoolsmodule.c
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -155,7 +155,7 @@ batched_next(batchedobject *bo)
return NULL;
}
for (i=0 ; i < bo->batch_size ; i++) {
item = (*Py_TYPE(it)->tp_iternext)(it);
item = PyIter_Next(it);
if (item == NULL) {
break;
}
Expand Down