Update test_econf_embd.py

deepmodeling · May 14, 2024 · 4f8175f · 4f8175f
1 parent c309cf1
commit 4f8175f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/source/tests/common/test_econf_embd.py b/source/tests/common/test_econf_embd.py
@@ -43,13 +43,13 @@ def test_fe(self):
     def test_fe_flatten(self):
         res = make_econf_embedding(["Fe"], flatten=True)["Fe"]
         # fmt: off
-        expected_res = [2,2,2,2,2,2,2,2,2,2,1,1,1,1,2,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0]
+        expected_res = [2,2,2,2,2,2,2,2,2,2,1,1,1,1,2,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0]
         # fmt: on
         self.assertEqual(list(res), expected_res)
 
     def test_dict(self):
         res = electronic_configuration_embedding["Fe"]
         # fmt: off
-        expected_res = [2,2,2,2,2,2,2,2,2,2,1,1,1,1,2,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0]
+        expected_res = [2,2,2,2,2,2,2,2,2,2,1,1,1,1,2,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0]
         # fmt: on
         self.assertEqual(list(res), expected_res)