使用 unittest 进行单元测试

单元测试是一种测试代码中独立单元、确保其按预期运行的方法。Python 的 unittest 模块是用于创建和运行单元测试的内置库。

在本课中，您将学习如何使用 unittest 框架创建和执行单元测试。

什么是单元测试

单元测试包括独立测试单个函数或方法，以确保它们运行正确。它有助于在开发过程的早期发现程序缺陷，并确保代码的可靠性。

设置单元测试

unittest 模块包含在标准库中，因此您无需安装任何内容。要使用它，请导入该模块，并通过继承 unittest.TestCase 创建测试用例。

# Setting up unittest
import unittest

class TestExample(unittest.TestCase):
    def test_addition(self):
        self.assertEqual(1 + 1, 2)  # Passes

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

编写测试用例

要编写测试用例，请在测试类中定义一个方法。每个方法都应以 test_ 开头，这样单元测试框架才能将其识别为测试用例。

# Writing test cases
class TestMathOperations(unittest.TestCase):
    def test_multiplication(self):
        self.assertEqual(2 * 3, 6)  # Passes

运行测试

请运行包含测试用例的脚本。单元测试框架会自动发现并执行所有测试方法。

# Running tests
# Save this file and run it using:
# python test_script.py

常用断言

unittest 模块提供了多种断言方法：

• assertEqual(a, b)：检查 a 是否等于 b。
• assertTrue(x)：检查 x 是否为真。
• assertIn(a, b)：检查 a 是否在 b 中。
• assertRaises(exception)：检查是否引发异常。

测试边界情况

测试边界情况非常重要，例如空输入、大数值或无效数据，以确保代码的健壮性。

# Testing edge cases
class TestEdgeCases(unittest.TestCase):
    def test_empty_input(self):
        self.assertEqual(len([]), 0)  # Passes

组织测试文件

对于较大的项目，请将测试用例放在单独的文件夹中，该文件夹通常命名为 tests/，并使用 test_ 前缀命名测试文件。

unittest 中的常见错误

以下是需要避免的一些错误：

• 未使用 test_ 前缀命名测试方法。
• 忘记在测试用例中处理边界情况。
• 开发过程中未定期运行测试。

我们学到了什么

在本课中，您学习了：

• 什么是单元测试，以及它为何重要。
• 如何使用 unittest 模块编写和执行测试用例。
• unittest 框架提供的常用断言。
• 组织测试文件和处理边界情况的最佳实践。

做得很好！让我们继续学习下一个主题。