mirror of
https://gitee.com/fastnlp/fastNLP.git
synced 2024-12-08 23:28:01 +08:00
18 lines
530 B
Python
18 lines
530 B
Python
import os
|
|
|
|
from fastNLP.envs.set_env import dump_fastnlp_backend
|
|
from tests.helpers.utils import Capturing
|
|
from fastNLP.core import synchronize_safe_rm
|
|
|
|
|
|
def test_dump_fastnlp_envs():
|
|
filepath = None
|
|
try:
|
|
with Capturing() as output:
|
|
dump_fastnlp_backend()
|
|
filepath = os.path.join(os.path.expanduser('~'), '.fastNLP', 'envs', os.environ['CONDA_DEFAULT_ENV']+'.json')
|
|
assert filepath in output[0]
|
|
assert os.path.exists(filepath)
|
|
finally:
|
|
synchronize_safe_rm(filepath)
|