Coverage for tests / test_field_sa_type.py: 100%

1from typing import Any, cast 9 ctx1jklmnopqr

2 

3from sqlalchemy import Integer, String 9 ctx1jklmnopqr

4from sqlmodel import Field, SQLModel 9 ctx1jklmnopqr

5 

6 

7def test_sa_type_class() -> None: 9 ctx1jklmnopqr

8 class Item(SQLModel, table=True): 9 ctx1stuvwxyzA

9 id: int = Field(primary_key=True, sa_type=Integer) 9 ctx1stuvwxyzA

10 

11 id_column = cast(Any, Item).id 9 ctx1stuvwxyzA

12 assert isinstance(id_column.type, Integer) 9 ctx1stuvwxyzA

13 

14 

15def test_sa_type_instance() -> None: 9 ctx1jklmnopqr

16 class Item(SQLModel, table=True): 9 ctx1abcdefghi

17 id: int = Field(primary_key=True) 9 ctx1abcdefghi

18 name: str = Field(sa_type=String(50)) 9 ctx1abcdefghi

19 

20 name_column = cast(Any, Item).name 9 ctx1abcdefghi

21 assert isinstance(name_column.type, String) 9 ctx1abcdefghi

22 assert name_column.type.length == 50 9 ctx1abcdefghi