python/sqlmodel
1
0
Fork 0
mirror of https://github.com/fastapi/sqlmodel.git synced 2026-03-13 09:29:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

✅ Simplify tests for code examples, one test file for multiple variants (#1664)

Browse Source 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Motov Yurii
2025-12-24 16:47:05 +01:00
committed by GitHub
parent 4c4d5b44ff
commit 4ea6a32f6f
186 changed files with 1769 additions and 14628 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
tests/test_tutorial/test_connect/test_create_connected_tables/test_tutorial001.py
View File

@@ -1,14 +1,32 @@
import importlib
from types import ModuleType
import pytest
from sqlalchemy import inspect
from sqlalchemy.engine.reflection import Inspector
from sqlmodel import create_engine
from tests.conftest import needs_py310
def test_tutorial001():
from docs_src.tutorial.connect.create_tables import tutorial001 as mod
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
mod.main()
insp: Inspector = inspect(mod.engine)
assert insp.has_table(str(mod.Hero.__tablename__))
assert insp.has_table(str(mod.Team.__tablename__))
@pytest.fixture(
name="module",
params=[
"tutorial001",
pytest.param("tutorial001_py310", marks=needs_py310),
],
)
def get_module(request: pytest.FixtureRequest) -> ModuleType:
module = importlib.import_module(
f"docs_src.tutorial.connect.create_tables.{request.param}"
)
return module
def test_tutorial001(module: ModuleType):
module.sqlite_url = "sqlite://"
module.engine = create_engine(module.sqlite_url)
module.main()
insp: Inspector = inspect(module.engine)
assert insp.has_table(str(module.Hero.__tablename__))
assert insp.has_table(str(module.Team.__tablename__))

17
tests/test_tutorial/test_connect/test_create_connected_tables/test_tutorial001_py310.py
View File

@@ -1,17 +0,0 @@
from sqlalchemy import inspect
from sqlalchemy.engine.reflection import Inspector
from sqlmodel import create_engine
from ....conftest import needs_py310
@needs_py310
def test_tutorial001():
from docs_src.tutorial.connect.create_tables import tutorial001_py310 as mod
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
mod.main()
insp: Inspector = inspect(mod.engine)
assert insp.has_table(str(mod.Hero.__tablename__))
assert insp.has_table(str(mod.Team.__tablename__))

33
tests/test_tutorial/test_connect/test_delete/test_tutorial001.py
View File

@@ -1,8 +1,10 @@
from unittest.mock import patch
import importlib
from types import ModuleType
import pytest
from sqlmodel import create_engine
from ....conftest import get_testing_print_function
from ....conftest import PrintMock, needs_py310
expected_calls = [
[
@@ -58,15 +60,22 @@ expected_calls = [
]
def test_tutorial():
from docs_src.tutorial.connect.delete import tutorial001 as mod
@pytest.fixture(
name="module",
params=[
"tutorial001",
pytest.param("tutorial001_py310", marks=needs_py310),
],
)
def get_module(request: pytest.FixtureRequest) -> ModuleType:
module = importlib.import_module(
f"docs_src.tutorial.connect.delete.{request.param}"
)
module.sqlite_url = "sqlite://"
module.engine = create_engine(module.sqlite_url)
return module
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
mod.main()
assert calls == expected_calls
def test_tutorial(print_mock: PrintMock, module: ModuleType):
module.main()
assert print_mock.calls == expected_calls

73
tests/test_tutorial/test_connect/test_delete/test_tutorial001_py310.py
View File

@@ -1,73 +0,0 @@
from unittest.mock import patch
from sqlmodel import create_engine
from ....conftest import get_testing_print_function, needs_py310
expected_calls = [
[
"Created hero:",
{
"age": None,
"id": 1,
"secret_name": "Dive Wilson",
"team_id": 2,
"name": "Deadpond",
},
],
[
"Created hero:",
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 1,
"name": "Rusty-Man",
},
],
[
"Created hero:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": None,
"name": "Spider-Boy",
},
],
[
"Updated hero:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 1,
"name": "Spider-Boy",
},
],
[
"No longer Preventer:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": None,
"name": "Spider-Boy",
},
],
]
@needs_py310
def test_tutorial():
from docs_src.tutorial.connect.delete import tutorial001_py310 as mod
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
mod.main()
assert calls == expected_calls

33
tests/test_tutorial/test_connect/test_insert/test_tutorial001.py
View File

@@ -1,8 +1,10 @@
from unittest.mock import patch
import importlib
from types import ModuleType
import pytest
from sqlmodel import create_engine
from ....conftest import get_testing_print_function
from ....conftest import PrintMock, needs_py310
expected_calls = [
[
@@ -38,15 +40,22 @@ expected_calls = [
]
def test_tutorial001():
from docs_src.tutorial.connect.insert import tutorial001 as mod
@pytest.fixture(
name="module",
params=[
"tutorial001",
pytest.param("tutorial001_py310", marks=needs_py310),
],
)
def get_module(request: pytest.FixtureRequest) -> ModuleType:
module = importlib.import_module(
f"docs_src.tutorial.connect.insert.{request.param}"
)
module.sqlite_url = "sqlite://"
module.engine = create_engine(module.sqlite_url)
return module
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
mod.main()
assert calls == expected_calls
def test_tutorial001(print_mock: PrintMock, module: ModuleType):
module.main()
assert print_mock.calls == expected_calls

53
tests/test_tutorial/test_connect/test_insert/test_tutorial001_py310.py
View File

@@ -1,53 +0,0 @@
from unittest.mock import patch
from sqlmodel import create_engine
from ....conftest import get_testing_print_function, needs_py310
expected_calls = [
[
"Created hero:",
{
"age": None,
"id": 1,
"secret_name": "Dive Wilson",
"team_id": 2,
"name": "Deadpond",
},
],
[
"Created hero:",
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 1,
"name": "Rusty-Man",
},
],
[
"Created hero:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": None,
"name": "Spider-Boy",
},
],
]
@needs_py310
def test_tutorial001():
from docs_src.tutorial.connect.insert import tutorial001_py310 as mod
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
mod.main()
assert calls == expected_calls

92
tests/test_tutorial/test_connect/test_select/test_tutorial001_py310_tutorial002_py310.py
View File

@@ -1,92 +0,0 @@
from unittest.mock import patch
from sqlmodel import create_engine
from ....conftest import get_testing_print_function, needs_py310
expected_calls = [
[
"Created hero:",
{
"age": None,
"id": 1,
"secret_name": "Dive Wilson",
"team_id": 2,
"name": "Deadpond",
},
],
[
"Created hero:",
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 1,
"name": "Rusty-Man",
},
],
[
"Created hero:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": None,
"name": "Spider-Boy",
},
],
[
"Hero:",
{
"age": None,
"id": 1,
"secret_name": "Dive Wilson",
"team_id": 2,
"name": "Deadpond",
},
"Team:",
{"id": 2, "name": "Z-Force", "headquarters": "Sister Margaret's Bar"},
],
[
"Hero:",
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 1,
"name": "Rusty-Man",
},
"Team:",
{"id": 1, "name": "Preventers", "headquarters": "Sharp Tower"},
],
]
@needs_py310
def test_tutorial001():
from docs_src.tutorial.connect.select import tutorial001_py310 as mod
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
mod.main()
assert calls == expected_calls
@needs_py310
def test_tutorial002():
from docs_src.tutorial.connect.select import tutorial002_py310 as mod
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
mod.main()
assert calls == expected_calls

58
tests/test_tutorial/test_connect/test_select/test_tutorial001_tutorial002.py
View File

@@ -1,8 +1,10 @@
from unittest.mock import patch
import importlib
from types import ModuleType
import pytest
from sqlmodel import create_engine
from ....conftest import get_testing_print_function
from ....conftest import PrintMock, needs_py310
expected_calls = [
[
@@ -62,29 +64,37 @@ expected_calls = [
]
def test_tutorial001():
from docs_src.tutorial.connect.select import tutorial001 as mod
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
mod.main()
assert calls == expected_calls
@pytest.fixture(name="module")
def get_module(request: pytest.FixtureRequest) -> ModuleType:
module = importlib.import_module(
f"docs_src.tutorial.connect.select.{request.param}"
)
module.sqlite_url = "sqlite://"
module.engine = create_engine(module.sqlite_url)
return module
def test_tutorial002():
from docs_src.tutorial.connect.select import tutorial002 as mod
@pytest.mark.parametrize(
"module",
[
"tutorial001",
pytest.param("tutorial001_py310", marks=needs_py310),
],
indirect=True,
)
def test_tutorial001(print_mock: PrintMock, module: ModuleType):
module.main()
assert print_mock.calls == expected_calls
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
mod.main()
assert calls == expected_calls
@pytest.mark.parametrize(
"module",
[
"tutorial002",
pytest.param("tutorial002_py310", marks=needs_py310),
],
indirect=True,
)
def test_tutorial002(print_mock: PrintMock, module: ModuleType):
module.main()
assert print_mock.calls == expected_calls

33
tests/test_tutorial/test_connect/test_select/test_tutorial003.py
View File

@@ -1,8 +1,10 @@
from unittest.mock import patch
import importlib
from types import ModuleType
import pytest
from sqlmodel import create_engine
from ....conftest import get_testing_print_function
from ....conftest import PrintMock, needs_py310
expected_calls = [
[
@@ -74,15 +76,22 @@ expected_calls = [
]
def test_tutorial():
from docs_src.tutorial.connect.select import tutorial003 as mod
@pytest.fixture(
name="module",
params=[
"tutorial003",
pytest.param("tutorial003_py310", marks=needs_py310),
],
)
def get_module(request: pytest.FixtureRequest) -> ModuleType:
module = importlib.import_module(
f"docs_src.tutorial.connect.select.{request.param}"
)
module.sqlite_url = "sqlite://"
module.engine = create_engine(module.sqlite_url)
return module
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
mod.main()
assert calls == expected_calls
def test_tutorial(print_mock: PrintMock, module: ModuleType):
module.main()
assert print_mock.calls == expected_calls

89
tests/test_tutorial/test_connect/test_select/test_tutorial003_py310.py
View File

@@ -1,89 +0,0 @@
from unittest.mock import patch
from sqlmodel import create_engine
from ....conftest import get_testing_print_function, needs_py310
expected_calls = [
[
"Created hero:",
{
"age": None,
"id": 1,
"secret_name": "Dive Wilson",
"team_id": 2,
"name": "Deadpond",
},
],
[
"Created hero:",
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 1,
"name": "Rusty-Man",
},
],
[
"Created hero:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": None,
"name": "Spider-Boy",
},
],
[
"Hero:",
{
"age": None,
"id": 1,
"secret_name": "Dive Wilson",
"team_id": 2,
"name": "Deadpond",
},
"Team:",
{"id": 2, "name": "Z-Force", "headquarters": "Sister Margaret's Bar"},
],
[
"Hero:",
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 1,
"name": "Rusty-Man",
},
"Team:",
{"id": 1, "name": "Preventers", "headquarters": "Sharp Tower"},
],
[
"Hero:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": None,
"name": "Spider-Boy",
},
"Team:",
None,
],
]
@needs_py310
def test_tutorial():
from docs_src.tutorial.connect.select import tutorial003_py310 as mod
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
mod.main()
assert calls == expected_calls

33
tests/test_tutorial/test_connect/test_select/test_tutorial004.py
View File

@@ -1,8 +1,10 @@
from unittest.mock import patch
import importlib
from types import ModuleType
import pytest
from sqlmodel import create_engine
from ....conftest import get_testing_print_function
from ....conftest import PrintMock, needs_py310
expected_calls = [
[
@@ -48,15 +50,22 @@ expected_calls = [
]
def test_tutorial():
from docs_src.tutorial.connect.select import tutorial004 as mod
@pytest.fixture(
name="module",
params=[
"tutorial004",
pytest.param("tutorial004_py310", marks=needs_py310),
],
)
def get_module(request: pytest.FixtureRequest) -> ModuleType:
module = importlib.import_module(
f"docs_src.tutorial.connect.select.{request.param}"
)
module.sqlite_url = "sqlite://"
module.engine = create_engine(module.sqlite_url)
return module
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
mod.main()
assert calls == expected_calls
def test_tutorial(print_mock: PrintMock, module: ModuleType):
module.main()
assert print_mock.calls == expected_calls

63
tests/test_tutorial/test_connect/test_select/test_tutorial004_py310.py
View File

@@ -1,63 +0,0 @@
from unittest.mock import patch
from sqlmodel import create_engine
from ....conftest import get_testing_print_function, needs_py310
expected_calls = [
[
"Created hero:",
{
"age": None,
"id": 1,
"secret_name": "Dive Wilson",
"team_id": 2,
"name": "Deadpond",
},
],
[
"Created hero:",
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 1,
"name": "Rusty-Man",
},
],
[
"Created hero:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": None,
"name": "Spider-Boy",
},
],
[
"Preventer Hero:",
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 1,
"name": "Rusty-Man",
},
],
]
@needs_py310
def test_tutorial():
from docs_src.tutorial.connect.select import tutorial004_py310 as mod
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
mod.main()
assert calls == expected_calls

33
tests/test_tutorial/test_connect/test_select/test_tutorial005.py
View File

@@ -1,8 +1,10 @@
from unittest.mock import patch
import importlib
from types import ModuleType
import pytest
from sqlmodel import create_engine
from ....conftest import get_testing_print_function
from ....conftest import PrintMock, needs_py310
expected_calls = [
[
@@ -50,15 +52,22 @@ expected_calls = [
]
def test_tutorial():
from docs_src.tutorial.connect.select import tutorial005 as mod
@pytest.fixture(
name="module",
params=[
"tutorial005",
pytest.param("tutorial005_py310", marks=needs_py310),
],
)
def get_module(request: pytest.FixtureRequest) -> ModuleType:
module = importlib.import_module(
f"docs_src.tutorial.connect.select.{request.param}"
)
module.sqlite_url = "sqlite://"
module.engine = create_engine(module.sqlite_url)
return module
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
mod.main()
assert calls == expected_calls
def test_tutorial(print_mock: PrintMock, module: ModuleType):
module.main()
assert print_mock.calls == expected_calls

65
tests/test_tutorial/test_connect/test_select/test_tutorial005_py310.py
View File

@@ -1,65 +0,0 @@
from unittest.mock import patch
from sqlmodel import create_engine
from ....conftest import get_testing_print_function, needs_py310
expected_calls = [
[
"Created hero:",
{
"age": None,
"id": 1,
"secret_name": "Dive Wilson",
"team_id": 2,
"name": "Deadpond",
},
],
[
"Created hero:",
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 1,
"name": "Rusty-Man",
},
],
[
"Created hero:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": None,
"name": "Spider-Boy",
},
],
[
"Preventer Hero:",
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 1,
"name": "Rusty-Man",
},
"Team:",
{"id": 1, "name": "Preventers", "headquarters": "Sharp Tower"},
],
]
@needs_py310
def test_tutorial():
from docs_src.tutorial.connect.select import tutorial005_py310 as mod
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
mod.main()
assert calls == expected_calls

33
tests/test_tutorial/test_connect/test_update/test_tutorial001.py
View File

@@ -1,8 +1,10 @@
from unittest.mock import patch
import importlib
from types import ModuleType
import pytest
from sqlmodel import create_engine
from ....conftest import get_testing_print_function
from ....conftest import PrintMock, needs_py310
expected_calls = [
[
@@ -48,15 +50,22 @@ expected_calls = [
]
def test_tutorial(clear_sqlmodel):
from docs_src.tutorial.connect.update import tutorial001 as mod
@pytest.fixture(
name="module",
params=[
"tutorial001",
pytest.param("tutorial001_py310", marks=needs_py310),
],
)
def get_module(request: pytest.FixtureRequest) -> ModuleType:
module = importlib.import_module(
f"docs_src.tutorial.connect.update.{request.param}"
)
module.sqlite_url = "sqlite://"
module.engine = create_engine(module.sqlite_url)
return module
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
mod.main()
assert calls == expected_calls
def test_tutorial(print_mock: PrintMock, module: ModuleType):
module.main()
assert print_mock.calls == expected_calls

63
tests/test_tutorial/test_connect/test_update/test_tutorial001_py310.py
View File

@@ -1,63 +0,0 @@
from unittest.mock import patch
from sqlmodel import create_engine
from ....conftest import get_testing_print_function, needs_py310
expected_calls = [
[
"Created hero:",
{
"age": None,
"id": 1,
"secret_name": "Dive Wilson",
"team_id": 2,
"name": "Deadpond",
},
],
[
"Created hero:",
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 1,
"name": "Rusty-Man",
},
],
[
"Created hero:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": None,
"name": "Spider-Boy",
},
],
[
"Updated hero:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 1,
"name": "Spider-Boy",
},
],
]
@needs_py310
def test_tutorial(clear_sqlmodel):
from docs_src.tutorial.connect.update import tutorial001_py310 as mod
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
mod.main()
assert calls == expected_calls