11 使用unittest进行测试之测试用例和测试套件

在上一篇文章中，我们讨论了如何使用 unittest 框架编写和运行测试。在这篇文章中，我们将深入探讨 unittest 中的测试用例和测试套件，帮助你更有效地组织和运行测试。

测试用例

什么是测试用例？

测试用例是对某个特定功能或代码片段进行检查的最小单元。它是通过 unittest.TestCase 类来定义的。每个测试用例都包含一个或多个测试方法，并且每个方法以 test_ 开头。

创建测试用例

下面是一个简单的示例，展示如何创建一个测试用例。假设我们有一个 Calculator 类，它包含一个 add 方法用于加法运算。

# calculator.py
class Calculator:
    def add(self, a, b):
        return a + b

现在，我们来为这个 add 方法创建一个测试用例。

# test_calculator.py
import unittest
from calculator import Calculator

class TestCalculator(unittest.TestCase):
    def setUp(self):
        """测试用例的准备工作，在每个测试方法调用之前执行"""
        self.calc = Calculator()

    def test_add(self):
        """测试加法功能"""
        self.assertEqual(self.calc.add(1, 2), 3)
        self.assertEqual(self.calc.add(-1, 1), 0)
        self.assertEqual(self.calc.add(-1, -1), -2)

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

在上面的代码中，我们定义了一个名为 TestCalculator 的测试用例。通过重写 setUp 方法，我们在每个测试方法调用前实例化一个 Calculator 对象。

运行测试用例

要运行测试用例，可以直接从命令行执行 python test_calculator.py。如果一切正常，你将看到测试通过的信息。

测试套件

什么是测试套件？

测试套件是一组测试用例的集合，允许你将多个测试组合在一起进行执行。这对于组织大量测试时非常有用。

创建测试套件

下面我们将通过创建一个包含多个测试用例的测试套件来进一步说明。

# test_suite.py
import unittest
from test_calculator import TestCalculator

def suite():
    suite = unittest.TestSuite()
    suite.addTest(unittest.makeSuite(TestCalculator))
    return suite

if __name__ == '__main__':
    runner = unittest.TextTestRunner()
    runner.run(suite())

在这个示例中，我们通过 unittest.TestSuite() 创建了一个测试套件，并使用 addTest 方法将 TestCalculator 测试用例添加到套件中。然后，使用 unittest.TextTestRunner() 来运行这个测试套件。

运行测试套件

与单个测试用例类似，要运行测试套件只需要执行 python test_suite.py。如果所有测试都通过，你同样会看到相应的结果输出。

总结

本文中我们详细讨论了如何使用 unittest 框架创建 测试用例 和 测试套件。通过正确地组织测试用例和使用测试套件，可以让你的测试工作更加高效和系统。掌握这些技巧后，你将为开发有质量保证的应用打下良好的基础。

下一篇文章中，我们将转向 pytest，讨论其安装和配置，帮助你更快速地进行自动化测试。如果你有任何问题或需要进一步的解释，欢迎在评论区交流！