添加单元测试
修复错误算法