调整测试代码结构