diff --git a/tests/test_extract_faces.py b/tests/test_extract_faces.py index 7ed228c..2d3fc78 100644 --- a/tests/test_extract_faces.py +++ b/tests/test_extract_faces.py @@ -207,6 +207,16 @@ def test_batch_extract_faces_with_nparray(detector_backend): ) +def test_batch_extract_faces_single_image(): + img_path = "dataset/couple.jpg" + imgs_objs_batch = DeepFace.extract_faces( + img_path=[img_path], + align=True, + ) + assert len(imgs_objs_batch) == 2 + assert [isinstance(obj, dict) for obj in imgs_objs_batch] + + def test_backends_for_enforced_detection_with_non_facial_inputs(): black_img = np.zeros([224, 224, 3]) for detector in detectors: