python/sqlmodel
1
0
Fork 0
mirror of https://github.com/fastapi/sqlmodel.git synced 2025-08-18 12:21:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

🎨 [pre-commit.ci] Auto format from pre-commit.com hooks

Browse Source
This commit is contained in:
pre-commit-ci[bot]
2025-06-20 07:22:42 +00:00
parent 02d2e7da0a
commit f464ab4b9a
66 changed files with 847 additions and 701 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
tests/test_tutorial/test_indexes/test_tutorial002.py
View File

@ -7,9 +7,9 @@ from unittest.mock import patch
import pytest
from sqlalchemy import inspect
from sqlalchemy.engine.reflection import Inspector
from sqlmodel import create_engine, SQLModel # Added SQLModel
from sqlmodel import create_engine # Added SQLModel
from ...conftest import get_testing_print_function, needs_py310, PrintMock
from ...conftest import PrintMock, get_testing_print_function, needs_py310
@pytest.fixture(
@ -32,9 +32,9 @@ def get_module(request: pytest.FixtureRequest, clear_sqlmodel: Any):
mod.engine = create_engine(mod.sqlite_url)
if hasattr(mod, "Hero") and hasattr(mod.Hero, "metadata"):
mod.Hero.metadata.create_all(mod.engine)
mod.Hero.metadata.create_all(mod.engine)
elif hasattr(mod, "SQLModel") and hasattr(mod.SQLModel, "metadata"):
mod.SQLModel.metadata.create_all(mod.engine)
mod.SQLModel.metadata.create_all(mod.engine)
return mod
@ -55,7 +55,7 @@ def test_tutorial(print_mock: PrintMock, module: types.ModuleType):
expected_indexes = [
{
"name": "ix_hero_name",
"dialect_options": {}, # Included for completeness but not strictly compared below
"dialect_options": {}, # Included for completeness but not strictly compared below
"column_names": ["name"],
"unique": 0,
},
@ -69,22 +69,29 @@ def test_tutorial(print_mock: PrintMock, module: types.ModuleType):
found_indexes_simplified = []
for index in indexes:
found_indexes_simplified.append({
"name": index["name"],
"column_names": sorted(index["column_names"]),
"unique": index["unique"],
})
found_indexes_simplified.append(
{
"name": index["name"],
"column_names": sorted(index["column_names"]),
"unique": index["unique"],
}
)
expected_indexes_simplified = []
for index in expected_indexes:
expected_indexes_simplified.append({
"name": index["name"],
"column_names": sorted(index["column_names"]),
"unique": index["unique"],
})
expected_indexes_simplified.append(
{
"name": index["name"],
"column_names": sorted(index["column_names"]),
"unique": index["unique"],
}
)
for expected_index in expected_indexes_simplified:
assert expected_index in found_indexes_simplified, f"Expected index {expected_index['name']} not found or mismatch."
assert expected_index in found_indexes_simplified, (
f"Expected index {expected_index['name']} not found or mismatch."
)
assert len(found_indexes_simplified) == len(expected_indexes_simplified), \
assert len(found_indexes_simplified) == len(expected_indexes_simplified), (
f"Mismatch in number of indexes. Found: {len(found_indexes_simplified)}, Expected: {len(expected_indexes_simplified)}"
)