def test_chunked(): """chunked should handle small chunk sizes""" examples = list(range(10)) assert list(chunked(iter(examples), 0)) == examples assert list(chunked(iter(examples), 1)) == [[i] for i in examples] assert list(chunked(iter(examples), 2)) == [[0, 1], [2, 3], [4, 5], [6, 7], [8, 9]] assert list(chunked(iter(examples), 3)) == [[0, 1, 2], [3, 4, 5], [6, 7, 8], [9]] assert list(chunked(iter(examples), 4)) == [[0, 1, 2, 3], [4, 5, 6, 7], [8, 9]] assert list(chunked(iter(examples), 5)) == [[0, 1, 2, 3, 4], [5, 6, 7, 8, 9]] assert list(chunked(iter(examples), 6)) == [[0, 1, 2, 3, 4, 5], [6, 7, 8, 9]] assert list(chunked(iter(examples), 7)) == [[0, 1, 2, 3, 4, 5, 6], [7, 8, 9]] assert list(chunked(iter(examples), 8)) == [[0, 1, 2, 3, 4, 5, 6, 7], [8, 9]] assert list(chunked(iter(examples), 9)) == [[0, 1, 2, 3, 4, 5, 6, 7, 8], [9]] assert list(chunked(iter(examples), 10)) == [examples] assert list(chunked(iter(examples), 11)) == [examples]
def test_chunked_empty(): """chunked should handle empty iterables""" assert list(chunked(iter([]), 0)) == [] assert list(chunked(iter([]), 1)) == [] assert list(chunked(iter([]), 2)) == []