some more check for find test

This commit is contained in:
Sefik Ilkin Serengil 2024-01-08 17:31:49 +00:00
parent 0eb1515e11
commit d35833e4e1

View File

@ -7,19 +7,42 @@ logger = Logger("tests/test_find.py")
def test_find_with_exact_path(): def test_find_with_exact_path():
dfs = DeepFace.find(img_path="dataset/img1.jpg", db_path="dataset", silent=True) img_path = "dataset/img1.jpg"
dfs = DeepFace.find(img_path=img_path, db_path="dataset", silent=True)
assert len(dfs) > 0
for df in dfs: for df in dfs:
assert isinstance(df, pd.DataFrame) assert isinstance(df, pd.DataFrame)
# one is img1.jpg itself
identity_df = df[df["identity"] == img_path]
assert identity_df.shape[0] > 0
# validate reproducability
assert identity_df["VGG-Face_cosine"].values[0] == 0
df = df[df["identity"] != img_path]
logger.debug(df.head()) logger.debug(df.head())
assert df.shape[0] > 0 assert df.shape[0] > 0
logger.info("✅ test find for exact path done") logger.info("✅ test find for exact path done")
def test_find_with_array_input(): def test_find_with_array_input():
img1 = cv2.imread("dataset/img1.jpg") img_path = "dataset/img1.jpg"
img1 = cv2.imread(img_path)
dfs = DeepFace.find(img1, db_path="dataset", silent=True) dfs = DeepFace.find(img1, db_path="dataset", silent=True)
assert len(dfs) > 0
for df in dfs: for df in dfs:
assert isinstance(df, pd.DataFrame)
# one is img1.jpg itself
identity_df = df[df["identity"] == img_path]
assert identity_df.shape[0] > 0
# validate reproducability
assert identity_df["VGG-Face_cosine"].values[0] == 0
df = df[df["identity"] != img_path]
logger.debug(df.head()) logger.debug(df.head())
assert df.shape[0] > 0 assert df.shape[0] > 0