update unit test

usdot-fhwa-OPS · Aug 21, 2024 · 467dc70 · 467dc70
1 parent 9e0b3fd
commit 467dc70
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/v2i-hub/MUSTSensorDriverPlugin/test/TestMUSTSensorDetection.cpp b/src/v2i-hub/MUSTSensorDriverPlugin/test/TestMUSTSensorDetection.cpp
@@ -18,6 +18,9 @@ TEST(TestMUSTSensorDetection, fromStringToDetectionClassification)
     EXPECT_EQ(DetectionClassification::SEDAN, fromStringToDetectionClassification("sedan"));
     EXPECT_EQ(DetectionClassification::VAN, fromStringToDetectionClassification("van"));
     EXPECT_EQ(DetectionClassification::TRUCK, fromStringToDetectionClassification("truck"));
+    EXPECT_EQ(DetectionClassification::TRUCK, fromStringToDetectionClassification("bus"));
+    EXPECT_EQ(DetectionClassification::TRUCK, fromStringToDetectionClassification("pickup truck"));
+    EXPECT_EQ(DetectionClassification::TRUCK, fromStringToDetectionClassification("pedestrain"));
     EXPECT_EQ(DetectionClassification::NA, fromStringToDetectionClassification("not_a_classification"));
 
 }
@@ -78,6 +81,9 @@ TEST(TestMUSTSensorDetection, detectionClassificationToSensorDetectedObjectType
     EXPECT_STRCASEEQ("SEDAN", detectionClassificationToSensorDetectedObjectType(DetectionClassification::SEDAN).c_str());
     EXPECT_STRCASEEQ("VAN", detectionClassificationToSensorDetectedObjectType(DetectionClassification::VAN).c_str());
     EXPECT_STRCASEEQ("TRUCK", detectionClassificationToSensorDetectedObjectType(DetectionClassification::TRUCK).c_str());
+    EXPECT_STRCASEEQ("BUS", detectionClassificationToSensorDetectedObjectType(DetectionClassification::BUS).c_str());
+    EXPECT_STRCASEEQ("PICKUP TRUCK", detectionClassificationToSensorDetectedObjectType(DetectionClassification::PICKUP_TRUCK).c_str());
+    EXPECT_STRCASEEQ("PEDESTRIAN", detectionClassificationToSensorDetectedObjectType(DetectionClassification::PEDESTRIAN).c_str());
     EXPECT_THROW(detectionClassificationToSensorDetectedObjectType(DetectionClassification::NA).c_str(), std::runtime_error);
 
 }