Coverage for tests / test_query.py: 100%

21 statements  

« prev     ^ index     » next       coverage.py v7.13.0, created at 2026-01-06 21:09 +0000

1from typing import Optional 1jklmnopqr

2 

3import pytest 1jklmnopqr

4from sqlmodel import Field, Session, SQLModel, create_engine 1jklmnopqr

5 

6 

7def test_query(clear_sqlmodel): 1jklmnopqr

8 class Hero(SQLModel, table=True): 1abcdefghi

9 id: Optional[int] = Field(default=None, primary_key=True) 1abcdefghi

10 name: str 1abcdefghi

11 secret_name: str 1abcdefghi

12 age: Optional[int] = None 1abcdefghi

13 

14 hero_1 = Hero(name="Deadpond", secret_name="Dive Wilson") 1abcdefghi

15 

16 engine = create_engine("sqlite://") 1abcdefghi

17 

18 SQLModel.metadata.create_all(engine) 1abcdefghi

19 with Session(engine) as session: 1abcdefghi

20 session.add(hero_1) 1abcdefghi

21 session.commit() 1abcdefghi

22 session.refresh(hero_1) 1abcdefghi

23 

24 with Session(engine) as session: 1abcdefghi

25 with pytest.warns(DeprecationWarning): 1abcdefghi

26 query_hero = session.query(Hero).first() 1abcdefghi

27 assert query_hero 1abcdefghi

28 assert query_hero.name == hero_1.name 1abcdefghi