Python中如何使用unittest模块？

在Python中使用unittest模块进行单元测试，需通过继承unittest.TestCase编写测试类，利用断言方法验证代码逻辑，并结合高级功能（如异常测试、环境管理）和最佳实践（如覆盖率检查、TDD）提升测试质量。

• 编写测试类：继承unittest.TestCase，测试方法需以test_开头。
• 断言方法：使用assertEqual、assertTrue等验证结果。import unittestdef add(a, b): return a + bclass TestAddFunction(unittest.TestCase): def test_add_positive_numbers(self): self.assertEqual(add(2, 3), 5) def test_add_negative_numbers(self): self.assertEqual(add(-1, -1), -2)if __name__ == '__main__': unittest.main()

• 测试异常：通过assertRaises验证是否抛出指定异常。def test_raise_exception(self): with self.assertRaises(ValueError): raise ValueError("Test exception")
• 捕获输出：重定向sys.stdout捕获打印内容。import io, sysdef test_output_to_stdout(self): captured_output = io.StringIO() sys.stdout = captured_output print("Hello, World!") sys.stdout = sys.__stdout__ self.assertEqual(captured_output.getvalue().strip(), "Hello, World!")
• 环境管理：

  setUp()：每个测试方法前执行，用于初始化资源。

  tearDown()：每个测试方法后执行，用于清理资源。

  def setUp(self): self.data = [1, 2, 3]def tearDown(self): del self.datadef test_data_manipulation(self): self.data.append(4) self.assertEqual(len(self.data), 4)

• 方法命名：测试方法必须以test_开头，否则不会被执行。
• 测试独立性：每个测试方法应独立运行，避免依赖其他测试的状态。
• 断言选择：根据场景选择合适的断言方法：

  assertEqual(a, b)：验证a == b。

  assertTrue(x)：验证x为True。

  assertRaises(exc, fun, *args)：验证函数是否抛出指定异常。

• 测试覆盖率：使用coverage.py检查代码覆盖率，确保测试覆盖所有分支。pip install coveragecoverage run -m unittest discovercoverage report
• 测试驱动开发（TDD）：先编写测试用例，再实现功能代码，确保代码可测试性。
• 持续集成（CI/CD）：将测试集成到CI/CD流程中（如GitHub Actions），自动运行测试。

• 分组组织：按功能模块或测试类型（单元测试、集成测试）分组测试文件和类，提高可维护性。tests/ ├── unit/ │ ├── test_math.py │ └── test_string.py └── integration/ └── test_api.py
• 测试发现：使用unittest.discover自动发现并运行测试。if __name__ == '__main__': unittest.main(module='tests.unit.test_math') # 运行指定模块

unittest是Python标准库中功能全面的单元测试框架，通过基础断言、高级功能（异常测试、环境管理）和最佳实践（覆盖率、TDD、CI/CD）的结合，可有效保障代码质量。合理组织测试用例、遵循命名规范和独立性原则，能进一步提升测试效率和可维护性。