Function test_polymorphic_serialization

tests/test_pydantic/test_model_dump.py:23–47

(
    polymorphic_serialization: bool | None, expected_result: dict[str, Any]
)

    ],
)
def test_polymorphic_serialization(
    polymorphic_serialization: bool | None, expected_result: dict[str, Any]
):

    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=False) == {
        "user": {"name": "pydantic"}
    }