fix: fix ut

wangzyphysics · Mar 30, 2024 · 987aaf1 · 987aaf1
1 parent 34ab952
commit 987aaf1
Show file tree

Hide file tree

Showing 7 changed files with 8 additions and 8 deletions.
diff --git a/tests/test_caly_evo_step.py b/tests/test_caly_evo_step.py
@@ -272,7 +272,7 @@ def tearDown(self):
             shutil.rmtree(i, ignore_errors=True)
         for i in Path().glob("caly_task*"):
             shutil.rmtree(i, ignore_errors=True)
-        shutil.rmtree("upload", ignore_errors=True)
+        # shutil.rmtree("upload", ignore_errors=True)
 
     @unittest.skip("only need to run test_01")
     def test_00(self):

diff --git a/tests/test_dpgen_loop.py b/tests/test_dpgen_loop.py
@@ -98,14 +98,14 @@
 )
 from dpgen2.utils.step_config import normalize as normalize_step_dict
 
-from .context import (
+from context import (
     default_host,
     default_image,
     skip_ut_with_dflow,
     skip_ut_with_dflow_reason,
     upload_python_packages,
 )
-from .mocked_ops import (
+from mocked_ops import (
     MockedCollectData,
     MockedCollectDataCheckOptParam,
     MockedCollectDataFailed,

diff --git a/tests/test_prep_run_caly.py b/tests/test_prep_run_caly.py
@@ -80,7 +80,7 @@
 )
 from dpgen2.utils.step_config import normalize as normalize_step_dict
 
-from .mocked_ops import (
+from mocked_ops import (
     MockedCollRunCaly,
     MockedRunCalyModelDevi,
     MockedRunDPOptim,

diff --git a/tests/test_prep_run_dp_train.py b/tests/test_prep_run_dp_train.py
@@ -59,7 +59,7 @@
 )
 from dpgen2.utils.step_config import normalize as normalize_step_dict
 
-from .mocked_ops import (
+from mocked_ops import (
     MockedModifyTrainScript,
     MockedPrepDPTrain,
     MockedRunDPTrain,

diff --git a/tests/test_prep_run_lmp.py b/tests/test_prep_run_lmp.py
@@ -76,7 +76,7 @@
 )
 from dpgen2.utils.step_config import normalize as normalize_step_dict
 
-from .mocked_ops import (
+from mocked_ops import (
     MockedRunLmp,
     mocked_numb_models,
 )

diff --git a/tests/test_prep_run_vasp.py b/tests/test_prep_run_vasp.py
@@ -65,7 +65,7 @@
 )
 from dpgen2.utils.step_config import normalize as normalize_step_dict
 
-from .mocked_ops import (
+from mocked_ops import (
     MockedPrepVasp,
     MockedRunVasp,
     mocked_incar_template,

diff --git a/tests/test_select_confs.py b/tests/test_select_confs.py
@@ -49,7 +49,7 @@
     SelectConfs,
 )
 
-from .mocked_ops import (
+from mocked_ops import (
     MockedConfSelector,
     MockedExplorationReport,
     MockedSelectConfs,