Expose individual prompt reference descriptions
This commit is contained in:
@@ -510,6 +510,11 @@ class DumasImageNodeTests(unittest.TestCase):
|
||||
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], "")
|
||||
|
||||
def test_h3_prompt_curator_renumbers_explicit_reference_tags(self):
|
||||
node = self.image_nodes.DumasH3PromptCuratorNode()
|
||||
@@ -647,11 +652,15 @@ 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), 25)
|
||||
self.assertEqual(len(node.RETURN_TYPES), 34)
|
||||
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)),
|
||||
)
|
||||
|
||||
def test_normalize_reference_upgrades_generic_summary_with_socket_picture_id(self):
|
||||
image = FakeTensorBatch()
|
||||
|
||||
Reference in New Issue
Block a user