Simplify test via parameterization

Author: YuriiMotov

tests/test_pydantic/test_model_dump.py

@@ -1,25 +1,20 @@
-from sqlmodel import SQLModel
-
-
-def test_polymorphic_serialization_default():
-
-    class User(SQLModel):
-        name: str
-
-    class UserLogin(User):
-        password: str
-
-    class OuterModel(SQLModel):
-        user: User
-
-    outer_model = OuterModel(
-        user=UserLogin(name="pydantic", password="password"),
-    )
+from typing import Any
 
-    assert outer_model.model_dump() == {"user": {"name": "pydantic"}}
+import pytest
+from sqlmodel import SQLModel
 
 
-def test_polymorphic_serialization_false():
+@pytest.mark.parametrize(
+    ("polymorphic_serialization", "expected_result"),
+    [
+        (None, {"user": {"name": "pydantic"}}),
+        (False, {"user": {"name": "pydantic"}}),
+        (True, {"user": {"name": "pydantic", "password": "password"}}),
+    ],
+)
+def test_polymorphic_serialization(
+    polymorphic_serialization: bool | None, expected_result: dict[str, Any]
+):
 
     class User(SQLModel):
         name: str
@@ -34,26 +29,7 @@ class OuterModel(SQLModel):
         user=UserLogin(name="pydantic", password="password"),
     )
 
-    assert outer_model.model_dump(polymorphic_serialization=False) == {
-        "user": {"name": "pydantic"}
-    }
-
-
-def test_polymorphic_serialization_true():
-
-    class User(SQLModel):
-        name: str
-
-    class UserLogin(User):
-        password: str
-
-    class OuterModel(SQLModel):
-        user: User
-
-    outer_model = OuterModel(
-        user=UserLogin(name="pydantic", password="password"),
+    assert (
+        outer_model.model_dump(polymorphic_serialization=polymorphic_serialization)
+        == expected_result
     )
-
-    assert outer_model.model_dump(polymorphic_serialization=True) == {
-        "user": {"name": "pydantic", "password": "password"}
-    }