diff --git a/tests/test_cli_tools.py b/tests/test_cli_tools.py index a4ca0a0bc..ebfd1d67d 100644 --- a/tests/test_cli_tools.py +++ b/tests/test_cli_tools.py @@ -199,9 +199,11 @@ def test_tools_install_covers_unknown_pypi_force_and_failures( monkeypatch.setattr( cli_tools.binaries, "resolve", - lambda name: (_ for _ in ()).throw(binaries.OfflineError("offline")) - if name == "broken" - else Path(f"C:\\cache\\{name}.exe"), + lambda name: ( + (_ for _ in ()).throw(binaries.OfflineError("offline")) + if name == "broken" + else Path(f"C:\\cache\\{name}.exe") + ), ) result = runner.invoke( diff --git a/tests/test_memory_handler_native_ops.py b/tests/test_memory_handler_native_ops.py index ef80a5a0d..982f3d0a6 100644 --- a/tests/test_memory_handler_native_ops.py +++ b/tests/test_memory_handler_native_ops.py @@ -895,8 +895,9 @@ async def test_memory_handler_misc_helpers(monkeypatch: pytest.MonkeyPatch, tmp_ __import__("sys").modules, "headroom.memory.tools", SimpleNamespace( - get_memory_tools_optimized=lambda: calls.__setitem__("count", calls["count"] + 1) - or [{"name": "tool"}] + get_memory_tools_optimized=lambda: ( + calls.__setitem__("count", calls["count"] + 1) or [{"name": "tool"}] + ) ), ) cache_handler = MemoryHandler(MemoryConfig(enabled=False), agent_type="codex") diff --git a/tests/test_memory_wrapper.py b/tests/test_memory_wrapper.py index ab9631023..c09a4c5e3 100644 --- a/tests/test_memory_wrapper.py +++ b/tests/test_memory_wrapper.py @@ -159,8 +159,9 @@ def test_wrapped_completions_create_injects_parses_and_stores( ) monkeypatch.setattr( "headroom.memory.wrapper.inject_memory_instruction", - lambda messages, short=True: messages - + [{"role": "system", "content": "memory-instruction"}], + lambda messages, short=True: ( + messages + [{"role": "system", "content": "memory-instruction"}] + ), ) monkeypatch.setattr( "headroom.memory.wrapper.parse_response_with_memory", diff --git a/tests/test_pipeline.py b/tests/test_pipeline.py index ce7d1b78a..1fec1f8b3 100644 --- a/tests/test_pipeline.py +++ b/tests/test_pipeline.py @@ -39,14 +39,16 @@ def test_discover_pipeline_extensions_handles_load_and_init_failures( monkeypatch.setattr( importlib.metadata, "entry_points", - lambda group=None: [ - FakeEntryPoint("working-instance", WorkingExtension()), - FakeEntryPoint("working-class", WorkingExtension), - FakeEntryPoint("bad-load", RuntimeError("bad load")), - FakeEntryPoint("bad-init", NeedsInit), - ] - if group == ENTRY_POINT_GROUP - else [], + lambda group=None: ( + [ + FakeEntryPoint("working-instance", WorkingExtension()), + FakeEntryPoint("working-class", WorkingExtension), + FakeEntryPoint("bad-load", RuntimeError("bad load")), + FakeEntryPoint("bad-init", NeedsInit), + ] + if group == ENTRY_POINT_GROUP + else [] + ), ) discovered = discover_pipeline_extensions() diff --git a/tests/test_proxy_handlers_batch.py b/tests/test_proxy_handlers_batch.py index d2af8784f..765e1164e 100644 --- a/tests/test_proxy_handlers_batch.py +++ b/tests/test_proxy_handlers_batch.py @@ -836,12 +836,14 @@ async def test_handle_google_batch_create_covers_passthrough_revert_and_store_fa pipeline_calls: list[dict[str, object]] = [] handler.openai_pipeline = SimpleNamespace( - apply=lambda **kwargs: pipeline_calls.append(kwargs) - or SimpleNamespace( - messages=[{"role": "user", "content": "inflated"}], - timing={}, - tokens_before=40, - tokens_after=80, + apply=lambda **kwargs: ( + pipeline_calls.append(kwargs) + or SimpleNamespace( + messages=[{"role": "user", "content": "inflated"}], + timing={}, + tokens_before=40, + tokens_after=80, + ) ) ) diff --git a/tests/test_relevance_extra.py b/tests/test_relevance_extra.py index 1dc25a29f..62a8fa895 100644 --- a/tests/test_relevance_extra.py +++ b/tests/test_relevance_extra.py @@ -107,9 +107,9 @@ def test_embedding_score_and_batch_with_fake_model(monkeypatch) -> None: monkeypatch.setattr( scorer, "_encode", - lambda texts: [[1.0, 0.0], [0.5, 0.5]] - if len(texts) == 2 - else [[1.0, 0.0], [0.0, 1.0], [1.0, 0.0]], + lambda texts: ( + [[1.0, 0.0], [0.5, 0.5]] if len(texts) == 2 else [[1.0, 0.0], [0.0, 1.0], [1.0, 0.0]] + ), ) monkeypatch.setattr( embedding, "_cosine_similarity", lambda a, b: 0.75 if a == [1.0, 0.0] else 0.25 diff --git a/tests/test_storage_backends.py b/tests/test_storage_backends.py index 0521c8ade..00860d556 100644 --- a/tests/test_storage_backends.py +++ b/tests/test_storage_backends.py @@ -130,7 +130,7 @@ def test_create_storage_builtin_entrypoint_and_fallback(monkeypatch, tmp_path: P monkeypatch.setattr( "importlib.metadata.entry_points", - lambda group: [SimpleNamespace(name="other", load=lambda: (lambda url: created))], + lambda group: [SimpleNamespace(name="other", load=lambda: lambda url: created)], ) missing_ep = create_storage("custom://missing.db") assert isinstance(missing_ep, FakeSQLiteStorage) @@ -289,6 +289,6 @@ def test_sqlite_storage_get_conn_reuses_connection_and_create_storage_entrypoint created = DummyStorage() monkeypatch.setattr( "importlib.metadata.entry_points", - lambda group: [SimpleNamespace(name="custom", load=lambda: (lambda url: created))], + lambda group: [SimpleNamespace(name="custom", load=lambda: lambda url: created)], ) assert create_storage("custom://db") is created