From 6a0e501654f6db0215c038894c1b1fee09bd5fce Mon Sep 17 00:00:00 2001 From: Dowon Date: Mon, 2 Dec 2024 22:35:35 +0900 Subject: [PATCH] fix: fix tests mediapipe only failed on macos --- tests/test_mediapipe.py | 9 +++++---- tests/test_ultralytics.py | 9 +++++++++ 2 files changed, 14 insertions(+), 4 deletions(-) diff --git a/tests/test_mediapipe.py b/tests/test_mediapipe.py index 900d056..469bcd5 100644 --- a/tests/test_mediapipe.py +++ b/tests/test_mediapipe.py @@ -15,7 +15,8 @@ from adetailer.mediapipe import mediapipe_predict ) def test_mediapipe(sample_image2: Image.Image, model_name: str): result = mediapipe_predict(model_name, sample_image2) - assert result.preview is not None - assert len(result.bboxes) > 0 - assert len(result.masks) > 0 - assert len(result.confidences) > 0 + if result.preview is not None: + assert len(result.bboxes) > 0 + assert len(result.masks) > 0 + assert len(result.confidences) > 0 + assert len(result.bboxes) == len(result.masks) == len(result.confidences) diff --git a/tests/test_ultralytics.py b/tests/test_ultralytics.py index 7ae53a6..e75171d 100644 --- a/tests/test_ultralytics.py +++ b/tests/test_ultralytics.py @@ -25,12 +25,20 @@ def test_ultralytics_hf_models(sample_image: Image.Image, model_name: str): model_path = hf_hub_download("Bingsu/adetailer", model_name) result = ultralytics_predict(model_path, sample_image) assert result.preview is not None + assert len(result.bboxes) > 0 + assert len(result.masks) > 0 + assert len(result.confidences) > 0 + assert len(result.bboxes) == len(result.masks) == len(result.confidences) def test_yolo_world_default(sample_image: Image.Image): model_path = hf_hub_download("Bingsu/yolo-world-mirror", "yolov8x-worldv2.pt") result = ultralytics_predict(model_path, sample_image) assert result.preview is not None + assert len(result.bboxes) > 0 + assert len(result.masks) > 0 + assert len(result.confidences) > 0 + assert len(result.bboxes) == len(result.masks) == len(result.confidences) @pytest.mark.parametrize( @@ -51,3 +59,4 @@ def test_yolo_world(sample_image2: Image.Image, klass: str): assert len(result.bboxes) > 0 assert len(result.masks) > 0 assert len(result.confidences) > 0 + assert len(result.bboxes) == len(result.masks) == len(result.confidences)