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

Author: pre-commit-ci[bot]

tests/test_advanced/test_decimal/test_tutorial001.py

@@ -1,12 +1,11 @@
 import importlib
-import types # Add import for types
+import types  # Add import for types
 from decimal import Decimal
-from unittest.mock import MagicMock # Keep MagicMock for type hint, though not strictly necessary for runtime
 
 import pytest
 from sqlmodel import create_engine
 
-from ...conftest import needs_py310, PrintMock # Import PrintMock for type hint
+from ...conftest import PrintMock, needs_py310  # Import PrintMock for type hint
 
 expected_calls = [
     [
@@ -45,8 +44,10 @@ def get_module(request: pytest.FixtureRequest):
     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
+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
+    assert print_mock.calls == expected_calls  # Use .calls instead of .mock_calls

tests/test_tutorial/test_connect/test_delete/test_tutorial001.py

@@ -69,9 +69,7 @@
 )
 def get_module(request: pytest.FixtureRequest) -> ModuleType:
     module_name = request.param
-    mod = importlib.import_module(
-        f"docs_src.tutorial.connect.delete.{module_name}"
-    )
+    mod = importlib.import_module(f"docs_src.tutorial.connect.delete.{module_name}")
     mod.sqlite_url = "sqlite://"
     mod.engine = create_engine(mod.sqlite_url)
     return mod

tests/test_tutorial/test_connect/test_insert/test_tutorial001.py

@@ -49,9 +49,7 @@
 )
 def get_module(request: pytest.FixtureRequest) -> ModuleType:
     module_name = request.param
-    mod = importlib.import_module(
-        f"docs_src.tutorial.connect.insert.{module_name}"
-    )
+    mod = importlib.import_module(f"docs_src.tutorial.connect.insert.{module_name}")
     mod.sqlite_url = "sqlite://"
     mod.engine = create_engine(mod.sqlite_url)
     return mod

tests/test_tutorial/test_connect/test_select/test_tutorial003.py

@@ -85,9 +85,7 @@
 )
 def get_module(request: pytest.FixtureRequest) -> ModuleType:
     module_name = request.param
-    mod = importlib.import_module(
-        f"docs_src.tutorial.connect.select.{module_name}"
-    )
+    mod = importlib.import_module(f"docs_src.tutorial.connect.select.{module_name}")
     mod.sqlite_url = "sqlite://"
     mod.engine = create_engine(mod.sqlite_url)
     return mod

tests/test_tutorial/test_connect/test_select/test_tutorial004.py

@@ -59,9 +59,7 @@
 )
 def get_module(request: pytest.FixtureRequest) -> ModuleType:
     module_name = request.param
-    mod = importlib.import_module(
-        f"docs_src.tutorial.connect.select.{module_name}"
-    )
+    mod = importlib.import_module(f"docs_src.tutorial.connect.select.{module_name}")
     mod.sqlite_url = "sqlite://"
     mod.engine = create_engine(mod.sqlite_url)
     return mod

tests/test_tutorial/test_connect/test_select/test_tutorial005.py

@@ -61,9 +61,7 @@
 )
 def get_module(request: pytest.FixtureRequest) -> ModuleType:
     module_name = request.param
-    mod = importlib.import_module(
-        f"docs_src.tutorial.connect.select.{module_name}"
-    )
+    mod = importlib.import_module(f"docs_src.tutorial.connect.select.{module_name}")
     mod.sqlite_url = "sqlite://"
     mod.engine = create_engine(mod.sqlite_url)
     return mod

tests/test_tutorial/test_connect/test_update/test_tutorial001.py

@@ -1,6 +1,6 @@
 import importlib
 from types import ModuleType
-from typing import Any # For clear_sqlmodel type hint
+from typing import Any  # For clear_sqlmodel type hint
 
 import pytest
 from sqlmodel import create_engine
@@ -60,14 +60,14 @@
 )
 def get_module(request: pytest.FixtureRequest) -> ModuleType:
     module_name = request.param
-    mod = importlib.import_module(
-        f"docs_src.tutorial.connect.update.{module_name}"
-    )
+    mod = importlib.import_module(f"docs_src.tutorial.connect.update.{module_name}")
     mod.sqlite_url = "sqlite://"
     mod.engine = create_engine(mod.sqlite_url)
     return mod
 
 
-def test_tutorial(clear_sqlmodel: Any, print_mock: PrintMock, module: ModuleType) -> None:
+def test_tutorial(
+    clear_sqlmodel: Any, print_mock: PrintMock, module: ModuleType
+) -> None:
     module.main()
     assert print_mock.calls == expected_calls

tests/test_tutorial/test_create_db_and_table/test_tutorial002.py

@@ -1,6 +1,6 @@
 import importlib
 from types import ModuleType
-from typing import Any # For clear_sqlmodel type hint
+from typing import Any  # For clear_sqlmodel type hint
 
 import pytest
 from sqlalchemy import inspect

tests/test_tutorial/test_create_db_and_table/test_tutorial003.py

@@ -1,6 +1,6 @@
 import importlib
 from types import ModuleType
-from typing import Any # For clear_sqlmodel type hint
+from typing import Any  # For clear_sqlmodel type hint
 
 import pytest
 from sqlalchemy import inspect

tests/test_tutorial/test_fastapi/test_app_testing/test_tutorial001_tests_main.py

@@ -1,15 +1,16 @@
 import importlib
-import sys # Add sys import
+import sys  # Add sys import
 from types import ModuleType
 from typing import Any, Generator
 
 import pytest
 from fastapi.testclient import TestClient
-from sqlmodel import Session, SQLModel, create_engine # Keep this for session_fixture
-from sqlmodel.pool import StaticPool # Keep this for session_fixture
+from sqlmodel import Session, SQLModel, create_engine  # Keep this for session_fixture
+from sqlmodel.pool import StaticPool  # Keep this for session_fixture
 
 from ....conftest import needs_py39, needs_py310
 
+
 # This will be our parametrized fixture providing the versioned 'main' module
 @pytest.fixture(
     name="module",
@@ -20,7 +21,9 @@
         pytest.param("tutorial001_py310", marks=needs_py310),
     ],
 )
-def get_module(request: pytest.FixtureRequest, clear_sqlmodel: Any) -> ModuleType: # clear_sqlmodel is autouse
+def get_module(
+    request: pytest.FixtureRequest, clear_sqlmodel: Any
+) -> ModuleType:  # clear_sqlmodel is autouse
     module_name = f"docs_src.tutorial.fastapi.app_testing.{request.param}.main"
 
     # Forcing reload to try to get a fresh state for models
@@ -30,6 +33,7 @@ def get_module(request: pytest.FixtureRequest, clear_sqlmodel: Any) -> ModuleTyp
         module = importlib.import_module(module_name)
     return module
 
+
 @pytest.fixture(name="session", scope="function")
 def session_fixture(module: ModuleType) -> Generator[Session, None, None]:
     # Store original engine-related attributes from the module
@@ -39,13 +43,13 @@ def session_fixture(module: ModuleType) -> Generator[Session, None, None]:
 
     # Force module to use a fresh in-memory SQLite DB for this test run
     module.sqlite_url = "sqlite://"
-    module.connect_args = {"check_same_thread": False} # Crucial for FastAPI + SQLite
+    module.connect_args = {"check_same_thread": False}  # Crucial for FastAPI + SQLite
 
     # Re-create the engine in the module to use these new settings
     test_engine = create_engine(
         module.sqlite_url,
         connect_args=module.connect_args,
-        poolclass=StaticPool  # Recommended for tests
+        poolclass=StaticPool,  # Recommended for tests
     )
     module.engine = test_engine
 
@@ -55,7 +59,9 @@ def session_fixture(module: ModuleType) -> Generator[Session, None, None]:
         # Fallback if the function isn't named create_db_and_tables
         SQLModel.metadata.create_all(module.engine)
 
-    with Session(module.engine) as session: # Use the module's (now test-configured) engine
+    with Session(
+        module.engine
+    ) as session:  # Use the module's (now test-configured) engine
         yield session
 
     # Teardown: drop tables from the module's engine
@@ -68,14 +74,16 @@ def session_fixture(module: ModuleType) -> Generator[Session, None, None]:
         module.connect_args = original_connect_args
     if original_engine is not None:
         module.engine = original_engine
-    else: # If engine didn't exist, remove the one we created
+    else:  # If engine didn't exist, remove the one we created
         if hasattr(module, "engine"):
             del module.engine
 
 
 @pytest.fixture(name="client", scope="function")
-def client_fixture(session: Session, module: ModuleType) -> Generator[TestClient, None, None]:
-    def get_session_override() -> Generator[Session, None, None]: # Must be a generator
+def client_fixture(
+    session: Session, module: ModuleType
+) -> Generator[TestClient, None, None]:
+    def get_session_override() -> Generator[Session, None, None]:  # Must be a generator
         yield session
 
     module.app.dependency_overrides[module.get_session] = get_session_override
@@ -140,7 +148,7 @@ def test_read_heroes(session: Session, client: TestClient, module: ModuleType):
 
 
 def test_read_hero(session: Session, client: TestClient, module: ModuleType):
-    hero_1 = module.Hero(name="Deadpond", secret_name="Dive Wilson") # Use module.Hero
+    hero_1 = module.Hero(name="Deadpond", secret_name="Dive Wilson")  # Use module.Hero
     session.add(hero_1)
     session.commit()
 
@@ -155,7 +163,7 @@ def test_read_hero(session: Session, client: TestClient, module: ModuleType):
 
 
 def test_update_hero(session: Session, client: TestClient, module: ModuleType):
-    hero_1 = module.Hero(name="Deadpond", secret_name="Dive Wilson") # Use module.Hero
+    hero_1 = module.Hero(name="Deadpond", secret_name="Dive Wilson")  # Use module.Hero
     session.add(hero_1)
     session.commit()
 
@@ -170,13 +178,13 @@ def test_update_hero(session: Session, client: TestClient, module: ModuleType):
 
 
 def test_delete_hero(session: Session, client: TestClient, module: ModuleType):
-    hero_1 = module.Hero(name="Deadpond", secret_name="Dive Wilson") # Use module.Hero
+    hero_1 = module.Hero(name="Deadpond", secret_name="Dive Wilson")  # Use module.Hero
     session.add(hero_1)
     session.commit()
 
     response = client.delete(f"/heroes/{hero_1.id}")
 
-    hero_in_db = session.get(module.Hero, hero_1.id) # Use module.Hero
+    hero_in_db = session.get(module.Hero, hero_1.id)  # Use module.Hero
 
     assert response.status_code == 200
     assert hero_in_db is None

tests/test_tutorial/test_fastapi/test_delete/test_tutorial001.py

@@ -1,12 +1,12 @@
 import importlib
 import sys
 from types import ModuleType
-from typing import Any # For clear_sqlmodel type hint
+from typing import Any  # For clear_sqlmodel type hint
 
 import pytest
 from dirty_equals import IsDict
 from fastapi.testclient import TestClient
-from sqlmodel import SQLModel, create_engine # Import SQLModel for metadata operations
+from sqlmodel import SQLModel, create_engine  # Import SQLModel for metadata operations
 from sqlmodel.pool import StaticPool
 
 from ....conftest import needs_py39, needs_py310
@@ -22,7 +22,7 @@
     ],
 )
 def get_module(request: pytest.FixtureRequest, clear_sqlmodel: Any) -> ModuleType:
-    module_name = f"docs_src.tutorial.fastapi.delete.{request.param}" # No .main here
+    module_name = f"docs_src.tutorial.fastapi.delete.{request.param}"  # No .main here
     if module_name in sys.modules:
         module = importlib.reload(sys.modules[module_name])
     else:
@@ -34,19 +34,23 @@ def get_module(request: pytest.FixtureRequest, clear_sqlmodel: Any) -> ModuleTyp
     module.sqlite_url = "sqlite://"
     module.engine = create_engine(
         module.sqlite_url,
-        connect_args={"check_same_thread": False}, # connect_args from original main.py
-        poolclass=StaticPool
+        connect_args={"check_same_thread": False},  # connect_args from original main.py
+        poolclass=StaticPool,
     )
     # Assuming the module has a create_db_and_tables or similar, or uses SQLModel.metadata directly
     if hasattr(module, "create_db_and_tables"):
         module.create_db_and_tables()
     else:
-        SQLModel.metadata.create_all(module.engine) # Fallback, ensure tables are created
+        SQLModel.metadata.create_all(
+            module.engine
+        )  # Fallback, ensure tables are created
 
     return module
 
 
-def test_tutorial(clear_sqlmodel: Any, module: ModuleType): # clear_sqlmodel is autouse but explicit for safety
+def test_tutorial(
+    clear_sqlmodel: Any, module: ModuleType
+):  # clear_sqlmodel is autouse but explicit for safety
     # The engine and tables are now set up by the 'module' fixture
     # The app's dependency overrides for get_session will use module.engine
 
@@ -56,7 +60,7 @@ def test_tutorial(clear_sqlmodel: Any, module: ModuleType): # clear_sqlmodel is
         hero2_data = {
             "name": "Spider-Boy",
             "secret_name": "Pedro Parqueador",
-            "id": 9000, # Note: ID is part of creation data here
+            "id": 9000,  # Note: ID is part of creation data here
         }
         hero3_data = {
             "name": "Rusty-Man",
@@ -65,13 +69,15 @@ def test_tutorial(clear_sqlmodel: Any, module: ModuleType): # clear_sqlmodel is
         }
         response = client.post("/heroes/", json=hero1_data)
         assert response.status_code == 200, response.text
-        hero1 = response.json() # Get actual ID of hero1
+        hero1 = response.json()  # Get actual ID of hero1
         hero1_id = hero1["id"]
 
         response = client.post("/heroes/", json=hero2_data)
         assert response.status_code == 200, response.text
         hero2 = response.json()
-        hero2_id = hero2["id"] # This will be the ID assigned by DB, not 9000 if 9000 is not allowed on POST
+        hero2_id = hero2[
+            "id"
+        ]  # This will be the ID assigned by DB, not 9000 if 9000 is not allowed on POST
 
         response = client.post("/heroes/", json=hero3_data)
         assert response.status_code == 200, response.text
@@ -86,8 +92,8 @@ def test_tutorial(clear_sqlmodel: Any, module: ModuleType): # clear_sqlmodel is
         # For robustness, let's check for a non-existent ID based on actual data.
         # If hero2_id is 1, check for 9000. If it's 9000, check for 1 (assuming hero1_id is 1).
         non_existent_id_check = 9000
-        if hero2_id == non_existent_id_check: # if DB somehow used 9000
-            non_existent_id_check = hero1_id + hero2_id + 100 # just some other ID
+        if hero2_id == non_existent_id_check:  # if DB somehow used 9000
+            non_existent_id_check = hero1_id + hero2_id + 100  # just some other ID
 
         response = client.get(f"/heroes/{non_existent_id_check}")
         assert response.status_code == 404, response.text
@@ -102,7 +108,9 @@ def test_tutorial(clear_sqlmodel: Any, module: ModuleType): # clear_sqlmodel is
         )
         assert response.status_code == 200, response.text
 
-        response = client.patch(f"/heroes/{non_existent_id_check}", json={"name": "Dragon Cube X"})
+        response = client.patch(
+            f"/heroes/{non_existent_id_check}", json={"name": "Dragon Cube X"}
+        )
         assert response.status_code == 404, response.text
 
         response = client.delete(f"/heroes/{hero2_id}")
@@ -111,7 +119,7 @@ def test_tutorial(clear_sqlmodel: Any, module: ModuleType): # clear_sqlmodel is
         response = client.get("/heroes/")
         assert response.status_code == 200, response.text
         data = response.json()
-        assert len(data) == 2 # After deleting one hero
+        assert len(data) == 2  # After deleting one hero
 
         response = client.delete(f"/heroes/{non_existent_id_check}")
         assert response.status_code == 404, response.text