python

custom_components/deepstack_object/tests.py

@@ -0,0 +1,84 @@
+"""The tests for the Deepstack object component."""
+from .image_processing import get_objects
+
+TARGET = "person"
+IMG_WIDTH = 960
+IMG_HEIGHT = 640
+
+MOCK_PREDICTIONS = [
+    {
+        "confidence": 0.9995428,
+        "label": "person",
+        "y_min": 95,
+        "x_min": 295,
+        "y_max": 523,
+        "x_max": 451,
+    },
+    {
+        "confidence": 0.9994912,
+        "label": "person",
+        "y_min": 99,
+        "x_min": 440,
+        "y_max": 531,
+        "x_max": 608,
+    },
+    {
+        "confidence": 0.9990447,
+        "label": "dog",
+        "y_min": 358,
+        "x_min": 647,
+        "y_max": 539,
+        "x_max": 797,
+    },
+]
+
+PARSED_PREDICTIONS = [
+    {
+        "bounding_box": {
+            "height": 0.669,
+            "width": 0.163,
+            "y_min": 0.148,
+            "x_min": 0.307,
+            "y_max": 0.817,
+            "x_max": 0.47,
+        },
+        "box_area": 0.109,
+        "centroid": {"x": 0.389, "y": 0.483},
+        "name": "person",
+        "confidence": 99.954,
+    },
+    {
+        "bounding_box": {
+            "height": 0.675,
+            "width": 0.175,
+            "y_min": 0.155,
+            "x_min": 0.458,
+            "y_max": 0.83,
+            "x_max": 0.633,
+        },
+        "box_area": 0.118,
+        "centroid": {"x": 0.545, "y": 0.493},
+        "name": "person",
+        "confidence": 99.949,
+    },
+    {
+        "bounding_box": {
+            "height": 0.283,
+            "width": 0.156,
+            "y_min": 0.559,
+            "x_min": 0.674,
+            "y_max": 0.842,
+            "x_max": 0.83,
+        },
+        "box_area": 0.044,
+        "centroid": {"x": 0.752, "y": 0.701},
+        "name": "dog",
+        "confidence": 99.904,
+    },
+]
+
+
+def test_get_objects():
+    objects = get_objects(MOCK_PREDICTIONS, IMG_WIDTH, IMG_HEIGHT)
+    assert len(objects) == 3
+    assert objects[0] == PARSED_PREDICTIONS[0]