remote/sqlite-utils
1
0
Fork 0
mirror of https://github.com/simonw/sqlite-utils.git synced 2026-10-23 09:47:06 +02:00
Code Issues Projects Releases Packages Wiki Activity

Merge d5b4235968 into 85b1be10c8

Browse source
This commit is contained in:
Koosha Paridehpour 2026-09-12 23:14:25 -07:00 • committed by GitHub
commit 8d63253f32
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 12 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
sqlite_utils/utils.py
View file

@ -577,6 +577,8 @@ def chunks(sequence: Iterable[T], size: int) -> Iterable[Iterable[T]]:
:param sequence: Any Python iterator
:param size: The size of each chunk
"""
if size < 1:
raise ValueError(f"chunk size must be at least 1, got {size}")
iterator = iter(sequence)
for item in iterator:
yield itertools.chain([item], itertools.islice(iterator, size - 1))

10
tests/test_utils.py
View file

@ -41,6 +41,16 @@ def test_chunks(size, expected):
assert chunks == expected
def test_chunks_size_zero_raises():
with pytest.raises(ValueError, match="chunk size must be at least 1"):
list(utils.chunks([1, 2, 3], 0))
def test_chunks_negative_size_raises():
with pytest.raises(ValueError, match="chunk size must be at least 1"):
list(utils.chunks([1, 2, 3], -1))
def test_hash_record():
expected = "d383e7c0ba88f5ffcdd09be660de164b3847401a"
assert utils.hash_record({"name": "Cleo", "twitter": "CleoPaws"}) == expected