METR
diff --git a/‎tests/test_scanners.py‎
Lines changed: 20 additions & 15 deletions b/‎tests/test_scanners.py‎
Lines changed: 20 additions & 15 deletions
@@ -75,33 +75,38 @@ async def test_citing_scanner_skips_items_without_ids_and_keeps_cites_contiguous
 
 
 @pytest.mark.parametrize(
-    ("with_messages", "with_events", "expected_message_cites", "expected_event_cites"),
+    ("messages", "events", "expected_message_cites", "expected_event_cites"),
     [
-        pytest.param(True, False, ["[M1]", "[M2]"], [], id="messages-but-no-events"),
-        pytest.param(False, True, [], ["[E1]"], id="events-but-no-messages"),
+        pytest.param(
+            [
+                ChatMessageUser(id="m1", content="hello"),
+                ChatMessageAssistant(id="m2", content="hi"),
+            ],
+            [],
+            ["[M1]", "[M2]"],
+            [],
+            id="messages-but-no-events",
+        ),
+        pytest.param(
+            [],
+            [InfoEvent(data={"note": "hello"})],
+            [],
+            ["[E1]"],
+            id="events-but-no-messages",
+        ),
     ],
 )
 @pytest.mark.asyncio
 async def test_citing_scanner_populates_reference_kinds_independently(
-    with_messages: bool,
-    with_events: bool,
+    messages: list[ChatMessage],
+    events: list[Event],
     expected_message_cites: list[str],
     expected_event_cites: list[str],
 ) -> None:
     # Each reference kind is populated from its own collection, so a transcript
     # carrying only one kind must yield references of that kind and an empty
     # list for the other. Exercised in both directions: the two must not be
     # coupled, and neither may suppress the other.
-    messages: list[ChatMessage] = (
-        [
-            ChatMessageUser(id="m1", content="hello"),
-            ChatMessageAssistant(id="m2", content="hi"),
-        ]
-        if with_messages
-        else []
-    )
-    events: list[Event] = [InfoEvent(data={"note": "hello"})] if with_events else []
-
     transcript = Transcript(transcript_id="t3", messages=messages, events=events)
     result = await citing_scanner()(transcript)
     assert isinstance(result, Result)