Expose per-beat H3 clip outputs
This commit is contained in:
@@ -276,6 +276,16 @@ class DumasH3LongVideosHelperTests(unittest.TestCase):
|
||||
for index in range(1, 10):
|
||||
self.assertIn(f"ref_{index}", optional)
|
||||
|
||||
def test_node_appends_per_beat_list_outputs_without_reordering_existing_slots(self):
|
||||
self.assertEqual(
|
||||
self.module.H3LongVideos.RETURN_NAMES[-2:],
|
||||
("beat_images", "beat_audio"),
|
||||
)
|
||||
self.assertEqual(
|
||||
self.module.H3LongVideos.OUTPUT_IS_LIST[-2:],
|
||||
(True, True),
|
||||
)
|
||||
|
||||
def test_reference_context_matches_character_names_and_location_tags(self):
|
||||
refs = [
|
||||
{"kind": "character", "image": "img1", "name": "Mara", "description": "silver hair", "wardrobe": "red jacket"},
|
||||
@@ -456,6 +466,7 @@ class DumasH3LongVideosHelperTests(unittest.TestCase):
|
||||
)
|
||||
self.assertIn("2 beat(s)", result[2])
|
||||
self.assertIn("2 shot(s)", result[2])
|
||||
self.assertEqual(result[-2:], ([], []))
|
||||
finally:
|
||||
module.torch = original_torch
|
||||
module.vram_gb = original_vram_gb
|
||||
|
||||
Reference in New Issue
Block a user