Coverage for tests / test_tutorial / test_connect / test_select / test_tutorial005.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
4import pytest 1abcdefghi
5from sqlmodel import create_engine 1abcdefghi
7from ....conftest import PrintMock, needs_py310 1abcdefghi
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 "Team:",
50 {"id": 1, "name": "Preventers", "headquarters": "Sharp Tower"},
51 ],
52]
55@pytest.fixture( 1abcdefghi
56 name="module",
57 params=[
58 "tutorial005_py39",
59 pytest.param("tutorial005_py310", marks=needs_py310),
60 ],
61)
62def get_module(request: pytest.FixtureRequest) -> ModuleType: 1abcdefghi
63 module = importlib.import_module( 1abcdefghi
64 f"docs_src.tutorial.connect.select.{request.param}"
65 )
66 module.sqlite_url = "sqlite://" 1abcdefghi
67 module.engine = create_engine(module.sqlite_url) 1abcdefghi
68 return module 1abcdefghi
71def test_tutorial(print_mock: PrintMock, module: ModuleType): 1abcdefghi
72 module.main() 1jklmnopqr
73 assert print_mock.calls == expected_calls 1jklmnopqr