Add test

src/curies/api.py

@@ -317,10 +317,12 @@ def _key(self) -> RecordKey:
 
     from pydantic import BaseModel
 
-    class Records(BaseModel):
+    class Records(BaseModel):  # type:ignore
         """A list of records."""
 
         class Config:
+            """Configuration for the records."""
+
             arbitrary_types_allowed = True
 
         __root__: List[Record]
@@ -331,10 +333,12 @@ class Config:
 
     from pydantic import RootModel
 
-    class Records(RootModel):
+    class Records(RootModel[List[Record]]):  # type:ignore
         """A list of records."""
 
-        root: List[Record]
+        def __iter__(self) -> Iterable[Record]:
+            """Iterate over records."""
+            return cast(Iterable[Record], iter(self.root))
 
 
 class DuplicateSummary(NamedTuple):

tests/test_api.py

@@ -21,6 +21,7 @@
     ExpansionError,
     PrefixStandardizationError,
     Record,
+    Records,
     Reference,
     ReferenceTuple,
     URIStandardizationError,
@@ -41,6 +42,16 @@
 GO_URI_PREFIX = "http://purl.obolibrary.org/obo/GO_"
 
 
+class TestStruct(unittest.TestCase):
+    """Test the data structures."""
+
+    def test_records(self):
+        """Test a list of records."""
+        records = Records.parse_obj([{"prefix": "chebi", "uri_prefix": CHEBI_URI_PREFIX}])
+        converter = Converter(records=records)
+        self.assertEqual({"chebi"}, converter.get_prefixes())
+
+
 class TestAddRecord(unittest.TestCase):
     """Test adding records."""