test: update parameter of unit test (#127)

Signed-off-by: ktro2828 <kotaro.uetake@tier4.jp>

Author: ktro2828

tests/viewer/conftest.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import pytest
 
 from t4_devkit.viewer import RerunViewer

tests/viewer/test_viewer.py

@@ -54,14 +54,16 @@ def test_render_box2ds(dummy_viewer, dummy_box2ds) -> None:
     dummy_viewer.render_box2ds(seconds, camera=camera, rois=rois, class_ids=class_ids, uuids=uuids)
 
 
-def test_render_segmentation2d(dummy_viewer) -> None:
+def test_render_segmentation2d(dummy_viewer, dummy_camera_calibration) -> None:
     """Test rendering 2D segmentation mask with `RerunViewer`."""
     seconds = 1.0  # [sec]
 
+    (width, height), _ = dummy_camera_calibration
+
     camera = "camera"
-    masks = [np.zeros((100, 200), dtype=np.uint8) for _ in range(2)]
-    masks[0][0:10, 0:10] = 1
-    masks[1][20:30, 20:30] = 1
+    masks = [np.zeros((height, width), dtype=np.uint8) for _ in range(2)]
+    masks[0][0:100, 0:100] = 1
+    masks[1][200:300, 200:300] = 1
     class_ids = [1, 2]
 
     dummy_viewer.render_segmentation2d(seconds, camera=camera, masks=masks, class_ids=class_ids)