openmv/scripts/unittest/tests/get_similarity.py
iabdalkader 4b1837f72e scripts: Update unit tests.
Co-authored-by: Kwabena W Agyeman <kwagyeman@users.noreply.github.com>
Signed-off-by: iabdalkader <i.abdalkader@gmail.com>
2025-10-18 17:14:36 +02:00

24 lines
661 B
Python

def unittest(data_path, temp_path):
import image
# Create two similar images
img1 = image.Image(50, 50, image.GRAYSCALE)
img2 = image.Image(50, 50, image.GRAYSCALE)
# Fill with similar patterns
for y in range(50):
for x in range(50):
img1.set_pixel(x, y, 100 + (x + y) % 50)
img2.set_pixel(x, y, 100 + (x + y) % 50)
# Get similarity (SSIM)
similarity = img1.get_similarity(img2)
# Verify similarity object
# Identical images should have high similarity
if similarity is not None:
# Typically returns a dictionary or object with metrics
return True
return True