Python 自动化测试入门：编写与运行测试用例

众所周知，自动化测试的优势是显而易见的，它可以大大节省我们的时间，提高我们的工作效率。那么 Python 如何做自动化测试呢？本文将用 Python 编写一个简单的测试用例，并指导大家编写自动化测试的代码。

1. 为什么需要自动化测试？

相信大家都遇到过这样的问题，你需要一遍一遍地在你的软件中输入命令，来测试软件的功能是否正常，这个过程是很枯燥无味的。如果能写一小段代码用来测试你的代码该有多好啊！因此我们就不得不需要用到自动化测试。试想一下，无论你对程序做了什么修改，或者添加了什么新东西，你只要'跑一下你的自动化测试'，而这些测试能确认程序依然能正确运行。这些自动测试不会抓到所有的 bug，但可以让你无需重复输入命令运行你的代码，从而为你节约很多时间。

2. 用 Python 编写测试用例

下面以一段非常简单的代码为例，用 Python 写一个简单的测试，这个测试将建立在一个项目骨架上面。首先从你的项目骨架创建一个叫做 ex47 的项目。下面是你要采取的步骤：

复制 skeleton 到 ex47
将所有的 NAME 重命名为 ex47
修改所有文件中 NAME 为 ex47
最后删除所有的 *.pyc 文件

接下来创建一个简单的 ex47/game.py 文件，里边放一些用来被测试的代码。我们现在放一个小 class 进去，用来作为我们的测试对象：

class Room(object):
    def __init__(self, name, description):
        self.name = name
        self.description = description
        self.paths = {}

    def go(self, direction):
        return self.paths.get(direction, None)

    def enter(self):
        print(self.description)

准备好了这个文件，接下来把测试骨架改成这样子：

import unittest
from game import Room

class TestGame(unittest.TestCase):
    def test_room_name(self):
        room = Room("North", "A dark tunnel")
        self.assertEqual(room.name, "North")

    def test_room_description(self):
        room = Room("South", "A sunny beach")
        self.assertEqual(room.description, "A sunny beach")

    def test_room_paths(self):
        room = Room("East", "To the castle")
        room.paths["east"] = "Castle"
        self.assertEqual(room.go("east"), "Castle")

这个文件导入了你在 ex47.game 创建的 Room 这个类，接下来我们要做的就是测试它。于是我们看到一系列的以 test_ 开头的测试函数，它们就是所谓的'测试用例 (test case)'，每一个测试用例里面都有一小段代码，它们会创建一个或者一些房间，然后去确认房间的功能和你期望的是否一样。它测试了基本的房间功能，然后测试了路径，最后测试了整个地图。

这里最重要的函数是 assert_equal，它保证了你设置的变量，以及你在 Room 里设置的路径和你的期望相符。如果你得到错误的结果的话，nosetests 将会打印出一个错误信息，这样你就可以找到出错的地方并且修正过来。

3. 测试指导

在写测试代码时，你可以照着下面这些不是很严格的指导来做：

测试脚本要放到 tests/ 目录下，并且命名为 BLAH_tests.py，否则 nosetests 就不会执行你的测试脚本了。这样做还有一个好处就是防止测试代码和别的代码互相混掉。
为你的每一个模组写一个测试。
测试用例（函数）保持简短，但如果看上去不怎么整洁也没关系，测试用例一般都有点乱。
就算测试用例有些乱，也要试着让他们保持整洁，把里边重复的代码删掉。创建一些辅助函数来避免重复的代码。当你下次在改完代码需要改测试的时候，你会感谢我这一条建议的。重复的代码会让修改测试变得很难操作。
最后一条是别太把测试当做一回事。有时候，更好的方法是把代码和测试全部删掉，然后重新设计代码。

4. 显示结果

$ nosetests
...
----------------------------------------------------------------------
Ran 3 tests in 0.008s
OK

如果一切工作正常的话，你看到的结果应该就是这样。试着把代码改错几个地方，然后看错误信息会是什么，再把代码改正确。

5. 现代测试工具推荐

虽然 nosetests 是一个经典的测试运行器，但在现代 Python 开发中，pytest 和标准库 unittest 更为流行。pytest 语法更简洁，支持丰富的插件；unittest 则是 Python 内置的标准库，无需安装额外依赖。对于新项目，建议优先选择 pytest，因为它对异常处理和参数化测试的支持更加友好。

此外，持续集成（CI）环境通常会自动运行这些测试脚本，确保每次代码提交都不会破坏现有功能。结合 Git 钩子（Git Hooks）可以在提交前自动触发测试，进一步提高代码质量。

6. 总结

自动化测试是软件开发中不可或缺的一环。通过编写可复用的测试脚本，开发者可以快速验证功能变更的正确性，减少回归测试的成本。掌握 Python 自动化测试的基本流程，包括项目结构搭建、测试用例编写、运行与调试，能够帮助团队构建更稳健的软件系统。