Add location helper reference outputs

tests/test_dumas_image_nodes.py

@@ -421,7 +421,17 @@ class DumasImageNodeTests(unittest.TestCase):
         self.assertIn("Coffee Shop is also known as Cafe Interior", result[2])
         self.assertIn("Warm tungsten lighting, narrow counter, rainy front window.", result[2])
         self.assertIn("Evening ambience, cramped but cozy.", result[2])
-        self.assertEqual(len(result), 3)
+        self.assertIs(result[3]["image"], image1)
+        self.assertIs(result[4]["image"], image2)
+        self.assertEqual(result[3]["kind"], "location")
+        self.assertEqual(result[4]["kind"], "location")
+        self.assertEqual(result[3]["id"], "coffee-shop-01")
+        self.assertEqual(result[4]["id"], "coffee-shop-01")
+        self.assertEqual(result[3]["name"], "Coffee Shop")
+        self.assertEqual(result[3]["aliases"], ["Cafe Interior"])
+        self.assertEqual(result[3]["description"], "Warm tungsten lighting, narrow counter, rainy front window")
+        self.assertEqual(result[3]["general"], "Evening ambience, cramped but cozy")
+        self.assertEqual(len(result), 5)
 
     def test_soundscape_helper_defaults_to_selected_preset_description(self):
         node = self.image_nodes.DumasSoundscapeHelperNode()
@@ -560,6 +570,40 @@ class DumasImageNodeTests(unittest.TestCase):
         self.assertIn("exactly one named character: <Picture 1> Dave", result[0])
         self.assertNotIn("exactly 2 named characters", result[0])
 
+    def test_h3_prompt_curator_uses_location_helper_references_by_name(self):
+        helper = self.image_nodes.DumasLocationHelperNode()
+        curator = self.image_nodes.DumasH3PromptCuratorNode()
+        image1 = FakeTensorBatch()
+        image2 = FakeTensorBatch()
+        helper_result = helper.build_location_text(
+            image1=image1,
+            image2=image2,
+            image1_picture_id="1",
+            image2_picture_id="2",
+            location_id="coffee_shop",
+            name="Coffee Shop",
+            alias="Cafe Interior",
+            description="Warm tungsten lighting, narrow counter, rainy front window",
+            general="Evening ambience, cramped but cozy",
+        )
+
+        result = curator.curate_prompt(
+            action_prompt="A slow push through the Coffee Shop as rain streaks the windows.",
+            anatomy_guard="on",
+            subject_count_guard="auto",
+            ref_1=helper_result[3],
+            ref_2=helper_result[4],
+        )
+
+        self.assertIs(result[1], image1)
+        self.assertIs(result[2], image2)
+        self.assertEqual(result[10], 2)
+        self.assertIn("<Picture 1> Coffee Shop", result[0])
+        self.assertIn("<Picture 2> Coffee Shop", result[0])
+        self.assertIn("Location context for <Picture 1> Coffee Shop", result[0])
+        self.assertIn("Warm tungsten lighting", result[0])
+        self.assertNotIn("Subject count guard:", result[0])
+
     def test_h3_prompt_curator_defaults_anatomy_guard_to_on(self):
         required = self.image_nodes.DumasH3PromptCuratorNode.INPUT_TYPES()["required"]