a h5_@sddlmZmZddlZddlZddlmZddlm Z m Z ddl m Z m Z ddlmZddlmZddlmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*dd l+m,Z,e*egegegegd d d ddd ddZ-GdddZ.Gddde Z/ej01deegddZ2ej01deegddZ3ddZ4ej01deeddddeeddddgd d!Z5ej01d"ed#d$dd%e6d&fedd'd(d%e6d)fedddd%e7d*feddd+d%e7d,fedd-dd%e7d.fgd/d0Z8d1d2Z9d3d4Z:ej01d5e;d6fed6fed7fejej01d=eeddddeeddddeeddd(dehd>eed d?e&ed@e eg dAdBZ?ej01dCeeddDd+deedEdFddfeeddDd+deedEdFd(dfeeddDd+deedGdFddfeeddDd+deedFdddfeeddDd+deedGdd(dfeedDdddeedEdFddfeedDdddeedEdFd(dfeedDdddeedd$ddfeedDdddeeddEddfeedDdddeeddHd(dfeedEdFddeeddd+dfeedEdFddeeddddfeedEdFddeedIdHd(dfeedEdFddeedIdHd+dfeedEdFd(deedJdKd(dfeedEdFd(deedJdKddfeed$dddeeddddfeeddddeeddddfgdLdMZ@ej01d=eeeAee"e$e%eeeddd(dgdNdOZBej01d=eeeeAe"e$e%ee&eed d?ehd>eehdPeeddd(deeddKd(deeddd(deeddd(deeddd(deeddddeeddd+ded@e egdQdRZCej01dSeedddddTfeeddd(ddUfehd>dVfeeDejEejFhejFfeGdWdXfdYdZdd$gdDdGggfdZeHdd$gdDdGggfd[edd$gdDdGggfg d\d]e,Dd^d_ejIJdfd`e.e.fe;dfedafdbdcddejKfed-fedefed$dffedfeeLddfeejKfedgfed@e/ddhfdidjdkZMej01dleeddddefedmdnhefeedTdohefdZefd[e%fd_e$fde"feGefe;efdpefdqe&fed d?efed@efdrefdde!fgdsdtZNdudvZOdwdxZPdydzZQd{d|ZRd}d~ZSddZTddZUddZVddZWddZXddZYddZZddZ[ddZ\ddZ]ddZ^ddZ_ddZ`ddZaej01dd dgddZbej01dgdddZcdS))IntegralRealN) csr_matrix)config_context get_config) BaseEstimator _fit_context) LeaveOneOut deprecated) HasMethodsHiddenIntervalInvalidParameterError MissingValuesOptions RealNotInt StrOptions _ArrayLikes _Booleans _Callables _CVObjects _InstancesOf_IterablesNotString_NanConstraint_NoneConstraint_PandasNAConstraint _RandomStates_SparseMatrices_VerboseHelpergenerate_invalid_param_valgenerate_valid_parammake_constraintvalidate_params)CSR_CONTAINERS)abcdTprefer_skip_nested_validation)r(cOsdS)z/A function to test the validation of functions.N)r%r&r'r(argskwargsr+r+e/var/www/html/assistant/venv/lib/python3.9/site-packages/sklearn/utils/tests/test_param_validation.py_func)sr/c@sJeZdZdZedegiddddZeedegiddddZd S) _ClasszJA class to test the _InstancesOf constraint and the validation of methods.r%Tr)cCsdS)zA validated methodNr+selfr%r+r+r._method4sz_Class._methodcCsdS)zA deprecated validated methodNr+r1r+r+r._deprecated_method8sz_Class._deprecated_methodN) __name__ __module__ __qualname____doc__r#rr3r r4r+r+r+r.r01s  r0c@s@eZdZUdZdegiZeed<ddZe ddd d d Z dS) _Estimatorzs r9 interval_typecCst|dddd}d|vsJd|vs(Jt|dddd}d|vsDJd|vsPJt|dddd}d|vslJd|vsxJt|dddd}d|vsJd|vsJdS) z.Check the range of values depending on closed.leftclosedrightbothneitherN)rrDintervalr+r+r.test_interval_rangeKs       rOcCst|dddd}dd|vs Jdd|vs0Jtdd|vsDJtdd|vsXJt|ddddd}dd|vs|Jdd|vsJtdd|vsJtdd|vsJdd|vsJdS) zbCheck that Interval constraint work with large integers. non-regression test for #26648. rrFrLrHAiN)rfloatrMr+r+r.test_interval_large_integers_srScCspttdddd}tj|vsJttdddd}tj |vs>Jttdddd}tj|vs\Jtj |vslJdS)zjCheck that inf is included iff a bound is closed and set to None. Only valid for real intervals. rNrJrHrGrL)rrnpinfrNr+r+r.test_interval_inf_in_boundstsrWrNrGrHrKcCstj|vsJdS)z)Check that np.nan is not in any interval.N)rTnanrVr+r+r.test_nan_not_in_intervalsrZzparams, error, match?rF)typerGrJrIz=Expecting left to be an int for an interval over the integersg@rLz>Expecting right to be an int for an interval over the integersz&left can't be None when closed == leftrJz(right can't be None when closed == rightzright can't be less than leftcCs@tj||dtfi|Wdn1s20YdS)zNCheck that informative errors are raised for invalid combination of parametersmatchN)pytestraisesr)paramserrorr_r+r+r.test_interval_errorss rdcCsPthddhd}|ds J|ds.J|dr._GoodcSsdSr;r+r2r+r+r.r%sz test_hasmethods.._Good.acSsdSr;r+rrr+r+r.r&sz test_hasmethods.._Good.bN)r5r6r7r%r&r+r+r+r._Goodsrsc@seZdZddZdS)ztest_hasmethods.._BadcSsdSr;r+rrr+r+r.r%sztest_hasmethods.._Bad.aN)r5r6r7r%r+r+r+r._Badsrtz"an object implementing 'a' and 'b'N)r rfrg)rprsrtr+r+r.test_hasmethodss  rurpre)Z numeric_onlyrAcCst|}||rJdS)z>Check that the value generated does not satisfy the constraintNr rf)rp bad_valuer+r+r.test_generate_invalid_param_valsrxzinteger_interval, real_intervali cCsPt|d}||rJ||r&Jt|d}||r>J||rLJdS)ztCheck that the value generated for an interval constraint does not satisfy any of the interval constraints. rpNrv)Zinteger_intervalZ real_intervalrwr+r+r.+test_generate_invalid_param_val_2_intervalss Q  rcCs6ttt|Wdn1s(0YdS)zmCheck that the function raises NotImplementedError when there's no invalid value for the constraint. N)r`raNotImplementedErrorr rr+r+r.)test_generate_invalid_param_val_all_validWs r>rXrFrycCst|}||sJdS)z;Check that the value generated does satisfy the constraint.N)r!rf)rpvaluer+r+r.test_generate_valid_paramlsrzconstraint_declaration, valuegzG?*r&cCs|dS)NrXr+)xr+r+r.r)NNz array-like sparse matrixcCs$g|]}d|ddgddggfqS)rrXrFryr|r+).0 containerr+r+r. sr) random_staterr)rNrk)booleanF)verboserXrYgimissingr<) cv_objectr{cCst|}||sJdS)z+Sanity check for the is_satisfied_by methodN)r"rf)constraint_declarationrrpr+r+r.test_is_satisfied_bys&rz1constraint_declaration, expected_constraint_classZoption1Zoption2rlrrrcCst|}|j|usJdS)zICheck that make_constraint dispatches to the appropriate constraint classN)r" __class__)rZexpected_constraint_classrpr+r+r.test_make_constraintsrcCs:tjtddtdWdn1s,0YdS)zNCheck that an informative error is raised when an unknown constraint is passedzUnknown constraintr^znot a valid constraintN)r`ra ValueErrorr"r+r+r+r.test_make_constraint_unknownsrcCstjtddtdddWdn1s00Ytjtdd"tddgddiWdn1sp0Ytjtd d"tdiddiWdn1s0Ytjtd dtdddd Wdn1s0Ytjtdd2td ggd RddiddiWdn1s>0Ytjtd d2td ggdRddiddiWdn1s0YdS)zGCheck that validate_params works no matter how the arguments are passedz"The 'a' parameter of _func must ber^wrongrXr'Nz"The 'b' parameter of _func must ber'"The 'c' parameter of _func must bez"The 'd' parameter of _func must be)r'r(r)rrFryr|er{rXrFryfour)rX)r`rarr/r+r+r+r.test_validate_paramss0*00,BrcCs(tdtgidddd}|dddS) zTCheck that no error is raised when there are parameters without constraints r%Tr)cSsdSr;r+)r%r&r+r+r.funcsz1test_validate_params_missing_params..funcrXrFN)r#rn)rr+r+r.#test_validate_params_missing_paramss rc Cstt}tjtdd|ddddWdn1s<0YtjtddLtjtdd|ddd dWdn1s0YWdn1s0YdS) z5Check that validate_params functions can be decoratedzFunction _func is deprecatedr^rXrFryrNrr)r r/r`warns FutureWarningrar)Zdecorated_functionr+r+r. test_decorate_validated_functions ,rc CstjtddtdWdn1s00YtjtddJtjtddtdWdn1sz0YWdn1s0YdS)z-Check that validate_params works with methodsz+The 'a' parameter of _Class._method must ber^rNz)Function _deprecated_method is deprecatedz6The 'a' parameter of _Class._deprecated_method must be)r`rarr0r3rrr4r+r+r+r.test_validate_params_methods*rcCsBtd}tjtdd|Wdn1s40YdS)z9Check that validate_params works with Estimator instancesrz'The 'a' parameter of _Estimator must ber^N)r9r`rarrA)Zestr+r+r.test_validate_params_estimator$s rcCsFtjtdd$thdddhdWdn1s80YdS)z@Check that the deprecated parameter must be a subset of options.z#deprecated options must be a subsetr^rer%r(r N)r`rarrr+r+r+r.!test_stroptions_deprecated_subset/srcCstdtttgidddd}|dddd |gd tjtd d }|d dWdn1sj0Yt|j}d|vsJd|vsJdS)zECheck that internal constraints are not exposed in the error message.paramTr)cSsdSr;r+rr+r+r.f8sz!test_hidden_constraint..frXrFry)r%r&r'rThe 'param' parameterr^badrNzan instance of 'dict'zan instance of 'list') r#r listrBr`rarrgrrexc_infoerr_msgr+r+r.test_hidden_constraint5s  (  rcCstdtdhttdhgidddd}|d|dtjtdd }|d d Wd n1sj0Yt|j}d|vsJd|vsJd S) zBCheck that we can have 2 StrOptions constraints, one being hidden.rautowarnTr)cSsdSr;r+rr+r+r.rPsz!test_hidden_stroptions..frr^rrN)r#rr r`rarrgrrr+r+r.test_hidden_stroptionsMs (  rcCs$ttdsJttjds JdS)zCheck that the validate_params decorator properly sets the parameter constraints as attribute of the decorated function/method. Z_skl_parameter_constraintsN)hasattrr/r0r3r+r+r+r.4test_validate_params_set_param_constraints_attributefsrcCs4tddgidddd}|d|tddS) zCheck that validate_params raise a deprecation message but still passes validation when using an int for a parameter accepting a boolean. rrTr)cSsdSr;r+rr+r+r.rssz1test_boolean_constraint_deprecated_int..fFN)r#rTZbool_rr+r+r.&test_boolean_constraint_deprecated_intns rcCsttdgddddddd}tjtdd |d d Wdn1sN0YGd d d }||d||ddS)z5Check that validation can be skipped for a parameter.NZ no_validationparam1param2Tr)cSsdSr;r+rr+r+r.rsztest_no_validation..fzThe 'param1' parameterr^r)rc@s eZdZdS)z$test_no_validation..SomeTypeN)r5r6r7r+r+r+r.SomeTypesr)r)NNr#rnr`rar)rrr+r+r.test_no_validation|s  ( rcCs<td}t}||js J|tgdr8JdS)z9Add a specific test for checking support for `pandas.NA`.ZpandasrN)r`Z importorskiprrfZNArTarray)pdZ na_constraintr+r+r.$test_pandas_na_constraint_with_pd_nas rcCs<t}|gdsJ|tds*J|dr8JdS)zGCheck that a string does not satisfy the _IterableNotString constraint.rrz some stringN)rrfrangerr+r+r.test_iterable_not_stringsrcCsnt}|dsJ|ts$J|ddgddgfddgddgfgsNJ|ds\J|drjJdS)zUCheck that the _CVObjects constraint accepts all current ways to pass cv objects.r{rXrFryr|Nznot a CV object)rrfr rr+r+r.test_cv_objectss *rcCs"Gdddt}|dddS)zCheck that the validation from a scikit-learn estimator inherited by a third party estimator does not impose a match between the dict of constraints and the parameters of the estimator. cs*eZdZfddZdfdd ZZS)z7test_third_party_estimator..ThirdPartyEstimatorcs||_tjdddS)Nrr<)r&superr=)r2r&rr+r.r=sz@test_third_party_estimator..ThirdPartyEstimator.__init__Ncst||dSr;)rrAr>rr+r.rAsz;test_third_party_estimator..ThirdPartyEstimator.fit)NN)r5r6r7r=rA __classcell__r+r+rr.ThirdPartyEstimators rrr&N)r9rA)rr+r+r.test_third_party_estimators rcCs0ttdddd}|dsJ|dr,JdS)z9Check for the type RealNotInt in the Interval constraint.rrXrKrHr[N)rrrfrr+r+r.test_interval_real_not_intsrcCsHtdtsJtdtrJttdts0JttdtrDJdS)zCheck for the RealNotInt type.r[rXN) isinstancerrTfloat64Zint64r+r+r+r.test_real_not_intsrcCstdtgidddd}tjtdd|dd Wd n1sH0Ytdd |dd Wd n1s|0Yd S) z@Check that param validation can be skipped using config_context.r%Tr)cSsdSr;r+r<r+r+r.rsz%test_skip_param_validation..fThe 'a' parameterr^1r<Nskip_parameter_validation)r#rnr`rarrrr+r+r.test_skip_param_validations  ( rr*Fcstdtgiddddtdtgi|dfdd}tjtd d |d d Wd n1sf0Y|r|dd n8tjtdd |dd Wd n1s0Yd S)z,Check that nested validation can be skipped.r%Tr)cSsdSr;r+r<r+r+r.rsz&test_skip_nested_validation..fr&cs ddS)Ninvalid_param_valuer<r+rrr+r.gsz&test_skip_nested_validation..gzThe 'b' parameterr^rrNrXrr)r*rr+rr.test_skip_nested_validations ( rzJskip_parameter_validation, prefer_skip_nested_validation, expected_skipped))TTT)TFT)FTT)FFFcCs\tdtgi|ddd}t|d|d}Wdn1sB0Y||ksXJdS)z5Check interaction between global skip and local skip.r%r)cSs tdS)Nr)rr<r+r+r.rsz9test_skip_nested_validation_and_config_context..grrXN)r#rnr)rr*Zexpected_skippedrZactual_skippedr+r+r..test_skip_nested_validation_and_config_contexts   &r)r)dnumbersrrnumpyrTr`Z scipy.sparserZsklearn._configrrZ sklearn.baserrZsklearn.model_selectionr Z sklearn.utilsr Zsklearn.utils._param_validationr r rrrrrrrrrrrrrrrrrrr r!r"r#Zsklearn.utils.fixesr$r/r0r9markZ parametrizerOrSrWrZ TypeErrorrrdrjrmrnZndarrayrqrurxrrrrr\Zfloat32rcallablerrandomZ RandomStaterYrRrrrrrrrrrrrrrrrrrrrrrrrr+r+r+r.s*   h                 M          $    "