github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[testing] move pytest to be inside the function (#4087)

pull/4105/head
Frank Lee 2023-06-27 11:02:25 +08:00 committed by GitHub
parent 4da324cd60
commit 95e95b6d58
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
colossalai/testing/pytest_wrapper.py
View File

@ -1,10 +1,9 @@
""" """
This file will not be automatically imported by `colossalai.testing` This file will not be automatically imported by `colossalai.testing`
as this file has a dependency on `pytest`. Therefore, you need to as this file has a dependency on `pytest`. Therefore, you need to
explicitly import this file `from colossalai.testing.pytest_wrapper import <func>`.from explicitly import this file `from colossalai.testing.pytest_wrapper import <func>`.from
""" """
import pytest
import os import os
@ -30,6 +29,12 @@ def run_on_environment_flag(name: str):
pytest test_for_something.py 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) assert isinstance(name, str)
flag = os.environ.get(name.upper(), '0') flag = os.environ.get(name.upper(), '0')