python 2/3 chunk_dict utility

chunk_dict.py

from types import GeneratorType

import six


def chunk_dict(dict_, size):
    chunk = {}
    for i, (key, val) in enumerate(six.iteritems(dict_), start=1):
        chunk[key] = val
        if i % size == 0:
            yield chunk
            chunk = {}
    if chunk:
        yield chunk

res1 = chunk_dict({}, 2)
assert isinstance(res1, GeneratorType)

res2 = list(chunk_dict({"a": 1, "b": 2, "c": 3}, 2))
assert len(res2) == 2
assert len(res2[0]) == 2
assert len(res2[1]) == 1

res3 = list(chunk_dict({"a": 1, "b": 2, "c": 3, "d": 4}, 2))
assert len(res3) == 2
assert len(res3[0]) == 2
assert len(res3[1]) == 2