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

Question

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

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

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

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 个函数在调用所有测试时看起来几乎相同。

我考虑的另一种方法是将模板实现为父类(super class)，然后为我要测试的每个类创建一个子类。子类可以提供用于实例化该类的函数。问题是默认的测试加载器会将(不可用的)基类视为有效的测试用例并尝试运行它，这会失败。

那么，你有什么建议？

附言:我使用的是 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