diff --git a/tests/core/utils/test_paddle_utils.py b/tests/core/utils/test_paddle_utils.py index ecabc1df..d0c4c972 100644 --- a/tests/core/utils/test_paddle_utils.py +++ b/tests/core/utils/test_paddle_utils.py @@ -64,6 +64,7 @@ class TestPaddleToDevice: # ############################################################################ +@pytest.mark.paddle class TestPaddleMoveDataToDevice: def check_gpu(self, tensor, idx): diff --git a/tests/envs/test_set_backend.py b/tests/envs/test_set_backend.py index c45acd7b..a283272c 100644 --- a/tests/envs/test_set_backend.py +++ b/tests/envs/test_set_backend.py @@ -9,10 +9,14 @@ 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) + if 'CONDA_DEFAULT_ENV' in os.environ: + 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) + else: + with pytest.raises(RuntimeError): + dump_fastnlp_backend(backend="torch") finally: rank_zero_rm(filepath)