import pytest
from PIL import Image

from adetailer.mediapipe import mediapipe_predict


@pytest.mark.parametrize(
    "model_name",
    [
        "mediapipe_face_short",
        "mediapipe_face_full",
        "mediapipe_face_mesh",
        "mediapipe_face_mesh_eyes_only",
    ],
)
def test_mediapipe(sample_image2: Image.Image, model_name: str):
    result = mediapipe_predict(model_name, sample_image2)
    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)