Keep curator auxiliary refs input-aligned

2026-09-17 16:02:27 +00:00
parent bfaa5fa733
commit b5eb941c4b
3 changed files with 59 additions and 15 deletions
@@ -503,13 +503,50 @@ class DumasImageNodeTests(unittest.TestCase):
        self.assertEqual(result[13], "steady rain")
        self.assertEqual(result[14], "low suspense music")
        self.assertIs(result[15], dave_image)
-        self.assertIs(result[16], cafe_image)
-        self.assertIsNone(result[17])
+        self.assertIs(result[16], van_image)
+        self.assertIs(result[17], cafe_image)
        self.assertIn("<Picture 1> Dave", result[24])
        self.assertNotIn("<Picture 2> Coffee Shop", result[24])
-        self.assertIn("<Picture 2> Coffee Shop", result[25])
-        self.assertIn("Location context for <Picture 2> Coffee Shop", result[25])
-        self.assertEqual(result[26], "")
+        self.assertIn("<Picture 2> Blue Van", result[25])
+        self.assertIn("scuffed blue delivery van", result[25])
+        self.assertIn("<Picture 3> Coffee Shop", result[26])
+        self.assertIn("Location context for <Picture 3> Coffee Shop", result[26])
+        self.assertNotIn("Blue Van", prompt)
+
+    def test_h3_prompt_curator_aux_reference_outputs_follow_input_sockets(self):
+        node = self.image_nodes.DumasH3PromptCuratorNode()
+        used_image = FakeTensorBatch()
+        unused_image = FakeTensorBatch()
+        used = self.image_nodes.make_reference(
+            kind="character",
+            image=used_image,
+            name="Maya",
+            description="short silver hair",
+        )
+        unused = self.image_nodes.make_reference(
+            kind="location",
+            image=unused_image,
+            name="Unused Warehouse",
+            description="rusted loading bay doors",
+        )
+
+        result = node.curate_prompt(
+            action_prompt="Maya waits in the rain.",
+            anatomy_guard="off",
+            subject_count_guard="off",
+            ref_1=used,
+            ref_2=unused,
+        )
+
+        self.assertIs(result[1], used_image)
+        self.assertIsNone(result[2])
+        self.assertEqual(result[10], 1)
+        self.assertNotIn("Unused Warehouse", result[0])
+        self.assertIs(result[15], used_image)
+        self.assertIs(result[16], unused_image)
+        self.assertIn("<Picture 1> Maya", result[24])
+        self.assertIn("<Picture 2> Unused Warehouse", result[25])
+        self.assertIn("rusted loading bay doors", result[25])

    def test_h3_prompt_curator_renumbers_explicit_reference_tags(self):
        node = self.image_nodes.DumasH3PromptCuratorNode()