#NEWLINE# def test_vectorized_query_multiple_neighbors( self ) : #NEWLINE# #INDENT# s = 23 #NEWLINE# kk = ( i . kdtree . n + s ) #NEWLINE# ( d , i ) = self . kdtree . query ( np . zeros ( ( 2 , 4 , 3 ) ) , k = kk ) #NEWLINE# assert_equal ( np . shape ( d ) , ( 2 , 4 , kk ) ) #NEWLINE# assert_equal ( np . shape ( i ) , ( 2 , 4 , kk ) ) #NEWLINE# assert_ ( np . all ( ( ~ np . isfinite ( d [ : , : , ( - s ) : ] ) ) ) ) #NEWLINE# assert_ ( np . all ( ( i [ : , : , ( - s ) : ] == self . kdtree . n ) ) )