diff --git a/tests/api/v1/test_reasoning_heartbeat_liveness.py b/tests/api/v1/test_reasoning_heartbeat_liveness.py index 2fc843ab..a1b198f7 100644 --- a/tests/api/v1/test_reasoning_heartbeat_liveness.py +++ b/tests/api/v1/test_reasoning_heartbeat_liveness.py @@ -365,7 +365,7 @@ class TestHeartbeatPumpsDuringReasoning: assert msg_id is not None # Drain the rest, snapshotting the row's heartbeat after each frame. - for _frame in gen: + for _ in gen: with pg_engine.connect() as conn: observed.append(_row(conn, str(msg_id))) diff --git a/tests/api/v1/test_v1_continuation_ownership.py b/tests/api/v1/test_v1_continuation_ownership.py index 757e5396..45b590e7 100644 --- a/tests/api/v1/test_v1_continuation_ownership.py +++ b/tests/api/v1/test_v1_continuation_ownership.py @@ -179,7 +179,7 @@ class TestForeignConversationOwnership: # write was rejected before any persistence. with pg_engine.connect() as conn: rows_after = _row_count(conn, conv_id) - assert rows_after == rows_before == 1 + assert rows_after == rows_before def test_owner_can_continue_into_own_conversation(self, pg_engine, monkeypatch): """Control: the SAME continuation into the caller's OWN conversation