Remove picture id from reference nodes
This commit is contained in:
@@ -245,7 +245,6 @@ class DumasImageNodeTests(unittest.TestCase):
|
||||
|
||||
result = node.build_reference(
|
||||
image=image,
|
||||
picture_id="2",
|
||||
character_id="char_dave",
|
||||
name="Dave",
|
||||
alias="The Locksmith",
|
||||
@@ -270,10 +269,10 @@ class DumasImageNodeTests(unittest.TestCase):
|
||||
"id": "char-dave",
|
||||
"name": "Dave",
|
||||
"aliases": ["The Locksmith"],
|
||||
"picture_id": 2,
|
||||
"picture_label": "<Picture 2>",
|
||||
"picture_id": None,
|
||||
"picture_label": "",
|
||||
"image": image,
|
||||
"summary": "Dave shown in <Picture 2>.",
|
||||
"summary": "Dave reference.",
|
||||
"description": "Square jaw, tired eyes, cropped brown hair.",
|
||||
"wardrobe": "weathered red flight jacket, grey cargo shorts, black boots",
|
||||
"general": "wears a long grey coat",
|
||||
@@ -289,13 +288,16 @@ class DumasImageNodeTests(unittest.TestCase):
|
||||
},
|
||||
)
|
||||
|
||||
def test_character_reference_input_types_do_not_expose_picture_id(self):
|
||||
required = self.image_nodes.DumasCharacterReferenceNode.INPUT_TYPES()["required"]
|
||||
self.assertNotIn("picture_id", required)
|
||||
|
||||
def test_character_reference_handles_missing_optional_fields(self):
|
||||
node = self.image_nodes.DumasCharacterReferenceNode()
|
||||
image = FakeTensorBatch()
|
||||
|
||||
result = node.build_reference(
|
||||
image=image,
|
||||
picture_id="4",
|
||||
character_id="",
|
||||
name="",
|
||||
alias="",
|
||||
@@ -313,8 +315,8 @@ class DumasImageNodeTests(unittest.TestCase):
|
||||
|
||||
reference = result[0]
|
||||
self.assertEqual(reference["kind"], "character")
|
||||
self.assertEqual(reference["picture_id"], 4)
|
||||
self.assertEqual(reference["picture_label"], "<Picture 4>")
|
||||
self.assertIsNone(reference["picture_id"])
|
||||
self.assertEqual(reference["picture_label"], "")
|
||||
self.assertEqual(reference["wardrobe"], "")
|
||||
self.assertEqual(reference["general"], "")
|
||||
self.assertEqual(reference["facts"]["age"], "")
|
||||
@@ -325,7 +327,6 @@ class DumasImageNodeTests(unittest.TestCase):
|
||||
|
||||
result = node.build_reference(
|
||||
image=image,
|
||||
picture_id="9",
|
||||
location_id="coffee-shop-01",
|
||||
name="Coffee Shop",
|
||||
alias="Cafe Interior",
|
||||
@@ -340,10 +341,10 @@ class DumasImageNodeTests(unittest.TestCase):
|
||||
"id": "coffee-shop-01",
|
||||
"name": "Coffee Shop",
|
||||
"aliases": ["Cafe Interior"],
|
||||
"picture_id": 9,
|
||||
"picture_label": "<Picture 9>",
|
||||
"picture_id": None,
|
||||
"picture_label": "",
|
||||
"image": image,
|
||||
"summary": "Coffee Shop shown in <Picture 9>.",
|
||||
"summary": "Coffee Shop reference.",
|
||||
"description": "Warm tungsten lighting, narrow counter, rainy front window.",
|
||||
"wardrobe": "",
|
||||
"general": "Evening ambience, cramped but cozy.",
|
||||
@@ -351,6 +352,10 @@ class DumasImageNodeTests(unittest.TestCase):
|
||||
},
|
||||
)
|
||||
|
||||
def test_location_reference_input_types_do_not_expose_picture_id(self):
|
||||
required = self.image_nodes.DumasLocationReferenceNode.INPUT_TYPES()["required"]
|
||||
self.assertNotIn("picture_id", required)
|
||||
|
||||
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