python 单元测试

参考：https://www.jianshu.com/p/d2bb64a4b819