Add seventh H3 scene image slot

This commit is contained in:
2026-08-13 09:33:05 +00:00
parent a2a9ff90f6
commit f261a048b7
3 changed files with 40 additions and 17 deletions
+5 -2
View File
@@ -348,12 +348,14 @@ class DumasImageNodeTests(unittest.TestCase):
plan = {"shots": [{"id": "one"}, {"id": "two"}]}
image_1 = FakeTensorBatch()
image_2 = FakeTensorBatch(width=12, height=9)
image_3 = FakeTensorBatch(width=8, height=8)
plan_after_first, _connected = attach_node.attach(plan=plan, scene_index=1, image2=image_1)
plan_after_second, _connected = attach_node.attach(
plan=plan_after_first,
scene_index=2,
image6=image_2,
image7=image_3,
)
scene1 = extract_node.extract(plan_after_second, 1)
@@ -362,10 +364,11 @@ class DumasImageNodeTests(unittest.TestCase):
self.assertIs(scene1[2], image_1)
self.assertEqual(scene1[-1], 1)
self.assertIs(scene2[6], image_2)
self.assertEqual(scene2[-1], 1)
self.assertIs(scene2[7], image_3)
self.assertEqual(scene2[-1], 2)
self.assertEqual(
plan_after_second["_dumas_scene_image_bindings"]["scene_counts"],
{"1": 1, "2": 1},
{"1": 1, "2": 2},
)
def test_h3_plan_scene_images_metadata_is_json_serializable(self):