改进单元测试代码