如何为不同的类运行相同的测试用例？

Question

我有几个类共享一些不变量并有一个公共接口，我想为每个类自动运行相同的测试。

例如，假设我有几个类实现了不同的数据集分区方法。这里的共同不变性是，对于所有这些类，所有分区上的并集应该等于原始数据集。

我目前拥有的看起来像这样：

class PartitionerInvariantsTests(unittest.TestCase):
    def setUp(self):
        self.testDataSet = range(100) # create test-data-set

    def impl(self, partitioner):
        self.assertEqual(self.testDataSet, 
                         chain.from_iterable(partitioner(self.testDataSet))

然后，我添加了一个不同的函数，该函数调用impl，为我想用该类的实例测试的每个类。当为多个测试功能执行此操作时，此问题变得很明显。假设我有 5 个测试函数和 5 个要测试的类。这将使调用所有测试的 25 个函数看起来几乎相同。

我正在考虑的另一种方法是将模板实现为超类，然后为我要测试的每个类创建一个子类。子类可以提供实例化类的功能。这样做的问题是默认的测试加载器会将（不可用的）基类视为有效的测试用例并尝试运行它，这会失败。

那么，你有什么建议呢？

P.S.：我使用的是 Python 2.6

Answer 1

您可以使用多重继承。

class PartitionerInvariantsFixture(object):
    def setUp(self):
        self.testDataSet = range(100) # create test-data-set
        super(PartitionInvariantsFixture, self).setUp()

    def test_partitioner(self):
        TestCase.assertEqual(self.testDataSet, 
                     chain.from_iterable(self.partitioner(self.testDataSet))

class MyClassTests(TestCase, PartitionerInvariantsFixture):
    partitioner = Partitioner

Answer 2

子类PartitionerInvariantsTests:

class PartitionerInvariantsTests(unittest.TestCase):
    def test_impl(self):
        self.assertEqual(self.testDataSet, 
                         chain.from_iterable(self.partitioner(self.testDataSet))

class PartitionerATests(PartitionerInvariantsTests):

对于您希望测试的每个 Partitioner 类。然后 test_impl 将通过继承为每个 Partitioner 类运行。

根据 Nathon 的评论，您可以通过仅从 object 继承基类来防止对其进行测试：

import unittest

class Test(object):
    def test_impl(self):
        print('Hi')

class TestA(Test,unittest.TestCase):
    pass

class TestB(Test,unittest.TestCase):
    pass

if __name__ == '__main__':
    unittest.sys.argv.insert(1,'--verbose')
    unittest.main(argv = unittest.sys.argv)    

运行 test.py 产生

test_impl (__main__.TestA) ... Hi
ok
test_impl (__main__.TestB) ... Hi
ok

----------------------------------------------------------------------
Ran 2 tests in 0.000s

OK