conftest.py

import shutil
import sys

import pytest


@pytest.fixture(autouse=True, scope="session")
def add_doctest_context(doctest_namespace):  # noqa: PT004
    from statemachine.utils import run_async_from_sync

    from statemachine import State
    from statemachine import StateChart
    from statemachine import StateMachine

    class ContribAsyncio:
        """
        Using `run_async_from_sync` to be injected in the doctests to better integration with an
        already running loop, as all of our examples are also automated executed as doctests.

        On real life code you should use standard `import asyncio; asyncio.run(main())`.
        """

        def __init__(self):
            self.run = run_async_from_sync

    doctest_namespace["State"] = State
    doctest_namespace["StateChart"] = StateChart
    doctest_namespace["StateMachine"] = StateMachine
    doctest_namespace["asyncio"] = ContribAsyncio()


def pytest_ignore_collect(collection_path, config):
    if sys.version_info >= (3, 10):  # noqa: UP036
        return None

    if "django_project" in str(collection_path):
        return True


@pytest.fixture(scope="session")
def has_dot_installed():
    return bool(shutil.which("dot"))


@pytest.fixture()
def requires_dot_installed(request, has_dot_installed):
    if not has_dot_installed:
        pytest.skip(f"Test {request.node.nodeid} requires 'dot' that is not installed.")