Q(hdZddlZddlZddlmZddlmZddlm Z m Z m Z ddl m Z ddgddgddgd d gd d gd d ggZgd Zddgd d gd d ggZgd Zgd ZgdZd dgdd gdd ggZej*Zej.j1d Zej5ej6j8Zej<ee_ej6ee_ej>jAde dZ!dZ"dZ#dZ$dZ%dZ&dZ'ej>jAde dZ(dZ)dZ*dZ+ej>jAdgdd Z,ej>jAd!ejZge zd"Z.y)#z* Testing for the nearest centroid module. N)datasets)NearestCentroid)assert_allcloseassert_array_almost_equalassert_array_equal)CSR_CONTAINERS)r r r r r r )r r r )i@Pg j =gHKbi):g68 csr_containerc`|t}|t}t}|jttt |j ttt|jttt|jtttd}|jttt |j ttt|jttt|jtttd}|jttt |j ttt|jttt|jtttddg}|jttt |j ttt}|j|tt |j |tt}|j|tt |j ttt}|jttt |j |tt}|j|jtt |j |jty)Nuniformpriors empiricalg?g?)XTrfityrpredict true_resultrdecision_functiontrue_discriminant_scores predict_proba true_probatrue_result_prior1tocootolil)rX_csrT_csrclfs k/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/sklearn/neighbors/tests/test_nearest_centroid.pytest_classification_toyr'%s ! E ! E  CGGAqMs{{1~{3c33A68PQc//2J?  +CGGAqMs{{1~{3c33A68PQc//2J?  -CGGAqMs{{1~{3c33A68PQc//2J? $ .CGGAqMs{{1~'9:  CGGE1s{{5);7  CGGE1s{{1~{3  CGGAqMs{{5);7  CGGEKKM1s{{5;;=1;?c8dD]}t|jtjtj}t j |jtjtjk(}|dkDrJdt|zy)N euclidean manhattanmetricg?Failed with score = rririsdatatargetnpmeanrstr)r.r%scores r& test_irisr8Ysq,@V,00DKKH DII.$++=>s{?2SZ??{@r(cLdD]}dD]}t||}|jtjtj}t j |jtjtjk(}|dkDrJdt|zy)Nr*)N皙?g?)r.shrink_thresholdg?r/r0)r.r;r%r7s r&test_iris_shrinkager<as,D 0 D !BRSC''$))T[[1CGGCKK 2dkkABE3; C 6U C C;  DDr(cddl}t}|jtjtj |j tjtj }|j|}|j|}t||jk(sJ|j tjtj }t||dy)Nrz>Failed to generate same score after pickling (classification).) picklerrr1r2r3r7dumpsloadstype __class__r)r>objr7sobj2score2s r& test_picklerGks  CGGDIIt{{# IIdii -E SA <<?D : && & ZZ 4;; /F Hr(c@tjddgddgddgddgddgg}tjgd}td}|j||tjd d gd d gg}tjj |j |y) Nrr r )r r r r r r:r;g;4,F]?gCMX?gJGp@gU@)r4arrayrrtestingr centroids_)rrr%expected_results r&test_shrinkage_correctrP~s 1a&1a&1a&1a&1a&9:A !A 3 /CGGAqMhhI 688LMNOJJ((Ir(ctd}tjt}d||dk(<|j t ||j }|j t tt||j y)Ng{Gz?rKrr )rr4asarrayrrrrNr)r%y_indcentroid_encodeds r&"test_shrinkage_threshold_decoded_yrUs[ 4 0C JJqMEE%2+GGAu~~GGAqM'8r(ctjjd}|jdd}|j ddd}|jd}t d}|j |||j|}t d}||z}|j |||j|}t||y)Nr2r r:rK) r4random RandomStaterandrandintrrrr)rngrrnoiser%y_initX_noise y_translates r&test_predict_translated_dataras ))   "C RA Aq"A HHRLE 3 /CGGAqM [[^F 3 /C%iGGGGQ++g&Kv{+r(c|t}td}|jtt|j}|j|tt |j|t |ddgddggy)Nr,r-r r )rrrrrNr)rr#r%dense_centroids r&test_manhattan_metricrdsa ! E  -CGGAqM^^NGGE1s~~~6~R1a&'9:r(ctjd}d|dddf<d|dddf<tjd}d|d<td}t j t 5|j||dddy#1swYyxYw) N) r g$b@rg_LOчr rfr:rK)r4emptyzerosrpytestraises ValueErrorr)rrr%s r&test_features_zero_varrlsz AAadGAadG "A AaD 3 /C z " 1 s 'BB ctddg}tjtd5|j t t dddy#1swYyxYw)zGCheck that we raise an error when the user-defined priors are negative.r rzpriors must be non-negativematchN)rrirjrkrrr)r%s r&test_negative_priors_errorrqsA "a )C z)F G 1 s AAcRddg}t|}tjtd5|j t t dddt|jtj|tj|jz y#1swYWxYw)zfCheck that we raise a warning and normalize the user-defined priors when they don't sum to 1. r rnrzAThe priors do not sum to 1. Normalizing such that it sums to one.roN) rriwarns UserWarningrrrr class_prior_r4rRsum)rr%s r&test_warn_non_normalized_priorsrwsVF  (C Q  1  C$$bjj&82::f;M;Q;Q;S&ST s BB&response_method)rrpredict_log_probactdjtt}t j t 5t||tdddy#1swYyxYw)z{Check that we raise an AttributeError with Manhattan metric when trying to call a non-thresholded response method. r,r-N) rrrrrirjAttributeErrorgetattrr)rxr%s r&(test_method_not_available_with_manhattanr}sM  - 1 1!Q 7C ~ &)%_%a()))s AA'array_constructorctjttdf}|dddfxxdzcc<||}t }t j td5|j|tdddy#1swYyxYw)zHCheck that we raise an error when the variance for all features is zero.r Nr zAll features have zero variancero) r4oneslenrrrirjrkr)r~rr%s r&test_error_zero_variancesrsn Q AadGqLG!A  C z)J K 1 s #BB )/__doc__numpyr4risklearnrsklearn.neighborsrsklearn.utils._testingrrrsklearn.utils.fixesrrrrrr rr load_irisr1rXrYr\ permutationr3sizepermr2mark parametrizer'r8r<rGrPrUrardrlrqrwr}rLrr(r&rs - / "XBx"bAq6Aq6Aq6:"X1v1v (+a 0;2B C xiiA t{{''( IIdO kk$ .90@:0@f@D& J9,".9 ;: ;  UR)),rxxj>.IJKr(