QA: Close images after comparing
This commit is contained in:
parent
ca8790fd8d
commit
ea648d67e6
|
@ -67,6 +67,7 @@ def textdiff_files( golden_filepath: str, new_filepath: str, skip: int = 0 ) ->
|
||||||
def image_is_blank( image_path: str ) -> bool:
|
def image_is_blank( image_path: str ) -> bool:
|
||||||
img = Image.open( image_path )
|
img = Image.open( image_path )
|
||||||
sum = np.sum( np.asarray( img ) )
|
sum = np.sum( np.asarray( img ) )
|
||||||
|
img.close()
|
||||||
|
|
||||||
return sum == 0
|
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 ) )
|
eroded_result_sum = np.sum( np.asarray( eroded_result ) )
|
||||||
retval = eroded_result_sum == 0
|
retval = eroded_result_sum == 0
|
||||||
|
|
||||||
# Save images
|
# Save images, cleanup
|
||||||
#if not retval:
|
#if not retval:
|
||||||
diff_name = image1.filename + ".diff.png"
|
diff_name = image1.filename + ".diff.png"
|
||||||
diff.save( diff_name ) # Note: if the image has alpha, the diff will be mostly transparent
|
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"
|
diff_name = image1.filename + ".binary_result.png"
|
||||||
binary_result.save( diff_name )
|
binary_result.save( diff_name )
|
||||||
|
binary_result.close()
|
||||||
|
|
||||||
diff_name = image1.filename + ".eroded_result_" + str( eroded_result_sum )+ ".png"
|
diff_name = image1.filename + ".eroded_result_" + str( eroded_result_sum )+ ".png"
|
||||||
eroded_result.save( diff_name )
|
eroded_result.save( diff_name )
|
||||||
|
eroded_result.close()
|
||||||
|
|
||||||
|
# Cleanup
|
||||||
|
image1.close()
|
||||||
|
image2.close()
|
||||||
|
|
||||||
return retval
|
return retval
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue