根据测试用例修改代码