修改为单元测试方式