mirror of
https://gitee.com/fastnlp/fastNLP.git
synced 2024-12-04 21:28:01 +08:00
19 lines
561 B
Python
19 lines
561 B
Python
import os
|
|
import pytest
|
|
|
|
from fastNLP.envs.set_backend import dump_fastnlp_backend
|
|
from tests.helpers.utils import Capturing
|
|
from fastNLP.envs.distributed import rank_zero_rm
|
|
|
|
|
|
def test_dump_fastnlp_envs():
|
|
filepath = None
|
|
try:
|
|
with Capturing() as output:
|
|
dump_fastnlp_backend(backend="torch")
|
|
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:
|
|
rank_zero_rm(filepath)
|