优化单元测试代码