Python 单元测试入门指南

单元测试是软件开发中非常重要的一环，它可以帮助我们确保代码的质量和稳定性。本文将为您介绍如何使用 Python 进行单元测试。

常见单元测试框架

在 Python 中，有几个常用的单元测试框架：

unittest: Python 标准库中的单元测试框架。
pytest: 一个成熟、功能强大的单元测试框架。
nose: 另一个流行的单元测试框架，但近年来已被 pytest 取代。

安装测试框架

以下是使用 pip 安装 pytest 的示例：

pip install pytest

编写测试用例

以下是一个简单的测试用例示例：

def test_add():
    assert 1 + 1 == 2

在这个例子中，我们测试了两个数字相加的结果是否等于 2。

运行测试

在命令行中，切换到包含测试文件的目录，然后运行以下命令：

pytest

如果一切正常，pytest 将输出所有通过和失败的测试用例。

断言

断言是单元测试的核心。以下是一些常用的断言方法：

assertEqual(a, b): 断言 a 和 b 是否相等。
assertNotEqual(a, b): 断言 a 和 b 不相等。
assertTrue(condition): 断言条件为真。
assertFalse(condition): 断言条件为假。

测试夹具（Fixtures）

测试夹具是用于设置测试环境的一些配置，例如数据库连接、文件系统等。pytest 提供了丰富的夹具功能。

以下是一个使用 pytest 的测试夹具示例：

def setup_module(module):
    # 在模块开始时执行
    print("Setting up module...")

def teardown_module(module):
    # 在模块结束时执行
    print("Tearing down module...")