Fix tag-driven H3 ref routing

2026-09-02 12:55:51 +00:00
parent 130b268d1e
commit 271e7de300
2 changed files with 72 additions and 14 deletions
@@ -458,6 +458,39 @@ class DumasH3LongVideosHelperTests(unittest.TestCase):
        self.assertEqual([self.module._reference_image(ref) for ref in references], ["img2"])
        self.assertEqual(dropped, [])

+    def test_resolve_tag_driven_prompt_refs_keeps_untagged_shot_on_handoff(self):
+        refs = [
+            {"kind": "character", "image": "img1", "name": "Mara"},
+            {"kind": "location", "image": "img2", "name": "Hangar"},
+        ]
+
+        text, references, dropped = self.module.resolve_tag_driven_prompt_refs(
+            "Mara waits in the hangar.",
+            refs,
+        )
+
+        self.assertEqual(text, "Mara waits in the hangar.")
+        self.assertEqual(references, [])
+        self.assertEqual(dropped, [])
+
+    def test_resolve_tag_driven_prompt_refs_keeps_named_refs_on_tagged_shot(self):
+        refs = [
+            {"kind": "character", "image": "img1", "name": "Mara"},
+            {"kind": "location", "image": "img2", "name": "Hangar"},
+        ]
+
+        text, references, dropped = self.module.resolve_tag_driven_prompt_refs(
+            "Mara waits in the hangar near <Picture 2>.",
+            refs,
+        )
+
+        self.assertEqual(text, "Mara waits in the hangar near <Picture 1>.")
+        self.assertEqual(
+            [self.module._reference_image(ref) for ref in references],
+            ["img2", "img1"],
+        )
+        self.assertEqual(dropped, [])
+
    def test_resolve_shot_references_uses_named_characters_without_picture_tags(self):
        refs = [
            {"kind": "character", "image": "img1", "name": "Mara"},