掌握Python单元测试（Python单元测试与unittest入门教程）

为什么需要单元测试？

• 提前发现 bug，减少后期调试成本
• 提高代码可维护性和可读性
• 支持重构：有测试保障，修改代码更安心
• 是实现自动化测试基础的关键一步

使用 unittest 编写第一个测试

Python 内置了 unittest 模块，无需安装第三方库即可开始unittest教程的学习。下面我们通过一个简单例子来演示如何编写和运行测试。

步骤1：创建被测试的函数

假设我们有一个计算两个数相加的函数：

# math_utils.pydef add(a, b):    return a + b

步骤2：编写测试用例

创建一个名为 test_math_utils.py 的文件，使用 unittest 来测试 add 函数：

# test_math_utils.pyimport unittestfrom math_utils import addclass TestMathUtils(unittest.TestCase):    def test_add_positive_numbers(self):        result = add(2, 3)        self.assertEqual(result, 5)    def test_add_negative_numbers(self):        result = add(-1, -1)        self.assertEqual(result, -2)    def test_add_zero(self):        result = add(0, 5)        self.assertEqual(result, 5)if __name__ == '__main__':    unittest.main()

运行测试

在终端中执行以下命令：

python test_math_utils.py

如果一切正常，你会看到类似如下的输出：

...----------------------------------------------------------------------Ran 3 tests in 0.001sOK

常用断言方法

unittest.TestCase 提供了多种断言方法，常用的包括：

• assertEqual(a, b)：判断 a 是否等于 b
• assertTrue(x)：判断 x 是否为 True
• assertFalse(x)：判断 x 是否为 False
• assertRaises(exception, callable, *args)：判断调用是否会抛出指定异常

小结

通过本教程，你已经掌握了 Python单元测试 的基本概念和使用方法。利用 unittest 模块，你可以轻松为自己的代码编写可靠的测试用例，这是迈向高质量软件开发的重要一步。记住，良好的测试习惯是每位开发者应具备的核心技能，也是构建 自动化测试基础 的关键。

现在就动手试试吧！为你的下一个 Python 项目添加单元测试，体验代码更健壮、开发更自信的感觉。