Chunky chunks

chunk.py

def chunks(iterable, chunk_size):
    """
    Yield successive chunk_size chunks from iterable
    """
    for i in range(0, len(iterable), chunk_size):
        yield iterable[i:i + chunk_size]


class Chunky:
    """
    Chunk up an iterable to list of lists, without the use of yield / generator.
    """
    def __init__(self, iterable, chunk_size):
        self.iterable = iterable
        self.chunk_size = int(chunk_size)
        self.chunks = []

        for i in range(0, len(self.iterable), self.chunk_size):
            self.chunks.append(list(self.iterable[i:i+self.chunk_size]))

            
from unittest import TestCase


class ChunkyTest(TestCase):

    def setUp(self):
        self.expected_result = [
            [1, 2, 3],
            [4, 5, 6],
            [7, 8, 9],
            [10]
        ]

    def test_returns_expected_result_from_list(self):
        input_list = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

        chunky = Chunky(input_list, 3)

        self.assertEqual(chunky.chunks, self.expected_result)

    def test_returns_expected_result_from_tuple(self):
        input_tuple = (1, 2, 3, 4, 5, 6, 7, 8, 9, 10)

        chunky = Chunky(input_tuple, 3)

        self.assertEqual(chunky.chunks, self.expected_result)