From c95b358d5fe36d22bbcd11d2b10f0f5900926d19 Mon Sep 17 00:00:00 2001 From: 9qeklajc Date: Sat, 26 Sep 2026 22:23:47 +0200 Subject: [PATCH] test: fix type errors and pin image_output in the model serialisation snapshot --- tests/integration/test_model_serialization.py | 2 ++ tests/unit/test_image_billing_units.py | 6 +++--- tests/unit/test_image_catalog.py | 2 +- 3 files changed, 6 insertions(+), 4 deletions(-) diff --git a/tests/integration/test_model_serialization.py b/tests/integration/test_model_serialization.py index b1477472..fa15305b 100644 --- a/tests/integration/test_model_serialization.py +++ b/tests/integration/test_model_serialization.py @@ -356,6 +356,7 @@ def _expected_serialised_model(provider_id: int) -> dict: "pricing": { "completion": 2.0e-7, "image": 0.0, + "image_output": 0.0, "input_cache_read": 0.0, "input_cache_write": 0.0, "internal_reasoning": 0.0, @@ -369,6 +370,7 @@ def _expected_serialised_model(provider_id: int) -> dict: "sats_pricing": { "completion": 2.0e-7 / _SATS, "image": 0.0, + "image_output": 0.0, "input_cache_read": 0.0, "input_cache_write": 0.0, "internal_reasoning": 0.0, diff --git a/tests/unit/test_image_billing_units.py b/tests/unit/test_image_billing_units.py index d6754966..147de22b 100644 --- a/tests/unit/test_image_billing_units.py +++ b/tests/unit/test_image_billing_units.py @@ -352,10 +352,10 @@ async def _settle( await _drain(response) async with AsyncSession(engine, expire_on_commit=False) as session: - key = await session.get(ApiKey, snapshot.key_hash) + settled = await session.get(ApiKey, snapshot.key_hash) record = await session.get(ReservationRelease, snapshot.release_id) - assert key is not None - return key.balance, key.total_spent, record.status if record else None + assert settled is not None + return settled.balance, settled.total_spent, record.status if record else None @pytest.mark.asyncio diff --git a/tests/unit/test_image_catalog.py b/tests/unit/test_image_catalog.py index e89fc6f1..d74a8e0d 100644 --- a/tests/unit/test_image_catalog.py +++ b/tests/unit/test_image_catalog.py @@ -296,7 +296,7 @@ def test_openrouter_provider_prices_image_models_from_the_image_api() -> None: def test_openai_provider_prices_gpt_image_from_its_token_rate() -> None: catalog = [ - dict(entry, id=entry["id"].removeprefix("openai/")) + dict(entry, id=str(entry["id"]).removeprefix("openai/")) for entry in OPENROUTER_CATALOG[:1] ] provider = OpenAIUpstreamProvider(api_key="k")