Use socket picture labels in ref summaries
This commit is contained in:
@@ -356,6 +356,41 @@ class DumasImageNodeTests(unittest.TestCase):
|
||||
required = self.image_nodes.DumasLocationReferenceNode.INPUT_TYPES()["required"]
|
||||
self.assertNotIn("picture_id", required)
|
||||
|
||||
def test_normalize_reference_upgrades_generic_summary_with_socket_picture_id(self):
|
||||
image = FakeTensorBatch()
|
||||
|
||||
reference = self.image_nodes.normalize_reference(
|
||||
{
|
||||
"kind": "character",
|
||||
"name": "Dave",
|
||||
"image": image,
|
||||
"summary": "Dave reference.",
|
||||
},
|
||||
picture_id=3,
|
||||
allow_image_fallback=False,
|
||||
)
|
||||
|
||||
self.assertEqual(reference["picture_id"], 3)
|
||||
self.assertEqual(reference["picture_label"], "<Picture 3>")
|
||||
self.assertEqual(reference["summary"], "Dave shown in <Picture 3>.")
|
||||
|
||||
def test_normalize_reference_keeps_custom_summary_when_socket_picture_id_is_added(self):
|
||||
image = FakeTensorBatch()
|
||||
|
||||
reference = self.image_nodes.normalize_reference(
|
||||
{
|
||||
"kind": "character",
|
||||
"name": "Dave",
|
||||
"image": image,
|
||||
"summary": "Primary hero look for the opening close-up.",
|
||||
},
|
||||
picture_id=3,
|
||||
allow_image_fallback=False,
|
||||
)
|
||||
|
||||
self.assertEqual(reference["picture_id"], 3)
|
||||
self.assertEqual(reference["summary"], "Primary hero look for the opening close-up.")
|
||||
|
||||
def test_anchor_style_node_exposes_requested_presets(self):
|
||||
input_types = self.image_nodes.DumasAnchorStyleNode.INPUT_TYPES()
|
||||
options = input_types["required"]["anchor_style"][0]
|
||||
|
||||
Reference in New Issue
Block a user