PiE ddlZddlZddlZddlmZddlZddlZddlm Z ddl m Z m Z ddl mZmZmZmZmZmZddlmZmZmZddlmZddlmZgd Zd;d ed ed efdZdZ dZ! dd)Z3ej+,d&eegej+,d'ej'ej(g d>d*Z4ej+,d+eej'feej(feej(feej'fgej+,d ed,Z5ej+,d&eeg d?d-Z6ej+,d eej+,d.d/ej+,d'ej'ej(gej+,d e d@d1Z7ej+,d eej+,d.d/ej+,d'ej'ej(g dAd2Z8ej+,d&eegej+,d d3d4gej+,d'ej'ej(gd5Z9ej+,d'ej'ej(gej+,d ed6Z:d7Z;ej+,d8gd9d:ZQ#n"5"5"9"9 /y / /.< / /.< / /% / /%  / / +  / / +  / / /   #:"9r-h㈵>ư>c "d}|j|jcxkr|jcxkr |jks nJd|j\}}t|D]} || } || } || } || } || s Jd| || s Jd| t| | | | | ||d|z tjtj| tj| z|z }t | | | | | |dS)aAssert that argkmin results are valid up to rounding errors. This function asserts that the results of argkmin queries are valid up to: - rounding error tolerance on distance values; - permutations of indices for distances values that differ up to the expected precision level. Furthermore, the distances must be sorted. To be used for testing neighbors queries on float32 datasets: we accept neighbors rank swaps only if they are caused by small rounding errors on the distance computations. cRtj|dd|ddkSNrr#allas r+z3assert_compatible_argkmin_results..$"&3B31QRR511r-z+Arrays of results have incompatible shapes.Distances aren't sorted on row rN)shaperangerCr#maximummaxrM)neighbors_dists_aneighbors_dists_bneighbors_indices_aneighbors_indices_br/r0 is_sorted n_queries_r7r8r9r:r;rHs r+!assert_compatible_argkmin_resultsrfs*21I   " % % % %  $ % % % %  $ % % % % %5  % % % %*LIq9%%& & &y1 &y1 +I6 +I6 y$$SS&S &S&SSS$y$$SS&S &S&SSS$2          &X F:  z 2 2" "    $           ?& & r- )XYrprecomputed_distsexpected_n_neighborsn_subsampled_queriesc | | Jd||J|Jt||fd|i|}n|d|}|d|dd|fS)Nz4Either metric or precomputed_dists must be provided.rraxis)rcopysortmean)rhrirrjrkrl metric_kwargs sampled_distss r+_non_trivial_radiusrus  (F,>,>>-?,> > }}}}}}*1aPPP-PP )*?+?*?@EEGG A 00 1 6 6 8 88r-Tc ~d}t|t|cxkr't|cxkrt|ksnJt|} t| D]M} || } || } || } || }|r0|| s Jd| || s Jd| t| t| ksJt| t|ksJt| dkr-tj| }||ksJd|d|d| t| dkr-tj| }||ksJd|d|d| t | | | | |||d|z |z|z }t | | | | ||OdS) a&Assert that radius neighborhood results are valid up to: - relative and absolute tolerance on computed distance values - permutations of indices for distances values that differ up to a precision level - missing or extra last elements if their distance is close to the radius To be used for testing neighbors queries on float32 datasets: we accept neighbors rank swaps only if they are caused by small rounding errors on the distance computations. Input arrays must be sorted w.r.t distances. cRtj|dd|ddkSrRrTrVs r+rXz2assert_compatible_radius_results.. rYr-rZrzLargest returned distance z not within requested radius z on row rN)rGr\r#r^rCrM)r_r`rarbradius check_sortedr/r0rcrdr7r8r9r:r; max_dist_a max_dist_brHs r+ assert_compatible_radius_resultsr|s021I   ! ! $ $ $ $ " # # $ $ $ $ " # # $ $ $ $ $ $ %&&I9%%- - &y1 &y1 +I6 +I6  X9Z(( W W*WI*W*W W W(9Z(( W W*WI*W*W W W(:#m"4"44444:#m"4"44444 z??Q   ++J'''7Z77!77+477('' z??Q   ++J'''7Z77!77+477('' 3          X'$. #           M- - r-Hz>r0r/g& .>c  d}d}t||}|dz }d|z }d|z}d|z }d|z}tjdd|d|g||d ||gg}tjgd gd g} t||| | |ttjdd|d|ggtjdd|d|ggtjgd gtjgd gfi|ttjddd d|ggtjddd |dggtjgd gtjgdgfi|ttj|d |||ggtjd d d d |ggtjgdgtjgdgfi|ttj|d |||ggtj|d d d |ggtjgdgtjgdgfi|t jd} t jt| 5ttjdd|d|ggtjdd|d|ggtjgd gtjgdgfi|dddn #1swxYwYt jd} t jt| 5ttjdd|d|ggtjdd|d|ggtjgd gtjgdgfi|dddn #1swxYwYt jd} t jt| 5ttj|d|d|ggtjdd|ddggtjgd gtjgdgfi|dddn #1swxYwYt jd} t jt| 5ttj|d|ddggtjdd|d|ggtjgdgtjgdgfi|dddn #1swxYwYd} t jt| 5ttjdd|d|ggtjdd|d|ggtjgd gtjgdgfi|ddddS#1swxYwYdS) Nr}r~r ?ffffff@333333?@rrrr  rg)rrrrr g@)rrr rr)rrrrgrrrrrrg)"r )*r r pQuery vector with index 0 lead to different distances for common neighbor with index 1: dist_a=1.2 vs dist_b=2.5matchrrr rrzFneighbors in b missing from a: [12] neighbors in a missing from b: [1])rrr r zDneighbors in b missing from a: [] neighbors in a missing from b: [3]r)rrrrrzDneighbors in b missing from a: [5] neighbors in a missing from b: [])rrr rr)rrrr r Distances aren't sorted on row 0rrrrr ) r&r#arrayrfreescapepytestraisesr6) r0r/tolseps_1m_1p_6_1m_6_1pref_dist ref_indicesmsgs r+&test_assert_compatible_argkmin_resultsr_s D D T % % %D (C )C )C #IE #IEx #uc5 ) #q#s # H ( OO    K&(Kd & 3UC/011 3UC/011 ///"## ///"##   & 3S#u-.// 3S%-.// ///"## ///"##   & 33S)*++ 1aAs#$%% """#$$ """#$$   & 33S)*++ 31a%&'' %%%&'' $$$%&&    ) -  C ~S 1 1 1  ) HsCU34 5 5 HsCU34 5 5 Hooo& ' ' Hooo& ' '                       )Q  C ~S 1 1 1  ) HsCU34 5 5 HsCU34 5 5 Hooo& ' ' H'''( ) )                       )O  C ~S 1 1 1  ) HsCU34 5 5 HsCQ/0 1 1 Hooo& ' ' H&&&' ( (                       )O  C ~S 1 1 1  ) HsCQ/0 1 1 HsCU34 5 5 H&&&' ( ( Hooo& ' '                       -C ~S 1 1 1  ) HsCU34 5 5 HsCU34 5 5 Hooo& ' ' Hooo& ' '                         s_A+K  KKA+M;;M?M?5A+P,,P03P0&A+SS!$S!A+U==VVryFc `d}d}t||}|dz }d|z }d|z}d|z }d|z}tjdd|d|gtj|d ||gg} tjgd tjgd g} t| | | | fd |d |ttjtjdd|d|ggtjtjdd|d|ggtjtjgd gtjtjgdgfd |d |ttjtj||d ||ggtjtj||d ||ggtjtjgdgtjtjgdgfd |d |t jd} t jt| 5ttjtjdd|d|ggtjtjdd|d|ggtjtjgd gtjtjgdgfd |d |dddn #1swxYwYttjtjdd|d||ggtjtjdd|dggtjtjgdgtjtjgdgf||d |t jd} t jt| 5ttjtjgdgtjtjddggtjtjgdgtjtjd dggfd|d |dddn #1swxYwYt jd} t jt| 5ttjtjgdgtjtjgdgtjtjgdgtjtjgdgfd|d |dddn #1swxYwYt jd} t jt| 5ttjtjdd|d|ggtjtjdd|ddggtjtjgd gtjtjgdgfd|d |dddn #1swxYwYt jt| 5ttjtjdd|ddggtjtjdd|d|ggtjtjgd gtjtjgdgfd|d |dddn #1swxYwY|rd } t jt| 5ttjtjdd|d|ggtjtjdd|d|ggtjtjgd gtjtjgdgf|d!d |ddddS#1swxYwYdSttjtjdd|d|ggtjtjdd|d|ggtjtjgd gtjtjgdgf|d"d |dS)#Nr}rr~r rrrrrr)rrrrg@)rxry)rrrrr rrrrr)rrr rrr)rrr rzQuery vector with index 0 lead to mismatched result indices: neighbors in b missing from a: [] neighbors in a missing from b: [3])rrr)rrr rzQuery vector with index 0 lead to mismatched result indices: neighbors in b missing from a: [4] neighbors in a missing from b: [2])rg@r)rrr)rrr zTLargest returned distance 6.100000033333333 not within requested radius 6.1 on row 0rrTF) r&r#rr|rrrrr6) ryr0r/rrrrrrrrrs r+%test_assert_compatible_radius_resultsrs D D T % % %D (C )C )C #IE #IE #sE3.// #q#s#$$H !! K %  !  % "(CeS%899:;; "(CeS%899:;; "(???++,-- "(???++,--  !  % "(Cac233455 "(Cac233455 "(+++,,-.. "(+++,,-..  !   ) -  C ~S 1 1 1   ( HbhS%e<==> ? ? HbhS%e<==> ? ? Hbh//0 1 1 Hbh//0 1 1  %                      % "(CeS%?@@ABB "(CeS122344 "(---../00 "(<<<(()**  !   ) B  C ~S 1 1 1   ( Hbh}}}--. / / HbhSz**+ , , Hbhyyy))* + + Hbh1v&&' ( (  %                       ) C  C ~S 1 1 1   ( Hbh//0 1 1 Hbh}}}--. / / Hbhyyy))* + + Hbhyyy))* + +  %                       )   C ~S 1 1 1   ( HbhS%e<==> ? ? HbhS%c:;;< = = Hbh//0 1 1 Hbh//0 1 1  %                       ~S 1 1 1   ( HbhS%c:;;< = = HbhS%e<==> ? ? Hbh//0 1 1 Hbh//0 1 1  %                       0 ]> 5 5 5   ,"(CeS%#@AABCC"(CeS%#@AABCC"(???33455"(???33455   !                         ) HbhS%e<==> ? ? HbhS%e<==> ? ? Hbh//0 1 1 Hbh//0 1 1          sr8B6K::K>K>)B0R%%R),R)B0VV"VB6ZZZ=B6]??^^)B6a,,a03a0 csr_containerc~tjd}|dd}|dd}||}||}d}t j|||sJt j|||sJt j|||sJt j|||sJt j|tj|tj|sJt j|tj|tj|sJt j|tj |tj |rJt j||drJt j|tj||rJt j||tj |rJt jtj |||rJt j||dsJt j||dsJt j||drJt j||drJ||dz}t j|||rJ||}|j tj |_ t j|||rJdS) Nrdrg manhattanpyfunc)rr sqeuclidean) r#r$r%r(r is_usable_forastypefloat64float32int64int32asfortranarrayindices) rr)rhriX_csrY_csrr X_csr_0_nnz X_csr_int64s r+/test_pairwise_distances_reduction_is_usable_forrsw )   " "C bA bA M!  E M!  E F , 9!Q G GGG G + 9% O OOO O + 9%F K KKK K + 9!UF K KKK K + 9 ahhrz22F    , 9 ahhrz22F   0= AHHRX.. 0=a8TTTTT T/= a 0= 188BH  v  0= !a  , 9%; W W WWW W + 9 5    0= u] 0= u[   -A&&K/=k1fUUUU U  -""K%-44RX>>K/=k1fUUUU UUUr-ctjd}|dd}|dd}d}d}d}t jt |5tj| tj |||dddn #1swxYwYd }t jt |5tj|| tj ||dddn #1swxYwYt jt d 5tj||d |dddn #1swxYwYt jt d 5tj||d |dddn #1swxYwYt jt d5tj|||ddddn #1swxYwYt jt d5tjtj ddg|||dddn #1swxYwYt jt d5tjtj ||||dddn #1swxYwYddi}d}t jt|5tj|||||dddn #1swxYwYdt!|dd}d}t jt|5tj|||||dddn #1swxYwYdt!|di}t#j5t#jdttj|||||dddn #1swxYwYt!|dt!|dd}t#j5t#jdttj|||||ddddS#1swxYwYdS) NrrrgrrkOnly float64 or float32 datasets pairs are supported at this time, got: X.dtype=float32 and Y.dtype=float64r)rhrikriOnly float64 or float32 datasets pairs are supported at this time, got: X.dtype=float64 and Y.dtype=int32k == -1, must be >= 1.rSk == 0, must be >= 1.rUnrecognized metric wrong metric;Buffer has wrong number of dimensions \(expected 2, got 1\)r@ndarray is not C-contiguousrr 4Some metric_kwargs have been passed \({'p': 3}\) but)rhrirrrsr num_threadsrY_norm_squared?Some metric_kwargs have been passed \({'p': 3, 'Y_norm_squared'X_norm_squarederrorcategoryrr)r#r$r%r(rr ValueErrorrcomputerrrrrwarns UserWarningr warningscatch_warnings simplefilter) r)rhrirrrunused_metric_kwargsmessagerss r+(test_argkmin_factory_method_wrong_usagesrs )   " "C bA bA A F 3 z - - -II!((2:..!qHHHHIIIIIIIIIIIIIII 1 z - - -GG!qxx11QvFFFFGGGGGGGGGGGGGGG z)A B B B77!qBv6666777777777777777 z)@ A A A66!qAf5555666666666666666 z)> ? ? ?>>!qAn====>>>>>>>>>>>>>>> X   II "(C:..!qHHHHIIIIIIIIIIIIIII z)F G G GII"+A..!qHHHHIIIIIIIIIIIIIII 8EG k 1 1 1  1&8L                   /qAAAM QG k 1 1 1SS!qAfMRRRRSSSSSSSSSSSSSSS /qAAAM  " "SSg <<<<!qAfMRRRRSSSSSSSSSSSSSSS 0qAAA/qAAAM  " "SSg <<<<!qAfMRRRRSSSSSSSSSSSSSSSSSSs-6B//B36B36DD D?E$$E(+E( F//F36F3G::G>G> -II I?+J66J:=J:"LL L M**M.1M.5OO #O 5QQ#&Q#c tjd}|dd}|dd}d}d}d}|ddd}tj|}d }t jt| 5tj | tj |||||| dddn #1swxYwYd }t jt| 5tj || tj ||||| dddn #1swxYwYt jtd 5tj ||d|||| dddn #1swxYwYt jtd 5tj ||d|||| dddn #1swxYwYt jtd 5tj |||d||| dddn #1swxYwYt jtd 5tj tjddg|||||| dddn #1swxYwYt jtd 5tj tj||||||| dddn #1swxYwYd} d| d} t jt| 5tj ||||| || ddddS#1swxYwYdS)Nrrrgrruniformrlowhighsizerr)rhrirrweightsY_labelsunique_Y_labelsrrrSrrrrrrrnon_existent_weights_strategy[Only the 'uniform' or 'distance' weights options are supported at this time. Got: weights=''.)r#r$r%r(randintuniquerrrrrrrrrr) r)rhrirrrrrrrrs r+2test_argkmin_classmode_factory_method_wrong_usagesrs )   " "C bA bA A FG{{qr{44Hi))O 3 z - - -    hhrz""+                     1 z - - -    hhrx  +                     z)A B B B    +                     z)@ A A A    +                     z)> ? ? ?    !+                     X        hSz""+                     z)F G G G    ""+                    %D! ;6 ; ; ;  z 1 1 1    1+                       s9C  C$'C$9E  EE3FF"FG))G-0G-H77H;>H;0JJ J?.K99K=K='MMMcBtjd}|dd}|dd}d}d}d}t jt |5tj| tj |||dddn #1swxYwYd }t jt |5tj|| tj ||dddn #1swxYwYt jt d 5tj||d |dddn #1swxYwYt jt d 5tj|||d dddn #1swxYwYt jt d5tjtj ddg|||dddn #1swxYwYt jt d5tjtj ||||dddn #1swxYwYddi}d}t jt|5tj|||||dddn #1swxYwYdt!|dd}d}t jt|5tj|||||dddn #1swxYwYt!|dt!|dd}t#j5t#jdttj|||||dddn #1swxYwYdt!|di}t#j5t#jdttj|||||ddddS#1swxYwYdS)Nrrrgrrrr)rhrirxrrradius == -1.0, must be >= 0.rSrrrrrrrr r)rhrirxrrsrrrrrrrr)r#r$r%r(rrrr rrrrrrrrr rrr) r)rhrirxrrrrrss r+1test_radius_neighbors_factory_method_wrong_usagesrs )   " "C bA bA F F 3       hhrz""av                    1    YY !qxx'9'9&QWXXXX YYYYYYYYYYYYYYY z)H I I IDD!qFCCCCDDDDDDDDDDDDDDD z)> ? ? ?PP!qOOOOPPPPPPPPPPPPPPP X      hSz""av                    z)F G G G  ""av                    8FG k 1 1 1  1VFBV                   /qAAAM QG k 1 1 1  1VF-                   0qAAA/qAAAM  " "  g <<<<1VF-                    /qAAAM  " "  g <<<<1VF-                      s-6B//B36B36DD D?E$$E(+E( F//F36F3-HHH4+I++I/2I/J==KK9LL#&L# 5N!!N%(N%5PPPc ,tjd}|dd}|dd}d}d}d}|ddd}tj|}d }t jt| 5tj | tj ||||||d dddn #1swxYwYd }t jt| 5tj || tj |||||d dddn #1swxYwYt jtd 5tj ||d||||d dddn #1swxYwYt jtd 5tj ||dd|||d dddn #1swxYwYt jtd 5tj tjddg||||||d dddn #1swxYwYt jtd 5tj tj|||||||d dddn #1swxYwYd} d| d}t jt| 5tj |||d| ||d ddddS#1swxYwYdS)Nrrrgrrrrrrr)rhrirxrrrr outlier_labelrrrSr wrong_metricrrrrrrr)r#r$r%r(rrrrrr rrrrrr) r)rhrirxrrrrrrs r+;test_radius_neighbors_classmode_factory_method_wrong_usagesrsN )   " "C bA bA F FG{{qr{44Hi))O 3 z - - -    (hhrz""+                  1 z - - -    (hhrx  +                  z)H I I I    (+                  z)> ? ? ?    (!+                  X       !(hSz""+                  z)F G G G    (""+                 %D! ;6 ; ; ; z - - -    (!1+                    s~:C!!C%(C% :EEE5FF"%F"G--G14G11III6/J11J58J5L  L L  Dispatcherdtyperctj|}d}|gddd\}}|||||z}|||||z} |t urd} i} i} nt|| d} | } d | i} d d i} |j|| | fd d d d| \}}|j|| | fdd d d| \}}t||f||||fi| dS)z3Check that results do not depend on the chunk size.rarerFrreplacergrrhrirrx sort_resultsTr) chunk_sizerreturn_distance)N) r#r$r%choicer(rrrur ASSERT_RESULT)global_random_seedrrrr)spread n_samples_X n_samples_Yrhri parametercheck_parameterscompute_parametersrxrrdistrs r+test_chunk_size_agnosticismr_s )   2 3 3C F"zz*=*=*=AuzUUK j))0077&@A j))0077&@AW $qAkBBB $f-,d3.J.    Hk'J&    MD':u%&$ W0@r-ctj|}|gddd\}}d}|||||z}|||||z} |t urd} i} i} nt|| d} | } d | i} d d i} |j|| | fd d d | \}}t dd5|j|| | fd d d | \}}dddn #1swxYwYt||f||||fi| dS)z:Check that results do not depend on the number of threads.rrFrrrgrrrxrTrrropenmp)limitsuser_apiN) r#r$r%rr(rrrurrlimitr)rrrrr)r r rrhrir r r rxrrrrs r+test_n_threads_agnosticismrs  )   2 3 3C"zz*=*=*=AuzUUK F j))0077&@A j))0077&@AW $qAkBBB $f-,d3.J.     Hk $ % % + +1x + H H  * *     !   g               :u%&$ W0@s<DD#&D#zDispatcher, dtypec^tj|}d}d\}}|||||z}|||||z} ||} || } |t urd} i} i}nt || d}|} d|i} ddi}|j|| | fd dd |\}}tj || f| | fD]<\}}||ur|| ur|j||| fd dd |\}}t||f||||fi| =d S) zLCheck that results do not depend on the format (dense, sparse) of the input.r)rrrgrrrxrT2rN) r#r$r%r(rrrur itertoolsproductr)rrrrr)r n_samplesrrhrirrr r r rx dist_dense indices_dense_X_Yrrs r+test_format_agnosticismr!s" )   2 3 3C F$Iz J''..u55>A J''..u55>A M!  E M!  EW %qAkBBB $f-,d3 2 2  ! !!  !!J #QJE ;;  B 77rQww * *       !   g z5)*              r-c ntj|}|tjgdt }|gddd\}}d}|||||z} |||||z} |dkr@tj| d d d df} tj| d d d df} |turd } i} i} nt| | | }|} d |i} d di} |j | | | f|t|||d|dzddd| \}}|j | | | f|t|||d|dzddd| \}}t||f||||fi| d S)z:Check that the results do not depend on the strategy used.)rrr haversinerrrFrrr#NrgrrxrTrrr parallel_on_X)rrsrstrategyr parallel_on_Y)r#r$r%rrobjectr(rascontiguousarrayrrurr,r)r global_dtyperrr)rr r rrhrir r r rx dist_par_X indices_par_X dist_par_Y indices_par_Ys r+test_strategies_consistencyr0s )   2 3 3C ZZ          F #zz*=*=*=AuzUUK F j))00>>GA j))00>>GA  111bqb5 * *  111bqb5 * *W $qAf=== $f-,d3 2 2  !- J%7    !# !! !!J !3 2  !- J%7    !# !! !!J :|,-J }@Pr-r')r&r(rc @tj|}|ddg} |ddg} d} | ||| || zz} | ||| || zz} || }|| }|dkr@tj| ddddf} tj| ddddf} t|| d}|dkrt| | }nt| | fd |i|}tj |d ddd|f}tj |j tj }t|j dD]}||||f||<tj| |f| |fD]F\}}t#j|||||d |dz|\}}t't"|f||||GdS)Nrrr.Ar#rrrrrnr$Tr)rrsrrr')r#r$r%rr(rr*r,rrargsortzerosr[rr\rrrrr)rrr'rrrdrrr)r translationrrhrirrrs dist_matrixargkmin_indices_refargkmin_distances_refrow_idxrr argkmin_distancesargkmin_indicess r+test_pairwise_distances_argkminr=Vsu )   2 3 3CRI&&J**aX&&K Fchhy*55<bjQQQ,21566  )4 (1 1* g&&#QJE ;;  B-4_   '  A~ . . . *? w&'  !         r-c tj|}|ddg}|ddg}d} |||||| zz} |||||| zz} t |||d} |dkrt| | } nt| | fd|i| } t| }g}g}| D]}tj |j d||k}||}tj |}||||}}| || |tj| | ||| d |d z|d \}}t!t|f|||||dS) Nrrrr2r3r%rr)rjTr)rrsrrr'r)r#r$r%rr(rr,rrruaranger[r4appendr rr)rrr'rrdrr)rr6rrhrirsr7rxneigh_indices_refneigh_distances_refrowindrrqneigh_distances neigh_indicess r+(test_pairwise_distances_radius_neighborsrGs  )   2 3 3CRI&&J**aX&&K Fchhy*55< & & &"O]?E*+,m=NPVr-rrctjd}d}d\}}|||||z}|||||z}t ||g\} } |t urd} i} i} n!dtj|z}|} d|i} ddi} |j||| f|dd| \}}|j| | | f|dd| \}}t||f||||fi| d S) zACheck that the results do not depend on the datasets writability.rr)rgrgrxrT)rrN) r#r$r%r(rrrlogrr)rrrr)rrrrhriX_mmY_mmr r r rxrrdist_mm indices_mms r+test_memmap_backed_datarOs )   " "C F#Iz J''..u55>A J''..u55>A+Aq622JD$W rvj))) $f-,d3.J.     Hk-*,     GZ:u%&'; 6Fr-ctj|}d}|gd}|gd}|gd}|||||z}||} tj|ddz} t||} t| |} t| | t| | tj t5tj |}t||ddddS#1swxYwYdS) Nr)rrrr3)rrgr)rrrrrnrr)r#r$r%rr(rlinalgnormr r rrrr) rrrr)rrrrrhrsq_row_norm_reference sq_row_normsq_row_norm_csrs r+test_sqeuclidean_row_normsrV s )   2 3 3C F ///00ILLL))J**YYY''K J''..u55>A M!  EINN11N55:'{CCCK+E{KKKO);777)?;;; z " "::  a a[9999::::::::::::::::::s&EEEc tjd}|dd}|dd}d}d}d}|ddd}tj|}t j|||||||d }t j|||||||d } t|| dS) Nrrrgrrrrrr&)rhrirrrrrr'r() r#r$r%r(rrrrr) r)rhrirrrrr results_X results_Ys r+*test_argkmin_classmode_strategy_consistentrZ)s )   " "C bA bA A FG{{qr{44Hi))O ( '    I!( '    Iy),,,,,r-r)Nrr rrc tjd}|dd}|dd}d}d}d}|ddd}tj|}t j||||||||d } t j||||||||d } t| | dS) Nrrrgrrrrrr&) rhrirxrrrrrr'r() r#r$r%r(rrr rr ) rr)rhrirxrrrrrXrYs r+3test_radius_neighbors_classmode_strategy_consistentr\Js )   " "C bA bA F FG{{qr{44Hi))O(0 '#    I)0 '#    IIy)))))r-)r)rNrO)TrNrO)r)rg)rrrg)rr)=rrr functoolsrnumpyr#rscipy.spatial.distancersklearn.metricsrr-sklearn.metrics._pairwise_distances_reductionrrr r r r sklearn.utils._testingr rrsklearn.utils.fixesrsklearn.utils.parallelr1CDIST_PAIRWISE_DISTANCES_REDUCTION_COMMON_METRICSstrintr,rCrMrfrur| FLOAT32_TOLS FLOAT64_TOLSrrrrmark parametrizerrrrrrrrr!r0r=rGrOrVrZr\r-r+rmsH  ((((((CCCCCCCC /.....======5551CS2###L! ! ! R  G G G G X  99999F Q Q Q Q j      bj77#DUU UU bj77#DUU UU w/@@<@@ w/@@<@@  H H H V$77_ _ 87_ D.99AVAV:9AVHJSJSJSZp p p lY Y Y xk k k \'ABB2:rz":;;  ---<;CB-`'ABB2:rz":;;  ,,,<;CB,^ "* "*% "* "*% .997 7 :97 t'ABB  IIICBI^#TUU%GHH2:rz":;;.99< < < :9<;IHVU< ~#TUU%GHH2:rz":;;  999<;IHVU9x'ABBK#=>>2:rz":;;..<;?>CB.b2:rz":;;.99:::9<;:6---B*<*<*<== * *>= * * *r-