test: Fix test_validate_rail_side scene creation

DSD-DBS · Nov 7, 2024 · 6d2a5e4 · 6d2a5e4
1 parent a1c61e1
commit 6d2a5e4
Showing 1 changed file with 12 additions and 12 deletions.
diff --git a/tests/test_raillabel_providerkit/validation/validate_rail_side/test_validate_rail_side.py b/tests/test_raillabel_providerkit/validation/validate_rail_side/test_validate_rail_side.py
@@ -11,16 +11,15 @@
 
 def test_validate_rail_side__no_errors(empty_scene, empty_frame):
     scene = empty_scene
-    scene.objects["a1082ef9-555b-4b69-a888-7da531d8a2eb"] = raillabel.format.Object(
-        uid="a1082ef9-555b-4b69-a888-7da531d8a2eb", name="track0001", type="track"
-    )
     object = raillabel.format.Object(
         uid="a1082ef9-555b-4b69-a888-7da531d8a2eb", name="track0001", type="track"
     )
+    scene.objects[object.uid] = object
     sensor = raillabel.format.Sensor(
         uid="rgb_center",
         type=raillabel.format.SensorType.CAMERA,
     )
+    scene.sensors[sensor.uid] = sensor
     frame = empty_frame
     frame.annotations["325b1f55-a2ef-475f-a780-13e1a9e823c3"] = raillabel.format.Poly2d(
         uid="325b1f55-a2ef-475f-a780-13e1a9e823c3",
@@ -44,23 +43,23 @@ def test_validate_rail_side__no_errors(empty_scene, empty_frame):
         closed=False,
         attributes={"railSide": "rightRail"},
     )
+    scene.frames[frame.uid] = frame
 
     actual = validate_rail_side(scene)
     assert len(actual) == 0
 
 
 def test_validate_rail_side__rail_sides_switched(empty_scene, empty_frame):
     scene = empty_scene
-    scene.objects["a1082ef9-555b-4b69-a888-7da531d8a2eb"] = raillabel.format.Object(
-        uid="a1082ef9-555b-4b69-a888-7da531d8a2eb", name="track0001", type="track"
-    )
     object = raillabel.format.Object(
         uid="a1082ef9-555b-4b69-a888-7da531d8a2eb", name="track0001", type="track"
     )
+    scene.objects[object.uid] = object
     sensor = raillabel.format.Sensor(
         uid="rgb_center",
         type=raillabel.format.SensorType.CAMERA,
     )
+    scene.sensors[sensor.uid] = sensor
     frame = empty_frame
     frame.annotations["325b1f55-a2ef-475f-a780-13e1a9e823c3"] = raillabel.format.Poly2d(
         uid="325b1f55-a2ef-475f-a780-13e1a9e823c3",
@@ -84,23 +83,23 @@ def test_validate_rail_side__rail_sides_switched(empty_scene, empty_frame):
         closed=False,
         attributes={"railSide": "leftRail"},
     )
+    scene.frames[frame.uid] = frame
 
     actual = validate_rail_side(scene)
     assert len(actual) == 1
 
 
 def test_validate_rail_side__two_left_rails(empty_scene, empty_frame):
     scene = empty_scene
-    scene.objects["a1082ef9-555b-4b69-a888-7da531d8a2eb"] = raillabel.format.Object(
-        uid="a1082ef9-555b-4b69-a888-7da531d8a2eb", name="track0001", type="track"
-    )
     object = raillabel.format.Object(
         uid="a1082ef9-555b-4b69-a888-7da531d8a2eb", name="track0001", type="track"
     )
+    scene.objects[object.uid] = object
     sensor = raillabel.format.Sensor(
         uid="rgb_center",
         type=raillabel.format.SensorType.CAMERA,
     )
+    scene.sensors[sensor.uid] = sensor
     frame = empty_frame
     frame.annotations["325b1f55-a2ef-475f-a780-13e1a9e823c3"] = raillabel.format.Poly2d(
         uid="325b1f55-a2ef-475f-a780-13e1a9e823c3",
@@ -124,23 +123,23 @@ def test_validate_rail_side__two_left_rails(empty_scene, empty_frame):
         closed=False,
         attributes={"railSide": "leftRail"},
     )
+    scene.frames[frame.uid] = frame
 
     actual = validate_rail_side(scene)
     assert len(actual) == 1
 
 
 def test_validate_rail_side__two_right_rails(empty_scene, empty_frame):
     scene = empty_scene
-    scene.objects["a1082ef9-555b-4b69-a888-7da531d8a2eb"] = raillabel.format.Object(
-        uid="a1082ef9-555b-4b69-a888-7da531d8a2eb", name="track0001", type="track"
-    )
     object = raillabel.format.Object(
         uid="a1082ef9-555b-4b69-a888-7da531d8a2eb", name="track0001", type="track"
     )
+    scene.objects[object.uid] = object
     sensor = raillabel.format.Sensor(
         uid="rgb_center",
         type=raillabel.format.SensorType.CAMERA,
     )
+    scene.sensors[sensor.uid] = sensor
     frame = empty_frame
     frame.annotations["325b1f55-a2ef-475f-a780-13e1a9e823c3"] = raillabel.format.Poly2d(
         uid="325b1f55-a2ef-475f-a780-13e1a9e823c3",
@@ -164,6 +163,7 @@ def test_validate_rail_side__two_right_rails(empty_scene, empty_frame):
         closed=False,
         attributes={"railSide": "rightRail"},
     )
+    scene.frames[frame.uid] = frame
 
     actual = validate_rail_side(scene)
     assert len(actual) == 1