【unittest，一个标准的 Python 库！】

在程序开发过程中，功能验证、bug 排查、代码稳定性保障是必不可少的环节。无论是开发小型工具、业务脚本，还是构建中型项目，都需要可靠的测试工具保证代码质量。unittest 是 Python 官方内置的标准单元测试框架，无需额外安装、无需复杂配置，遵循业界通用的测试规范，支持用例管理、断言验证、前置后置处理等完整测试能力，是 Python 开发者最基础、最稳定、最通用的测试工具。它不依赖第三方环境，兼容性极强，从日常脚本测试到企业项目回归测试，都能稳定支撑，是 Python 测试体系中最标准、最不可或缺的核心库。

一、库的简介：实际生活中的核心作用

unittest 是 Python 内置标准库，也是官方推荐的单元测试框架，核心作用是对代码进行自动化验证，确保函数、接口、业务逻辑符合预期。在实际生活与开发中，它的用处无处不在：开发计算器时自动验证计算结果、写数据处理脚本时校验数据准确性、开发小工具时确保功能稳定、维护旧代码时快速检测修改是否引发问题。它遵循标准化测试流程，结构清晰、易于维护，所有 Python 环境都能直接运行，是学习测试、编写规范代码、保障项目稳定的必备工具，也是所有测试框架的基础标准。

二、安装库

unittest 是 Python 官方内置库，完全不需要安装，安装 Python 后即可直接导入使用，跨平台、跨版本通用，这是它最大的优势。

三、基本用法（分 4 个核心步骤）

步骤 1：编写待测试函数

新建 tool.py 文件，编写需要测试的业务代码：

python

运行

# 计算两数之和
def add(a, b):
    return a + b

# 判断是否为偶数
def is_even(n):
    return n % 2 == 0

步骤 2：创建测试类并继承 unittest.TestCase

新建测试文件 test_tool.py，创建测试类，必须继承 unittest.TestCase：

python

运行

import unittest
from tool import add, is_even

# 测试类必须继承 TestCase
class TestTool(unittest.TestCase):

步骤 3：编写测试方法（以 test 开头）

所有测试方法必须以 test 开头，使用断言验证结果：

python

运行

    # 测试加法函数
    def test_add(self):
        self.assertEqual(add(1, 2), 3)
        self.assertEqual(add(-1, 1), 0)

    # 测试偶数判断
    def test_is_even(self):
        self.assertTrue(is_even(4))
        self.assertFalse(is_even(5))

步骤 4：运行测试并查看结果

在文件末尾添加运行代码，直接执行文件即可启动测试：

python

运行

if __name__ == '__main__':
    unittest.main()

运行后会自动展示成功、失败、错误的用例数量。

四、高级用法

unittest 提供完整的高级测试能力，满足复杂场景：

1. 前置后置方法：setUp() 和 tearDown() 实现用例前置准备与后置清理；
2. 丰富断言：支持相等、包含、真假、异常等多种断言方式；
3. 测试跳过：用 @unittest.skip 跳过不需要执行的用例；
4. 异常验证：断言函数是否抛出指定异常；
5. 批量执行：自动发现并执行目录下所有测试用例。

高级用法示例（前置 + 异常断言）：

python

运行

import unittest

class TestAdvanced(unittest.TestCase):
    # 前置：每条用例执行前运行
    def setUp(self):
        self.data = [1, 2, 3]

    # 测试异常
    def test_exception(self):
        with self.assertRaises(IndexError):
            self.data[10]

if __name__ == '__main__':
    unittest.main()

五、实际应用场景

unittest 作为标准库，适用场景极广：

1. 个人工具测试：计算器、脚本、小工具功能验证；
2. 学习教学：Python 入门标准测试工具，规范代码习惯；
3. 小型项目：接口、函数、业务逻辑自动化回归测试；
4. 跨环境项目：无第三方依赖，可在任何环境运行测试；
5. 开源项目：标准化测试，兼容性强，便于他人维护。

---

深度案例代码：用户登录功能完整测试（可直接运行）

python

运行

# 业务逻辑：user.py
def login(username, password):
    if not username or not password:
        raise ValueError("用户名和密码不能为空")
    if username == "admin" and password == "123456":
        return "登录成功"
    return "用户名或密码错误"

# 测试用例：test_user.py
import unittest
from user import login

class TestLogin(unittest.TestCase):
    def test_login_success(self):
        self.assertEqual(login("admin", "123456"), "登录成功")

    def test_login_fail(self):
        self.assertEqual(login("admin", "wrong"), "用户名或密码错误")

    def test_empty_input(self):
        with self.assertRaises(ValueError):
            login("", "")

if __name__ == '__main__':
    unittest.main()

unittest 作为 Python 官方标准测试库，凭借无需安装、兼容性强、规范稳定的特点，成为所有开发者必须掌握的基础工具。它结构清晰、使用简单，既能满足日常小脚本测试，也能支撑项目级自动化测试，是编写高质量、高稳定性 Python 代码的基石。掌握 unittest，不仅能提升代码质量，还能建立标准化开发思维，为学习其他测试框架打下坚实基础。

看完这篇文章，你已经掌握了 unittest 的核心用法，它能帮你快速写出稳定可靠的测试用例，让代码更安全。你平时写代码时会写测试用例吗？最想测试自己写的哪个小功能呢？不妨用 unittest 动手试试，感受标准测试带来的高效与安心。