Added tests for predefined row factories

Signed-off-by: chandr-andr (Kiselev Aleksandr) <[email protected]>

Author: chandr-andr

python/psqlpy/_internal/row_factories.pyi

@@ -27,7 +27,28 @@ def tuple_row(row: dict[str, Any]) -> Tuple[Tuple[str, Any]]:
     """
 
 class class_row(Generic[_CustomClass]):  # noqa: N801
-    """Row converter to specified class."""
+    """Row converter to specified class.
+
+    ### Example:
+    ```python
+    from psqlpy.row_factories import class_row
+
+
+    class ValidationModel:
+        name: str
+        views_count: int
+
+
+    async def main:
+        res = await db_pool.execute(
+            "SELECT * FROM users",
+        )
+
+        results: list[ValidationModel] = res.row_factory(
+            class_row(ValidationModel),
+        )
+    ```
+    """
 
     def __init__(self: Self, class_: Type[_CustomClass]) -> None:
         """Construct new `class_row`.

python/tests/test_row_factories.py

@@ -0,0 +1,68 @@
+from dataclasses import dataclass
+from typing import Any, Callable, Dict, Type
+
+import pytest
+
+from psqlpy import ConnectionPool
+from psqlpy.row_factories import class_row, tuple_row
+
+pytestmark = pytest.mark.anyio
+
+
+async def test_tuple_row(
+    psql_pool: ConnectionPool,
+    table_name: str,
+    number_database_records: int,
+) -> None:
+    conn_result = await psql_pool.execute(
+        querystring=f"SELECT * FROM {table_name}",
+    )
+    tuple_res = conn_result.row_factory(row_factory=tuple_row)
+
+    assert len(tuple_res) == number_database_records
+    assert isinstance(tuple_res[0], tuple)
+
+
+async def test_class_row(
+    psql_pool: ConnectionPool,
+    table_name: str,
+    number_database_records: int,
+) -> None:
+    @dataclass
+    class ValidationTestModel:
+        id: int
+        name: str
+
+    conn_result = await psql_pool.execute(
+        querystring=f"SELECT * FROM {table_name}",
+    )
+    class_res = conn_result.row_factory(row_factory=class_row(ValidationTestModel))
+    assert len(class_res) == number_database_records
+    assert isinstance(class_res[0], ValidationTestModel)
+
+
+async def test_custom_row_factory(
+    psql_pool: ConnectionPool,
+    table_name: str,
+    number_database_records: int,
+) -> None:
+    @dataclass
+    class ValidationTestModel:
+        id: int
+        name: str
+
+    def to_class(
+        class_: Type[ValidationTestModel],
+    ) -> Callable[[Dict[str, Any]], ValidationTestModel]:
+        def to_class_inner(row: Dict[str, Any]) -> ValidationTestModel:
+            return class_(**row)
+
+        return to_class_inner
+
+    conn_result = await psql_pool.execute(
+        querystring=f"SELECT * FROM {table_name}",
+    )
+    class_res = conn_result.row_factory(row_factory=to_class(ValidationTestModel))
+
+    assert len(class_res) == number_database_records
+    assert isinstance(class_res[0], ValidationTestModel)