Revert "rm shift from rot test"

This reverts commit 07b611e.
deepmodeling · May 29, 2024 · ebd9289 · ebd9289
1 parent 07b611e
commit ebd9289
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/source/tests/pt/model/test_dipole_fitting.py b/source/tests/pt/model/test_dipole_fitting.py
@@ -194,7 +194,7 @@ def test_rot(self):
                     _,
                     nlist,
                 ) = extend_input_and_build_neighbor_list(
-                    xyz, atype, self.rcut, self.sel, False, box=self.cell
+                    xyz + self.shift, atype, self.rcut, self.sel, False, box=self.cell
                 )
 
                 rd0, gr0, _, _, _ = self.dd0(

diff --git a/source/tests/pt/model/test_polarizability_fitting.py b/source/tests/pt/model/test_polarizability_fitting.py
@@ -205,7 +205,7 @@ def test_rot(self):
                     _,
                     nlist,
                 ) = extend_input_and_build_neighbor_list(
-                    xyz, atype, self.rcut, self.sel, False, box=self.cell
+                    xyz + self.shift, atype, self.rcut, self.sel, False, box=self.cell
                 )
 
                 rd0, gr0, _, _, _ = self.dd0(