python单元测试怎么写?自动化脚本系列(6)

通过前面几期的学习，我们基本掌握了如何应对工作中，需要重复执行，手动书写shell命令和浪费时间的操作。想必大家都已经开始实验编写python自动化脚本去节省时间，提高工作效率了。

不过工作中，或者编写python脚本时，还是存在一个痛点，那就是调试。。想必如果大家看过前面几期的内容，跟着编写过，尝试过写一些脚本的话，那么应该都能体会到这一点，那就是，每次都得写完脚本，使用python 命令尝试执行一下，然后如果不通过，再去排除是哪里写的有问题，这无疑很浪费时间。

因此，这一期，我们来谈论一下，如何使用单元测试和其它测试手段来保证python 脚本在编写过程中的正确性。

先看下面的简单示例：

如上图，是我们在linux 上做的一个最简单的单元测试，我们有一个rearrange_name的函数，功能是，对给定格式的字符串进行前后替换，输出结果。

然后我们创建了test.py ， 在test.py中对函数进行测试。

一般在python 单元测试中，我们为了区分正式文件和测试文件，经常对测试文件使用"_test" 后缀进行命名。

但是其中有两个痛点：

1. 我们自己的test文件看起来可以随意执行，无限制，就是个可执行脚本
2. 只能使用print 等一些简单的验证语句，无法大规模进行测试

下面，就推出我们python自带的unittest 模块，它是专为单元测试准备的，里面包含了所有我们想要的功能，可以解决上面的所有痛点。

示例如下图所示：

如上图代表着一个正规的单元测试，我们使用了python自带的unittest 模块，其中有几个需要注意的点，我把源码贴在下面，并通过注释的方式给大家讲解。（不考虑缩进，因为头条代码缩进太长了，看不清）

import unittest 
// 首先就是导入unittest模块
// 然后我们自定义一个类，继承unittest.TestCase
class TestRearrange(unittest.TestCase):
// 这个类中所有以test开头的自定义函数，均
// 会在我们执行单元测试时，被当作测试用例
def test_basic(self):
// ....	省略部分
   self.assertEqual
// 使用单元测试模块提供的assert 方法

unittest.main()  // 执行单元测试入口函数

上面测试的执行结果如下图所示：

大家可以看到，我们有 一些运行测试的基本信息打印，然后也有你跑了几个测试，失败了几个，成功了几个等信息。

然后，我们尝试在上面的基础上，再加一个测试，具体如下：

Edge Condtion（边缘场景或者边缘条件）， 是我们编写单元测试经常需要考虑和测试的重点。

然后运行测试后，结果如下：

我们发现新写的测试，测出来一个bug， 我们的代码的确没有对入参为空字符串的场景做任何处理，导致产生了None。

然后我们需要修改实际生产代码如下：

然后我们再次执行测试结果如下：

我们发现bug 被修复了。

以上就是单元测试模块的简单使用方法，今天就先讲到这里，如果大家有兴趣，可以给我关注留言，留下你想听的内容，我随时安排。

Tips: 我其实本职工作目前是在做数据库内核开发和Linux 操作系统等开发工作，对C++，Rust都很熟悉，工作中常用的也是这些技术栈，Python等其它工具也是偶尔工作中使用，给大家介绍python也只是有一篇文章发了shell脚本，发现大家貌似很喜欢看，才给大家讲解的。。

所以大家不用客气，只要加我，关注我，给我留言想学的内容，我都会安排起来的。。。我可以说啥都会，图像渲染，多线程编程，cuda，还有我本职工作相关内容，如果大家感兴趣，关注我给我留言即可。