修改单元测试方法