fix invalid condition in compLabeledSys in unittest

deepmodeling · Feb 1, 2024 · 5301b53 · 5301b53
1 parent e43f00e
commit 5301b53
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/comp_sys.py b/tests/comp_sys.py
@@ -92,8 +92,8 @@ def test_virial(self):
         # if len(self.system_1['virials']) == 0:
         #     self.assertEqual(len(self.system_1['virials']), 0)
         #     return
-        if "virials" not in self.system_1:
-            self.assertFalse("virials" in self.system_2)
+        if not self.system_1.has_virial():
+            self.assertFalse(self.system_2.has_virial())
             return
         np.testing.assert_almost_equal(
             self.system_1["virials"],