python - 通过 python Pytest 从一个 test.py 文件运行多个测试文件

Question

我在 test 文件夹中有多个测试文件。结构类似于这样:

/test
----test_abc.py
----test_bcd.py
----test_cde.py
----conftest.py

conftest.py 包含运行单元测试所需的所有 spark 上下文初始化。我的问题是我想要一个 test.py内部触发所有 test_abc.py 的文件, test_bcd.py和 test_cde.py .当我们处理 时就变得很容易了utit_test python 模块，但我不知道如何通过 获得它pytest 模块。请让我知道是否需要对该问题进行更多说明。

conftest.py 看起来像这样:

import pytest
from pyspark import SQLContext
from pyspark import SparkConf
from pyspark import SparkContext
from pyspark.streaming import StreamingContext

@pytest.fixture(scope="session")
def spark_context(request):
    conf = (SparkConf().setMaster("local[2]").setAppName("pytest-pyspark-local-testing"))
    request.addfinalizer(lambda: sc.stop())
    sc = SparkContext(conf=conf).getOrCreate()
    return sc

其中一个 test_abc.py 看起来像这样:

import pytest
import os
from pyspark.sql import SQLContext
pytestmark = pytest.mark.usefixtures("spark_context")
def test_get_pc_browser_sql(spark_context):
    "assert something"

Answer 1

我建议只使用 bash 脚本并使用它来调用命令行 python 调用。例如，在你的 bash 脚本中，你可以只写:

pytest test/

运行 test 中的所有测试目录。您还可以添加所有参数命令和在命令行上使用的任何命令。然后，只需为要测试的特定文件或目录集执行 bash 脚本。看 pytest Documentation以供引用。