我如何（或能够）从鼻子测试中调用生成器以由鼻子处理？

Question

我正在尝试创建一个通用的数据驱动测试处理程序，我可以从我的鼻子测试中调用它。我的测试文件看起来像：

import ScenarioHandler

def test_foo():
    scenario = ScenarioHandler(__test_foo, [1, 2])
    scenario.run()

def __test_foo(var):
    assert var % 2 == 0, 'Odd!'

ScenarioHandler 会是这样的：

class ScenarioHandler(object):

    def __init__(self, test, args):
        self.test = test
        self.args = args

    def run(self):
        for arg in self.args:
            yield self.test, arg

我遇到的问题是我无法弄清楚如何从 ScenarioHandler.run() 备份生成器鼻子。我尝试过从 test_foo() 中的 run() 返回生成器，但这也不好。这可能吗？

Answer 1

我不知道这个鼻子的东西到底是如何工作的。

但你给出的例子不能工作。

您有调用树

test_foo() -> 运行()。

由于 run() 返回一个生成器，该生成器直到迭代开始才启动，因此您应该这样做。

要么做

def test_foo():
    scenario = ScenarioHandler(__test_foo, [1, 2])
    for f, a in scenario.run():
        f(a)

要么更好，因为nose已经通过使用func.func_code.co_flags & 显式测试给定函数的标志来提供这一点。 CO_GENERATOR != 0，

def test_foo():
    scenario = ScenarioHandler(__test_foo, [1, 2])
    for f, a in scenario.run():
        yield f, a

以便让 nose 进行这些调用。

（可能有一个更短的事情，即使用

test_meth = ScenarioHandler(__test_foo, [1, 2]).run

它，唉，不起作用，因为方法显然没有被正确检测为有效的调用目标。）

但是您可以添加一个“测试制作者”到你的班级。

# as method:
    def make_test(self):
        def test_generator():
            for item in self.run():
                yield item
        return test_generator

所以你可以这样做，

test_foo = ScenarioHandler(__test_foo, [1, 2]).make_test()

但可能 nose 提供了更好的方法，通过子类化 TestCase 和 TestSuite 将测试封装在类中。

Answer 2

为什么不使用类似的东西：

def test_foo():
    for var in [1, 2]:
        yield __test_foo, var

另请参阅我的答案中的add_tests(generator)函数。