Skip to content

Commit a3044bb

Browse files
author
John Lyu
committed
add needs_pydanticv2 mark to test
1 parent 4aade03 commit a3044bb

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tests/test_polymorphic_model.py

+5
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,10 @@
44
from sqlalchemy.orm import mapped_column
55
from sqlmodel import Field, Session, SQLModel, create_engine, select
66

7+
from tests.conftest import needs_pydanticv2
78

9+
10+
@needs_pydanticv2
811
def test_polymorphic_joined_table(clear_sqlmodel) -> None:
912
class Hero(SQLModel, table=True):
1013
__tablename__ = "hero"
@@ -47,6 +50,7 @@ class DarkHero(Hero):
4750
assert isinstance(result[0].dark_power, str)
4851

4952

53+
@needs_pydanticv2
5054
def test_polymorphic_joined_table_sm_field(clear_sqlmodel) -> None:
5155
class Hero(SQLModel, table=True):
5256
__tablename__ = "hero"
@@ -90,6 +94,7 @@ class DarkHero(Hero):
9094
assert isinstance(result[0].dark_power, str)
9195

9296

97+
@needs_pydanticv2
9398
def test_polymorphic_single_table(clear_sqlmodel) -> None:
9499
class Hero(SQLModel, table=True):
95100
__tablename__ = "hero"

0 commit comments

Comments
 (0)