mirror of
https://github.com/blakeblackshear/frigate.git
synced 2024-11-21 19:07:46 +01:00
28 lines
731 B
Python
28 lines
731 B
Python
import cv2
|
|
import numpy as np
|
|
from unittest import TestCase, main
|
|
from frigate.video import box_overlaps, reduce_boxes
|
|
|
|
|
|
class TestBoxOverlaps(TestCase):
|
|
def test_overlap(self):
|
|
assert box_overlaps((100, 100, 200, 200), (50, 50, 150, 150))
|
|
|
|
def test_overlap_2(self):
|
|
assert box_overlaps((50, 50, 150, 150), (100, 100, 200, 200))
|
|
|
|
def test_no_overlap(self):
|
|
assert not box_overlaps((100, 100, 200, 200), (250, 250, 350, 350))
|
|
|
|
|
|
class TestReduceBoxes(TestCase):
|
|
def test_cluster(self):
|
|
clusters = reduce_boxes(
|
|
[(144, 290, 221, 459), (225, 178, 426, 341), (343, 105, 584, 250)]
|
|
)
|
|
assert len(clusters) == 2
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main(verbosity=2)
|