mypy init

alembic/env.py

@@ -65,9 +65,8 @@ def run_migrations_online() -> None:
     and associate a connection with the context.
 
     """
-    if get_set_engine() is not None:
-        connectable = get_set_engine()
-    else:
+    connectable = get_set_engine()
+    if connectable is None:
         connectable = engine_from_config(
             config.get_section(config.config_ini_section, {}),
             prefix="sqlalchemy.",

pyproject.toml

@@ -29,9 +29,16 @@ module-root = ""
 requires = ["uv_build>=0.12.9,<0.13.0"]
 build-backend = "uv_build"
 
+[tool.mypy]
+explicit_package_bases = true
+mypy_path = "."
+
 [tool.ruff]
 target-version = "py314"
 
+[tool.black]
+target-version = [ "py314" ]
+
 [tool.ruff.lint]
 ignore = ["EXE"]
 

tests/test_transactions.py

@@ -112,6 +112,7 @@ def test_consolidate_b4w_signed_amount_and_category(db_session: Session):
         CdtDbtInd="DBIT",
     )
     staging = upsert_b4w(db_session, export, do_commit=False)
+    assert staging is not None
     consolidate_transactions(
         db_session, "b4w", B4wDb, map_b4w_to_transaction, do_commit=False
     )
@@ -130,10 +131,12 @@ def test_ingest_b4w_e2e_idempotent(db_session: Session):
     for _ in range(2):
         ingest_b4w(B4W_PATH, db_session)
 
-    staging_count = db_session.execute(select(func.count()).select_from(B4wDb)).scalar()
+    staging_count = db_session.execute(
+        select(func.count()).select_from(B4wDb)
+    ).scalar_one()
     transaction_count = db_session.execute(
         select(func.count()).select_from(TransactionDb)
-    ).scalar()
+    ).scalar_one()
     assert staging_count > 0
     assert transaction_count == staging_count
 
@@ -153,10 +156,10 @@ def test_ingest_comdirect_e2e_idempotent(db_session: Session):
 
     staging_count = db_session.execute(
         select(func.count()).select_from(ComdirectGiroDb)
-    ).scalar()
+    ).scalar_one()
     transaction_count = db_session.execute(
         select(func.count()).select_from(TransactionDb)
-    ).scalar()
+    ).scalar_one()
     assert staging_count > 0
     assert transaction_count == staging_count
 
@@ -174,13 +177,13 @@ def test_ingest_mixed_sources_into_one_table(db_session: Session):
         account_currency="EUR",
     )
 
-    b4w_count = db_session.execute(select(func.count()).select_from(B4wDb)).scalar()
+    b4w_count = db_session.execute(select(func.count()).select_from(B4wDb)).scalar_one()
     giro_count = db_session.execute(
         select(func.count()).select_from(ComdirectGiroDb)
-    ).scalar()
+    ).scalar_one()
     transaction_count = db_session.execute(
         select(func.count()).select_from(TransactionDb)
-    ).scalar()
+    ).scalar_one()
 
     assert b4w_count > 0
     assert giro_count > 0