修改测试代码

修改测试代码