administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-08 20:25:26 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Fix tests

Browse Source
This commit is contained in:
Nicolas Mowen 2024-03-03 05:53:51 -07:00
parent 0975394280
commit f1e3ba8f80
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
frigate/test/test_record_retention.py
View File

@ -7,27 +7,27 @@ from frigate.record.maintainer import SegmentInfo
class TestRecordRetention(unittest.TestCase): class TestRecordRetention(unittest.TestCase):
def test_motion_should_keep_motion_not_object(self): def test_motion_should_keep_motion_not_object(self):
segment_info = SegmentInfo( segment_info = SegmentInfo(
motion_box_count=1, active_object_count=0, average_dBFS=0 motion_area=1, active_object_count=0, average_dBFS=0
) )
assert not segment_info.should_discard_segment(RetainModeEnum.motion) assert not segment_info.should_discard_segment(RetainModeEnum.motion)
assert segment_info.should_discard_segment(RetainModeEnum.active_objects) assert segment_info.should_discard_segment(RetainModeEnum.active_objects)
def test_object_should_keep_object_not_motion(self): def test_object_should_keep_object_not_motion(self):
segment_info = SegmentInfo( segment_info = SegmentInfo(
motion_box_count=0, active_object_count=1, average_dBFS=0 motion_area=0, active_object_count=1, average_dBFS=0
) )
assert segment_info.should_discard_segment(RetainModeEnum.motion) assert segment_info.should_discard_segment(RetainModeEnum.motion)
assert not segment_info.should_discard_segment(RetainModeEnum.active_objects) assert not segment_info.should_discard_segment(RetainModeEnum.active_objects)
def test_all_should_keep_all(self): def test_all_should_keep_all(self):
segment_info = SegmentInfo( segment_info = SegmentInfo(
motion_box_count=0, active_object_count=0, average_dBFS=0 motion_area=0, active_object_count=0, average_dBFS=0
) )
assert not segment_info.should_discard_segment(RetainModeEnum.all) assert not segment_info.should_discard_segment(RetainModeEnum.all)
def test_should_keep_audio_in_motion_mode(self): def test_should_keep_audio_in_motion_mode(self):
segment_info = SegmentInfo( segment_info = SegmentInfo(
motion_box_count=0, active_object_count=0, average_dBFS=1 motion_area=0, active_object_count=0, average_dBFS=1
) )
assert not segment_info.should_discard_segment(RetainModeEnum.motion) assert not segment_info.should_discard_segment(RetainModeEnum.motion)
assert segment_info.should_discard_segment(RetainModeEnum.active_objects) assert segment_info.should_discard_segment(RetainModeEnum.active_objects)