correct name of the class

deepmodeling · Feb 15, 2022 · c7ccc08 · c7ccc08
1 parent fc7229f
commit c7ccc08
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/source/tests/test_neighbor_stat.py b/source/tests/test_neighbor_stat.py
@@ -20,7 +20,7 @@ def gen_sys(nframes):
     return data
 
 
-class TestEnerShift(unittest.TestCase):
+class TestNeighborStat(unittest.TestCase):
     def setUp(self):
         data0 = gen_sys(1)
         sys0 = dpdata.LabeledSystem()
@@ -30,7 +30,7 @@ def setUp(self):
     def tearDown(self):
         shutil.rmtree('system_0')
 
-    def test_ener_shift(self):
+    def test_neighbor_stat(self):
         # set rcut to 0. will cause a core dumped
         # TODO: check what is wrong
         for rcut in (3., 6., 11.):