Coverage for tests / test_tutorial / test_connect / test_select / test_tutorial004.py: 100%

15 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 sqlmodel import create_engine 1abcdefghi

6 

7from ....conftest import PrintMock, needs_py310 1abcdefghi

8 

9expected_calls = [ 1abcdefghi

10 [ 

11 "Created hero:", 

12 { 

13 "age": None, 

14 "id": 1, 

15 "secret_name": "Dive Wilson", 

16 "team_id": 2, 

17 "name": "Deadpond", 

18 }, 

19 ], 

20 [ 

21 "Created hero:", 

22 { 

23 "age": 48, 

24 "id": 2, 

25 "secret_name": "Tommy Sharp", 

26 "team_id": 1, 

27 "name": "Rusty-Man", 

28 }, 

29 ], 

30 [ 

31 "Created hero:", 

32 { 

33 "age": None, 

34 "id": 3, 

35 "secret_name": "Pedro Parqueador", 

36 "team_id": None, 

37 "name": "Spider-Boy", 

38 }, 

39 ], 

40 [ 

41 "Preventer Hero:", 

42 { 

43 "age": 48, 

44 "id": 2, 

45 "secret_name": "Tommy Sharp", 

46 "team_id": 1, 

47 "name": "Rusty-Man", 

48 }, 

49 ], 

50] 

51 

52 

53@pytest.fixture( 1abcdefghi

54 name="module", 

55 params=[ 

56 "tutorial004_py39", 

57 pytest.param("tutorial004_py310", marks=needs_py310), 

58 ], 

59) 

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

61 module = importlib.import_module( 1abcdefghi

62 f"docs_src.tutorial.connect.select.{request.param}" 

63 ) 

64 module.sqlite_url = "sqlite://" 1abcdefghi

65 module.engine = create_engine(module.sqlite_url) 1abcdefghi

66 return module 1abcdefghi

67 

68 

69def test_tutorial(print_mock: PrintMock, module: ModuleType): 1abcdefghi

70 module.main() 1jklmnopqr

71 assert print_mock.calls == expected_calls 1jklmnopqr