/
githubmirror
/
ColossalAI
Обзор
Документация
Войти
/
githubmirror
/
ColossalAI
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
colossalai/testing/pytest_wrapper.py
46 строк
1 KB
Hongxin Liu
[misc] update pre-commit and run all files (#4752)
19 сен 2023, 09:20
Не верифицирован
19 сен 2023, 09:20
079bf3c
Код
Авторство
О чём код?
""" This file will not be automatically imported by `colossalai.testing` as this file has a dependency on `pytest`. Therefore, you need to explicitly import this file `from colossalai.testing.pytest_wrapper import <func>`.from """ import os def run_on_environment_flag(name: str): """ Conditionally run a test based on the environment variable. If this environment variable is set to 1, this test will be executed. Otherwise, this test is skipped. The environment variable is default to 0. Args: name (str): the name of the environment variable flag. Usage: # in your pytest file @run_on_environment_flag(name='SOME_FLAG') def test_for_something(): do_something() # in your terminal # this will execute your test SOME_FLAG=1 pytest test_for_something.py # this will skip your test pytest test_for_something.py """ try: import pytest except ImportError: raise ImportError( "This function requires `pytest` to be installed, please do `pip install pytest` and try again." ) assert isinstance(name, str) flag = os.environ.get(name.upper(), "0") reason = f"Environment variable {name} is {flag}" if flag == "1": return pytest.mark.skipif(False, reason=reason) else: return pytest.mark.skipif(True, reason=reason)