From 27dea80248be128942420bd860c6f964ad91e38c Mon Sep 17 00:00:00 2001
From: galthran-wq <levmorozov900@gmail.com>
Date: Tue, 18 Feb 2025 10:13:03 +0000
Subject: [PATCH] batch test add other detector models

---
 tests/test_extract_faces.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/tests/test_extract_faces.py b/tests/test_extract_faces.py
index 3ac1dcb..11c37fa 100644
--- a/tests/test_extract_faces.py
+++ b/tests/test_extract_faces.py
@@ -84,15 +84,20 @@ def test_different_detectors():
     "ssd",
     "mtcnn",
     "retinaface",
+    "yunet",
+    "centerface",
+    # optional
+    # "yolov11s",
+    # "mediapipe",
+    # "dlib",
 ])
 def test_batch_extract_faces(detector_backend):
     detector_backend_to_rtol = {
         "opencv": 0.1,
-        "ssd": 0.01,
         "mtcnn": 0.2,
-        "retinaface": 0.01,
+        "yolov11s": 0.03,
     }
-    rtol = detector_backend_to_rtol[detector_backend]
+    rtol = detector_backend_to_rtol.get(detector_backend, 0.01)
     img_paths = [
         "dataset/img2.jpg",
         "dataset/img3.jpg",