hWJddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl mZmZmZmZmZmZmZmZmZmZmZmZmZmZddlmZddlmZdZ d Z!d Z"d Z#d Z$Gd dej%Z&dZ'dZ(dZ)dZ*dZ+dZ,GddZ-GddZ.GddZ/dZ0GddZ1d7dZ2d Z3ej45d!d"d#gd$Z6ej45d%d&e7fd'e8fd(ej9fd)ej:fd*ej:fd+ej;fd,d-fd.d/fd0d1fd2efej?ej>fej@ejAfejBejAfgd4ZCd5ZDd6ZEdS)8N)sparse)LinearDiscriminantAnalysis)DecisionTreeClassifier) TempMemmap_convert_container_delete_folderassert_allcloseassert_allclose_dense_sparseassert_no_warningsassert_raise_message assert_raisesassert_raises_regexcheck_docstring_parameterscreate_memmap_backed_dataignore_warningsraisesset_random_state) deprecated) available_ifct}t}t|dt|d|jdksJdS)N)rrr random_state)ldatrees h/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/sklearn/utils/tests/test_testing.pytest_set_random_stater sR $ & &C ! # #DS!T1   ! ! ! ! ! !c tjddd}d}tj|}||fD]X}t jt|5t||dzdddn #1swxYwYt||Yt jtd5t||dddn #1swxYwYtj tj dd  }tj tj d }t jtd 5t||ddddS#1swxYwYdS) N rzNot equal to tolerance matchzCan only compare two sparser)offsets)r#zArrays are not equal)nparangereshaper csc_matrixpytestrAssertionErrorr ValueErrordiagsonestocsr csr_matrix)xmsgyXABs r!test_assert_allclose_dense_sparser7)s ! Q""A #C!AV++ ]> 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"'&//**A ~-C D D D++$Q***++++++++++++++++++s6 BB B 7CCCE88E<?E<cttd5ttd5 dddn #1swxYwYddddS#1swxYwYdS)Nz Hello world)r2)rr+r r,rrtest_assert_raises_msgr:<s ^] ; ; := 9 9 9                  s.A; A? A? AAAcbd}d}ttd|dttttd|dttttd|dttttd|ttttt fd|dS)Nc t|N)r,)messages r_raise_ValueErrorz4test_assert_raise_message.._raise_ValueErrorCs!!!rcdSr=r9r9rr _no_raisez,test_assert_raise_message.._no_raiseF rtestzsomething else)r r,r r+ TypeErrorAttributeError)r?rAs rtest_assert_raise_messagerFBs"""   V->GGG  ."6 FIVVV ^$ rc> ddttttttjt5tt dddn #1swxYwYtjt 5tt dddn #1swxYwYtjt5tt dddn #1swxYwYtttt 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} tjt| 5t| }|dddn #1swxYwYtjt| 5t| d}ddddS#1swxYwYdS)Nc:tjdtdSNzdeprecation warningwarningswarnDeprecationWarningr9rr_warning_functionz.test_ignore_warning.._warning_functionls +-?@@@@@rcbtjdttjddSrIrJr9rr_multiple_warning_functionz7test_ignore_warning.._multiple_warning_functionos- +-?@@@ +,,,,,rcategoryc0dSr=r9)rPrNsrdecorator_no_warningz1test_ignore_warning..decorator_no_warnings'""$$$$$rcdSr=r9rPsrdecorator_no_warning_multiplez:test_ignore_warning..decorator_no_warning_multiple""$$$$$rcdSr=r9rNsr decorator_no_deprecation_warningz=test_ignore_warning..decorator_no_deprecation_warningrcdSr=r9rZsrdecorator_no_user_warningz6test_ignore_warning..decorator_no_user_warningr\rcdSr=r9rVsr)decorator_no_deprecation_multiple_warningzFtest_ignore_warning..decorator_no_deprecation_multiple_warningrXrcdSr=r9rVsr"decorator_no_user_multiple_warningz?test_ignore_warning..decorator_no_user_multiple_warningrXrcjt5ddddS#1swxYwYdSr=)rrZsrcontext_manager_no_warningz7test_ignore_warning..context_manager_no_warnings                           s (,,ctttf5ddddS#1swxYwYdSNrQ)rrM UserWarningrVsr#context_manager_no_warning_multiplez@test_ignore_warning..context_manager_no_warning_multiples '9;&G H H H ) ) & & ( ( ( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )s 6::cxtt5ddddS#1swxYwYdSrfrrMrZsr&context_manager_no_deprecation_warningzCtest_ignore_warning..context_manager_no_deprecation_warnings &8 9 9 9                        /33cxtt5ddddS#1swxYwYdSrfrrgrZsrcontext_manager_no_user_warningz.context_manager_no_user_warnings k 2 2 2                        rlcxtt5ddddS#1swxYwYdSrfrjrVsr/context_manager_no_deprecation_multiple_warningzLtest_ignore_warning..context_manager_no_deprecation_multiple_warnings &8 9 9 9 ) ) & & ( ( ( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )rlcxtt5ddddS#1swxYwYdSrfrnrVsr(context_manager_no_user_multiple_warningzEtest_ignore_warning..context_manager_no_user_multiple_warnings k 2 2 2 ) ) & & ( ( ( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )rlzA'obj' should be a callable.+you should use 'category=UserWarning'r cdSr=r9r9rrrCz!test_ignore_warning..tests Dr) r rrMr*warnsrg FutureWarningrr,)rTrWr[r^r`rbrdrhrkrorqrs warning_classr!silence_warnings_funcrCrPrNs @@rtest_ignore_warningryisOAAA--- '899:::'8CUVVVWWW ( ) )CC@)K@@@BBBCCCCCCCCCCCCCCC k " "NNK2]KKKMMMNNNNNNNNNNNNNNN ( ) )LLI2[IIIKKKLLLLLLLLLLLLLLL)5G4UVVV %%%%%_%1;?@@@%%%%A@%011121k***+*0111%%%%21%k***%%%%+*%+,,,45557888 ( ) )$$!!###$$$$$$$$$$$$$$$ k " "4411333444444444444444 ( ) )--**,,,---------------     )))))          ))))))))))1222:;;;=>>> ( ) )**'')))*************** k " "::77999::::::::::::::: ( ) )3300222333333333333333 M OE z / / /  > > >?P Q Q                z / / /  ' '   ( ' s!B  BB0CC"C?D**D.1D. I66I:=I: J11J58J5 K,,K03K0 M22M69M6 N--N14N1 O((O,/O,#QQ  Q ,RRRceZdZdZdS) TestWarnscd}tjt5t|dddn #1swxYwYtdddksJdS)Nc.tjddS)Nyor)rKrLr9rrfzTestWarns.test_warn..fs M$   1rc|Sr=r9)r1s rz%TestWarns.test_warn..sArr%)r*rr+r )selfrs r test_warnzTestWarns.test_warns   ]> * * " " q ! ! ! " " " " " " " " " " " " " " "!++q11Q666666s 9==N)__name__ __module__ __qualname__rr9rrr{r{s#77777rr{c||z}|SzFunction f Parameters ---------- a : int Parameter a b : float Parameter b Returns ------- c : list Parameter c r9abcs rf_okr AA Hrc||z}|S)zFunction f Parameters ---------- a : int Parameter a b : float Parameter b Results ------- c : list Parameter c r9rs rf_bad_sectionsrrrc||z}|Srr9)rrrs r f_bad_orderr rrc||z}|S)zFunction f Parameters ---------- a : int Parameter a b : int Parameter b c : int Parameter c Returns ------- d : list Parameter c r9)rrds rf_too_many_param_docstringrs" AA Hrc||z}|S)zFunction f Parameters ---------- a : int Parameter a Returns ------- c : list Parameter c r9rs r f_missingr2s AA Hrc||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 r9)rrrres rf_check_param_definitionrCs q519q=rceZdZdZdZdS)KlasscdSr=r9rr4r3s rrzKlass.f_missingWrBrcdS)zFunction f Parameter --------- a : int Parameter a b : float Parameter b Results ------- c : list Parameter c Nr9rs rrzKlass.f_bad_sectionsZs  rN)rrrrrr9rrrrVs2        rrc,eZdZdZdZdZdZdZdS)MockEstcdS) MockEstimatorNr9rs r__init__zMockEst.__init__mrc|Sr=r9rs rfitz MockEst.fitprc|Sr=r9rr4s rpredictzMockEst.predictsrrc|Sr=r9rs r predict_probazMockEst.predict_probavrrcdS)N?r9rs rscorez MockEst.scoreyssrN)rrrrrrrrr9rrrrls_rrceZdZdZeddZededdZeddZed d Z d S) MockMetaEstimatorc||_dS)zMetaEstimator to check if doctest on delegated methods work. Parameters --------- delegate : estimator Delegated estimator. Ndelegate)rrs rrzMockMetaEstimator.__init__~s! rc,t|jdS)Nrhasattrrrs rrzMockMetaEstimator.swt}i@@rc6|j|S)zThis is available only if delegate has predict. Parameters ---------- y : ndarray Parameter y )rrrs rrzMockMetaEstimator.predicts}$$Q'''rc,t|jdS)Nrrrs rrzMockMetaEstimator.swt}g>>rz%Testing a deprecated delegated methodcdS)zThis is available only if delegate has score. Parameters --------- y : ndarray Parameter y Nr9rs rrzMockMetaEstimator.scorerrc,t|jdS)Nrrrs rrzMockMetaEstimator.swt}oFFrc|S)zThis is available only if delegate has predict_proba. Parameters --------- X : ndarray Parameter X r9rs rrzMockMetaEstimator.predict_probas rz-Testing deprecated function with wrong paramscdS)z,Incorrect docstring but should not be testedNr9rs rrzMockMetaEstimator.fitrrN) rrrrrrrrrrr9rrrr}s!!!\@@AA((BA(\>>??Z788  98@? \FFGGHGZ?@@;;A@;;;rrc 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 "")r* importorskiprrrr RuntimeErrorrrrrr __class__rziprrrrrr) incorrect mock_metamock_meta_namemessagesr2rs rtest_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 rrzRegistrationCounter.__init__7s  rcF|xjdz c_|jtusJdS)Nr%)rfuncr)rto_register_funcs r__call__zRegistrationCounter.__call__:s,  $666666rN)rrrrrr9rrrr6s277777rrrct|tjsJ|dk}|jj|usJtj||dS)Nr) isinstancer&memmapflags writeabletestingassert_array_equal) input_array mmap_data mmap_moders r check_memmapr?sX i + ++++S I ? $ 1 1 1 1J!!+y99999rct}|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&r.rrospathdirnamefilenamenameexistsr) monkeypatchregistration_counterrdata temp_folderrs rtest_tempmemmaprFs.00 ,@AAA'!**K K 5D[$'''goodm44 555555555555555 w$7>>+.....  (A - - - -I K9 5 5 55[$)<<<<goodm44 555555555555555 w$7>>+.....  (A - - - - - -s$5BBB+7D..D25D2alignedFTc&t}|td|tjd}t ||}t |||jdksJt |d|\}}t |||tj |j ksJ|jdksJd}t ||| }t ||||jdksJ||dz|dzg}t ||}t||D]\}}t |||jd ksJtjtd 5t |d gdddddS#1swxYwYdS)Nrr)rr%T) return_folderrr"r)rrz`When creating aligned memmap-backed arrays, input must be a single array or a sequence of arraysr z not-an-array)rrrr&r.rrrrrrrrr*rr,) rrrrrfolderr input_listmmap_data_lists rtest_create_memmap_backed_datar[s+.00 ,@AAA'!**K $[' B B BDd###  (A - - - -,4LD&d### RW__T]33 3 3 3 3  (A - - - -I $[Iw W W WDdI...  (A - - - -{Q a@J.z7KKKN ^<<(( T[$''''  (A - - - -  '   OO ";"?NNNNOOOOOOOOOOOOOOOOOOs%FF  F z constructor_name, container_typelisttuplearrayr sparse_csr sparse_csc dataframec4tjdjSNpandas)r*r DataFramer9rrrrsf1(;;Erseriesc4tjdjSr )r*rSeriesr9rrrrs6.x88?rindexc4tjdjSr )r*rIndexr9rrrrs&-h77=rslicezdtype, 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.)rr rrr%dtype)rrrrdtypesN)rrr& issubdtypetyperrr)constructor_namecontainer_typer superdtype containercontainer_converteds rtest_convert_containerrs@;;;())AI, )> : ::::555}T"5a"899:FFFFFFF $g . .6"(E111111 $h / /6")!,55556655rcztt5t#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!r z Did not raiseT)r!may_pass)rrDraised_and_matchedr*r,r+)cms r test_raisesr%s   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㈵?rrgX?gc^?g55?gh㈵>ggh㈵>)atol)r r&rfloat32r*rr+float64r9rr"test_float32_aware_assert_allcloser*sEBHj\<<r;sI  DDDDDD////// 100000555555"""+++& $$$NjjjZ77777!777   &   &   &   *   "&        ,".;.;.;.;.;.;.;.;bEMEMEMP77777777::::...*UDM22"O"O32"OJ&  % "* 6$% v() v() EEF ??@ ==> %  2: 2: R[! R[! 6606