python/sqlmodel
1
0
Fork 0
mirror of https://github.com/fastapi/sqlmodel.git synced 2025-08-19 04:41:39 +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_tutorial002.py
View File

@ -4,13 +4,13 @@ import types
from typing import Any
import pytest
from sqlmodel import create_engine, SQLModel, Session, select
from sqlmodel import Session, SQLModel, create_engine, select
from ...conftest import needs_py310, clear_sqlmodel as clear_sqlmodel_fixture # Use aliased import
from ...conftest import needs_py310 # Use aliased import
@pytest.fixture(
name="module", # Fixture provides the main module to be tested (tutorial002 variant)
name="module", # Fixture provides the main module to be tested (tutorial002 variant)
params=[
"tutorial002",
pytest.param("tutorial002_py310", marks=needs_py310),
@ -76,8 +76,10 @@ def module_fixture(request: pytest.FixtureRequest, clear_sqlmodel_fixture: Any):
return mod_tut002
def test_tutorial(module: types.ModuleType, clear_sqlmodel_fixture: Any): # `module` is tutorial002 with .Team attached
module.main() # Executes the tutorial002's data insertion logic
def test_tutorial(
module: types.ModuleType, clear_sqlmodel_fixture: Any
): # `module` is tutorial002 with .Team attached
module.main() # Executes the tutorial002's data insertion logic
with Session(module.engine) as session:
hero_spider_boy = session.exec(
@ -88,7 +90,9 @@ def test_tutorial(module: types.ModuleType, clear_sqlmodel_fixture: Any): # `mod
select(module.Team).where(module.Team.name == "Preventers")
).one()
assert hero_spider_boy.team_id == team_preventers.id
assert hero_spider_boy.team == team_preventers # This checks the relationship resolves
assert (
hero_spider_boy.team == team_preventers
) # This checks the relationship resolves
heroes = session.exec(select(module.Hero)).all()