Expose H3 prompt curator components
This commit is contained in:
@@ -499,6 +499,17 @@ class DumasImageNodeTests(unittest.TestCase):
|
||||
self.assertIsNone(result[3])
|
||||
self.assertEqual(result[10], 2)
|
||||
self.assertIn("input 3-><Picture 2> Coffee Shop", result[11])
|
||||
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.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])
|
||||
|
||||
def test_h3_prompt_curator_renumbers_explicit_reference_tags(self):
|
||||
node = self.image_nodes.DumasH3PromptCuratorNode()
|
||||
@@ -636,8 +647,11 @@ 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), 12)
|
||||
self.assertEqual(len(node.RETURN_TYPES), 25)
|
||||
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")
|
||||
|
||||
def test_normalize_reference_upgrades_generic_summary_with_socket_picture_id(self):
|
||||
image = FakeTensorBatch()
|
||||
|
||||
Reference in New Issue
Block a user