Question

我目前正在实现一种在值，边界处理等之间进行插值的表，并希望为它编写单元测试。

有很多可能的情况，例如这些表只能有一个/多个行/列和不同的边界条件。所有这些场景中的表都应该通过相同的单元测试集。

现在，我正在为其中一个案例编写一个基类，并从中推导出覆盖setUpClass()方法。然而，这对于测试所有组合来说是乏味的。

是否生成TestCase类，使用不同的参数动态运行它们。

在搜索问题时，我发现最好的事情是重载load_tests()并逐个添加所有测试方法（这意味着在运行每个测试方法之前设置对象，而不是每个方案一次）。

Answer 1

感谢@jonrsharpe的提示，我设法做到了。

我的解决方案是动态创建类，然后使用load_tests将它们添加到TestSuite：

def create_test_case(testcase_name, scenario_data):
    class ScenarioTestCase(BaseTestCase):
        @classmethod
        def setUpClass(cls):
            cls.fillClassVariables(scenario_data)
    return_class = ScenarioTestCase
    return_class.__name__ = testcase_name #for separating the results
    return return_class


def load_tests(loader, tests, pattern):
    list_scenario_names = [...]
    list_scenario_data = [...]

    loader = unittest.TestLoader()
    tests = TestSuite()
    for scenario_name, scenario_data in zip(list_scenario_names, list_scenario_data):
        tests.addTests(loader.loadTestsFromTestCase(
            create_test_case(scenario_name, scenario_data)))
    return tests

这样，TestCase是用不同的参数动态创建的，并在PyCharm的Test Runner选项卡中单独列出。

Answer 2

尽管这是一个可以接受的答案，但我想记录一下如何通过子类进行此操作，因为花了很长时间才能找到这种解决方案的细节。在我的案例中，目标是使TestCase类具有几种方法来具有类固定装置（在文件系统上设置文件以测试cli工具），并在参数化方面稍加改动。

# Inherit from object to not run this abstract class
class TemporaryFolderClassSetup(object):

    @classmethod
    def setUpClass(cls):
        try:
            cls._root = tempfile.mkdtemp()
            # .. create some files here ...
            # allow subclasses to manipulate file tre
            cls.prepare_directory(cls._root)
        except Exception as e:
            cls.tearDownClass()
            raise

    @classmethod
    def tearDownClass(cls):
        shutil.rmtree(cls._root)

    @classmethod
    def prepare_directory(cls, root):
        pass

    # test methods that will run in each subclass
    def testX():
        pass

# scenariotest, must inherit from superclass first, before unittest.TestCase)
class ScenarioTests(TemporaryFolderClassSetup, unittest.TestCase):

    # ... called by parent class
    @classmethod
    def prepare_directory(cls, root):
        pass

    # test only for this scenario
    def testY(self):
        pass

如何在Python（2.7）单元测试中参数化setUpClass（）

2 个答案: