fix UT issues: can be run in the tests folder and remove temp file (#130

)

- the UTs can be run in the `tests` folder.
- remove a temp folder.

---------

Co-authored-by: Han Wang <wang_han@iapcm.ac.cn>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

tests/exploration/test_conf_filter.py

@@ -3,6 +3,9 @@
 
 import dpdata
 import numpy as np
+from fake_data_set import (
+    fake_system,
+)
 from mock import (
     patch,
 )
@@ -12,9 +15,6 @@
     ConfFilters,
 )
 
-from ..fake_data_set import (
-    fake_system,
-)
 from .context import (
     dpgen2,
 )

tests/fp/test_prep_vasp.py

@@ -19,6 +19,10 @@
     OPIOSign,
     TransientError,
 )
+from fake_data_set import (
+    fake_multi_sys,
+    fake_system,
+)
 
 from dpgen2.constants import (
     fp_task_pattern,
@@ -35,10 +39,6 @@
     dump_object_to_file,
 )
 
-from ..fake_data_set import (
-    fake_multi_sys,
-    fake_system,
-)
 from .context import (
     dpgen2,
 )

tests/op/test_collect_data.py

@@ -14,6 +14,10 @@
     OPIOSign,
     TransientError,
 )
+from fake_data_set import (
+    fake_multi_sys,
+    fake_system,
+)
 from mock import (
     call,
     mock,
@@ -34,11 +38,6 @@
     CollectData,
 )
 
-from ..fake_data_set import (
-    fake_multi_sys,
-    fake_system,
-)
-
 
 class TestRunLmp(unittest.TestCase):
     def setUp(self):

tests/op/test_run_dp_train.py

@@ -15,6 +15,10 @@
     OPIOSign,
     TransientError,
 )
+from fake_data_set import (
+    fake_multi_sys,
+    fake_system,
+)
 from mock import (
     call,
     patch,
@@ -31,11 +35,6 @@
     RunDPTrain,
 )
 
-from ..fake_data_set import (
-    fake_multi_sys,
-    fake_system,
-)
-
 
 class TestRunDPTrain(unittest.TestCase):
     def setUp(self):

tests/test_prep_run_dp_labeling.py

@@ -84,7 +84,7 @@ def tearDown(self):
 
         shutil.rmtree(deepmd_input_path, ignore_errors=True)
 
-        shutil.rmtree(deepmd_teacher_model, ignore_errors=True)
+        shutil.rmtree(deepmd_temp_path, ignore_errors=True)
 
         shutil.rmtree(deepmd_teacher_model, ignore_errors=True)