44 lines
1.3 KiB
C++
44 lines
1.3 KiB
C++
#include <gtest/gtest.h>
|
|
|
|
#include <memory>
|
|
|
|
#include "behavior/behavior_event_format.h"
|
|
#include "frame/frame.h"
|
|
|
|
namespace rk3588 {
|
|
namespace {
|
|
|
|
TEST(BehaviorEventModelTest, FrameStoresBehaviorEventsSeparatelyFromDetections) {
|
|
auto frame = std::make_shared<Frame>();
|
|
frame->det = std::make_shared<DetectionResult>();
|
|
frame->behavior_events = std::make_shared<BehaviorEventResult>();
|
|
|
|
BehaviorEventItem item;
|
|
item.event_id = 7;
|
|
item.type = BehaviorEventType::Fall;
|
|
item.status = BehaviorEventStatus::Active;
|
|
item.track_ids = {42};
|
|
item.duration_ms = 1600;
|
|
|
|
frame->behavior_events->items.push_back(item);
|
|
|
|
ASSERT_NE(frame->det, nullptr);
|
|
ASSERT_NE(frame->behavior_events, nullptr);
|
|
ASSERT_EQ(frame->behavior_events->items.size(), 1u);
|
|
ASSERT_EQ(frame->behavior_events->items[0].track_ids.size(), 1u);
|
|
EXPECT_EQ(frame->behavior_events->items[0].track_ids[0], 42);
|
|
EXPECT_EQ(frame->behavior_events->items[0].duration_ms, 1600u);
|
|
}
|
|
|
|
TEST(BehaviorEventModelTest, FormatsBehaviorEventLabelForOsd) {
|
|
BehaviorEventItem item;
|
|
item.type = BehaviorEventType::Intrusion;
|
|
item.track_ids = {9};
|
|
item.duration_ms = 2100;
|
|
|
|
EXPECT_EQ(FormatBehaviorEventLabel(item), "intrusion #9 2.1s");
|
|
}
|
|
|
|
} // namespace
|
|
} // namespace rk3588
|