o &ѹgn@sddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z mZmZmZmZmZmZmZmZmZmZmZmZmZmZddlmZddlmZm Z m!Z!m"Z"m#Z#ddl$m%Z%d d Z&ej'(d e d d Z)ddZ*ddZ+ddZ,ddZ-ddZ.ddZ/ddZ0GdddZ1GdddZ2Gd d!d!Z3ed"d#Z4d$d%Z5d&d'Z6d(d)Z7ed*d+Z8eej'(d,e5e6gd-gd.gd/d0fe5e6gd1d2gd3d4fgd5d6Z9eej'(d7ej:e5e6gd8d-gid9d9d:d;ej:e5e6gdd;ej:e5e6e7gd8d-gid?d9d@d;ej:e5e6e7gd8d.gidAd9dBd;ej:e6e7gd8dCgidDd9dEd;ej:e5e6gd=d.gd/d9dFdGd;gdHdIZ;dJdKZedPdQZ?edRdSZ@GdTdUdUZAddWdXZBdYdZZCej'jDed[d\d]d^ZEej'(d_d`eFfdaeGfdbejHfdcejIfgeJdddege!eJdfdgge dhdidjfdkdldjfdmdndjfdoeKfej'(dpejLejMfejNejMfejOejPfejQejPfgdqdrZRdsdtZSdudvZTdwdxZUej'jVe#e"dykdzd\ej'(d{dedggej'(d|ejLejNejOejQgd}d~ZWddZXddZYej'jDedd\ddZZej'(d{dddfej:deej'jVe#e"dykdzd\dej:dgej'jVe#e"dykdzd\dgddZ[ddZ\ej'(deddZ]ej'(deddZ^dS)N)sparse)LinearDiscriminantAnalysis)DecisionTreeClassifier) TempMemmap_convert_container_delete_folder_get_warnings_filters_info_listassert_allcloseassert_allclose_dense_sparseassert_docstring_consistency'assert_run_python_script_without_outputcheck_docstring_parameterscreate_memmap_backed_dataignore_warningsraisesset_random_stateskip_if_no_numpydocturn_warnings_into_errors) deprecated)_IS_WASMCSC_CONTAINERSCSR_CONTAINERS parse_version sp_version) available_ifcCs2t}t}t|dt|d|jdksJdS)N)rrrZ random_state)Zldatreerd/Users/admin/.pyenv/versions/3.10.0/lib/python3.10/site-packages/sklearn/utils/tests/test_testing.pytest_set_random_state's   r csr_containerc Cstddd}d}||}||fD]%}tjt|dt||dWdn1s.wYt||qtjtdd t||Wdn1sPwYtj t ddd  }|t d }tjtd dt||WddS1swYdS) N rzNot equal to tolerance matchzCan only compare two sparser)offsets)r%zArrays are not equal) npZarangeZreshapepytestrAssertionErrorr ValueErrorrZdiagsonesZtocsr)r xmsgyXABrrr!test_assert_allclose_dense_sparse0s     "r3csddddttdtttdWdn1s&wYttttdWdn1sBwYt}ttdWdn1s]wYt |dksjJt |dj tstJt |d j ts~Jt}ttdWdn1swYt |d ksJt |dj tsJttdtttfdWdn1swYtfd d }tttfdfd d }ttdfdd}ttdfdd}ttdfdd}ttdfdd}ttd|||Wdn 1s,wYtt |Wdn 1sEwYtt |Wdn 1s^wYtt |Wdn 1swwYfdd}fdd}fdd} fdd} fdd} fd d!} ttd||| Wdn 1swYtt | Wdn 1swYtt | Wdn 1swYtt | Wdn 1s wYt} d"}tj t |d#t| }|Wdn 1s.wYtj t |d#t| d$d%}WddS1sOwYdS)&NcSstdtdSNzdeprecation warningwarningswarnDeprecationWarningrrrr_warning_functionGsz.test_ignore_warning.._warning_functioncSstdttddSr4r5rrrr_multiple_warning_functionJs z7test_ignore_warning.._multiple_warning_functionerrorcategoryr$rr'csdSNrrr:r9rrdecorator_no_warningis z1test_ignore_warning..decorator_no_warningc dSr>rrr:rrdecorator_no_warning_multiplen z:test_ignore_warning..decorator_no_warning_multiplecrAr>rrr9rr decorator_no_deprecation_warningrrDz=test_ignore_warning..decorator_no_deprecation_warningcrAr>rrrErrdecorator_no_user_warningvrDz6test_ignore_warning..decorator_no_user_warningcrAr>rrrBrr)decorator_no_deprecation_multiple_warningzrDzFtest_ignore_warning..decorator_no_deprecation_multiple_warningcrAr>rrrBrr"decorator_no_user_multiple_warning~rDz?test_ignore_warning..decorator_no_user_multiple_warningcs2t WddS1swYdSr>)rrrErrcontext_manager_no_warnings"z7test_ignore_warning..context_manager_no_warningcs:tttfd WddS1swYdSNr<)rr8 UserWarningrrBrr#context_manager_no_warning_multiples"z@test_ignore_warning..context_manager_no_warning_multiplec6ttd WddS1swYdSrKrr8rrErr&context_manager_no_deprecation_warning "zCtest_ignore_warning..context_manager_no_deprecation_warningcrNrKrrLrrErrcontext_manager_no_user_warningrQz.context_manager_no_user_warningcrNrKrOrrBrr/context_manager_no_deprecation_multiple_warningrQzLtest_ignore_warning..context_manager_no_deprecation_multiple_warningcrNrKrRrrBrr(context_manager_no_user_multiple_warningrQzEtest_ignore_warning..context_manager_no_user_multiple_warningzA'obj' should be a callable.+you should use 'category=UserWarning'r"cSdSr>rrrrrtestsz!test_ignore_warning..test)r6catch_warnings simplefilterrr8r)warnsrL FutureWarninglen isinstancemessagerr+)recordr@rCrFrGrHrIrJrMrPrSrTrUZ warning_classr#Zsilence_warnings_funcrWrr?rtest_ignore_warningDs                          $r`cC ||}|SzFunction f Parameters ---------- a : int Parameter a b : float Parameter b Returns ------- c : list Parameter c Nrabcrrrf_okrgcCra)zFunction f Parameters ---------- a : int Parameter a b : float Parameter b Results ------- c : list Parameter c Nrrcrrrf_bad_sectionsrhricCs ||}|Srbr)rerdrfrrr f_bad_orderrhrjcCra)zFunction f Parameters ---------- a : int Parameter a b : int Parameter b c : int Parameter c Returns ------- d : list Parameter c Nr)rdredrrrf_too_many_param_docstringsrlcCra)zFunction f Parameters ---------- a : int Parameter a Returns ------- c : list Parameter c Nrrcrrr f_missings rmcCs||||S)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 Nr)rdrerfrkerrrf_check_param_definition(sroc@eZdZddZddZdS)KlasscCrVr>rselfr0r/rrrrm<zKlass.f_missingcCdS)zFunction f Parameter --------- a : int Parameter a b : float Parameter b Results ------- c : list Parameter c Nrrrrrrri?szKlass.f_bad_sectionsN)__name__ __module__ __qualname__rmrirrrrrq; rqc@s4eZdZddZddZddZddZd d Zd S) MockEstcCru)Z MockEstimatorNrrsrrr__init__RzMockEst.__init__cC|Sr>rrrrrrfitUrtz MockEst.fitcCr~r>rrsr0rrrpredictXrtzMockEst.predictcCr~r>rrrrr predict_proba[rtzMockEst.predict_probacCru)N?rrrrrscore^rtz MockEst.scoreN)rvrwrxr|rrrrrrrrrzQs  rzc@sheZdZddZeddddZeddedd d Zed dd d ZedddZ dS)MockMetaEstimatorcCs ||_dS)zMetaEstimator to check if doctest on delegated methods work. Parameters --------- delegate : estimator Delegated estimator. Ndelegate)rsrrrrr|cs zMockMetaEstimator.__init__cC t|jdS)Nrhasattrrr{rrrm zMockMetaEstimator.cCs |j|S)zThis is available only if delegate has predict. Parameters ---------- y : ndarray Parameter y N)rrrrrrrms zMockMetaEstimator.predictcCr)Nrrr{rrrrxrz%Testing a deprecated delegated methodcCru)zThis is available only if delegate has score. Parameters --------- y : ndarray Parameter y Nrrrrrrxr}zMockMetaEstimator.scorecCr)Nrrr{rrrrrcCr~)zThis is available only if delegate has predict_proba. Parameters --------- X : ndarray Parameter X Nrrrrrrs zMockMetaEstimator.predict_probaz-Testing deprecated function with wrong paramscCru)z,Incorrect docstring but should not be testedNrrrrrrrr}zMockMetaEstimator.fitN) rvrwrxr|rrrrrrrrrrrbs     rc Cstt}|gks Jttdgd}|gksJttdgd}|gks$Jtjtdd ttWdn1s:wYtjtdd ttjWdn1sVwYtt}t t d}|j j }|gdksqJgdgd gd gd d d |dddddddgd d |dddddgd d |ddddgd d |dddddgg}t |ttttj|j|j|j|jgD]\}}t|}||ksJd||fqdS)Nre)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 "%s" not in "%s")r rgrmr)r RuntimeErrorrirqrorrz __class__rvziprjrlrrrr)Z incorrectZ mock_metaZmock_meta_namemessagesr.frrrtest_check_docstring_parameterss            S  rcCru)zFunction one. Parameters ---------- a : int, float Parameter a. Second line. b : str Parameter b. Returns ------- c : int Returning d : int Returning Nrrdrerrrf_onesrcCru)aFunction two. Parameters ---------- a : int, float Parameter a. Second line. b : str Parameter bb. e : int Extra parameter. Returns ------- c : int Returning d : int Returning Nrrrrrf_two-srcCru)zFunction two. Parameters ---------- a : int, float Parameter a. b : str Parameter B! e : Extra parameter. Returns ------- c : int Returning. d : int Returning Nrrrrrf_threeGsrcCs@tjtddtdtgWddS1swYdS)z1Check error raised when `objects` incorrect type.zAll 'objects' must be one ofr"stringN)r)r TypeErrorr rrrrr-test_assert_docstring_consistency_object_type`s"rzobjects, kwargs, errorrdre)include_paramsZexclude_paramszThe 'exclude_params' argumentFrf)include_returnsZexclude_returnszThe 'exclude_returns' argumentcCsDtjt|dt|fi|WddS1swYdS)z?Check `assert_docstring_consistency` argument checking correct.r"N)r)rrr )objectskwargsr;rrr,test_assert_docstring_consistency_arg_checksgs"rzobjects, kwargs, error, warnr whitespace)idrTZincl_allzNThe description of Parameter 'a' is inconsistent between \['f_one',\n'f_two'\]z 2-1 groupzYThe description of Parameter 'b' is inconsistent between \['f_one'\] and\n\['f_two'\] andz 1-1-1 grouprnzPThe type specification of Parameter 'e' is inconsistent between\n\['f_two'\] andz empty typez,Checking was skipped for Parameters: \['e'\]z skip warncCs|r$tjt|dt|fi|WddS1swYdS|rHtjt|dt|fi|WddS1sAwYdSt|fi|dS)z;Check `assert_docstring_consistency` gives correct results.r"N)r)rr*r rZrL)rrr;r7rrr!test_assert_docstring_consistency}s1""rcCru)zFunction four. Parameters ---------- labels : array-like, default=None The set of labels to include when `average != 'binary'`, and their order if `average is None`. Labels present in the data can be excluded. Nrlabelsrrrf_four rcCru)aFunction five. Parameters ---------- labels : array-like, default=None The set of labels to include when `average != 'binary'`, and their order if `average is None`. This is an extra line. Labels present in the data can be excluded. Nrrrrrf_fives rcCru)zFunction six. Parameters ---------- labels : array-like, default=None The group of labels to add when `average != 'binary'`, and the order if `average is None`. Labels present on them datas can be excluded. Nrrrrrf_sixrrcCsJd}tjt|dttttgddWddS1swYdS)z8Check `assert_docstring_consistency` difference message.aThe description of Parameter 'labels' is inconsistent between \['f_four'\] and \['f_five'\] and \['f_six'\]: \*\*\* \['f_four'\] --- \['f_five'\] \*\*\*\*\*\*\*\*\*\*\*\*\*\*\* \*\*\* 10,25 \*\*\*\* --- 10,30 ---- 'binary'`, and their order if `average is None`. \+ This is an extra line. Labels present in the data can be excluded. \*\*\* \['f_four'\] --- \['f_six'\] \*\*\*\*\*\*\*\*\*\*\*\*\*\*\* \*\*\* 1,25 \*\*\*\* The ! set of labels to ! include when `average != 'binary'`, and ! their order if `average is None`. Labels present ! in the data can be excluded. --- 1,25 ---- The ! group of labels to ! add when `average != 'binary'`, and ! the order if `average is None`. Labels present ! on them datas can be excluded.r"T)rN)r)rr*r rrr)r.rrr+test_assert_docstring_consistency_error_msgs*"rcCsd}ttttgdd|dd}ttttgdd|dd}d}tjt|dttttgdd|dWd d S1sHwYd S) zACheck `assert_docstring_consistency` `descr_regex_pattern` works.zThe (set|group) of labels to (include|add) when `average \!\= 'binary'`, and (their|the) order if `average is None`\.[\s\w]*\.* Labels present (on|in) (them|the) datas? can be excluded\.T )rZdescr_regex_patternz(labels|average|binary)z#The set of labels to include when.+zCThe description of Parameter 'labels' in \['f_six'\] does not matchr"N) r rrrjoinsplitr)rr*)Z regex_fullZ regex_wordsZ regex_errorr.rrr5test_assert_docstring_consistency_descr_regex_patterns,    "rc@rp)RegistrationCountercCs d|_dS)Nr)nb_callsr{rrrr|8s zRegistrationCounter.__init__cCs |jd7_|jtusJdS)Nr')rfuncr)rsZto_register_funcrrr__call__;szRegistrationCounter.__call__N)rvrwrxr|rrrrrr7ryrrcCs:t|tjsJ|dk}|jj|usJtj||dS)Nr)r]r(Zmemmapflags writeableZtestingZassert_array_equal) input_arrayZ mmap_data mmap_moderrrr check_memmap@srcCst}|td|td}t|}t||tj |j }Wdn1s*wYtj dkrJd}t||d}t||||jdksSJ||d|dg}t|}t ||D] \}}t||qe|jd ksvJt|d g\}} t||| d ksJdS) Nrrr'T)Z return_folderr$rrz not-an-array) rrrr(r,rrrrrrrr) rrrrfolderrZ input_listZmmap_data_listZ output_dataotherrrrtest_create_memmap_backed_data\s,       rz constructor_name, container_typelisttuplearrayrZ sparse_csrZsparse_csr_arrayZ sparse_cscZsparse_csc_array dataframecC tdjSNpandas)r) importorskipZ DataFramerrrrrrrseriescCrr)r)rZSeriesrrrrrrindexcCrr)r)rIndexrrrrrrslicezdtype, superdtypecCs|dvr|}ddg}t|||d}t||sJ|dvr+tt|d|s)JdSt|dr9|j|ks7JdSt|drG|jd|ksIJdSdS) z\Check that we convert the container to the right type of array with the right data type.)rpolarsrZ polars_seriesrrr'dtype)rrrrdtypesN)rr]r(Z issubdtypetyperrr)constructor_nameZcontainer_typerZ superdtype containerZcontainer_convertedrrrtest_convert_container{s""  rcCs:tdtdggddgdgd}|jjddksJdS)Nrr-rr1Zcategorical_feature_namesrr=)r)rrrZiloc)dfrrr(test_convert_container_categories_pandass rcCs<td}tdggddgdgd}|jd|ksJdS)Nrr-r1r)r)rrschemaZ Categorical)plrrrr(test_convert_container_categories_polarss rcCs@td}tdggddgdgd}t|jdj|jusJdS)NZpyarrowr-r1rr)r)rrrrZDictionaryType)parrrr)test_convert_container_categories_pyarrows rz1.8z-sparse arrays are available as of scipy 1.8.0rrcCsPddg}tjtdtdt|||dWddS1s!wYdS)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"rN)r)rr+rr)rrrrrr7test_convert_container_raise_when_sparray_not_availables "rc Csttt1s wYttdd}td1swY|js(Jttddgd}td1s8wY|jsBJtjtddtt}td1sVwYWdn1sewY|jroJtjtddttdd}t1swYWdn1swY|jrJtjtddttdd}td1swYWdn1swY|jrJtjtd dttd d}td 1swYWdn1swY|jrJtjtddttd dd }td 1swYWdn 1s#wY|jr.Jtjtd dtt}Wdn 1sFwYWdn 1sVwY|jraJttd dd}Wdn 1stwY|jrJtttft1swYtttft1swYtt(tttfWdn1swYWddSWddS1swYdS)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 patternsZhelloz Bad message)r#rz Did not raiseT)r#Zmay_pass)rrZraised_and_matchedr)r+r*)cmrrr test_raisess              $rcCsttjdgtjddttttjdgtjddWdn1s(wYttjdgtjddttttjdgtjddWdn1sUwYttttjdgtjddWdn1svwYttjdgtjddd d dS) Ng㈵?rrgX?gc^?g55?gh㈵>ggh㈵>)Zatol)r r(rfloat32r)rr*float64rrrr"test_float32_aware_assert_allclose s    rzcannot start subprocesscCsd}t|d}tjtdd t|Wdn1swYd}tjtddt|ddWdn1s=wYdd d g}tjtd dt|d dWddS1sbwYdS) Nzx = 1zprint('something to stdout')zExpected no outputr"z:output was not supposed to match.+got.+something to stdoutz to.+stdout)pattern z import sysz-print('something to stderr', file=sys.stderr)z:output was not supposed to match.+got.+something to stderrz to.+stderr)r r)rr*r)coderrr,test_assert_run_python_script_without_output2s( "r)ZmarkscCs tjddddd}t||dS)zyNon-regression test to check that we can still convert a sparse container from a given format to another format. g?Zcsr)ZdensityformatN)rrandomr)rZX_sparserrr'test_convert_container_sparse_to_sparseJsrcCs|jdkr)|r)tj|j|jdtj|j|jdWdn1s$wY|jdkrtjddD}|j}d|vr>d}tj||jd|rPt|d ksQJ |rht |d j|ks^J|d j|jkspJWddSWddS1s{wYdSdS) Nr;r")r^r=rT)r_ZPyarrowz* Pyarrow will become a required dependencyrr') actionr)rr=r^r6r7rXr\str) warning_infowarnings_as_errorsr_r^rrrcheck_warnings_as_errorsgs2  "rrcCs"tjdddk}t||ddS)NZSKLEARN_WARNINGS_AS_ERRORS0r)renvirongetr)rrrrrtest_sklearn_warnings_as_errorssrcCs@ttt|ddWddS1swYdS)NTr)r6rXrr)rrrrtest_turn_warnings_into_errorss "r )r)_rrr6Znumpyr(r)ZscipyrZsklearn.discriminant_analysisrZ sklearn.treerZsklearn.utils._testingrrrrr r r r r rrrrrrZsklearn.utils.deprecationrZsklearn.utils.fixesrrrrrZsklearn.utils.metaestimatorsrrmarkZ parametrizer3r`rgrirjrlrmrorqrzrrrrrrrparamrrrrrrrrrZxfailrrrZndarrayZ csr_matrixrrZint32integerZint64rZfloatingrrrrrZskipifrrrrrrrr rrrrsN   D   1          .   0 %               F