加入测试代码