VimPlug/typeshed
1
0
Fork 0
mirror of https://github.com/davidhalter/typeshed.git synced 2026-06-24 17:54:01 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

[itertools] More specific type hints for batched(..., strict=True) (#15277)

Browse Source
This commit is contained in:
Liam DeVoe
2026-01-14 05:50:40 -05:00
committed by GitHub
parent 77f8e8f2a5
commit e887eae0d2
2 changed files with 39 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
stdlib/@tests/test_cases/itertools/check_batched.py
+21
View File
@@ -0,0 +1,21 @@
from __future__ import annotations
import sys
from typing_extensions import assert_type
if sys.version_info >= (3, 13):
from itertools import batched
assert_type(batched([0], 1, strict=True), batched[tuple[int]])
assert_type(batched([0, 0], 2, strict=True), batched[tuple[int, int]])
assert_type(batched([0, 0, 0], 3, strict=True), batched[tuple[int, int, int]])
assert_type(batched([0, 0, 0, 0], 4, strict=True), batched[tuple[int, int, int, int]])
assert_type(batched([0, 0, 0, 0, 0], 5, strict=True), batched[tuple[int, int, int, int, int]])
assert_type(batched([0], 2), batched[tuple[int, ...]])
assert_type(batched([0], 2, strict=False), batched[tuple[int, ...]])
def f() -> int:
return 3
assert_type(batched([0, 0, 0], f(), strict=True), batched[tuple[int, ...]])