python - pytest - 如何知道选择了哪些标记

Question

在 pytest 中有什么方法可以知道从命令行选择了哪些标记？

我有一些标记为“缓慢”的测试需要大量处理。我只想在标记慢被激活时处理治疗。

heavy_var = None

def setup_module(module):
    global heavy_var

    # Need help here!?
    if markers["slow"]:
        heavy_var = treatment()


def test_simple():
    pass


@pytest.mark.slow():
def test_slow():
    assert heavy_var.x == "..."

我怎么知道是否选择了慢速标记？当我使用 -m not slow 调用 pytest 时，markers["slow"] 将为 False，否则为 True。

Answer 1

如果您只需要在测试标记为 slow 时运行一些代码被选中时，您可以通过在替换 setup_module 的模块范围的 fixture 中过滤测试项目来做到这一点.示例:

@pytest.fixture(scope='module', autouse=True)
def init_heavy_var(request):
    for item in request.session.items:
        if item.get_closest_marker('slow') is not None:
            # found a test marked with the 'slow' marker, invoke heavy lifting
            heavy_var = treatment()
            break