去掉main调用，改为单元测试