From 8bd4163579837d68b852650fedeab9b2c57c002b Mon Sep 17 00:00:00 2001 From: Yurii Motov Date: Fri, 20 Feb 2026 22:37:06 +0100 Subject: [PATCH] Update test to Python 3.10+ syntax --- tests/test_pydantic/test_field.py | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/tests/test_pydantic/test_field.py b/tests/test_pydantic/test_field.py index f52a1cc1b..1956f55bf 100644 --- a/tests/test_pydantic/test_field.py +++ b/tests/test_pydantic/test_field.py @@ -1,5 +1,5 @@ from decimal import Decimal -from typing import Annotated, Any, Literal, Optional, Union +from typing import Annotated, Any, Literal import pytest from pydantic import ValidationError @@ -38,7 +38,7 @@ def test_discriminator(): scales: bool class Model(SQLModel): - pet: Union[Cat, Dog, Lizard] = Field(..., discriminator="pet_type") + pet: Cat | Dog | Lizard = Field(..., discriminator="pet_type") n: int Model(pet={"pet_type": "dog", "barks": 3.14}, n=1) # type: ignore[arg-type] @@ -66,10 +66,9 @@ def test_discriminator_callable(): return getattr(v, "fruit", getattr(v, "filling", None)) class ThanksgivingDinner(SQLModel): - dessert: Union[ - Annotated[ApplePie, Tag("apple")], - Annotated[PumpkinPie, Tag("pumpkin")], - ] = Field( + dessert: ( + Annotated[ApplePie, Tag("apple")] | Annotated[PumpkinPie, Tag("pumpkin")] + ) = Field( discriminator=Discriminator(get_discriminator_value), ) @@ -82,7 +81,7 @@ def test_discriminator_callable(): def test_repr(): class Model(SQLModel): - id: Optional[int] = Field(primary_key=True) + id: int | None = Field(primary_key=True) foo: str = Field(repr=False) instance = Model(id=123, foo="bar")