Example #1
0
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():
    """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]
Example #3
0
def test_chunked_empty():
    """chunked should handle empty iterables"""
    assert list(chunked(iter([]), 0)) == []
    assert list(chunked(iter([]), 1)) == []
    assert list(chunked(iter([]), 2)) == []
def test_chunked_empty():
    """chunked should handle empty iterables"""
    assert list(chunked(iter([]), 0)) == []
    assert list(chunked(iter([]), 1)) == []
    assert list(chunked(iter([]), 2)) == []