mirror of
https://github.com/openmv/openmv.git
synced 2025-11-04 14:49:50 +08:00
Co-authored-by: Kwabena W Agyeman <kwagyeman@users.noreply.github.com> Signed-off-by: iabdalkader <i.abdalkader@gmail.com>
29 lines
734 B
Python
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
|