8^j\$ddlZddlZddlZddlmZddlmZmZm Z ddl m Z ejjZedddfZej \ZZdZdZdZdZd Zd Zd Zd Zej6j9d dgdZy)N)datasets) MinCovDetempirical_covariancefast_mcd)assert_array_almost_equalc tdddddd|tdddddd |tddd ddd |td dd ddd|tdddddd|tdddddd|tdddddd|y)Ndrg{Gz?皙?J333333?A(2iiiii ifii^)launch_mcd_on_dataset)global_random_seeds z/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/sklearn/covariance/tests/test_robust_covariance.pytest_mcdrs#q!T34FG#q"c34FG#q"c34FG$3S#7IJ$3S#7IJ#q#tT38JK"b!S#r3EFctjd}d}tjt|5t |dddy#1swYyxYwNr z'Expected 2D array, got 1D array insteadmatch)nparangepytestraises ValueErrorr)Xmsgs rtest_fast_mcd_on_invalid_inputr$)s> #A 3C z - s AActjd}t}d}tjt |5|j |dddy#1swYyxYwr)rrrrr r!fit)r"mcdr#s rtest_mcd_class_on_invalid_inputr(0sI #A +C 3C z -  s AA!ctjj|}|j||}|j |d|} d|j d||fdz z} || xx| z cc<tj |jt} d| | <|| } t|j|} | j}| j}| j}tj| jd|z dz}||ksJtjt| |z dz}||ksJtj ||k\sJt#| j%|| j&y)Ng$@size?F random_stater)rrandom RandomStaterandn permutationrandintonesastypeboolrr& location_ covariance_support_meanrsumr mahalanobisdist_) n_samples n_features n_outlierstol_loctol_cov tol_supportseedrand_gendataoutliers_indexoutliers_offset inliers_mask pure_datamcd_fitTSHerror_location error_covs rrr8sayy$$T*H >>)Z 0D)))4[jANh..q J7O.PSVVWOO+779%,,T2L#(L \"IT*..t4GAAAWWinnQ/!39:N G ## #-i81<BCI w   66!9 ## #g11$7Grctjjd}|jd}t }|j |y)Nr)rr+)rr0r1normalrr&)rndr"r's rtest_mcd_issue1127rVSs: ))   "C  A +CGGAJrctjj|}tjdddj }tj t tj||}tj|tj|jddff}t|j|y)Nr rrr.)rr0r1linspacetolistarraylist itertoolsproducthstackzerosshaperr&)rrF data_valuesrGs rtest_mcd_issue3367rd\syy$$%78H++b!R(//1K 88D**; DE FD 99dBHHdjjmQ%789 :D"8$((.rcPtjgd}|jdd}tjgd}|jdd}d}||fD]?}tjt |5t j|dddAy#1swYLxYw)N) r-r r r 9v?r r r Cl?r r) r-rrrrfrrrrgrzYThe covariance matrix of the support data is equal to 0, try to increase support_fractionr)rr\reshaperr r!rr&)X_1X_2r#r"s r#test_mcd_support_covariance_is_zerorl|s ((J KC ++b! C ((J KC ++b! C $3Z ]]:S 1  KOOA     s 7BB% cgdgdgdgdgdgdgdgdgd gd gd gd gd gdgdgdgdgdgdg}td|}d}tjt|5|j |dddy#1swYyxYw)N)ffffff@ @ffffff?皙?)皙@@rprq)g@g @g?rq)ffffff@@?rq)@ @rprq)rt333333 @rpr)rwryrvrq)g@g333333@rprq)rrrurvr )皙@g @rvrq)333333@ry皙?rq)r{rsrpr )g333333@rsg?r )rnrorpr)g@gffffff@333333?r)rzryr}rq)rtrx?rq)rwrsr|rq)g@rorvrqr-)support_fractionr/zDeterminant has increasedr)rrwarnsRuntimeWarningr&)rr"r'warn_msgs rtest_mcd_increasing_det_warningrs ' A, S7I JC*H nH 5  s &BB zn_samples,n_features)irYcd}tjj|}|j||f}t |j dj }tddd|j|j j}tj|}tj|tj|z } | |kDsJdy ) zCheck that MinCovDet does not underestimate the empirical variance on Gaussian data. A large sample size and n_features makes the test robust. Non-regression test for: https://github.com/scikit-learn/scikit-learn/issues/23162 gQ?r+T)assume_centeredr~F)rstore_precisionrr/z.MinCovDet underestimates the Gaussian varianceN) rr0 default_rngrTrrMdiagonalrr&r9diagr<) r?r@r thresholdrngxvar_empcov_mcdvar_mcdmean_var_ratios rtest_mincovdet_bias_on_normalrsI ))   2 3C Y/ 0A"133=FFHG ! +  QSS  gggGVVG_rvvg6N I %W'WW %r)r^numpyrrsklearnrsklearn.covariancerrrsklearn.utils._testingr load_irisrGr"X_1drbr?r@rr$r(rrVrdrlrmark parametrizerrrrs HH<HAw :G.H6/@  D/*>X?Xr