conftest

This module contains the fixtures for the tests.

spark()

Fixture for creating a Spark session.

Yields:

Name	Type	Description
SparkSession	Any	Spark session object.

Source code in code\tests\conftest.py

@pytest.fixture(scope="session")
def spark() -> Generator[Any, Any, Any]:
    """Fixture for creating a Spark session.

    Yields:
        SparkSession: Spark session object.
    """
    findspark.init()
    spark = SparkSession.builder.appName("pytest").getOrCreate()

    yield spark
    spark.stop()