按自定义顺序运行 pytest 类

Question

我正在用pycharm中的pytest编写测试。测试分为不同的类别。
我想指定必须在其他类之前运行的某些类。
我已经看到了关于stackoverflow的各种问题（例如指定pytest测试从文件运行以及如何在所有其他测试之前运行方法）。
这些问题和其他各种问题都希望选择特定的功能来按顺序运行。我知道，这可以使用夹具或pytest排序来完成。
我不在乎每个类中的哪个函数首先运行。我所关心的只是类按我指定的顺序运行。这可能吗？

Answer 1

可以使用 pytest_collection_modifyitems 挂钩就地修改收集的测试（项）的顺序。这还有一个额外的好处，即不必安装任何第三方库。

使用一些自定义逻辑，这允许按类排序。

假设我们有三个测试类：

1. 测试提取
2. 测试转换
3. 测试加载

还要说，默认情况下，测试的执行顺序是按字母顺序排列的，即：

测试提取 -

由于测试类的相互依赖关系，这对我们不起作用。

我们可以按如下方式向 conftest.py 添加pytest_collection_modifyitems，以强制执行我们想要的执行顺序：

# conftest.py
def pytest_collection_modifyitems(items):
    """Modifies test items in place to ensure test classes run in a given order."""
    CLASS_ORDER = ["TestExtract", "TestTransform", "TestLoad"]
    class_mapping = {item: item.cls.__name__ for item in items}

    sorted_items = items.copy()
    # Iteratively move tests of each class to the end of the test queue
    for class_ in CLASS_ORDER:
        sorted_items = [it for it in sorted_items if class_mapping[it] != class_] + [
            it for it in sorted_items if class_mapping[it] == class_
        ]
    items[:] = sorted_items

关于实施细节的一些评论：

• 测试类可以存在于不同的模块中
• CLASS_ORDER不必详尽无遗。您可以只对那些要强制排序的类进行重新排序（但请注意：如果重新排序，任何未重新排序的类将在任何重新排序的类之前执行）
• 类中的测试顺序保持不变
• 假定测试类具有唯一的名称
• 项目必须就地修改，因此最终项目[：] 分配