test: fix label test for TLR

Signed-off-by: ktro2828 <[email protected]>
tier4 · Dec 27, 2023 · 97f9e2d · 97f9e2d
1 parent b41e743
commit 97f9e2d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/perception_eval/perception_eval/common/label.py b/perception_eval/perception_eval/common/label.py
@@ -381,6 +381,7 @@ def _get_traffic_light_paris(
         ]
     else:
         pair_list: List[Tuple[TrafficLightLabel, str]] = [
+            (TrafficLightLabel.TRAFFIC_LIGHT, "traffic_light"),
             (TrafficLightLabel.TRAFFIC_LIGHT, "green"),
             (TrafficLightLabel.TRAFFIC_LIGHT, "green_straight"),
             (TrafficLightLabel.TRAFFIC_LIGHT, "green_left"),

diff --git a/perception_eval/test/common/test_label.py b/perception_eval/test/common/test_label.py
@@ -65,10 +65,10 @@
     (TrafficLightLabel.YELLOW, "yellow"),
     (TrafficLightLabel.RED_STRAIGHT, "red_straight"),
     (TrafficLightLabel.RED_LEFT, "red_left"),
-    (TrafficLightLabel.RED_STRAIGHT_LEFT, "red_left_straight"),
+    (TrafficLightLabel.RED_STRAIGHT_LEFT, "red_straight_left"),
     (TrafficLightLabel.RED_LEFT_DIAGONAL, "red_left_diagonal"),
     (TrafficLightLabel.RED_RIGHT, "red_right"),
-    (TrafficLightLabel.RED_STRAIGHT_RIGHT, "red_right_straight"),
+    (TrafficLightLabel.RED_STRAIGHT_RIGHT, "red_straight_right"),
     (TrafficLightLabel.RED_RIGHT_DIAGONAL, "red_right_diagonal"),
     (TrafficLightLabel.YELLOW_RIGHT, "yellow_right"),
     (TrafficLightLabel.UNKNOWN, "unknown"),