update test

zincware · Jan 19, 2024 · 369b657 · 369b657
1 parent 100c517
commit 369b657
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_znframe.py b/tests/test_znframe.py
@@ -72,9 +72,9 @@ def test_to_json(ammonia):
 
 def test_water_with_calc(waterWithCalc):
     frame = Frame.from_atoms(waterWithCalc)
-    duplicates = list(set(frame.info["calc"].keys()) & set(frame.arrays.keys()))
-    if duplicates:
-        raise ValueError(f"Duplicate keys: {duplicates}")
+    intersection = set(frame.info) & set(frame.arrays)
+    if intersection:
+        raise ValueError(f"Duplicate keys: {intersection}")
 
     assert "forces" in frame.arrays
     assert "stress" in frame.info