修改测试函数