From b29faf5f6ef218c181177a00ea0843e4c43a1eb9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Alex=20Gr=C3=B6nholm?= Date: Mon, 30 Dec 2024 17:37:09 +0200 Subject: [PATCH] Fixed mypy errors --- tests/test_component.py | 6 +++--- tests/test_utils.py | 4 ++-- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/test_component.py b/tests/test_component.py index bc714ea..2117f46 100644 --- a/tests/test_component.py +++ b/tests/test_component.py @@ -4,7 +4,7 @@ from contextlib import AsyncExitStack from pathlib import Path from threading import Thread, current_thread -from typing import Any +from typing import Any, cast import pytest from asphalt.core import ( @@ -151,7 +151,7 @@ async def test_close_twice_sync(psycopg_url: str) -> None: await component.start() session = get_resource_nowait(Session) assert isinstance(session.bind, Engine) - pool = session.bind.pool + pool = cast(QueuePool, session.bind.pool) assert isinstance(pool, QueuePool) session.execute(text("SELECT 1")) assert pool.checkedout() == 1 @@ -170,7 +170,7 @@ async def test_close_twice_async(psycopg_url_async: str) -> None: await component.start() session = get_resource_nowait(AsyncSession) assert isinstance(session.bind, AsyncEngine) - pool = session.bind.pool + pool = cast(QueuePool, session.bind.pool) assert isinstance(pool, AsyncAdaptedQueuePool) await session.execute(text("SELECT 1")) assert pool.checkedout() == 1 diff --git a/tests/test_utils.py b/tests/test_utils.py index 5e21c70..d7d1a05 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -19,7 +19,7 @@ def connection(sync_engine: Engine) -> Generator[Connection, Any, None]: Table("table", metadata, Column("column1", Integer, primary_key=True)) Table("table2", metadata, Column("fk_column", ForeignKey("table.column1"))) if conn.dialect.name != "sqlite": - conn.execute(CreateSchema("altschema")) # type: ignore[no-untyped-call] + conn.execute(CreateSchema("altschema")) Table("table3", metadata, Column("fk_column", Integer), schema="altschema") metadata.create_all(conn) @@ -28,7 +28,7 @@ def connection(sync_engine: Engine) -> Generator[Connection, Any, None]: if conn.dialect.name != "sqlite": metadata.drop_all(conn) - conn.execute(DropSchema("altschema")) # type: ignore[no-untyped-call] + conn.execute(DropSchema("altschema")) def test_clear_database(connection: Connection) -> None: