chris.dumas/DumasNodes
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

Simplify H3 prompt curator component outputs

Browse Source
This commit is contained in:
chris.dumas
2026-09-10 08:26:39 +00:00
parent 1a4cfbfd7e
commit 22398c5213
3 changed files with 7 additions and 78 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_dumas_image_nodes.py
+3 -31
View File
@@ -502,27 +502,9 @@ class DumasImageNodeTests(unittest.TestCase):
self.assertEqual(result[12], "grounded handheld thriller")
self.assertEqual(result[13], "steady rain")
self.assertEqual(result[14], "low suspense music")
self.assertEqual(result[15]["name"], "Dave")
self.assertIs(result[15]["image"], dave_image)
self.assertEqual(result[16]["name"], "Coffee Shop")
self.assertIs(result[16]["image"], cafe_image)
self.assertIs(result[15], dave_image)
self.assertIs(result[16], cafe_image)
self.assertIsNone(result[17])
self.assertIn("<Picture 1> Dave", result[24])
self.assertIn("<Picture 2> Coffee Shop", result[24])
self.assertIn("Location context for <Picture 2> Coffee Shop", result[24])
self.assertIn("<Picture 1> Dave", result[25])
self.assertNotIn("<Picture 2> Coffee Shop", result[25])
self.assertIn("<Picture 2> Coffee Shop", result[26])
self.assertIn("Location context for <Picture 2> Coffee Shop", result[26])
self.assertEqual(result[27], "")
self.assertIs(result[34], dave_image)
self.assertIs(result[35], cafe_image)
self.assertIsNone(result[36])
self.assertIn("<Picture 1> Dave", result[43])
self.assertNotIn("<Picture 2> Coffee Shop", result[43])
self.assertIn("<Picture 2> Coffee Shop", result[44])
self.assertIn("Location context for <Picture 2> Coffee Shop", result[44])
self.assertEqual(result[45], "")
def test_h3_prompt_curator_renumbers_explicit_reference_tags(self):
node = self.image_nodes.DumasH3PromptCuratorNode()
@@ -660,20 +642,10 @@ class DumasImageNodeTests(unittest.TestCase):
def test_h3_prompt_curator_uses_documented_reference_limits(self):
node = self.image_nodes.DumasH3PromptCuratorNode()
self.assertEqual(len(node.RETURN_TYPES), 52)
self.assertEqual(len(node.RETURN_TYPES), 24)
self.assertEqual(node.RETURN_NAMES[1:10], tuple(f"ref_image_{i}" for i in range(1, 10)))
self.assertEqual(node.RETURN_NAMES[12:15], ("anchor", "sounds", "bgm"))
self.assertEqual(node.RETURN_NAMES[15:24], tuple(f"original_ref_{i}" for i in range(1, 10)))
self.assertEqual(node.RETURN_NAMES[24], "reference_description")
self.assertEqual(
node.RETURN_NAMES[25:34],
tuple(f"original_ref_description_{i}" for i in range(1, 10)),
)
self.assertEqual(node.RETURN_NAMES[34:43], tuple(f"compiled_ref_image_{i}" for i in range(1, 10)))
self.assertEqual(
node.RETURN_NAMES[43:52],
tuple(f"compiled_ref_description_{i}" for i in range(1, 10)),
)
def test_normalize_reference_upgrades_generic_summary_with_socket_picture_id(self):
image = FakeTensorBatch()