更新时间:2024-05-16 来源:黑马程序员 浏览量:
在Python中进行单元测试通常使用内置的unittest模块。下面是一个简单的示例,演示如何使用unittest编写和运行单元测试:
import unittest # 要测试的函数 def add(x, y): return x + y # 编写测试用例 class TestAddFunction(unittest.TestCase): def test_add_positive_numbers(self): self.assertEqual(add(2, 3), 5) def test_add_negative_numbers(self): self.assertEqual(add(-2, -3), -5) def test_add_mixed_numbers(self): self.assertEqual(add(2, -3), -1) # 运行测试 if __name__ == '__main__': unittest.main()
在这个示例中,我们定义了一个名为add的函数,然后使用unittest.TestCase编写了一些测试用例。每个测试用例都是一个方法,方法名以test开头。在每个测试方法中,我们使用断言来检查函数的输出是否符合预期。
当运行这个脚本时,unittest会自动运行所有测试用例,并输出测试结果。
我们也可以使用其他的单元测试框架,比如pytest,它提供了更简洁、更灵活的语法来编写测试用例。