wangyu
e338f0ece3
Signed-off-by: wangyu <410167048@qq.com> Signed-off-by: [Your Name] <your.email@example.com>
25 lines
664 B
Python
25 lines
664 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project
|
|
|
|
"""Pytest CLI options and the fixtures that expose them."""
|
|
|
|
from __future__ import annotations
|
|
|
|
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")
|