67 lines
2.4 KiB
C++
67 lines
2.4 KiB
C++
#include <gtest/gtest.h>
|
|
|
|
#include <string>
|
|
|
|
#include "face/face_recog_debug.h"
|
|
#include "utils/simple_json.h"
|
|
|
|
namespace rk3588 {
|
|
namespace {
|
|
|
|
TEST(FaceRecogDebugTest, ParsesDebugConfigWithDefaultsAndOverrides) {
|
|
FaceRecogDebugConfig defaults;
|
|
EXPECT_FALSE(defaults.enabled);
|
|
EXPECT_FALSE(defaults.log_matches);
|
|
EXPECT_EQ(defaults.min_log_interval_ms, 2000);
|
|
|
|
SimpleJson::Object debug_obj;
|
|
debug_obj.emplace("enabled", SimpleJson(true));
|
|
debug_obj.emplace("log_matches", SimpleJson(true));
|
|
debug_obj.emplace("min_log_interval_ms", SimpleJson(350.0));
|
|
|
|
SimpleJson::Object root_obj;
|
|
root_obj.emplace("debug", SimpleJson(std::move(debug_obj)));
|
|
SimpleJson cfg(std::move(root_obj));
|
|
|
|
FaceRecogDebugConfig parsed = ParseFaceRecogDebugConfig(cfg, defaults);
|
|
EXPECT_TRUE(parsed.enabled);
|
|
EXPECT_TRUE(parsed.log_matches);
|
|
EXPECT_EQ(parsed.min_log_interval_ms, 350);
|
|
}
|
|
|
|
TEST(FaceRecogDebugTest, BuildsSummaryLineForUncertainCandidate) {
|
|
FaceRecogItem item;
|
|
item.bbox = Rect{10.0f, 20.0f, 30.0f, 40.0f};
|
|
item.candidate_person_id = 7;
|
|
item.candidate_name = "alice";
|
|
item.best_sim = 0.44f;
|
|
item.second_sim = 0.41f;
|
|
item.state = FaceRecogState::Uncertain;
|
|
|
|
const std::string line = BuildFaceRecogDebugSummaryLine("face_recog", 42, item);
|
|
|
|
EXPECT_NE(line.find("[ai_face_recog] match"), std::string::npos);
|
|
EXPECT_NE(line.find("id=face_recog"), std::string::npos);
|
|
EXPECT_NE(line.find("frame=42"), std::string::npos);
|
|
EXPECT_NE(line.find("status=uncertain"), std::string::npos);
|
|
EXPECT_NE(line.find("candidate=alice"), std::string::npos);
|
|
EXPECT_NE(line.find("candidate_id=7"), std::string::npos);
|
|
EXPECT_NE(line.find("best_sim=0.44"), std::string::npos);
|
|
EXPECT_NE(line.find("second_sim=0.41"), std::string::npos);
|
|
EXPECT_NE(line.find("sim_margin=0.03"), std::string::npos);
|
|
EXPECT_NE(line.find("bbox=(10,20,30,40)"), std::string::npos);
|
|
}
|
|
|
|
TEST(FaceRecogDebugTest, BuildsFrameLineForRecognitionBatch) {
|
|
const std::string line = BuildFaceRecogDebugFrameLine("face_recog", 42, 2, 2);
|
|
|
|
EXPECT_NE(line.find("[ai_face_recog] frame"), std::string::npos);
|
|
EXPECT_NE(line.find("id=face_recog"), std::string::npos);
|
|
EXPECT_NE(line.find("frame=42"), std::string::npos);
|
|
EXPECT_NE(line.find("faces_in=2"), std::string::npos);
|
|
EXPECT_NE(line.find("recog_items=2"), std::string::npos);
|
|
}
|
|
|
|
} // namespace
|
|
} // namespace rk3588
|