Added support for any custom python class in QueryResult

Author: chandr-andr

python/tests/conftest.py

@@ -3,10 +3,26 @@
 from typing import AsyncGenerator
 
 import pytest
+from pydantic import BaseModel
 
 from psqlpy import Cursor, PSQLPool
 
 
+class DefaultPydanticModel(BaseModel):
+    """Validation model for test data based on Pydantic."""
+
+    id: int
+    name: str
+
+
+class DefaultPythonModelClass:
+    """Validation model for test data based on default Python class."""
+
+    def __init__(self, id: int, name: str) -> None:
+        self.id = id
+        self.name = name
+
+
 @pytest.fixture()
 def anyio_backend() -> str:
     """

python/tests/test_value_converter.py

@@ -0,0 +1,34 @@
+import pytest
+from tests.conftest import DefaultPydanticModel, DefaultPythonModelClass
+
+from psqlpy import PSQLPool
+
+pytestmark = pytest.mark.anyio
+
+
+async def test_as_class(
+    psql_pool: PSQLPool,
+    table_name: str,
+    number_database_records: int,
+) -> None:
+    """Test `as_class()` method."""
+    select_result = await psql_pool.execute(
+        f"SELECT * FROM {table_name}",
+    )
+
+    as_pydantic = select_result.as_class(
+        as_class=DefaultPydanticModel,
+    )
+    assert len(as_pydantic) == number_database_records
+
+    for single_record in as_pydantic:
+        assert isinstance(single_record, DefaultPydanticModel)
+
+    as_py_class = select_result.as_class(
+        as_class=DefaultPythonModelClass,
+    )
+
+    assert len(as_py_class) == number_database_records
+
+    for single_py_record in as_py_class:
+        assert isinstance(single_py_record, DefaultPythonModelClass)