从由 pytest 夹具生成的工厂产生

Question

以下 pytest 夹具

@pytest.fixture(scope="module")
def gen_factory():
    def _gen_factory(owner: str):

        value = ...

        yield value

        print('gen teardown')

    yield _gen_factory

产生多次使用的工厂。

在一个测试用例中，我使用该夹具创建两个工厂并使用它们生成一些值：

@pytest.mark.asyncio
def test_case(gen_factory):
    gen1 = gen_factory('owner1')
    gen2 = gen_factory('owner2')
    val1 = next(gen1)
    val2 = next(gen2)

    ...

    next(gen1)
    next(gen2)

发生的情况是 print('gen teardown') 仅被调用一次，然后循环关闭，第二个 next() 调用引发 StopIteration 错误。

我在这里缺少什么？为什么第二次打印没有发生？

Answer 1

错误发生在第二次调用next(gen1) 的（最后一个）行上，在它执行最后一个语句（next(gen2)）之前。迭代器已用尽。这就是生成器的工作原理。 StopIteration 在生成器末尾引发，（这就是打印第一个的原因！）一旦它们没有要迭代的项目，除非将默认值传递给内置 @987654325 @ 功能。 StopIteration 使 for 循环停止。尽管生成器可以以不同的方式使用，但它们的使用主要是在 for 循环中进行迭代。这就是为什么默认情况下会引发此“异常”。

如果您不想更改夹具中的任何内容，只需将最后一次调用更改为next：

next(gen1, None)

使用默认值不会引发StopIteration。

另一种方法是让_gen_factory 成为上下文管理器：

from contextlib import contextmanager

def gen_factory():
    @contextmanager
    def _gen_factory(owner):
        value = owner
        yield value
        print('gen teardown')
    yield _gen_factory
    

def test_case(genfactory):   
    with genfactory('owner1') as own1, genfactory('owner2') as own2:
        print(own1, own2)

看到这个tutorial on generators。