Update H3 character helper outputs
This commit is contained in:
@@ -260,6 +260,7 @@ class DumasImageNodeTests(unittest.TestCase):
|
||||
height_inches="2",
|
||||
accent="English",
|
||||
general="wears a long grey coat",
|
||||
wardrobe="weathered red flight jacket, grey cargo shorts, black boots",
|
||||
)
|
||||
|
||||
self.assertIs(result[0], image1)
|
||||
@@ -274,6 +275,10 @@ class DumasImageNodeTests(unittest.TestCase):
|
||||
"wears a long grey coat."
|
||||
),
|
||||
)
|
||||
self.assertEqual(
|
||||
result[3],
|
||||
"Dave = weathered red flight jacket, grey cargo shorts, black boots",
|
||||
)
|
||||
|
||||
def test_character_helper_handles_missing_optional_fields(self):
|
||||
node = self.image_nodes.DumasCharacterHelperNode()
|
||||
@@ -296,6 +301,7 @@ class DumasImageNodeTests(unittest.TestCase):
|
||||
height_inches="",
|
||||
accent="",
|
||||
general="",
|
||||
wardrobe="",
|
||||
)
|
||||
|
||||
self.assertEqual(
|
||||
@@ -306,6 +312,41 @@ class DumasImageNodeTests(unittest.TestCase):
|
||||
"<Picture 6> is a frontal facial reference for the character."
|
||||
),
|
||||
)
|
||||
self.assertEqual(result[3], "")
|
||||
|
||||
def test_character_helper_allows_picture_one_and_preserves_full_sheet_wardrobe(self):
|
||||
node = self.image_nodes.DumasCharacterHelperNode()
|
||||
image1 = FakeTensorBatch()
|
||||
image2 = FakeTensorBatch()
|
||||
|
||||
result = node.build_character_text(
|
||||
image1=image1,
|
||||
image2=image2,
|
||||
image1_picture_id="1",
|
||||
image2_picture_id="9",
|
||||
character_id="char_kristy",
|
||||
name="Kristy",
|
||||
alias="",
|
||||
gender="",
|
||||
age="",
|
||||
nationality="",
|
||||
occupation="",
|
||||
height_feet="",
|
||||
height_inches="",
|
||||
accent="",
|
||||
general="",
|
||||
wardrobe="Kristy = black coat, silver boots",
|
||||
)
|
||||
|
||||
self.assertEqual(
|
||||
result[2],
|
||||
(
|
||||
"<Picture 1> and <Picture 9> reference the same character who is called Kristy.\n"
|
||||
"<Picture 1> is the primary full-body reference for Kristy.\n"
|
||||
"<Picture 9> is a frontal facial reference for Kristy."
|
||||
),
|
||||
)
|
||||
self.assertEqual(result[3], "Kristy = black coat, silver boots")
|
||||
|
||||
def test_save_image_returns_ui_entries_for_output_folder(self):
|
||||
node = self.image_nodes.DumasSaveImageNode()
|
||||
|
||||
Reference in New Issue
Block a user