def test_n_at_a_time(self): result = list(u.n_at_a_time(range(100), 9)) self.assertEqual([9] * 11 + [1], list(map(len, result))) result = list(u.n_at_a_time(range(100), 10)) self.assertEqual([10] * 10, list(map(len, result))) result = list(u.n_at_a_time(range(100), 11)) self.assertEqual([11] * 9 + [1], list(map(len, result))) result = list(u.n_at_a_time(range(100), 12)) self.assertEqual([12] * 8 + [4], list(map(len, result)))