Update test_api.py

biopragmatics · Oct 30, 2024 · 9c4daf0 · 9c4daf0
1 parent ae2b507
commit 9c4daf0
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tests/test_api.py b/tests/test_api.py
@@ -63,6 +63,18 @@ def test_sort(self):
         ]
         self.assertEqual(expected, sorted(start))
 
+    def test_set_membership(self):
+        """Test membership in sets."""
+        collection = {
+            Reference.from_curie("def:1234"),
+            Reference.from_curie("abc:1234"),
+            Reference.from_curie("abc:1235"),
+        }
+        self.assertIn(Reference.from_curie("def:1234"), collection)
+        self.assertNotIn(Reference.from_curie("xyz:1234"), collection)
+        self.assertNotIn(Reference.from_curie(":1234"), collection)
+        self.assertNotIn(Reference.from_curie("abc:"), collection)
+
 
 class TestAddRecord(unittest.TestCase):
     """Test adding records."""