部分改进了自动生成单元测试代码