修改测试main函数