conftest.py

from __future__ import annotations

import sys
from pathlib import Path

import pytest

from manim import config, tempconfig


def pytest_addoption(parser):
    parser.addoption(
        "--skip_slow",
        action="store_true",
        default=False,
        help="Will skip all the slow marked tests. Slow tests are arbitrarily marked as such.",
    )
    parser.addoption(
        "--show_diff",
        action="store_true",
        default=False,
        help="Will show a visual comparison if a graphical unit test fails.",
    )
    parser.addoption(
        "--set_test",
        action="store_true",
        default=False,
        help="Will create the control data for EACH running tests. ",
    )


def pytest_configure(config):
    config.addinivalue_line("markers", "skip_end_to_end: mark test as end_to_end test")


def pytest_collection_modifyitems(config, items):
    if not config.getoption("--skip_slow"):
        return
    else:
        slow_skip = pytest.mark.skip(
            reason="Slow test skipped due to --disable_slow flag.",
        )
        for item in items:
            if "slow" in item.keywords:
                item.add_marker(slow_skip)


@pytest.fixture(scope="session")
def python_version():
    # use the same python executable as it is running currently
    # rather than randomly calling using python or python3, which
    # may create problems.
    return sys.executable


@pytest.fixture
def reset_cfg_file():
    cfgfilepath = Path(__file__).parent / "test_cli" / "manim.cfg"
    original = cfgfilepath.read_text()
    yield
    cfgfilepath.write_text(original)


@pytest.fixture
def using_opengl_renderer():
    """Standard fixture for running with opengl that makes tests use a standard_config.cfg with a temp dir."""
    with tempconfig({"renderer": "opengl"}):
        yield
    # as a special case needed to manually revert back to cairo
    # due to side effects of setting the renderer
    config.renderer = "cairo"