forked from ccf-ai-infra/Intro-ops
25 lines
544 B
Python
25 lines
544 B
Python
from __future__ import annotations
|
|
|
|
import os
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
PYTHON_DIR = ROOT / "python"
|
|
if str(PYTHON_DIR) not in sys.path:
|
|
sys.path.insert(0, str(PYTHON_DIR))
|
|
if str(ROOT) not in sys.path:
|
|
sys.path.insert(0, str(ROOT))
|
|
|
|
|
|
def pytest_addoption(parser):
|
|
parser.addoption("--backend", action="store", default=os.environ.get("CAMP_TEST_BACKEND", "nvidia"))
|
|
|
|
|
|
@pytest.fixture
|
|
def backend(request) -> str:
|
|
return request.config.getoption("--backend")
|
|
|