From fefd273a0f5bb70de8740b7284a13db2cba8e836 Mon Sep 17 00:00:00 2001 From: Hardik Dava Date: Tue, 3 Oct 2023 19:45:38 +0200 Subject: [PATCH] removing class list from test inputs --- test/detection/test_utils.py | 9 +-------- 1 file changed, 1 insertion(+), 8 deletions(-) diff --git a/test/detection/test_utils.py b/test/detection/test_utils.py index 63a2c33c..bd99aab5 100644 --- a/test/detection/test_utils.py +++ b/test/detection/test_utils.py @@ -264,7 +264,6 @@ def test_filter_polygons_by_area( [ ( {"predictions": [], "image": {"width": 1000, "height": 1000}}, - ["person", "car", "truck"], (np.empty((0, 4)), np.empty(0), np.empty(0), None), DoesNotRaise(), ), # empty result @@ -282,7 +281,6 @@ def test_filter_polygons_by_area( ], "image": {"width": 1000, "height": 1000}, }, - ["person", "car", "truck"], ( np.array([[175.0, 275.0, 225.0, 325.0]]), np.array([0.9]), @@ -313,7 +311,6 @@ def test_filter_polygons_by_area( ], "image": {"width": 1000, "height": 1000}, }, - ["person", "car", "truck"], ( np.array([[175.0, 275.0, 225.0, 325.0], [450.0, 450.0, 550.0, 550.0]]), np.array([0.9, 0.8]), @@ -337,7 +334,6 @@ def test_filter_polygons_by_area( ], "image": {"width": 1000, "height": 1000}, }, - ["person", "car", "truck"], (np.empty((0, 4)), np.empty(0), np.empty(0), None), DoesNotRaise(), ), # single incorrect instance segmentation result with no points @@ -356,7 +352,6 @@ def test_filter_polygons_by_area( ], "image": {"width": 1000, "height": 1000}, }, - ["person", "car", "truck"], (np.empty((0, 4)), np.empty(0), np.empty(0), None), DoesNotRaise(), ), # single incorrect instance segmentation result with no enough points @@ -380,7 +375,6 @@ def test_filter_polygons_by_area( ], "image": {"width": 1000, "height": 1000}, }, - ["person", "car", "truck"], ( np.array([[175.0, 275.0, 225.0, 325.0]]), np.array([0.9]), @@ -418,7 +412,6 @@ def test_filter_polygons_by_area( ], "image": {"width": 1000, "height": 1000}, }, - ["person", "car", "truck"], ( np.array([[175.0, 275.0, 225.0, 325.0]]), np.array([0.9]), @@ -437,7 +430,7 @@ def test_process_roboflow_result( ) -> None: with exception: result = process_roboflow_result( - roboflow_result=roboflow_result, class_list=class_list + roboflow_result=roboflow_result ) assert np.array_equal(result[0], expected_result[0]) assert np.array_equal(result[1], expected_result[1])