📝 Add tests for type alias handling in test_field_sa_type.py

Author: stickm4n

sqlmodel/_compat.py

@@ -15,6 +15,7 @@
     Optional,
     Set,
     Type,
+    TypeAliasType,
     TypeVar,
     Union,
 )
@@ -74,8 +75,6 @@ def partial_init() -> Generator[None, None, None]:
 
 
 if IS_PYDANTIC_V2:
-    from typing import TypeAliasType
-
     from annotated_types import MaxLen
     from pydantic import ConfigDict as BaseConfig
     from pydantic._internal._fields import PydanticMetadata

tests/test_field_sa_type.py

@@ -4,9 +4,6 @@
 import pytest
 from sqlmodel import Field, SQLModel
 
-Type5: TypeAlias = str
-Type6: TypeAlias = Annotated[str, "Just a comment"]
-
 
 @pytest.mark.skipif(version_info[1] < 12, reason="Language feature of Python 3.12+")
 def test_sa_type_1() -> None:
@@ -26,33 +23,39 @@ class Hero(SQLModel, table=True):
         weapon: Type2 = "sword"
 
 
+Type3: TypeAlias = str
+
+
 @pytest.mark.skipif(version_info[1] < 12, reason="Language feature of Python 3.12+")
 def test_sa_type_3() -> None:
-    type Type3 = str
-
     class Hero(SQLModel, table=True):
         pk: int = Field(primary_key=True)
         weapon: Type3 = "sword"
 
 
+Type4: TypeAlias = Annotated[str, "Just a comment"]
+
+
 @pytest.mark.skipif(version_info[1] < 12, reason="Language feature of Python 3.12+")
 def test_sa_type_4() -> None:
-    type Type4 = Annotated[str, "Just a comment"]
-
     class Hero(SQLModel, table=True):
         pk: int = Field(primary_key=True)
         weapon: Type4 = "sword"
 
 
 @pytest.mark.skipif(version_info[1] < 12, reason="Language feature of Python 3.12+")
 def test_sa_type_5() -> None:
+    type Type5 = str  # type: ignore[invalid-syntax]
+
     class Hero(SQLModel, table=True):
         pk: int = Field(primary_key=True)
         weapon: Type5 = "sword"
 
 
 @pytest.mark.skipif(version_info[1] < 12, reason="Language feature of Python 3.12+")
 def test_sa_type_6() -> None:
+    type Type6 = Annotated[str, "Just a comment"]  # type: ignore[invalid-syntax]
+
     class Hero(SQLModel, table=True):
         pk: int = Field(primary_key=True)
         weapon: Type6 = "sword"