administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-03 09:45:22 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Formatting

Browse Source
This commit is contained in:
Nick Mowen 2022-05-27 13:31:04 -06:00
parent b351bebfc8
commit d487f84792
1 changed files with 22 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
frigate/test/test_util.py
View File

@ -3,6 +3,7 @@ import unittest
from frigate.config import FrigateConfig from frigate.config import FrigateConfig
from frigate.util import BoundingBoxTriggerEnum from frigate.util import BoundingBoxTriggerEnum
class TestConfig(unittest.TestCase): class TestConfig(unittest.TestCase):
def setUp(self): def setUp(self):
self.config = { self.config = {
@ -33,9 +34,15 @@ class TestConfig(unittest.TestCase):
box = (150, 150, 350, 350) box = (150, 150, 350, 350)
zone = frigate_config.cameras["back"].zones["test"] zone = frigate_config.cameras["back"].zones["test"]
assert BoundingBoxTriggerEnum.bottom_center.is_in_zone(centroid, box, zone.contour) assert BoundingBoxTriggerEnum.bottom_center.is_in_zone(
assert BoundingBoxTriggerEnum.left_center.is_in_zone(centroid, box, zone.contour) centroid, box, zone.contour
assert BoundingBoxTriggerEnum.right_center.is_in_zone(centroid, box, zone.contour) )
assert BoundingBoxTriggerEnum.left_center.is_in_zone(
centroid, box, zone.contour
)
assert BoundingBoxTriggerEnum.right_center.is_in_zone(
centroid, box, zone.contour
)
assert BoundingBoxTriggerEnum.top_center.is_in_zone(centroid, box, zone.contour) assert BoundingBoxTriggerEnum.top_center.is_in_zone(centroid, box, zone.contour)
def test_bounding_box_trigger_points_outside_zone(self): def test_bounding_box_trigger_points_outside_zone(self):
@ -44,7 +51,15 @@ class TestConfig(unittest.TestCase):
box = (0, 0, 500, 500) box = (0, 0, 500, 500)
zone = frigate_config.cameras["back"].zones["test"] zone = frigate_config.cameras["back"].zones["test"]
assert not BoundingBoxTriggerEnum.bottom_center.is_in_zone(centroid, box, zone.contour) assert not BoundingBoxTriggerEnum.bottom_center.is_in_zone(
assert not BoundingBoxTriggerEnum.left_center.is_in_zone(centroid, box, zone.contour) centroid, box, zone.contour
assert not BoundingBoxTriggerEnum.right_center.is_in_zone(centroid, box, zone.contour) )
assert not BoundingBoxTriggerEnum.top_center.is_in_zone(centroid, box, zone.contour) assert not BoundingBoxTriggerEnum.left_center.is_in_zone(
centroid, box, zone.contour
)
assert not BoundingBoxTriggerEnum.right_center.is_in_zone(
centroid, box, zone.contour
)
assert not BoundingBoxTriggerEnum.top_center.is_in_zone(
centroid, box, zone.contour
)