具有可变数量测试的 PyUnit

Question

我想做的是创建一个文件夹，人们可以在其中放入文件进行测试，并让 pyunit 自动扩展以便将测试作为单独的测试运行。目前，我正在做的是：

class TestName(unittest.testcase):
    def setUp(self):
        for file in os.listdir(DIRECTORY):
            # Setup Tests

    def test_comparison(self):
        for file in os.listdir(DIRECTORY):
            # Run the tests

def suite():
    return unittest.TestSuite([
unittest.TestLoader().loadTestsFromTestCase(TestName),
])

if __name__ == "__main__":
    unittest.TextTestRunner(verbosity=2).run(suite())

显然存在很多问题，例如如果任何测试失败，整个事情都会失败等等。我想做的是设置 pyunit，以便它将运行测试（或测试用例），对于放置在目录中的每个文件。现在，我想我可以做的是创建上面列出的类，仅使用两种方法，但为了成功地做到这一点，我必须添加一个上下文参数，例如：

def setUp(self, filepath):
    # Do stuff

然后我可以将循环放入代码的主要块，并像这样运行测试：

def suite():
    return unittest.TestSuite([
unittest.TestLoader().loadTestsFromTestCase(TestName, file),
])

if __name__ == "__main__":
    for file in DIRECTORY:
        unittest.TextTestRunner(verbosity=2).run(suite(file))

但我不知道如何在不几乎重写漏洞的unittest.TestCase()类的情况下做到这一点。有什么方法可以做到这一点，或者有没有其他方法可以根据文件夹中有多少文件来动态扩展测试/测试用例集？谢谢。

Answer 1

您可以使用 nose 支持 测试生成器。这看起来像这样：

def test_comparison():
    for file in os.listdir(DIRECTORY):
        yield run_cmp_test, file

def run_cmp_test(file):
    # Run the tests

或者你也可以在 py.test 中 funcargs 。

Answer 2

好吧，我想我找到了问题的答案。我想说我正在使用全局命名空间，但由于某种原因，我通常对命名空间的了解在表面上是平淡的。

class TestImage(unittest.TestCase)
    pass #Do stuff, as well as using the filename variable

def suite():
    return unittest.TestSuite([
unittest.TestLoader().loadTestsFromTestCase(TestImage),
])

if __name__ == "__main__":
    for filename in os.listdir(sys.path[0]):
        unittest.TextTestRunner(verbosity=2).run(suite())

这将为目录中的每个文件名运行一个测试用例。 （和文件夹名称）。虽然我不知道为什么我可以在单元测试类中使用文件名。它一定与这些宏加载和运行类的方式有关。

Answer 3

您可以使用 testscenarios 模块来完成此操作，如下所示:

class MyTest(unittest.TestCase):

    scenarios = [os.listdir(sys.path[0])]

    ...