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