From 6a412f51037724509e27513ef00964f74de25572 Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Fri, 15 Mar 2024 09:15:56 -0600 Subject: [PATCH] Fix test --- frigate/test/test_record_retention.py | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/frigate/test/test_record_retention.py b/frigate/test/test_record_retention.py index 81230449d..4620c9a3e 100644 --- a/frigate/test/test_record_retention.py +++ b/frigate/test/test_record_retention.py @@ -6,20 +6,28 @@ from frigate.record.maintainer import SegmentInfo class TestRecordRetention(unittest.TestCase): def test_motion_should_keep_motion_not_object(self): - segment_info = SegmentInfo(motion_area=1, active_object_count=0, average_dBFS=0) + segment_info = SegmentInfo( + motion_area=1, active_object_count=0, region_count=0, average_dBFS=0 + ) assert not segment_info.should_discard_segment(RetainModeEnum.motion) assert segment_info.should_discard_segment(RetainModeEnum.active_objects) def test_object_should_keep_object_not_motion(self): - segment_info = SegmentInfo(motion_area=0, active_object_count=1, average_dBFS=0) + segment_info = SegmentInfo( + motion_area=0, active_object_count=1, region_count=0, average_dBFS=0 + ) assert segment_info.should_discard_segment(RetainModeEnum.motion) assert not segment_info.should_discard_segment(RetainModeEnum.active_objects) def test_all_should_keep_all(self): - segment_info = SegmentInfo(motion_area=0, active_object_count=0, average_dBFS=0) + segment_info = SegmentInfo( + motion_area=0, active_object_count=0, region_count=0, average_dBFS=0 + ) assert not segment_info.should_discard_segment(RetainModeEnum.all) def test_should_keep_audio_in_motion_mode(self): - segment_info = SegmentInfo(motion_area=0, active_object_count=0, average_dBFS=1) + segment_info = SegmentInfo( + motion_area=0, active_object_count=0, region_count=0, average_dBFS=1 + ) assert not segment_info.should_discard_segment(RetainModeEnum.motion) assert segment_info.should_discard_segment(RetainModeEnum.active_objects)