本系列是对入门书籍《Python编程：从入门到实践》的笔记整理，属于初级内容。标题顺序采用书中标题。

本章主要学习如何使用Python标准库中的unittest模块对代码进行简单的测试。

作为初学者，并非必须为你尝试的所有项目编写测试；但参与工作量较大的项目时，你应对自己编写的函数和类的重要行为进行测试。这样你就能够更加确定自己所做的工作不会破坏项目的其他部分，你就能够随心所欲地改进即有代码。如果不小心破坏了原来的功能，你马上就会知道，从而能够轻松地修复问题。相比于等到Bug出现后再去改，在测试未通过时采取措施要容易得多。而且，如果你想要分享你的项目，有测试的代码更容易让人接受。

一个能通过的测试

以下是一个将用户输入的姓与名拼接的函数：

当然你也可以将if语句下面的代码单独放在一个文件中，并在该文件开头带入get_formatted_name()函数。

对 if __name__ == “__main__” 的补充：

在Python中，模块就是对象，所有模块都有一个内置属性__name__，当该模块被导入时，该模块的__name__属性会被置为模块名，当直接运行该模块，或者说直接运行该文件时，该属性就会使用默认值"__main__"，可以用一句经典的话总结这个用法：

Make a script both importable and executable.

if语句下面的代码相当于对上面的函数的测试，不过这样的测试每次都需要我们自己输入数据，并自己根据结果判断代码是否工作正常，如果代码稍微多一点，稍微复杂一点，这样的测试方法将会很繁琐，所以，我们使用unittest模块了测试代码。

这里先明确两个概念：

单元测试：用于核实函数在某个方面没有问题

测试用例：一组单元测试，这些单元测试一起核实函数在各种情况下的行为都符合要求。

也就是说，你可以将上述代码中的test_first_last_name看做单元测试，而将NamesTestCase看做测试用例。

一般测试文件单独放在一个文件夹中，也可以将测试都放在一个文件中。

为函数编写测试用例，可先导入unittest模块和要测试的函数，再创建一个继承unittest.TestCase的类，并编写一系列方法对函数行为的不同方面进行测试。在测试用，我们使用断言self.assertEqual()(并不是只有这一个断言函数)来判断结果与期望是否相同。在测试类中的每一个测试方法都必须以 test_ 开头，否则将不会被认定是一个单元测试。最后我们通过unittest.main()来运行这个文件中的所有测试。当测试通过时，结果中会先输出一个实心句点，输出几个句点表示通过了几个单元测试，然后输出单元测试数目，最后输出”OK”。

一个不能通过的测试

外国人的名字还有中间名，以上代码并未考虑这个情况。我们通过将上述代码改成含有中间名的版本来演示测试不通过的情况：

第一行输出了一个字母E，traceback指出缺少了参数。如果你检查的条件没错，测试通过了意味着函数的行为是对的，而测试未通过意味着你编写的新代码有错。因此，测试未通过时，不是去修改测试代码，而失去修改你编写的代码。

添加新测试

以下我们将上述的get_formatted_name()函数修改为能自动处理中间名的函数，并在测试文件中添加一个单元测试：