diff --git a/test/dataset/test_utils.py b/test/dataset/test_utils.py index 589a8f38..70105d16 100644 --- a/test/dataset/test_utils.py +++ b/test/dataset/test_utils.py @@ -204,6 +204,12 @@ def test_build_class_index_mapping( mock_detections(xyxy=[[0, 0, 10, 10]], class_id=[1]), DoesNotRaise() ), # single mapping + ( + {0: 1, 1: 2}, + Detections.empty(), + Detections.empty(), + DoesNotRaise() + ), # empty detections ( {0: 1, 1: 2}, mock_detections(xyxy=[[0, 0, 10, 10]], class_id=[0]),