o ?Hhn@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)rrr random_state)ldatreer`/home/air/sanwanet/gpt-api/venv/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) nparangereshapepytestrAssertionErrorr ValueErrorrdiagsonestocsr)r"xmsgyXABrrr !test_assert_allclose_dense_sparse0s     "r9csddddttdtttdWdn1s&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_functioncSstdttddSr:r;rrrr _multiple_warning_functionJs z7test_ignore_warning.._multiple_warning_functionerrorcategoryr&rr)csdSNrrr@r?rr decorator_no_warningis z1test_ignore_warning..decorator_no_warningc dSrDrrr@rr decorator_no_warning_multiplen z:test_ignore_warning..decorator_no_warning_multiplecrGrDrrr?rr decorator_no_deprecation_warningrrJz=test_ignore_warning..decorator_no_deprecation_warningcrGrDrrrKrr decorator_no_user_warningvrJz6test_ignore_warning..decorator_no_user_warningcrGrDrrrHrr )decorator_no_deprecation_multiple_warningzrJzFtest_ignore_warning..decorator_no_deprecation_multiple_warningcrGrDrrrHrr "decorator_no_user_multiple_warning~rJz?test_ignore_warning..decorator_no_user_multiple_warningcs2t WddS1swYdSrD)rrrKrr context_manager_no_warnings"z7test_ignore_warning..context_manager_no_warningcs:tttfd WddS1swYdSNrB)rr> UserWarningrrHrr #context_manager_no_warning_multiples"z@test_ignore_warning..context_manager_no_warning_multiplec6ttd WddS1swYdSrQrr>rrKrr &context_manager_no_deprecation_warning "zCtest_ignore_warning..context_manager_no_deprecation_warningcrTrQrrRrrKrr context_manager_no_user_warningrWz.context_manager_no_user_warningcrTrQrUrrHrr /context_manager_no_deprecation_multiple_warningrWzLtest_ignore_warning..context_manager_no_deprecation_multiple_warningcrTrQrXrrHrr (context_manager_no_user_multiple_warningrWzEtest_ignore_warning..context_manager_no_user_multiple_warningzA'obj' should be a callable.+you should use 'category=UserWarning'r$cSdSrDrrrrr testsz!test_ignore_warning..test)r<catch_warnings simplefilterrr>r-warnsrR FutureWarninglen isinstancemessagerr/)recordrFrIrLrMrNrOrPrSrVrYrZr[ warning_classr%silence_warnings_funcr]rrEr test_ignore_warningDs                          $rhcC ||}|SzFunction f Parameters ---------- a : int Parameter a b : float Parameter b Returns ------- c : list Parameter c rabcrrr f_okrocCri)zFunction f Parameters ---------- a : int Parameter a b : float Parameter b Results ------- c : list Parameter c rrkrrr f_bad_sectionsrprqcCs ||}|Srjr)rmrlrnrrr f_bad_orderrprrcCri)zFunction f Parameters ---------- a : int Parameter a b : int Parameter b c : int Parameter c Returns ------- d : list Parameter c r)rlrmdrrr f_too_many_param_docstringsrtcCri)zFunction f Parameters ---------- a : int Parameter a Returns ------- c : list Parameter c rrkrrr f_missings rucCs||||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 r)rlrmrnrserrr f_check_param_definition(srwc@eZdZddZddZdS)KlasscCr\rDrselfr6r5rrr ru<zKlass.f_missingcCdS)zFunction f Parameter --------- a : int Parameter a b : float Parameter b Results ------- c : list Parameter c Nrrzrrr rq?szKlass.f_bad_sectionsN)__name__ __module__ __qualname__rurqrrrr ry; ryc@s4eZdZddZddZddZddZd d Zd S) MockEstcCr}) MockEstimatorNrr{rrr __init__RzMockEst.__init__cC|SrDrrzrrr fitUr|z MockEst.fitcCrrDrr{r6rrr predictXr|zMockEst.predictcCrrDrrrrr predict_proba[r|zMockEst.predict_probacCr})N?rrrrr score^r|z MockEst.scoreN)r~rrrrrrrrrrr rQs  rc@sheZdZddZeddddZeddedd d Zed dd d ZedddZ dS)MockMetaEstimatorcCs ||_dS)zMetaEstimator to check if doctest on delegated methods work. Parameters --------- delegate : estimator Delegated estimator. Ndelegate)r{rrrr rcs zMockMetaEstimator.__init__cC t|jdS)Nrhasattrrrrrr m zMockMetaEstimator.cCs |j|S)zThis is available only if delegate has predict. Parameters ---------- y : ndarray Parameter y )rrrrrr rms zMockMetaEstimator.predictcCr)Nrrrrrr rxrz%Testing a deprecated delegated methodcCr})zThis is available only if delegate has score. Parameters --------- y : ndarray Parameter y Nrrrrr rxrzMockMetaEstimator.scorecCr)Nrrrrrr rrcCr)zThis is available only if delegate has predict_proba. Parameters --------- X : ndarray Parameter X rrrrr rs zMockMetaEstimator.predict_probaz-Testing deprecated function with wrong paramscCr})z,Incorrect docstring but should not be testedNrrzrrr rrzMockMetaEstimator.fitN) r~rrrrrrrrrrrrr rbs     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)Nrm)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 rorur-r RuntimeErrorrqryrwrr __class__r~ziprrrtrrrr) incorrect mock_metamock_meta_namemessagesr4frrr test_check_docstring_parameterss            S  rcCr})zFunction one. Parameters ---------- a : int, float Parameter a. Second line. b : str Parameter b. Returns ------- c : int Returning d : int Returning Nrrlrmrrr f_onesrcCr})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 Nrrrrr f_two-srcCr})zFunction two. Parameters ---------- a : int, float Parameter a. b : str Parameter B! e : Extra parameter. Returns ------- c : int Returning. d : int Returning Nrrrrr f_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, errorrlrm)include_paramsexclude_paramszThe 'exclude_params' argumentFrn)include_returnsexclude_returnszThe 'exclude_returns' argumentcCsDtjt|dt|fi|WddS1swYdS)z?Check `assert_docstring_consistency` argument checking correct.r$N)r-rrr )objectskwargsrArrr ,test_assert_docstring_consistency_arg_checksgs"rzobjects, kwargs, error, warnr whitespace)idrTincl_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 grouprvzPThe 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 r`rR)rrrAr=rrr !test_assert_docstring_consistency}s1""rcCr})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. Nrlabelsrrr f_four rcCr})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. Nrrrrr f_fives rcCr})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. Nrrrrr f_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)r4rrr +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 )rdescr_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.) regex_full regex_words regex_errorr4rrr 5test_assert_docstring_consistency_descr_regex_patterns,    "rc@rx)RegistrationCountercCs d|_dS)Nr)nb_callsrrrr r8s zRegistrationCounter.__init__cCs |jd7_|jtusJdS)Nr))rfuncr)r{to_register_funcrrr __call__;szRegistrationCounter.__call__N)r~rrrrrrrr r7rrrcCs:t|tjsJ|dk}|jj|usJtj||dS)Nr)rcr*memmapflags writeabletestingassert_array_equal) input_array 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) return_folderr&rrz not-an-array) rrrr*r1rrrrrrrr) rrrrfolderr input_listmmap_data_list output_dataotherrrr test_create_memmap_backed_data\s,       rz constructor_name, container_typelisttuplearrayr sparse_csrsparse_csr_array sparse_cscsparse_csc_array dataframecC tdjSNpandas)r- importorskip DataFramerrrr rrrseriescCrr)r-rSeriesrrrr rrindexcCrr)r-rIndexrrrr rrslicezdtype, 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.)rpolarsr polars_seriesrrr)dtype)rrrrdtypesN)rrcr* issubdtypetyperrr)constructor_namecontainer_typer superdtype containercontainer_convertedrrr test_convert_container{s""  rcCs:tdtdggddgdgd}|jjddksJdS)Nrr3rr7categorical_feature_namesrrC)r-rrriloc)dfrrr (test_convert_container_categories_pandass rcCs<td}tdggddgdgd}|jd|ksJdS)Nrr3r7r)r-rrschema Categorical)plrrrr (test_convert_container_categories_polarss rcCs@td}tdggddgdgd}t|jdj|jusJdS)Npyarrowr3r7rr)r-rrr rDictionaryType)parrrr )test_convert_container_categories_pyarrows rz1.8z-sparse arrays are available as of scipy 1.8.0r rcCsPddg}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)r rr rrr 7test_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 patternshelloz Bad message)r%rz Did not raiseT)r%may_pass)rrraised_and_matchedr-r/r.)cmrrr test_raisess              $r#cCsttjdgtjddttttjdgtjddWdn1s(wYttjdgtjddttttjdgtjddWdn1sUwYttttjdgtjddWdn1svwYttjdgtjddd d dS) Ng㈵?rrgX?gc^?g55?gh㈵>ggh㈵>)atol)r r*rfloat32r-rr.float64rrrr "test_float32_aware_assert_allclose s    r'zcannot 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+)markscCs tjddddd}t||dS)zyNon-regression test to check that we can still convert a sparse container from a given format to another format. g?csr)densityformatN)rrandomr)r X_sparserrr 'test_convert_container_sparse_to_sparseJsr3cCs|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) NrAr$)rdrCrT)rePyarrowz* Pyarrow will become a required dependencyr) actionr-rrCrdr<r=r^rbstr) warning_infowarnings_as_errorsrerdrrr check_warnings_as_errorsgs2  "r9r7cCs"tjdddk}t||ddS)NSKLEARN_WARNINGS_AS_ERRORS0r8)renvirongetr9)r7r8rrr test_sklearn_warnings_as_errorssr?cCs@ttt|ddWddS1swYdS)NTr<)r<r^rr9)r7rrr test_turn_warnings_into_errorss "r@)r)_rrr<numpyr*r-scipyrsklearn.discriminant_analysisr sklearn.treersklearn.utils._testingrrrrr r r r r rrrrrrsklearn.utils.deprecationrsklearn.utils.fixesrrrrrsklearn.utils.metaestimatorsrr!mark parametrizer9rhrorqrrrtrurwryrrrrrrrrparamrrrrrrrrrxfailrrrndarray csr_matrixrrint32integerint64r%floatingr&rrrrskipifrr#r'r+r3r9r?r@rrrr sN   D   1          .   0 %               F