From 5301b537e4b89ac51ac483154feb8ac4124663e9 Mon Sep 17 00:00:00 2001
From: robinzyb <38876805+robinzyb@users.noreply.github.com>
Date: Thu, 1 Feb 2024 21:00:06 +0100
Subject: [PATCH] fix invalid condition in compLabeledSys in unittest

---
 tests/comp_sys.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/comp_sys.py b/tests/comp_sys.py
index cfa92c49..f4663780 100644
--- 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"],