generate binning with a size, that accumulates to the count

generate_bins.py

#!/usr/bin/env python
import math


def generate_bins(size: int, count: int):
    """
    >>> list(generate_bins(size=100, count=300))
    [100, 100, 100]
    >>> list(generate_bins(size=100, count=301))
    [100, 100, 100, 1]
    >>> list(generate_bins(size=10, count=1))
    [1]
    >>> list(generate_bins(size=10, count=19))
    [10, 9]
    >>> list(generate_bins(size=-10, count=-19))
    Traceback (most recent call last):
        ...
    ValueError: size and count > 0
    >>> list(generate_bins(size=-10, count=19))
    Traceback (most recent call last):
        ...
    ValueError: size > 0
    >>> list(generate_bins(size=10, count=-19))
    Traceback (most recent call last):
        ...
    ValueError: count > 0
    """
    if size <= 0 and count <= 0: 
        raise ValueError(f"size and count > 0")
    if size <= 0: 
        raise ValueError(f"size > 0")
    if count<= 0: 
        raise ValueError(f"count > 0")
    number_bins = math.ceil(count / size)
    for i in range(1, number_bins):
        yield size

    remainder = count % size
    if remainder == 0:
        yield size
    else:
        yield remainder

if __name__ == "__main__":
    import doctest
    doctest.testmod()