python - 如何将要测试的函数名称传递给单元 test.TestCase？

Question

我需要测试两个几乎相同的功能，并且看不到复制粘贴我为function1. 我可以以某种方式将函数名称传递给派生自的类unittest.TestCase并设置需要在其中测试的函数setUp吗？

Answer 1

这没有什么诀窍。函数是 Python 中的一等对象，可以像任何其他参数一样传递。所以这会起作用，假设被测函数是'func_one'和'func_two'：

class MyTestCase(unittest.TestCase):

    def generic_test(self, func):
        result = func()
        ... assertions about result....

    def test_func_one(self):
        self.generic_test(func_one)

    def test_func_two(self):
        self.generic_test(func_two)