h$,f^ ddlmZmZddlZddlZddlmZddlm 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*e*egegegegd d dzdd d Z+GddZ,Gdde Z-ej\j_deegdZ0ej\j_deegdZ1dZ2ej\j_deedddeedddgdZ3ej\j_dedddde4d fedd!d"de4d#fedddde5d$feddd%de5d&fedd'dde5d(fgd)Z6d*Z7d+Z8ej\j_d,e9d-fed-fed.fejtd/fgd0Z;d1Zd?d"feed9d:d"eed>d?dfeedddeedddfeedddeedddfgd@Z>ej\j_d2eee?ee"e$e%eeeddd"gdAZ@ej\j_d2eeee?e"e$e%ee&eed 4ehd3eehdBeeddd"eedd?d"eeddd"eeddd"eeddd"eedddeeddd%ed5e egdCZAej\j_dDeeddddEfeeddd"dFfehd3dGfeeBejejhejfeEdHfdIdJddgd8d;ggfdJejddgd8d;ggfdKeddgd8d;ggfdLdMejjdfdNe,e,fe9dfedOfdPdQdRejfed'fedSfeddTzfedfeeJdRfeejfedUfed5e-dVfdWgdXZKej\j_dYeedddefedZd[hefeedEd\hefdJefdKe%fdMe$fde"feEefe9efd]efd^e&fed 4efed5efd_efdRe!fgd`ZLdaZMdbZNdcZOddZPdeZQdfZRdgZSdhZTdiZUdjZVdkZWdlZXdmZYdnZZdoZ[dpZ\dqZ]drZ^dsZ_ej\j_dtd dugdvZ`ej\j_dwgdxdyZay){)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)abcdTprefer_skip_nested_validation)r(cy)z/A function to test the validation of functions.N)r%r&r'r(argskwargss Ilib/python3.12/site-packages/sklearn/utils/tests/test_param_validation.py_funcr0(cpeZdZdZedegiddZeedegiddZy)_ClasszJA class to test the _InstancesOf constraint and the validation of methods.r%Tr)cy)zA validated methodNr,selfr%s r/_methodz_Class._method3r1r2cy)zA deprecated validated methodNr,r6s r/_deprecated_methodz_Class._deprecated_method7r1r2N) __name__ __module__ __qualname____doc__r$rr8r r:r,r2r/r4r40sQTcD6]$G!H!\cD6]$G,H,r2r4cLeZdZUdZdegiZeed<dZe dd dZ y) _EstimatorzrrAdict__annotations__rEr rIr,r2r/r@r@=s4F$'$=D05 6 r2r@ interval_typect|ddd}d|vsJd|vsJt|ddd}d|vsJd|vsJt|ddd}d|vsJd|vsJt|ddd}d|vsJd|vsJy) z.Check the range of values depending on closed.leftclosedrightbothneitherN)rrNintervals r/test_interval_rangerZJs r1V> H   r1W=H X   == r1V> == r1Y?H X   H  r2c0t|ddd}ddz|vsJddz|vsJtddz|vsJtddz|vsJt|dddzd}ddz|vsJddz|vsJtddz|vsJtddz|vsJddz|vsJy) zbCheck that Interval constraint work with large integers. non-regression test for #26648. rrQrWrSAiN)rfloatrXs r/test_interval_large_integersr_^s  q!I>H b5  c6 !! ! BExpecting right to be an int for an interval over the integersz&left can't be None when closed == leftrUz(right can't be None when closed == rightzright can't be less than leftcrtj||5tdi|dddy#1swYyxYw)zNCheck that informative errors are raised for invalid combination of parametersmatchNr,)pytestraisesr)paramserrorrns r/test_interval_errorsrss3@ uE *6s -6cthddh}|jdsJ|jdsJ|jdrJdt|vsJy)z*Sanity check for the StrOptions constraint>r%r&r'r'r r%r(z'c' (deprecated)N)ris_satisfied_bystroptionss r/test_stroptionsrys^cU;G  " "3 '' '  " "3 '' '&&s++ + W -- -r2cttddtjhdh}|j dsJ|j tjsJ|j drJdt |vsJy)z'Sanity check for the Options constraintg?r Gz?z-0.5 (deprecated)N)rrrarbrurvrws r/ test_optionsr}sodT3/TFCG  " "4 (( (  " "266 ** *&&t,, , #g, .. .r2ztype, expected_type_nameintr^z numpy.ndarraycBt|}t|d|dk(sJy)z?Check the string representation of the _InstancesOf constraint.zan instance of ''N)rrv)rjexpected_type_name constraints r/%test_instances_of_type_human_readablers,d#J z? 01C0DAF FF Fr2ctddg}Gdd}Gdd}|j|sJ|j|rJt|dk(sJy) z Check the HasMethods constraint.r%r&ceZdZdZdZy)test_hasmethods.._GoodcyrCr,r7s r/r%z test_hasmethods.._Good.a r2cyrCr,rs r/r&z test_hasmethods.._Good.brr2N)r;r<r=r%r&r,r2r/_Goodrs   r2rceZdZdZy)test_hasmethods.._BadcyrCr,rs r/r%ztest_hasmethods.._Bad.arr2N)r;r<r=r%r,r2r/_Badrs r2rz"an object implementing 'a' and 'b'N)r rurv)rrrs r/test_hasmethodsrseS#J'J  % %eg .. .))$&11 1 z?B BB Br2r>r%r&r') numeric_onlyrIc@t|}|j|rJy)z>Check that the value generated does not satisfy the constraintNr!ru)r bad_values r/test_generate_invalid_param_valrs%"+:6I)))44 44r2zinteger_interval, real_intervali ct|}|j|rJ|j|rJt|}|j|rJ|j|rJy)ztCheck that the value generated for an interval constraint does not satisfy any of the interval constraints. rNr)integer_interval real_intervalrs r/+test_generate_invalid_param_val_2_intervalsrsmb+mDI,,Y77 7// :: :*6FGI,,Y77 7// :: ::r2cvtjt5t|dddy#1swYyxYw)zmCheck that the function raises NotImplementedError when there's no invalid value for the constraint. N)rorpNotImplementedErrorr!rs r/)test_generate_invalid_param_val_all_validrVs." * +/":.///s /8>rerQrc@t|}|j|sJy)z;Check that the value generated does satisfy the constraint.N)r"ru)rvalues r/test_generate_valid_paramrks": ! ,E  % %e ,, ,r2zconstraint_declaration, valuegzG?*r&c |dzS)Nrer,)xs r/rs QUr2rKz array-likez sparse matrix) random_staterr)rNr{)booleanF)verbosererggimissingrD) cv_objectrc@t|}|j|sJy)z+Sanity check for the is_satisfied_by methodN)r#ru)constraint_declarationrrs r/test_is_satisfied_byrs$D!!78J  % %e ,, ,r2z1constraint_declaration, expected_constraint_classoption1option2r|rrrc:t|}|j|usJy)zICheck that make_constraint dispatches to the appropriate constraint classN)r# __class__)rexpected_constraint_classrs r/test_make_constraintrs$,!!78J   #< << 201222s 1:ctjtd5tdddddtjtd5tddgddidddtjtd 5tdiddidddtjtd 5tddd dddtjtd5td ggd ddiddidddtjtd 5td ggdddiddidddy#1swYxYw#1swYxYw#1swYxYw#1swYxYw#1swYtxYw#1swYyxYw)zGCheck that validate_params works no matter how the arguments are passedz"The 'a' parameter of _func must bermwrongrer'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)rrQrrerrerQrfour)re)rorprr0r,r2r/test_validate_paramsrss %I  g %I " 7|!q!" %I # "C>"# %I ! a1 ! %I 4 a3/3Q33(34 %I 3 a2)2v2#q2335 "" ## !! 44 33sGD9 E>E/E!E*E69EEEE'*E36E?cJtdtgidd}|ddy)zTCheck that no error is raised when there are parameters without constraints r%Tr)cyrCr,)r%r&s r/funcz1test_validate_params_missing_params..funcrJr2rerQN)r$r~)rs r/#test_validate_params_missing_paramsrs/ cC5\F G  AJr2cxtt}tjtd5|ddddddtjtd5tj t d5|ddd ddddddy#1swY]xYw#1swYxYw#1swYyxYw) z5Check that validate_params functions can be decoratedzFunction _func is deprecatedrmrerQrrNrr)r r0rowarns FutureWarningrpr)decorated_functions r/ test_decorate_validated_functionrs%e, m+I J&1a1%& m+I J0 ]] !)N  0 q!w / 000 &&  0 000s/ BB0; B$B0B!$B- )B00B9ctjtd5tj ddddtj t d5tjtd5tjdddddddy#1swYkxYw#1swYxYw#1swYyxYw)z-Check that validate_params works with methodsz+The 'a' parameter of _Class._method must bermrNz)Function _deprecated_method is deprecatedz6The 'a' parameter of _Class._deprecated_method must be)rorprr4r8rrr:r,r2r/test_validate_params_methodrs %R " !" m+V W1 ]] !J  1 H ' ' 0  111 "" 1 111s/B B85B,B8 B),B5 1B88Cctd}tjtd5|j dddy#1swYyxYw)z9Check that validate_params works with Estimator instancesrz'The 'a' parameter of _Estimator must bermN)r@rorprrI)ests r/test_validate_params_estimatorrs@ W C %N   s AA ctjtd5thdddhdddy#1swYyxYw)z@Check that the deprecated parameter must be a subset of options.z#deprecated options must be a subsetrm>r%r&r'r%r(r N)rorprrr,r2r/!test_stroptions_deprecated_subsetr*s4 z)N O;?Sz:;;;s 7Ac4tdtttgidd}|dddd|gd t j t d 5}|d dddtj}d|vsJd|vsJy#1swY+xYw)zECheck that internal constraints are not exposed in the error message.paramTr)cyrCr,rs r/fz!test_hidden_constraint..f3s r2rerQr)r%r&r'rThe 'param' parameterrmbadrNzan instance of 'dict'zan instance of 'list') r$rlistrLrorprrvrrexc_infoerr_msgs r/test_hidden_constraintr0s 6$<&'t  AAAiL %<    (..!G "g -- - "' 11 1s  BBc @tdtdhttdhgidd}|d|dtjt d5}|d d d d t j}d|vsJd|vsJy #1swY+xYw) zBCheck that we can have 2 StrOptions constraints, one being hidden.rautowarnTr)cyrCr,rs r/rz!test_hidden_stroptions..fK r2rrmrrN)r$rrrorprrvrrs r/test_hidden_stroptionsrHs :vh' F80D)EFG&*  fIfI %<    (..!G W     s BBchttdsJttjdsJy)zCheck that the validate_params decorator properly sets the parameter constraints as attribute of the decorated function/method. _skl_parameter_constraintsN)hasattrr0r4r8r,r2r/4test_validate_params_set_param_constraints_attributeras. 56 77 7 68##%A BB Br2cvtddgidd}|d|tjdy)zCheck that validate_params raise a deprecation message but still passes validation when using an int for a parameter accepting a boolean. rrTr)cyrCr,rs r/rz1test_boolean_constraint_deprecated_int..fnrJr2FN)r$rabool_rs r/&test_boolean_constraint_deprecated_intris> g {+4P Q dGbhhuor2cttdgddddd}tjtd5|d dddGd d }|| || y#1swY+xYw)z5Check that validation can be skipped for a parameter.N no_validationparam1param2Tr)cyrCr,rs r/rztest_no_validation..fzrr2zThe 'param1' parameterrmr)rc eZdZy)$test_no_validation..SomeTypeN)r;r<r=r,r2r/SomeTypers r2r)rrKr$r~rorpr)rrs r/test_no_validationrwsw;/:&*   ,4L M   XXZs A--A6ctjd}t}|j|jsJ|jt j gdrJy)z9Add a specific test for checking support for `pandas.NA`.pandasrN)ro importorskiprruNAraarray)pd na_constraints r/$test_pandas_na_constraint_with_pd_narsR   X &B')M  ( ( // /,,RXXi-@AA AAr2ct}|jgdsJ|jtdsJ|jdrJy)zGCheck that a string does not satisfy the _IterableNotString constraint.rrz some stringN)rrurangers r/test_iterable_not_stringrsM$&J  % %i 00 0  % %eBi 00 0))-88 88r2ct}|jdsJ|jtsJ|jddgddgfddgddgfgsJ|jdsJ|jdrJy)zUCheck that the _CVObjects constraint accepts all current ways to pass cv objects.rrerQrrNznot a CV object)rrur rs r/test_cv_objectsrsJ  % %a (( (  % %km 44 4  % %AA'71a&1a&9I&J KK K  % %d ++ +))*;<< <.ThirdPartyEstimatorc4||_t| dy)NrrD)r&superrE)r7r&rs r/rEz@test_third_party_estimator..ThirdPartyEstimator.__init__sDF G q  !r2c&t|||yrC)r rI)r7rGrHrs r/rIz;test_third_party_estimator..ThirdPartyEstimator.fits GK1 r2rK)r;r<r=rErI __classcell__)rs@r/ThirdPartyEstimatorr s "  r2rrr&N)r@rI)rs r/test_third_party_estimatorrs! j!  "r2cvttddd}|jdsJ|jdrJy)z9Check for the type RealNotInt in the Interval constraint.rrerVrSriN)rrrurs r/test_interval_real_not_intrs=*a6:J  % %c ** *))!,, ,,r2ctdtsJtdtrJttjdtsJttjdtrJy)zCheck for the RealNotInt type.rireN) isinstancerrafloat64int64r,r2r/test_real_not_intrsU c: && &!Z(( ( bjjmZ 00 0"((1+z22 22r2ctdtgidd}tjtd5|dd d d t d 5|dd d d y #1swY(xYw#1swYy xYw) z@Check that param validation can be skipped using config_context.r%Tr)cyrCr,rDs r/rz%test_skip_param_validation..frJr2The 'a' parameterrm1rDNskip_parameter_validation)r$r~rorprrrs r/test_skip_param_validationrs}cC5\F G  ,4G H C $ 7 C s A' A3'A03A<r*Fcntdtgiddtdtgi|fd}tjtd5|d d d d |r |d y tjtd 5|d d d d y #1swYCxYw#1swYy xYw)z,Check that nested validation can be skipped.r%Tr)cyrCr,rDs r/rz&test_skip_nested_validation..frJr2r&cdS)Ninvalid_param_valuerDr,)r&rs r/gz&test_skip_nested_validation..gs ())r2zThe 'b' parameterrmr#rNrerr)r*r$rs @r/test_skip_nested_validationr%scC5\F G  se &C* * ,4G H# !"#% A ]]08K L  F   ##   s B B+B(+B4zJskip_parameter_validation, prefer_skip_nested_validation, expected_skipped))TTT)TFT)FTT)FFFctdtgi|d}t|5|d}ddd|k(sJy#1swYxYw)z5Check interaction between global skip and local skip.r%r)ctdS)Nr)rrDs r/r$z9test_skip_nested_validation_and_config_context..gs|788r2rreN)r$r~r)rr*expected_skippedr$actual_skippeds r/.test_skip_nested_validation_and_config_contextr*sa se 4Q99 2K L1 - -- -s AA )r)bnumbersrrnumpyraro scipy.sparsersklearn._configrr sklearn.baserr sklearn.model_selectionr sklearn.utilsr sklearn.utils._param_validationr rrrrrrrrrrrrrrrrrrr r!r"r#r$r0r4r@mark parametrizerZr_rdrh TypeErrorrrsryr}r~ndarrayrrrrrrrrjfloat32rcallablerrandom RandomStatergr^rrrrrrrrrrrrrrrrrrrrrr%r*r,r2r/r;s " #64/$: &vTF$8"&:a: : , ,    8T*:;<&8T*:;#<#( #  dAq((4tF*ST" " sQ& I  L q3) L  L ta6 J  5 q47 K  7 q2 H  , +<=< ./ e 5 w _% GG C(tQv.q$v.tT)4?#4(5  55 % XtQw 7 ZQv 6 XtQw 7 ZQy 9 XtQw 7 ZAf 5 XtQw 7 ZD 8 XtQw 7 ZD ; Xq$v 6 ZQv 6 Xq$v 6 ZQy 9 Xq$v 6 ZAf 5 Xq$v 6 Zr& 9 Xq$v 6 Zr) < Xr1V 4 Zq 9 Xr1V 4 ZD 8 Xr1V 4 Zb ; Xr1V 4 Zb 9 Xr1Y 7 ZBy 9 Xr1Y 7 ZBv 6 Xq$v 6 ZAf 5 Xq$v 6 ZAf 5 KILZ ;[LZ ; T  4i8  / /  T 4(?#)$4i81b31d95495q!I.q$v.tQw/5 -6-76- # $1V ,d3 (AtI 6; O $c*  BJJ/ 0"**= ?# AA'( xrxx!Q!Q 012 *q!fq!f%567 ..q12  a s   " $ !T'" $ %,' "&&! )$ E JO,7@-A@- 7 $1V ,h7 Y * +Z8 d| $g. {# /* '  : l I N# D )=9 E J' j! (=)(= 2 3D  0 1 ; 20!2C ,B9=#&-3 84-HI4P..r2