add simple test for entity_id

BradyAJohnston · Aug 3, 2023 · 7b850a0 · 7b850a0
1 parent 74c5d9b
commit 7b850a0
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/tests/test_attributes.py b/tests/test_attributes.py
@@ -0,0 +1,9 @@
+import MolecularNodes as mn
+import numpy as np
+
+def test_entity_id():
+    mol = mn.load.molecule_rcsb('1cd3')
+    ents = mn.obj.get_attribute(mol, 'entity_id')
+
+    assert np.all(np.isin(ents, np.array(range(4))))
+