Allow population of SparqlResult pydantic models by name

cimsparql/sparql_result_json.py

@@ -1,24 +1,28 @@
 from polyfactory.decorators import post_generated
 from polyfactory.factories import pydantic_factory
-from pydantic import BaseModel, Field
+from pydantic import BaseModel, ConfigDict, Field
 
 
-class SparqlResultHead(BaseModel):
+class CimsparqlBaseModel(BaseModel):
+    model_config = ConfigDict(populate_by_name=True)
+
+
+class SparqlResultHead(CimsparqlBaseModel):
     link: list[str] = Field(default_factory=list)
     variables: list[str] = Field(default_factory=list, alias="vars")
 
 
-class SparqlResultValue(BaseModel):
+class SparqlResultValue(CimsparqlBaseModel):
     value_type: str = Field(alias="type")
     value: str
     datatype: str = ""
 
 
-class SparqlData(BaseModel):
+class SparqlData(CimsparqlBaseModel):
     bindings: list[dict[str, SparqlResultValue]]
 
 
-class SparqlResultJson(BaseModel):
+class SparqlResultJson(CimsparqlBaseModel):
     """
     Data model for rest api resonse of MIME type
 

tests/test_sparql_result_json.py

@@ -0,0 +1,15 @@
+from cimsparql.sparql_result_json import SparqlResultHead, SparqlResultValue
+
+
+def test_populate_sparql_result_head_by_name():
+    sparql_head_1 = SparqlResultHead(variables=["a", "b"], links=["l1", "l2"])
+
+    sparql_head_2 = SparqlResultHead(vars=["a", "b"], links=["l1", "l2"])
+    assert sparql_head_1 == sparql_head_2
+
+
+def test_populate_sparql_result_value_by_name():
+    sparql_value_1 = SparqlResultValue(value_type="literal", value="value")
+
+    sparql_value_2 = SparqlResultValue(type="literal", value="value")
+    assert sparql_value_1 == sparql_value_2