openmv/scripts/unittest/tests/lens_correction.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

29 lines
734 B
Python

def unittest(data_path, temp_path):
import image
# Create test image
img = image.Image(64, 64, image.GRAYSCALE)
# Fill with grid pattern
for y in range(64):
for x in range(64):
if (x % 8 == 0) or (y % 8 == 0):
img.set_pixel(x, y, 255)
else:
img.set_pixel(x, y, 0)
# Apply lens distortion correction
img.lens_corr(strength=1.0, zoom=1.0)
# Verify image still has valid dimensions
if img.width() != 64 or img.height() != 64:
return False
# Verify image was modified
stats = img.get_statistics()
if stats.min() == 0 and stats.max() == 255:
# Still has contrast, good
return True
return True