单元测试是确保代码质量的重要手段。本文将介绍如何在 Python 中进行单元测试。

1. 选择测试框架

Python 中有许多单元测试框架，如 unittest、pytest 等。以下是一些常用的测试框架：

  • unittest：Python 标准库中的单元测试框架。
  • pytest：强大的单元测试框架，易于使用。

2. 编写测试用例

编写测试用例时，应遵循以下原则：

  • 单一职责：每个测试用例只测试一个功能点。
  • 可复现：测试用例应能在任何环境下复现。
  • 快速执行：测试用例应尽量快速执行。

以下是一个使用 unittest 框架编写的测试用例示例：

import unittest

def add(a, b):
    return a + b

class TestAddFunction(unittest.TestCase):
    def test_add(self):
        self.assertEqual(add(1, 2), 3)
        self.assertEqual(add(-1, -2), -3)

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

3. 运行测试

运行测试有多种方法，以下是一些常见方法：

  • 使用命令行运行：python -m unittest my_test_module.py
  • 使用 IDE 运行：大多数 IDE 都支持单元测试的运行。

4. 扩展阅读

希望这篇文章能帮助您更好地了解 Python 单元测试。😊

Python unittest