mirror of
https://github.com/huggingface/transformers.git
synced 2025-08-01 18:51:14 +06:00
32 lines
844 B
Python
32 lines
844 B
Python
# content of conftest.py
|
|
|
|
import pytest
|
|
|
|
|
|
def pytest_addoption(parser):
|
|
parser.addoption(
|
|
"--runslow", action="store_true", default=False, help="run slow tests"
|
|
)
|
|
parser.addoption(
|
|
"--use_cuda", action="store_true", default=False, help="run tests on gpu"
|
|
)
|
|
|
|
|
|
def pytest_configure(config):
|
|
config.addinivalue_line("markers", "slow: mark test as slow to run")
|
|
|
|
|
|
def pytest_collection_modifyitems(config, items):
|
|
if config.getoption("--runslow"):
|
|
# --runslow given in cli: do not skip slow tests
|
|
return
|
|
skip_slow = pytest.mark.skip(reason="need --runslow option to run")
|
|
for item in items:
|
|
if "slow" in item.keywords:
|
|
item.add_marker(skip_slow)
|
|
|
|
@pytest.fixture
|
|
def use_cuda(request):
|
|
""" Run test on gpu """
|
|
return request.config.getoption("--use_cuda")
|