#NEWLINE# def test_cublasIsamax( self ) : #NEWLINE# #INDENT# x = np . random . rand ( 5 ) . astype ( np . float32 ) #NEWLINE# x_gpu = gpuarray . to_gpu ( x ) #NEWLINE# result = cublas . cublasIsamax ( self . cublas_handle , x_gpu . size , x_gpu . gpudata , 1 ) #NEWLINE# assert np . allclose ( result , np . argmax ( x ) )