Coverage for tests / test_tutorial / test_relationship_attributes / test_back_populates / test_tutorial003.py: 100%

1import importlib 9 ctx1abcdefghi

2from types import ModuleType 9 ctx1abcdefghi

3 

4import pytest 9 ctx1abcdefghi

5from sqlalchemy import inspect 9 ctx1abcdefghi

6from sqlalchemy.engine.reflection import Inspector 9 ctx1abcdefghi

7from sqlmodel import create_engine 9 ctx1abcdefghi

8 

9from ....conftest import needs_py310 9 ctx1abcdefghi

10 

11 

12@pytest.fixture( 9 ctx1abcdefghi

13 name="mod", 

14 params=[ 

15 pytest.param("tutorial003_py39"), 

16 pytest.param("tutorial003_py310", marks=needs_py310), 

17 ], 

18) 

19def get_module(request: pytest.FixtureRequest) -> ModuleType: 9 ctx1abcdefghi

20 mod = importlib.import_module( 9 ctx1abcdefghi

21 f"docs_src.tutorial.relationship_attributes.back_populates.{request.param}" 

22 ) 

23 mod.sqlite_url = "sqlite://" 9 ctx1abcdefghi

24 mod.engine = create_engine(mod.sqlite_url) 9 ctx1abcdefghi

25 return mod 9 ctx1abcdefghi

26 

27 

28def test_tutorial(mod: ModuleType): 9 ctx1abcdefghi

29 mod.main() 9 ctx1jklmnopqr

30 insp: Inspector = inspect(mod.engine) 9 ctx1jklmnopqr

31 assert insp.has_table(str(mod.Hero.__tablename__)) 9 ctx1jklmnopqr

32 assert insp.has_table(str(mod.Weapon.__tablename__)) 9 ctx1jklmnopqr

33 assert insp.has_table(str(mod.Power.__tablename__)) 9 ctx1jklmnopqr

34 assert insp.has_table(str(mod.Team.__tablename__)) 9 ctx1jklmnopqr