Coverage for tests / test_tutorial / test_connect / test_create_connected_tables / test_tutorial001.py: 100%

18 statements  

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

1import importlib 1abcdefghi

2from types import ModuleType 1abcdefghi

3 

4import pytest 1abcdefghi

5from sqlalchemy import inspect 1abcdefghi

6from sqlalchemy.engine.reflection import Inspector 1abcdefghi

7from sqlmodel import create_engine 1abcdefghi

8 

9from tests.conftest import needs_py310 1abcdefghi

10 

11 

12@pytest.fixture( 1abcdefghi

13 name="module", 

14 params=[ 

15 "tutorial001_py39", 

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

17 ], 

18) 

19def get_module(request: pytest.FixtureRequest) -> ModuleType: 1abcdefghi

20 module = importlib.import_module( 1abcdefghi

21 f"docs_src.tutorial.connect.create_tables.{request.param}" 

22 ) 

23 return module 1abcdefghi

24 

25 

26def test_tutorial001(module: ModuleType): 1abcdefghi

27 module.sqlite_url = "sqlite://" 1jklmnopqr

28 module.engine = create_engine(module.sqlite_url) 1jklmnopqr

29 module.main() 1jklmnopqr

30 insp: Inspector = inspect(module.engine) 1jklmnopqr

31 assert insp.has_table(str(module.Hero.__tablename__)) 1jklmnopqr

32 assert insp.has_table(str(module.Team.__tablename__)) 1jklmnopqr