From 4a4e7066822d52c6f1e8c71ba4bc5275b2a82373 Mon Sep 17 00:00:00 2001 From: Sefik Ilkin Serengil Date: Sat, 13 Apr 2024 08:10:43 +0100 Subject: [PATCH] load base64 function renamed and moved --- tests/test_extract_faces.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/test_extract_faces.py b/tests/test_extract_faces.py index dfbc7cf..c841eb3 100644 --- a/tests/test_extract_faces.py +++ b/tests/test_extract_faces.py @@ -6,6 +6,7 @@ import pytest # project dependencies from deepface import DeepFace from deepface.modules import preprocessing +from deepface.commons import image_utils from deepface.commons import logger as log logger = log.get_singletonish_logger() @@ -60,12 +61,12 @@ def test_file_types_while_loading_base64(): img1_base64 = image_to_base64(image_path=img1_path) with pytest.raises(ValueError, match="input image can be jpg or png, but it is"): - _ = preprocessing.load_base64(uri=img1_base64) + _ = image_utils.load_image_from_base64(uri=img1_base64) img2_path = "dataset/img1.jpg" img2_base64 = image_to_base64(image_path=img2_path) - img2 = preprocessing.load_base64(uri=img2_base64) + img2 = image_utils.load_image_from_base64(uri=img2_base64) # 3 dimensional image should be loaded assert len(img2.shape) == 3