#NEWLINE# def test_identity( ) : #NEWLINE# #INDENT# x = np . ones ( ( 1 , 1 , 2 , 1 ) ) #NEWLINE# def get_idx( idx ) : #NEWLINE# #INDENT# return x [ idx ] #NEWLINE# #UNINDENT# y = parakeet . imap ( get_idx , x . shape ) #NEWLINE# assert eq ( x , y )