Ugl,ddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZddlm Z ddl!m"Z"m#Z#m$Z$m%Z%m&Z&ddl'm(Z(d Z)ej*+d e#d Z,d Z-d Z.dZ/Gddej0Z1dZ2dZ3dZ4dZ5dZ6dZ7GddZ8GddZ9GddZ:dZ;GddZej*?e"d#$d%Z@ej*+d&d'eAfd(eBfd)ejCfd*ejDfgeEd+d,ge$eEd-d.ge#d/d0fd1d2fd3d4fd5eFfej*+d6ejGejHfejIejHfejJejKfejLejKfgd7ZMd8ZNd9ZOd:ZPej*Qe&e%d;kd<$ej*+d=d,d.gej*+d>ejGejIejJejLgd?ZRd@ZSdAZTej*?e"dB$dCZUej*+d=d+d-ejVd,ej*Qe&e%d;kd<$DejVd.ej*Qe&e%d;kd<$DgdEZWdFZXej*+dGedHZYej*+dGedIZZdS)KN)sparse)LinearDiscriminantAnalysis)DecisionTreeClassifier) TempMemmap_convert_container_delete_folder_get_warnings_filters_info_listassert_allcloseassert_allclose_dense_sparseassert_no_warningsassert_raise_message assert_raisesassert_raises_regex'assert_run_python_script_without_outputcheck_docstring_parameterscreate_memmap_backed_dataignore_warningsraisesset_random_stateturn_warnings_into_errors) deprecated)_IS_WASMCSC_CONTAINERSCSR_CONTAINERS parse_version sp_version) available_ifct}t}t|dt|d|jdksJdS)N)rrr random_state)ldatrees _/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/sklearn/utils/tests/test_testing.pytest_set_random_stater$*sR $ & &C ! # #DS!T1   ! ! ! ! ! ! csr_containerctjddd}d}||}||fD]X}tjt |5t ||dzdddn #1swxYwYt ||Ytjtd5t ||dddn #1swxYwYtj tj dd  }|tj d }tjt d 5t ||ddddS#1swxYwYdS) N rzNot equal to tolerance matchzCan only compare two sparser)offsets)r,zArrays are not equal) nparangereshapepytestrAssertionErrorr ValueErrorrdiagsonestocsr)r&xmsgyXABs r#!test_assert_allclose_dense_sparser>3s ! Q""A #C aAV++ ]> 5 5 5 3 3 (AE 2 2 2 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3$Q**** z)F G G G++$Q***+++++++++++++++  RWQZZ+++1133A bgfoo&&A ~-C D D D++$Q***++++++++++++++++++s6A77A; >A; .C  CCE&&E*-E*cttd5ttd5 dddn #1swxYwYddddS#1swxYwYdS)Nz Hello world)r9)rr3rr4r%r#test_assert_raises_msgrAGs ^] ; ; := 9 9 9                  s.A; A? A? AAAcbd}d}ttd|dttttd|dttttd|dttttd|ttttt fd|dS)Nc t|N)r4)messages r#_raise_ValueErrorz4test_assert_raise_message.._raise_ValueErrorNs!!!r%cdSrDr@r@r%r# _no_raisez,test_assert_raise_message.._no_raiseQ r%testzsomething else)r r4rr3 TypeErrorAttributeError)rFrHs r#test_assert_raise_messagerMMs"""   V->GGG  ."6 FIVVV ^$ r%cF ddttttttjt5tt dddn #1swxYwYtj5}tt dddn #1swxYwYt|dksJt|dj tsJt|dj t sJtj5}tt dddn #1swxYwYt|dksJt|dj tsJtttt ftfd}ttt ffd}ttfd }tt fd }ttfd }tt fd }t|t|t|tjt5|dddn #1swxYwYtjt 5|dddn #1swxYwYtjt5|dddn #1swxYwYfd }fd}fd} fd} fd} fd} t|t|t| tjt5| dddn #1swxYwYtjt 5| dddn #1swxYwYtjt5| dddn #1swxYwYt } d}tj t|5t| }|dddn #1swxYwYtj t|5t| d}ddddS#1swxYwYdS)Nc:tjdtdSNzdeprecation warningwarningswarnDeprecationWarningr@r%r#_warning_functionz.test_ignore_warning.._warning_functionws +-?@@@@@r%cbtjdttjddSrPrQr@r%r#_multiple_warning_functionz7test_ignore_warning.._multiple_warning_functionzs- +-?@@@ +,,,,,r%categoryr+rr.c0dSrDr@)rWrUsr#decorator_no_warningz1test_ignore_warning..decorator_no_warnings'""$$$$$r%cdSrDr@rWsr#decorator_no_warning_multiplez:test_ignore_warning..decorator_no_warning_multiple""$$$$$r%cdSrDr@rUsr# decorator_no_deprecation_warningz=test_ignore_warning..decorator_no_deprecation_warningr%cdSrDr@rasr#decorator_no_user_warningz6test_ignore_warning..decorator_no_user_warningrcr%cdSrDr@r]sr#)decorator_no_deprecation_multiple_warningzFtest_ignore_warning..decorator_no_deprecation_multiple_warningr_r%cdSrDr@r]sr#"decorator_no_user_multiple_warningz?test_ignore_warning..decorator_no_user_multiple_warningr_r%cjt5ddddS#1swxYwYdSrD)rrasr#context_manager_no_warningz7test_ignore_warning..context_manager_no_warnings                           s (,,ctttf5ddddS#1swxYwYdSNrX)rrT UserWarningr]sr##context_manager_no_warning_multiplez@test_ignore_warning..context_manager_no_warning_multiples '9;&G H H H ) ) & & ( ( ( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )s 6::cxtt5ddddS#1swxYwYdSrmrrTrasr#&context_manager_no_deprecation_warningzCtest_ignore_warning..context_manager_no_deprecation_warnings &8 9 9 9                        /33cxtt5ddddS#1swxYwYdSrmrrnrasr#context_manager_no_user_warningz.context_manager_no_user_warnings k 2 2 2                        rscxtt5ddddS#1swxYwYdSrmrqr]sr#/context_manager_no_deprecation_multiple_warningzLtest_ignore_warning..context_manager_no_deprecation_multiple_warnings &8 9 9 9 ) ) & & ( ( ( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )rscxtt5ddddS#1swxYwYdSrmrur]sr#(context_manager_no_user_multiple_warningzEtest_ignore_warning..context_manager_no_user_multiple_warnings k 2 2 2 ) ) & & ( ( ( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )rszA'obj' should be a callable.+you should use 'category=UserWarning'r)cdSrDr@r@r%r#rJz!test_ignore_warning..tests Dr%) r rrTr2warnsrn FutureWarninglen isinstancerErr4)recordr[r^rbrergrirkrorrrvrxrz warning_classr*silence_warnings_funcrJrWrUs @@r#test_ignore_warningrtsAAA--- '899:::'8CUVVVWWW ( ) )CC@)K@@@BBBCCCCCCCCCCCCCCC N6K2]KKKMMMNNNNNNNNNNNNNNN v;;!     fQi'); < <<<< fQi' 5 5555 L6I2[IIIKKKLLLLLLLLLLLLLLL v;;!     fQi'); < <<<<)5G4UVVV %%%%%_%1;?@@@%%%%A@%011121k***+*0111%%%%21%k***%%%%+*%+,,,45557888 ( ) )$$!!###$$$$$$$$$$$$$$$ k " "4411333444444444444444 ( ) )--**,,,---------------     )))))          ))))))))))1222:;;;=>>> ( ) )**'')))*************** k " "::77999::::::::::::::: ( ) )3300222333333333333333 M OE z / / /  > > >?P Q Q                z / / /  ' '   ( ' s!B  BB*CCC E77E;>E;# K::K>K> L55L9<L9 M00M47M4 O66O:=O: P11P58P5 Q,,Q03Q0#S  SS0TTTceZdZdZdS) TestWarnscd}tjt5t|dddn #1swxYwYtdddksJdS)Nc.tjddS)Nyor)rRrSr@r%r#fzTestWarns.test_warn..fs M$   1r%c|SrDr@)r8s r#z%TestWarns.test_warn..sAr%r.)r2rr3r )selfrs r# test_warnzTestWarns.test_warns   ]> * * " " q ! ! ! " " " " " " " " " " " " " " "!++q11Q666666s 9==N)__name__ __module__ __qualname__rr@r%r#rrs#77777r%rc||z}|SzFunction f Parameters ---------- a : int Parameter a b : float Parameter b Returns ------- c : list Parameter c r@abcs r#f_okr AA Hr%c||z}|S)zFunction f Parameters ---------- a : int Parameter a b : float Parameter b Results ------- c : list Parameter c r@rs r#f_bad_sectionsr rr%c||z}|Srr@)rrrs r# f_bad_orderrrr%c||z}|S)zFunction f Parameters ---------- a : int Parameter a b : int Parameter b c : int Parameter c Returns ------- d : list Parameter c r@)rrds r#f_too_many_param_docstringr0s" AA Hr%c||z}|S)zFunction f Parameters ---------- a : int Parameter a Returns ------- c : list Parameter c r@rs r# f_missingrEs AA Hr%c||z|z|zS)zFunction f Parameters ---------- a: int Parameter a b: Parameter b c : This is parsed correctly in numpydoc 1.2 d:int Parameter d e No typespec is allowed without colon r@)rrrres r#f_check_param_definitionrVs q519q=r%ceZdZdZdZdS)KlasscdSrDr@rr;r:s r#rzKlass.f_missingjrIr%cdS)zFunction f Parameter --------- a : int Parameter a b : float Parameter b Results ------- c : list Parameter c Nr@rs r#rzKlass.f_bad_sectionsms  r%N)rrrrrr@r%r#rris2        r%rc,eZdZdZdZdZdZdZdS)MockEstcdS) MockEstimatorNr@rs r#__init__zMockEst.__init__r%c|SrDr@rs r#fitz MockEst.fitr%c|SrDr@rr;s r#predictzMockEst.predictrr%c|SrDr@rs r# predict_probazMockEst.predict_probarr%cdS)N?r@rs r#scorez MockEst.scoressr%N)rrrrrrrrr@r%r#rrs_r%rceZdZdZeddZededdZeddZed d Z d S) MockMetaEstimatorc||_dS)zMetaEstimator to check if doctest on delegated methods work. Parameters --------- delegate : estimator Delegated estimator. Ndelegate)rrs r#rzMockMetaEstimator.__init__s! r%c,t|jdS)Nrhasattrrrs r#rzMockMetaEstimator.swt}i@@r%c6|j|S)zThis is available only if delegate has predict. Parameters ---------- y : ndarray Parameter y )rrrs r#rzMockMetaEstimator.predicts}$$Q'''r%c,t|jdS)Nrrrs r#rzMockMetaEstimator.swt}g>>r%z%Testing a deprecated delegated methodcdS)zThis is available only if delegate has score. Parameters --------- y : ndarray Parameter y Nr@rs r#rzMockMetaEstimator.scorerr%c,t|jdS)Nrrrs r#rzMockMetaEstimator.swt}oFFr%c|S)zThis is available only if delegate has predict_proba. Parameters --------- X : ndarray Parameter X r@rs r#rzMockMetaEstimator.predict_probas r%z-Testing deprecated function with wrong paramscdS)z,Incorrect docstring but should not be testedNr@rs r#rzMockMetaEstimator.fitrr%N) rrrrrrrrrrr@r%r#rrs!!!\@@AA((BA(\>>??Z788  98@? \FFGGHGZ?@@;;A@;;;r%rc tjdddtt}|gksJttdg}|gksJttdg}|gksJtjt d5ttdddn #1swxYwYtjt d 5ttjdddn #1swxYwYtt}tt }|j j }|gd ksJgd gd gdgddd|dzddddddgdd|dzdzdddgdd|dzdddgdd|dzddd d!gg}t|tt ttj|j|j|j|jgD]+\}}t|}||ksJd"|d#|d$,dS)%Nnumpydocz+numpydoc is required to test the docstringsz1.2.0)reason minversionr)ignorezUnknown section Resultsr)zUnknown section Parameterr)zxsklearn.utils.tests.test_testing.f_check_param_definition There was no space between the param name and colon ('a: int')ztsklearn.utils.tests.test_testing.f_check_param_definition There was no space between the param name and colon ('b:')zwsklearn.utils.tests.test_testing.f_check_param_definition There was no space between the param name and colon ('d:int'))z9In function: sklearn.utils.tests.test_testing.f_bad_orderznThere's a parameter name mismatch in function docstring w.r.t. function signature, at index 0 diff: 'b' != 'a' Full diff:z - ['b', 'a']z + ['a', 'b'])zHIn function: sklearn.utils.tests.test_testing.f_too_many_param_docstringz_Parameters in function docstring have more items w.r.t. function signature, first extra item: cr - ['a', 'b']z+ ['a', 'b', 'c']z? +++++)z7In function: sklearn.utils.tests.test_testing.f_missingzaParameters in function docstring have less items w.r.t. function signature, first missing item: brrz+ ['a'])z=In function: sklearn.utils.tests.test_testing.Klass.f_missingaParameters in function docstring have less items w.r.t. function signature, first missing item: Xr - ['X', 'y']+ []z In function: z!sklearn.utils.tests.test_testing.z.predictznThere's a parameter name mismatch in function docstring w.r.t. function signature, at index 0 diff: 'X' != 'y'rz- ['X']z? ^z+ ['y'].rz&potentially wrong underline length... z Parameters z --------- in z.scorez.fitrrrz "z " not in "")r2 importorskiprrrr RuntimeErrorrrrrr __class__rziprrrrrr) incorrect mock_metamock_meta_namemessagesr9rs r#test_check_docstring_parametersrs_ < +400I ????*4>>>I ????*9cUCCCI ???? |+D E E E33">222333333333333333 |+F G G G99"5#7888999999999999999++CDDI!799555I(1N             J.JJJ KC       C.CCC D  5     H.HHH I 4     V.VVV V4    OQHf  &  O    # O M   MMQ/q11 iSSS)))!LMMs$ B..B25B2C::C>C>ceZdZdZdZdS)RegistrationCountercd|_dS)Nr)nb_callsrs r#rzRegistrationCounter.__init__Js  r%cF|xjdz c_|jtusJdS)Nr.)rfuncr)rto_register_funcs r#__call__zRegistrationCounter.__call__Ms,  $666666r%N)rrrrrr@r%r#rrIs277777r%rrct|tjsJ|dk}|jj|usJtj||dS)Nr)rr/memmapflags writeabletestingassert_array_equal) input_array mmap_data mmap_moders r# check_memmaprRsX i + ++++S I ? $ 1 1 1 1J!!+y99999r%ct}|td|tjd}t |5}t ||tj |j }dddn #1swxYwYtj dkr!tj |rJ|j dksJd}t ||5}t |||tj |j }dddn #1swxYwYtj dkr!tj |rJ|j dksJdS)Nregisterrntr.r+rr+)rsetattratexitr/r6rrospathdirnamefilenamenameexistsr) monkeypatchregistration_counterrdata temp_folderrs r#test_tempmemmaprYs.00 ,@AAA'!**K K 5D[$'''goodm44 555555555555555 w$7>>+.....  (A - - - -I K9 5 5 55[$)<<<<goodm44 555555555555555 w$7>>+.....  (A - - - - - -s$5BBB+7D..D25D2zmemmap not fully supported)rct}|td|tjd}t |}t |||jdksJt |d\}}t |||tj |j ksJ|jdksJd}t ||}t ||||jdksJ||dz|dzg}t |}t||D]\}}t |||jd ksJt |d g\}} t ||| d ksJdS) Nrrr.T) return_folderr+rrz not-an-array) rrrr/r6rrrrrrrr) rrrrfolderr input_listmmap_data_list output_dataothers r#test_create_memmap_backed_datar ns.00 ,@AAA'!**K $[ 1 1Dd###  (A - - - -,[MMMLD&d### RW__T]33 3 3 3 3  (A - - - -I $[I F F FDdI...  (A - - - -{Q a@J.z::N ^<<(( T[$''''  (A - - - -2K3PQQKk*** N " " " " " "r%z constructor_name, container_typelisttuplearrayr sparse_csrsparse_csr_array sparse_cscsparse_csc_array dataframec4tjdjSNpandas)r2r DataFramer@r%r#rrsf1(;;Er%seriesc4tjdjSr)r2rSeriesr@r%r#rrs6.x88?r%indexc4tjdjSr)r2rIndexr@r%r#rrs&-h77=r%slicezdtype, superdtypecZ|dvr |}ddg}t|||}t||sJ|dvr,tjt |d|sJdSt |dr|j|ksJdSt |dr|jd|ksJdSdS) z\Check that we convert the container to the right type of array with the right data type.)rpolarsr polars_seriesrrr.dtype)r rrr$dtypesN)rrr/ issubdtypetyperr$r%)constructor_namecontainer_typer$ superdtype containercontainer_converteds r#test_convert_containerr-sDVVV())AI, )> : ::::555}T"5a"899:FFFFFFF $g . .6"(E111111 $h / /6")!,55556655r%ctjdtdggddgdg}|jjddksJdS)Nrr8rr<categorical_feature_namesrrY)r2rrr%iloc)dfs r#(test_convert_container_categories_pandasr3sZ !!!  seu   B 9>!  * * * * * *r%ctjd}tdggddgdg}|jd|ksJdS)Nr!r8r<r/)r2rrschema Categorical)plr2s r#(test_convert_container_categories_polarsr8sY  X & &B cUGXuQTPU V V VB 9S>R^^-- - - - - - -r%ctjd}tdggddgdg}t|jdj|jusJdS)Npyarrowr8r<r/r)r2rrr'r5DictionaryType)par2s r#)test_convert_container_categories_pyarrowr=s]  Y ' 'B cUGYRUQV W W WB  ! ! " "b&7 7 7 7 7 7 7r%z1.8z-sparse arrays are available as of scipy 1.8.0r(r$cddg}tjtdt5t |||ddddS#1swxYwYdS)zCheck that if we convert to sparse array but sparse array are not supported (scipy<1.8.0), we should raise an explicit error.rr.z&only available with scipy>=1.8.0, got r)r#N)r2rr4rr)r(r$r+s r#7test_convert_container_raise_when_sparray_not_availabler?sAI CzCC   EE 9&6eDDDD EEEEEEEEEEEEEEEEEEsAA A cztt5t#1swxYwYttd5}td#1swxYwY|jsJttddg5}td#1swxYwY|jsJtjtd5tt5}t d#1swxYwY dddn #1swxYwY|jrJtjt d5ttd5}t #1swxYwY dddn #1swxYwY|jrJtjtd5ttd5}t d#1swxYwY dddn #1swxYwY|jrJtjt d 5ttd 5}td #1swxYwY dddn #1swxYwY|jrJtjt d5ttd d 5}td #1swxYwY dddn #1swxYwY|jrJtjt d 5tt5} dddn #1swxYwYdddn #1swxYwY|jrJttd d5} dddn #1swxYwY|jrJtttf5t#1swxYwYtttf5t #1swxYwYtjt 5tttf5 dddn #1swxYwYddddS#1swxYwYdS)Nz how are your)zhello how are youz not this onezthis will be raisedzthe failure message)err_msgzthis is ignoredz,should contain one of the following patternshelloz Bad message)r*rAz Did not raiseT)r*may_pass)rrKraised_and_matchedr2r4r3)cms r# test_raisesrFs   kk   / / /-2+,,,--------      .-!@ A A A-R+,,,--------     z)> ? ? ?44 I   4"233 3 4 4 4 4 4 4 4 4 4444444444444444$$$$ ~-B C C C I'< = = = ,,          $$$$ z)> ? ? ?44 I%6 7 7 7 42233 3 4 4 4 4 4 4 4 4 4444444444444444$$$$ L   ++IW - - - +M** * + + + + + + + + ++++++++++++++++ $$$$ ~-B C C C++ IW6K L L L +PRM** * + + + + + + + + ++++++++++++++++$$$$ ~_ = = = I   "                $$$$  4 8 8 8 B                $$$$ J' ( (kk J' ( (ll ~ & & Y + , ,                  s$((AAABBBC=C%%C) )C=,C) -C==DD,E*EE E*E E**E.1E.G0GG GG GGGIH..H2 2I5H2 6II  I 5J5 JJ! !J5$J! %J55J9<J9$L9L; LL LL LL"%L" MMMNNN;O  OO.P0 P P0P P0P P00P47P4cdttjdgtjdt jt 5ttjdgtjddddn #1swxYwYttjdgtjdt jt 5ttjdgtjddddn #1swxYwYt jt 5ttjdgtjddddn #1swxYwYttjdgtjdd dS) Ng㈵?r#rgX?gc^?g55?gh㈵>ggh㈵>)atol)r r/rfloat32r2rr3float64r@r%r#"test_float32_aware_assert_allcloserK2sEBHj\<<PQQQQQQr%ctj5tt|dddddS#1swxYwYdS)NTrg)rRr_rrc)ras r#test_turn_warnings_into_errorsrls  " "HH!### $GGGGHHHHHHHHHHHHHHHHHHs AAA)r)[rrunittestrRnumpyr/r2scipyrsklearn.discriminant_analysisr sklearn.treersklearn.utils._testingrrrr r r r r rrrrrrrrrsklearn.utils.deprecationrsklearn.utils.fixesrrrrrsklearn.utils.metaestimatorsrr$mark parametrizer>rArMrTestCaserrrrrrrrrrrrrrxfailr r rndarray csr_matrixrrint32integerint64rIfloatingrJr-r3r8r=skipifr?rFrKrQparamrZrcrjrlr@r%r#rsi  DDDDDD//////&100000655555""".99++:9+& $$$Nrrrj77777!777   &   &   &   *   "&        ,".;.;.;.;.;.;.;.;bEMEMEMP77777777::::...*8$@AA##BA#<&  % "* 6$%  l. / @ @  l. / @ @  EEF  ??@  ==>  % " 2: 2: R[! R[! 66#46>+++... 888 --&&& :+.@BT-UVV28RXrz2:"NOO E EPOWV  ECCCLHHH$8$=>>LL?>L. +$$]]5111F%     +$$]]5111F%   *33+*3CCC0)H)H)J)JKKRRLKR )H)H)J)JKKHHLKHHHr%