20 lines
515 B
Python
20 lines
515 B
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
import pytest
|
|
|
|
from mmdeploy.codebase import import_codebase
|
|
from mmdeploy.utils import Codebase
|
|
|
|
|
|
def pytest_ignore_collect(*args, **kwargs):
|
|
import importlib
|
|
return importlib.util.find_spec('mmseg') is None
|
|
|
|
|
|
@pytest.fixture(autouse=True, scope='package')
|
|
def import_all_modules():
|
|
codebase = Codebase.MMSEG
|
|
try:
|
|
import_codebase(codebase)
|
|
except ImportError:
|
|
pytest.skip(f'{codebase} is not installed.', allow_module_level=True)
|