Feature/test and clean

添加了一些单元测试 & 修改了一些代码