Refactor unittest

Unittest感觉很混乱.所以简单重构了下