Format tests

This commit is contained in:
Nicolas Mowen 2024-02-05 15:24:54 -07:00
parent 78ed912a20
commit d1d9db33e9

View File

@ -288,7 +288,10 @@ class TestObjectBoundingBoxes(unittest.TestCase):
assert len(consolidated_detections) == len(detections) assert len(consolidated_detections) == len(detections)
def test_vert_stacked_cars_not_reduced(self): def test_vert_stacked_cars_not_reduced(self):
detections = [("car", 0.8, (954, 312, 1247, 475), 498512, 1.48, (800, 200, 1400, 600)), ("car", 0.85, (970, 380, 1273, 610), 698752, 1.56, (800, 200, 1400, 700))] detections = [
("car", 0.8, (954, 312, 1247, 475), 498512, 1.48, (800, 200, 1400, 600)),
("car", 0.85, (970, 380, 1273, 610), 698752, 1.56, (800, 200, 1400, 700)),
]
frame_shape = (720, 1280) frame_shape = (720, 1280)
consolidated_detections = reduce_detections(frame_shape, detections) consolidated_detections = reduce_detections(frame_shape, detections)
assert len(consolidated_detections) == len(detections) assert len(consolidated_detections) == len(detections)