def test_add_img_on_cluster(self):
"""
Test if the image is add on the cluster
"""
irl = ImageRetrieveLocal("../../holiday_dataset")
path_img = "../../cluster_res_1/0/126401.jpg"
num_cluster = 0
assert (irl.add_img_on_cluster(path_img, num_cluster) == 0)
def test_holiday_images(self):
"""
Test if 50 images are selected
"""
nb_imgs = 50
irl = ImageRetrieveLocal("../../holiday_dataset")
image_list = irl.holiday_images(nb_imgs)
assert (len(image_list) == nb_imgs)
def test_save_des_with_img_into_new_cluster(self):
"""
Test if the function to save descriptor and image on a new directory works.
"""
irl = ImageRetrieveLocal("../holiday_dataset")
path_img = "../../cluster_res_1/0/126400.jpg"
des = irl.extract_descriptor_from_path_img(path_img)
assert (irl.save_des_in_descriptor_directory(des, path_img) == 0)
def test_compare_path_des_and_des_false(self):
"""
Test if the function compare descriptor works with false case.
"""
irl = ImageRetrieveLocal("../../holiday_dataset")
path_des = str(0) + "/des"
path_img = "../../cluster_res_1/1/126402.jpg"
des = irl.extract_descriptor_from_path_img(path_img)
assert (not irl.compare_path_des_and_des(path_des, des))
def test_get_des_from_path_des(self):
"""
Test if the function to get a descriptor using a path of the file containing
the descriptor.
"""
path_img = "../../cluster_res_1/0/126400.jpg"
irl = ImageRetrieveLocal("../../holiday_dataset")
path_des = str(0) + "/des"
des1 = irl.get_des_from_path_des(path_des)
des2 = irl.extract_descriptor_from_path_img(path_img)
assert (np.array_equal(des1, des2))
def test_extract_descriptor_from_path_img(self):
"""
Test if the function extract descriptor from path image
return the correct descriptor.
"""
irl = ImageRetrieveLocal("../../holiday_dataset")
path_img = "../../cluster_res_1/0/126400.jpg"
des1 = irl.extract_descriptor_from_path_img(path_img)
des2 = irl.get_des_from_path_des("0/des")
assert (np.array_equal(des1, des2))
def test_exec(self):
"""
Test if execution algorithm to retrieve local image works.
"""
irl = ImageRetrieveLocal("../../holiday_dataset")
assert (irl.exec(812, 250) == 0)