Python grouper Examples

Example #1

File: utils_test.py Project: wehlutyk/brainscopypaste

def test_grouper_fillvalue():
    base = range(13)
    blocks = [list(block) for block in grouper(base, 5, fillvalue=-1)]
    assert len(blocks) == 3
    assert blocks[0] == list(range(5))
    assert blocks[1] == list(range(5, 10))
    assert blocks[2] == [10, 11, 12, -1, -1]

Example #2

File: utils_test.py Project: wehlutyk/brainscopypaste

def test_grouper():
    base = range(13)
    blocks = [list(block) for block in grouper(base, 5)]
    assert len(blocks) == 3
    assert blocks[0] == list(range(5))
    assert blocks[1] == list(range(5, 10))
    assert blocks[2] == [10, 11, 12, None, None]