Make test more robust

coudertlab · Mar 27, 2024 · 7150359 · 7150359
1 parent 5772575
commit 7150359
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_structures_2.py b/tests/test_structures_2.py
@@ -11,7 +11,8 @@ def func_check_random_structs(dataset):
         assert s.num_sites > 1
         assert s.n_elems > 1
         assert s.volume > 10
-        assert (6 in s.atomic_numbers) or (1 in s.atomic_numbers) or (8 in s.atomic_numbers)
+        assert ((1 in s.atomic_numbers) or (6 in s.atomic_numbers)
+                or (7 in s.atomic_numbers) or (8 in s.atomic_numbers))
 
 
 @pytest.mark.filterwarnings("ignore:.*rounded to ideal values")