稍微修改了一下参数以及添加了部分注释
修改了py的测试用例