fix ut

wanghan-iapcm · Feb 5, 2024 · 2899b31 · 2899b31
1 parent eaf28dd
commit 2899b31
Show file tree

Hide file tree

Showing 5 changed files with 20 additions and 15 deletions.
diff --git a/source/tests/common/dpmodel/test_descriptor_se_e2_a.py b/source/tests/common/dpmodel/test_descriptor_se_e2_a.py
@@ -2,14 +2,15 @@
 import unittest
 
 import numpy as np
-from case_single_frame_with_nlist import (
-    TestCaseSingleFrameWithNlist,
-)
 
 from deepmd.dpmodel.descriptor import (
     DescrptSeA,
 )
 
+from .case_single_frame_with_nlist import (
+    TestCaseSingleFrameWithNlist,
+)
+
 
 class TestDescrptSeA(unittest.TestCase, TestCaseSingleFrameWithNlist):
     def setUp(self):

diff --git a/source/tests/common/dpmodel/test_dp_atomic_model.py b/source/tests/common/dpmodel/test_dp_atomic_model.py
@@ -2,9 +2,6 @@
 import unittest
 
 import numpy as np
-from case_single_frame_with_nlist import (
-    TestCaseSingleFrameWithNlist,
-)
 
 from deepmd.dpmodel.descriptor import (
     DescrptSeA,
@@ -16,6 +13,10 @@
     DPAtomicModel,
 )
 
+from .case_single_frame_with_nlist import (
+    TestCaseSingleFrameWithNlist,
+)
+
 
 class TestDPAtomicModel(unittest.TestCase, TestCaseSingleFrameWithNlist):
     def setUp(self):

diff --git a/source/tests/common/dpmodel/test_dp_model.py b/source/tests/common/dpmodel/test_dp_model.py
@@ -2,9 +2,6 @@
 import unittest
 
 import numpy as np
-from case_single_frame_with_nlist import (
-    TestCaseSingleFrameWithNlist,
-)
 
 from deepmd.dpmodel.descriptor import (
     DescrptSeA,
@@ -16,6 +13,10 @@
     DPModel,
 )
 
+from .case_single_frame_with_nlist import (
+    TestCaseSingleFrameWithNlist,
+)
+
 
 class TestDPModel(unittest.TestCase, TestCaseSingleFrameWithNlist):
     def setUp(self):

diff --git a/source/tests/common/dpmodel/test_env_mat.py b/source/tests/common/dpmodel/test_env_mat.py
@@ -2,14 +2,15 @@
 import unittest
 
 import numpy as np
-from case_single_frame_with_nlist import (
-    TestCaseSingleFrameWithNlist,
-)
 
 from deepmd.dpmodel.utils import (
     EnvMat,
 )
 
+from .case_single_frame_with_nlist import (
+    TestCaseSingleFrameWithNlist,
+)
+
 
 class TestEnvMat(unittest.TestCase, TestCaseSingleFrameWithNlist):
     def setUp(self):

diff --git a/source/tests/common/dpmodel/test_fitting_invar_fitting.py b/source/tests/common/dpmodel/test_fitting_invar_fitting.py
@@ -3,9 +3,6 @@
 import unittest
 
 import numpy as np
-from case_single_frame_with_nlist import (
-    TestCaseSingleFrameWithNlist,
-)
 
 from deepmd.dpmodel.descriptor import (
     DescrptSeA,
@@ -14,6 +11,10 @@
     InvarFitting,
 )
 
+from .case_single_frame_with_nlist import (
+    TestCaseSingleFrameWithNlist,
+)
+
 
 class TestInvarFitting(unittest.TestCase, TestCaseSingleFrameWithNlist):
     def setUp(self):