feat(traffic_light_fine_detector): add test case

Signed-off-by: Shin-kyoto <[email protected]>
autowarefoundation · May 30, 2024 · afb9953 · afb9953
1 parent 4c941e3
commit afb9953
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/perception/traffic_light_fine_detector/test/test_nodelet.cpp b/perception/traffic_light_fine_detector/test/test_nodelet.cpp
@@ -40,3 +40,11 @@ tensorrt_yolox::Object createYoloxBbox(
   bbox.type = type;
   return bbox;
 }
+
+TEST(CalWeightedIouTest, NoOverlap)
+{
+  const sensor_msgs::msg::RegionOfInterest map_based_bbox = createMapBasedBbox(0, 0, 10, 10);
+  const tensorrt_yolox::Object yolox_bbox = createYoloxBbox(20, 20, 10, 10, 0.9f, 0);
+
+  EXPECT_FLOAT_EQ(traffic_light::calWeightedIou(map_based_bbox, yolox_bbox), 0.0f);
+}