单元测试中常用的TestLoader类详解

在单元测试中，TestLoader类是一个非常常用的工具，它用于加载和发现测试用例，并将其分组以便于执行，以及生成测试套件供测试运行器使用。

TestLoader类的主要功能有：

1. 自动发现测试用例：TestLoader可以自动扫描指定目录下的测试模块，并根据一定的规则发现其中的测试用例。例如，可以通过指定目录名、文件名前缀、后缀等方式来进行匹配。

2. 加载测试用例：TestLoader根据发现的测试用例生成TestCase对象，并将其添加到测试套件中。这样就可以方便地将多个测试用例组织在一起执行。

3. 测试用例分组：TestLoader可以根据指定的规则对测试用例进行分组，例如按照模块、类名、方法名等进行分组。这样在执行测试用例时，可以方便地只运行某个特定的分组。

4. 动态加载：TestLoader支持在运行时动态加载测试用例。这意味着可以根据实际情况在程序运行过程中动态加载新的测试用例，从而实现更灵活的测试。

下面是一个使用TestLoader的例子：

import unittest

class MyTest(unittest.TestCase):
    def test_add(self):
        self.assertEqual(1 + 1, 2)
    
    def test_subtract(self):
        self.assertEqual(2 - 1, 1)

class MyOtherTest(unittest.TestCase):
    def test_multiply(self):
        self.assertEqual(2 * 3, 6)

# 创建一个TestLoader对象
loader = unittest.TestLoader()

# 使用discover方法自动发现测试用例
suite = loader.discover(start_dir='.', pattern='test_*.py')

# 使用loadTestsFromTestCase方法加载测试用例
suite.addTests(loader.loadTestsFromTestCase(MyOtherTest))

# 运行测试套件
runner = unittest.TextTestRunner()
result = runner.run(suite)

在上面的例子中，首先创建了一个TestLoader对象loader。然后使用loader的discover方法自动发现当前目录下所有名字以test_开头的.py文件，并将其中的测试用例组成一个测试套件suite。

接着，使用loader的loadTestsFromTestCase方法加载了MyOtherTest类中的测试用例，并将它们添加到之前生成的测试套件中。这样一来，执行测试套件时就会包含这两个测试类中的所有测试用例。

最后，使用unittest的TextTestRunner运行测试套件，并将结果输出到控制台。