QA: Close images after comparing

This commit is contained in:
Roberto Fernandez Bautista 2023-09-12 09:40:30 +00:00
parent ca8790fd8d
commit ea648d67e6
1 changed files with 9 additions and 1 deletions

View File

@ -67,6 +67,7 @@ def textdiff_files( golden_filepath: str, new_filepath: str, skip: int = 0 ) ->
def image_is_blank( image_path: str ) -> bool:
img = Image.open( image_path )
sum = np.sum( np.asarray( img ) )
img.close()
return sum == 0
@ -106,16 +107,23 @@ def images_are_equal( image1_path: str, image2_path: str ) -> bool:
eroded_result_sum = np.sum( np.asarray( eroded_result ) )
retval = eroded_result_sum == 0
# Save images
# Save images, cleanup
#if not retval:
diff_name = image1.filename + ".diff.png"
diff.save( diff_name ) # Note: if the image has alpha, the diff will be mostly transparent
diff.close()
diff_name = image1.filename + ".binary_result.png"
binary_result.save( diff_name )
binary_result.close()
diff_name = image1.filename + ".eroded_result_" + str( eroded_result_sum )+ ".png"
eroded_result.save( diff_name )
eroded_result.close()
# Cleanup
image1.close()
image2.close()
return retval