dataset/problem_5/test.py

import unittest
from dataset.problem_5.buggy import chunk_list


class TestChunkList(unittest.TestCase):
    def test_even_chunks(self):
        self.assertEqual(chunk_list([1, 2, 3, 4], 2), [[1, 2], [3, 4]])

    def test_partial_tail_chunk(self):
        self.assertEqual(chunk_list([1, 2, 3, 4, 5], 2), [[1, 2], [3, 4], [5]])


if __name__ == "__main__":
    unittest.main()