Add seventh H3 scene image slot
This commit is contained in:
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user