refine paddle unitests

Author: HydrogenSulfate

source/tests/pd/model/test_descriptor.py

@@ -13,6 +13,9 @@
     Path,
 )
 
+from deepmd.common import (
+    expand_sys_str,
+)
 from deepmd.pd.model.descriptor import (
     prod_env_mat,
 )
@@ -31,9 +34,6 @@
 from deepmd.pd.utils.nlist import (
     extend_input_and_build_neighbor_list,
 )
-from deepmd.tf.common import (
-    expand_sys_str,
-)
 from deepmd.tf.env import (
     op_module,
 )

source/tests/pd/model/test_embedding_net.py

@@ -18,6 +18,9 @@
     Path,
 )
 
+from deepmd.common import (
+    expand_sys_str,
+)
 from deepmd.pd.model.descriptor import (
     DescrptSeA,
 )
@@ -34,9 +37,6 @@
 from deepmd.pd.utils.nlist import (
     extend_input_and_build_neighbor_list,
 )
-from deepmd.tf.common import (
-    expand_sys_str,
-)
 from deepmd.tf.descriptor import DescrptSeA as DescrptSeA_tf
 
 from ..test_finetune import (

source/tests/pd/model/test_model.py

@@ -19,6 +19,9 @@
     Path,
 )
 
+from deepmd.common import (
+    expand_sys_str,
+)
 from deepmd.dpmodel.utils.learning_rate import LearningRateExp as MyLRExp
 from deepmd.pd.loss import (
     EnergyStdLoss,
@@ -32,9 +35,6 @@
 from deepmd.pd.utils.env import (
     DEVICE,
 )
-from deepmd.tf.common import (
-    expand_sys_str,
-)
 from deepmd.tf.descriptor import DescrptSeA as DescrptSeA_tf
 from deepmd.tf.fit import (
     EnerFitting,

source/tests/pd/model/test_saveload_dpa1.py

@@ -13,6 +13,9 @@
     DataLoader,
 )
 
+from deepmd.common import (
+    expand_sys_str,
+)
 from deepmd.pd.loss import (
     EnergyStdLoss,
 )
@@ -32,9 +35,6 @@
 from deepmd.pd.utils.stat import (
     make_stat_input,
 )
-from deepmd.tf.common import (
-    expand_sys_str,
-)
 
 
 def get_dataset(config):

source/tests/pd/model/test_saveload_se_e2_a.py

@@ -13,6 +13,9 @@
     DataLoader,
 )
 
+from deepmd.common import (
+    expand_sys_str,
+)
 from deepmd.pd.loss import (
     EnergyStdLoss,
 )
@@ -32,9 +35,6 @@
 from deepmd.pd.utils.stat import (
     make_stat_input,
 )
-from deepmd.tf.common import (
-    expand_sys_str,
-)
 
 
 def get_dataset(config):

source/tests/pd/test_training.py

@@ -136,11 +136,14 @@ def test_trainable(self) -> None:
     def tearDown(self) -> None:
         for f in os.listdir("."):
             if f.startswith("model") and f.endswith(".pd"):
-                os.remove(f)
+                if os.path.exists(f):
+                    os.remove(f)
             if f in ["lcurve.out"]:
-                os.remove(f)
+                if os.path.exists(f):
+                    os.remove(f)
             if f in ["stat_files"]:
-                shutil.rmtree(f)
+                if os.path.exists(f):
+                    shutil.rmtree(f)
 
 
 class TestEnergyModelSeA(unittest.TestCase, DPTrainTest):