🧪 tests added

This commit is contained in:
SkalskiP 2023-07-14 15:06:55 +02:00
parent bfe058c307
commit 3a8c4d63ea
1 changed files with 6 additions and 0 deletions

View File

@ -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]),