def test_report_dhash(self, sample_image_square_rgb):
log_dict = defaultdict(dict)
image_path = SAMPLE_DATA_DIR / 'wildebeest_rgb.jpg'
result = report_dhash(
sample_image_square_rgb, inpath=image_path, log_dict=log_dict
)
assert log_dict[image_path]['dhash'] == calculate_dhash(sample_image_square_rgb)
assert (result == sample_image_square_rgb).all()
def test_record_dhash_hash_is_different_for_non_duplicates(
self, sample_image_square_rgb, sample_image_tall_grayscale
):
hash1 = calculate_dhash(sample_image_square_rgb, sqrt_hash_size=8)
hash2 = calculate_dhash(sample_image_tall_grayscale, sqrt_hash_size=8)
assert _hamming_dist(hash1, hash2) > 10
def test_record_dhash_hash_robust_to_resize_gray(self, sample_image_tall_grayscale):
hash = calculate_dhash(sample_image_tall_grayscale, sqrt_hash_size=8)
resized_hash = calculate_dhash(
resize(sample_image_tall_grayscale, shape=(24, 24)), sqrt_hash_size=8
)
assert _hamming_dist(hash, resized_hash) < 10
def test_record_dhash_hash_robust_to_resize_rgba(self, sample_image_square_rgba):
hash = calculate_dhash(sample_image_square_rgba, sqrt_hash_size=8)
resized_hash = calculate_dhash(
resize(sample_image_square_rgba, shape=(24, 24)), sqrt_hash_size=8
)
assert _hamming_dist(hash, resized_hash) < 10
def test_calculate_dhash_rgb_hashlen_gray(self, sample_image_tall_grayscale):
hash = calculate_dhash(sample_image_tall_grayscale, sqrt_hash_size=8)
assert _hashlen(hash) in [63, 64]
def test_calculate_dhash_rgb_hashlen_rgba(self, sample_image_square_rgba):
hash = calculate_dhash(sample_image_square_rgba, sqrt_hash_size=8)
assert _hashlen(hash) in [63, 64]