如何在Python中为不同实现复用同一单元测试？

这个需求完全可以实现，而且有几种简洁的方式来搞定，我帮你梳理下最适合的两种方案：

方案一：抽象基类+子类指定测试函数（适合有特定测试的场景） #

你的思路方向是对的，只是原代码里有几个小问题需要调整：比如抽象测试类必须继承unittest.TestCase，还有继承顺序的问题。优化后的写法可以让子类只需要指定要测试的函数，不用写setUp，代码更清爽：

import unittest

# 先定义你的两个测试函数示例
def function_a(input_str):
    return input_str == "input 2" or input_str == "specific input"

def function_b(input_str):
    return input_str == "input 2" or input_str == "specific input"

# 抽象测试基类，核心是继承unittest.TestCase
class BaseFunctionTest(unittest.TestCase):
    # 子类必须覆盖这个类属性，指定要测试的目标函数
    target_function = None

    def setUp(self):
        # 做个校验，防止子类忘记指定函数
        self.assertIsNotNone(self.target_function, 
                            "请在子类中设置target_function属性指定要测试的函数")

    # 通用测试用例
    def test_generic_input_one(self):
        result = self.target_function("input 1")
        self.assertFalse(result)

    def test_generic_input_two(self):
        result = self.target_function("input 2")
        self.assertTrue(result)

# 测试function_a的子类，只需要指定target_function即可
class TestFunctionA(BaseFunctionTest):
    target_function = function_a

    # 可以添加function_a专属的测试用例
    def test_specific_input_for_a(self):
        result = self.target_function("specific input")
        self.assertTrue(result)

# 测试function_b的子类，同理
class TestFunctionB(BaseFunctionTest):
    target_function = function_b

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

这个方案的好处是：

• 通用测试逻辑完全复用，子类只需要关注专属测试
• 基类的setUp可以做校验，避免子类漏写目标函数
• 结构清晰，每个函数的测试用例都集中在对应的子类里

方案二：参数化测试（适合无专属测试的场景） #

如果所有函数的测试用例完全一致，不需要专属测试，那用参数化的方式会更紧凑，甚至不用写多个子类。可以用unittest自带的subTest来实现（不需要额外安装库）：

import unittest

def function_a(input_str):
    return input_str == "input 2" or input_str == "specific input"

def function_b(input_str):
    return input_str == "input 2" or input_str == "specific input"

class TestAllFunctions(unittest.TestCase):
    # 把所有要测试的函数放在一个列表里
    FUNCTIONS_TO_TEST = [function_a, function_b]
    # 通用测试用例集合
    TEST_CASES = [
        ("input 1", False),
        ("input 2", True),
        ("specific input", True)
    ]

    def test_all_functions(self):
        for func in self.FUNCTIONS_TO_TEST:
            # 用subTest标记当前测试的函数，出错时能明确看到是哪个函数的问题
            with self.subTest(function_name=func.__name__):
                for input_str, expected in self.TEST_CASES:
                    result = func(input_str)
                    self.assertEqual(result, expected, 
                                    f"测试输入[{input_str}]时失败")

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

这个方案的优势是：

• 代码量更少，所有测试逻辑集中在一个类里
• 新增函数只需要往FUNCTIONS_TO_TEST里加就行，不用新建子类
• subTest会帮你区分每个函数的测试结果，出错时能快速定位

对你原代码的小修正 #

顺便提下你原代码里的两个小问题：

1. 抽象测试类必须继承unittest.TestCase，否则unittest不会识别里面的测试方法
2. test_specific_input里的写法有误：result = self.assertTrue(...)是错的，self.assertTrue没有返回值，应该先调用函数拿到结果，再断言：

   def test_specific_input(self):
       result = self.function("specific input")
       self.assertTrue(result)
   

内容的提问来源于stack exchange，提问作者Sylhare