GitHub-Proxy/sqlmodel-fix
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add needs_pydanticv2 mark to test

Browse Source
This commit is contained in:
John Lyu 2024-11-26 11:24:00 +08:00
parent 4aade03081
commit a3044bbf68
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/test_polymorphic_model.py
View File

@ -4,7 +4,10 @@ from sqlalchemy import ForeignKey
from sqlalchemy.orm import mapped_column
from sqlmodel import Field, Session, SQLModel, create_engine, select
from tests.conftest import needs_pydanticv2
@needs_pydanticv2
def test_polymorphic_joined_table(clear_sqlmodel) -> None:
class Hero(SQLModel, table=True):
__tablename__ = "hero"
@ -47,6 +50,7 @@ def test_polymorphic_joined_table(clear_sqlmodel) -> None:
assert isinstance(result[0].dark_power, str)
@needs_pydanticv2
def test_polymorphic_joined_table_sm_field(clear_sqlmodel) -> None:
class Hero(SQLModel, table=True):
__tablename__ = "hero"
@ -90,6 +94,7 @@ def test_polymorphic_joined_table_sm_field(clear_sqlmodel) -> None:
assert isinstance(result[0].dark_power, str)
@needs_pydanticv2
def test_polymorphic_single_table(clear_sqlmodel) -> None:
class Hero(SQLModel, table=True):
__tablename__ = "hero"