简化代码；完善测试用例