#NEWLINE# def test_sieve_out( basic_image ) : #NEWLINE# #INDENT# ' Output array passed in should match the returned array ' #NEWLINE# with Env ( ) : #NEWLINE# #INDENT# output = numpy . zeros_like ( basic_image ) #NEWLINE# output [ 1 : 3 , 1 : 3 ] = 5 #NEWLINE# sieved_image = sieve ( basic_image , basic_image . sum ( ) , out = output ) #NEWLINE# assert numpy . array_equal ( basic_image , sieved_image ) #NEWLINE# assert numpy . array_equal ( output , sieved_image )