From b351bebfc848169ee52c69cb456cc3afffdf7acf Mon Sep 17 00:00:00 2001
From: Nick Mowen <nickmowen213@gmail.com>
Date: Fri, 27 May 2022 13:29:06 -0600
Subject: [PATCH] Add tests for bounding box

---
 frigate/test/test_util.py | 50 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 50 insertions(+)
 create mode 100644 frigate/test/test_util.py

diff --git a/frigate/test/test_util.py b/frigate/test/test_util.py
new file mode 100644
index 000000000..b17705431
--- /dev/null
+++ b/frigate/test/test_util.py
@@ -0,0 +1,50 @@
+import unittest
+
+from frigate.config import FrigateConfig
+from frigate.util import BoundingBoxTriggerEnum
+
+class TestConfig(unittest.TestCase):
+    def setUp(self):
+        self.config = {
+            "mqtt": {"host": "mqtt"},
+            "record": {
+                "events": {"retain": {"default": 20, "objects": {"person": 30}}}
+            },
+            "cameras": {
+                "back": {
+                    "ffmpeg": {
+                        "inputs": [
+                            {"path": "rtsp://10.0.0.1:554/video", "roles": ["detect"]}
+                        ]
+                    },
+                    "detect": {
+                        "height": 1080,
+                        "width": 1920,
+                        "fps": 5,
+                    },
+                    "zones": {"test": {"coordinates": "100,100,400,400"}},
+                }
+            },
+        }
+
+    def test_bounding_box_trigger_points_in_zone(self):
+        frigate_config = FrigateConfig(**self.config)
+        centroid = (200, 200)
+        box = (150, 150, 350, 350)
+        zone = frigate_config.cameras["back"].zones["test"]
+
+        assert BoundingBoxTriggerEnum.bottom_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)
+
+    def test_bounding_box_trigger_points_outside_zone(self):
+        frigate_config = FrigateConfig(**self.config)
+        centroid = (200, 200)
+        box = (0, 0, 500, 500)
+        zone = frigate_config.cameras["back"].zones["test"]
+
+        assert not BoundingBoxTriggerEnum.bottom_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)