Python 中如何编写测试用例 (TestCase)？-JavaScript中文网-JavaScript教程资源分享门户

推荐答案

在 Python 中，编写测试用例通常使用 unittest 模块。以下是一个简单的示例，展示了如何编写测试用例：

// www.javascriptcn.com code example
import unittest

# 被测试的函数
def add(a, b):
    return a + b

# 测试用例类
class TestAddFunction(unittest.TestCase):
    
    def test_add_positive_numbers(self):
        self.assertEqual(add(1, 2), 3)
    
    def test_add_negative_numbers(self):
        self.assertEqual(add(-1, -2), -3)
    
    def test_add_zero(self):
        self.assertEqual(add(0, 0), 0)

# 运行测试
if __name__ == '__main__':
    unittest.main()

本题详细解读

1. 导入 `unittest` 模块

unittest 是 Python 标准库中的一个模块，专门用于编写和运行测试用例。首先需要导入该模块。

import unittest

2. 定义被测试的函数

在编写测试用例之前，需要先定义要测试的函数。例如，我们定义了一个简单的 add 函数，用于计算两个数的和。

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

3. 创建测试用例类

测试用例类需要继承 unittest.TestCase。在这个类中，可以定义多个测试方法，每个方法对应一个测试用例。

// www.javascriptcn.com code example
class TestAddFunction(unittest.TestCase):
    
    def test_add_positive_numbers(self):
        self.assertEqual(add(1, 2), 3)
    
    def test_add_negative_numbers(self):
        self.assertEqual(add(-1, -2), -3)
    
    def test_add_zero(self):
        self.assertEqual(add(0, 0), 0)

4. 使用断言方法

在测试方法中，使用 self.assertEqual() 等断言方法来验证函数的输出是否符合预期。assertEqual 是 unittest 提供的一个常用断言方法，用于比较两个值是否相等。

5. 运行测试

最后，使用 unittest.main() 来运行测试用例。当脚本直接运行时，unittest.main() 会自动查找并执行所有继承自 unittest.TestCase 的测试类中的测试方法。

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

6. 运行结果

运行上述代码后，如果所有测试用例都通过，输出将显示 OK。如果有测试用例失败，输出将显示失败的详细信息，帮助开发者定位问题。

...
----------------------------------------------------------------------
Ran 3 tests in 0.001s

OK

通过这种方式，可以系统地编写和运行测试用例，确保代码的正确性和稳定性。

Python 中如何编写测试用例 (TestCase)？