Files
vllm-omni/tests/helpers/fixtures/run_args.py
T

18 lines
454 B
Python

import pytest
def pytest_addoption(parser):
parser.addoption(
"--run-level",
action="store",
default="core_model",
choices=["core_model", "advanced_model", "full_model"],
help="Test level to run: L2, L3, L4",
)
@pytest.fixture(scope="session")
def run_level(request) -> str:
"""Session test level from ``--run-level`` (see CI five-level docs)."""
return request.config.getoption("--run-level")