python/sqlmodel
1
0
Fork 0
mirror of https://github.com/fastapi/sqlmodel.git synced 2025-08-17 20:01:41 +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

16
tests/test_tutorial/test_insert/test_tutorial003.py
View File

@ -4,7 +4,7 @@ import types
from typing import Any
import pytest
from sqlmodel import create_engine, SQLModel, Session, select
from sqlmodel import Session, create_engine, select
from ...conftest import needs_py310
@ -30,12 +30,16 @@ def get_module(request: pytest.FixtureRequest, clear_sqlmodel: Any):
# Create tables. Tutorial003.py in insert focuses on refresh, so tables and initial data are key.
# It's likely main() handles this. If not, direct creation is a fallback.
if hasattr(mod, "create_db_and_tables"): # Some tutorials use this helper
if hasattr(mod, "create_db_and_tables"): # Some tutorials use this helper
mod.create_db_and_tables()
elif hasattr(mod, "Hero") and hasattr(mod.Hero, "metadata"): # Check for Hero model metadata
mod.Hero.metadata.create_all(mod.engine)
elif hasattr(mod, "SQLModel") and hasattr(mod.SQLModel, "metadata"): # Generic fallback
mod.SQLModel.metadata.create_all(mod.engine)
elif hasattr(mod, "Hero") and hasattr(
mod.Hero, "metadata"
): # Check for Hero model metadata
mod.Hero.metadata.create_all(mod.engine)
elif hasattr(mod, "SQLModel") and hasattr(
mod.SQLModel, "metadata"
): # Generic fallback
mod.SQLModel.metadata.create_all(mod.engine)
return mod