修改get_nb_groups的分组算法，并添加测试