mirror of
https://github.com/fastapi/sqlmodel.git
synced 2025-08-18 12:21:35 +08:00
54 lines
1.3 KiB
Python
54 lines
1.3 KiB
Python
import importlib
|
|
import types # Add import for types
|
|
from decimal import Decimal
|
|
|
|
import pytest
|
|
from sqlmodel import create_engine
|
|
|
|
from ...conftest import PrintMock, needs_py310 # Import PrintMock for type hint
|
|
|
|
expected_calls = [
|
|
[
|
|
"Hero 1:",
|
|
{
|
|
"name": "Deadpond",
|
|
"age": None,
|
|
"id": 1,
|
|
"secret_name": "Dive Wilson",
|
|
"money": Decimal("1.100"),
|
|
},
|
|
],
|
|
[
|
|
"Hero 2:",
|
|
{
|
|
"name": "Rusty-Man",
|
|
"age": 48,
|
|
"id": 3,
|
|
"secret_name": "Tommy Sharp",
|
|
"money": Decimal("2.200"),
|
|
},
|
|
],
|
|
["Total money: 3.300"],
|
|
]
|
|
|
|
|
|
@pytest.fixture(
|
|
name="module",
|
|
params=[
|
|
"tutorial001",
|
|
pytest.param("tutorial001_py310", marks=needs_py310),
|
|
],
|
|
)
|
|
def get_module(request: pytest.FixtureRequest):
|
|
module_name = request.param
|
|
return importlib.import_module(f"docs_src.advanced.decimal.{module_name}")
|
|
|
|
|
|
def test_tutorial(
|
|
print_mock: PrintMock, module: types.ModuleType
|
|
): # Use PrintMock for type hint and types.ModuleType
|
|
module.sqlite_url = "sqlite://"
|
|
module.engine = create_engine(module.sqlite_url)
|
|
module.main()
|
|
assert print_mock.calls == expected_calls # Use .calls instead of .mock_calls
|