-iW_ SSKJrJr SSKrSSKrSSKJr SSKJ r J r SSK J r J r SSKJr SSKJr SSKJrJrJrJrJrJrJrJrJrJrJrJrJrJ r J!r!J"r"J#r#J$r$J%r%J&r&J'r'J(r(J)r)J*r* SS K+J,r, \*"\/\/\/\/S .S S 9S{SS .Sjj5r-"SS5r."SS\ 5r/\R`RcS\\/5S5r2\R`RcS\\/5S5r3Sr4\R`RcS\"\SSSS9\"\SSSS9/5S5r5\R`RcS\SSSS .\6S!4\SS"S#S .\6S$4\SSSS .\7S%4\SSS&S .\7S'4\SS(SS .\7S)4/5S*5r8S+r9S,r:\R`RcS-\;S.4\S.4\S/4\RxS04/5S15r=S2r>\R`RcS3\"\SSSS9\"\SSSS9\"\SSS#S9\"1S4k5\"5\"S S59\&"5\"S65\ "5\"5/ 5S75r?\R`RcS8\"\SS9S&S9\"\S:S;SS94\"\SS9S&S9\"\S:S;S#S94\"\SS9S&S9\"\SS=S#S94\"\S:S;SS9\"\S>S=S&S94\"\S:S;S#S9\"\S?S@S#S94\"\S:S;S#S9\"\S?S@SS94\"\SSSS9\"\SSSS94\"\SSSS9\"\SSSS94/5SA5r@\R`RcS3\"5\"\A5\"5\""5\$"5\%"5\"5\"\SSS#S9/5SB5rB\R`RcS3\"5\"5\"\A5\""5\$"5\%"5\"5\&"5\"5\"S S59\"1S4k5\"\1SCk5\"\SSS#S9\"\SS@S#S9\"\SSS#S9\"\SSS#S9\"\SSS#S9\"\SSSS9\"\SSS&S9\"S65\ "5\"5/5SD5rC\R`RcSE\"\SSSS9SF4\"\SSS#S9SG4\"1S4k5SH4\"\D\R\R15\R4\GSI4SJSKSS/S9Szs4F$'$=D05 6 r3rF interval_typec[USSSS9nSU;deSU;de[USSSS9nSU;deSU;de[USSSS9nSU;deSU;de[USSSS9nSU;deSU;deg) z.Check the range of values depending on closed.leftclosedrightbothneitherN)rrWintervals r0test_interval_rangercKs r1V> H   r1W=H X   == r1V> == r1Y?H X   H  r3cH[USSSS9nSS-U;deSS-U;de[SS-5U;de[SS-5U;de[USSS-SS9nSS-U;deSS-U;de[SS-5U;de[SS-5U;deSS-U;deg) zZCheck that Interval constraint work with large integers. non-regression test for #26648. rrZr`r\AiN)rfloatras r0test_interval_large_integersrh_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 == leftr^z(right can't be None when closed == rightzright can't be less than leftcv[R"XS9 [S0UD6 SSS5 g!,(df  g=f)zNCheck that informative errors are raised for invalid combination of parametersmatchNr-)pytestraisesr)paramserrorrws r0test_interval_errorsr|s(@ u *6 + * *s * 8c[1SkS1S9nURS5(deURS5(deURS5(aeS[U5;deg)z*Sanity check for the StrOptions constraintr&r'r(r(r r&r)z'c' (deprecated)N)ris_satisfied_bystroptionss r0test_stroptionsrsgcU;G  " "3 ' '' '  " "3 ' '' '&&s+++ + W -- -r3c[[SS[R1S1S9nUR S5(deUR [R5(deUR S5(aeS[ U5;deg)z'Sanity check for the Options constraintg?r Gz?z-0.5 (deprecated)N)rrrjrkrrrs r0 test_optionsrsxdT3/TFCG  " "4 ( (( (  " "266 * ** *&&t,,, , #g, .. .r3ztype, expected_type_nameintrgz numpy.ndarraycD[U5n[U5SUS3:Xdeg)z?Check the string representation of the _InstancesOf constraint.zan instance of ''N)rr)rsexpected_type_name constraints r0%test_instances_of_type_human_readablers,d#J z? 01C0DAF FF Fr3c[SS/5n"SS5n"SS5nURU"55(deURU"55(ae[U5S:Xdeg) z Check the HasMethods constraint.r&r'c \rSrSrSrSrSrg)test_hasmethods.._GoodcgrJr-r9s r0r& test_hasmethods.._Good.a r3cgrJr-rs r0r' test_hasmethods.._Good.brr3r-N)r?r@rArBr&r'rDr-r3r0_Goodrs   r3rc\rSrSrSrSrg)test_hasmethods.._BadcgrJr-rs r0r&test_hasmethods.._Bad.arr3r-N)r?r@rArBr&rDr-r3r0_Badrs r3rz"an object implementing 'a' and 'b'N)r rr)rrrs r0test_hasmethodsrskS#J'J  % %eg . .. .))$&111 1 z?B BB Br3rr~) numeric_onlyrQcJ[U5nURU5(aeg)z>Check that the value generated does not satisfy the constraintNr!r)r bad_values r0test_generate_invalid_param_valrs("+:6I)))444 44r3zinteger_interval, real_intervali c[US9nURU5(aeURU5(ae[US9nURU5(aeURU5(aeg)zlCheck that the value generated for an interval constraint does not satisfy any of the interval constraints. rNr)integer_interval real_intervalrs r0+test_generate_invalid_param_val_2_intervalsrsyb+mDI,,Y777 7// ::: :*6FGI,,Y777 7// ::: ::r3c[R"[5 [U5 SSS5 g!,(df  g=f)zeCheck that the function raises NotImplementedError when there's no invalid value for the constraint. N)rxryNotImplementedErrorr!rs r0)test_generate_invalid_param_val_all_validrWs%" * +":. , + +s 0 >>rnrZrcJ[U5nURU5(deg)z;Check that the value generated does satisfy the constraint.N)r"r)rvalues r0test_generate_valid_paramrls%: ! ,E  % %e , ,, ,r3zconstraint_declaration, valuegzG?*r'c US-$)Nrnr-)xs r0rsQUr3rTz array-likez sparse matrix) random_staterr)rNr)booleanF)verbosernrpgimissingrK) cv_objectrcJ[U5nURU5(deg)z+Sanity check for the is_satisfied_by methodN)r#r)constraint_declarationrrs r0test_is_satisfied_byrs'L!!78J  % %e , ,, ,r3z1constraint_declaration, expected_constraint_classoption1option2rrrrc<[U5nURULdeg)zICheck that make_constraint dispatches to the appropriate constraint classN)r# __class__)rexpected_constraint_classrs r0test_make_constraintrs$.!!78J   #< << 01 ? > >s / =c[R"[SS9 [SSS9 SSS5 [R"[SS9 [SS/SS06 SSS5 [R"[S S9 [S0SS0D6 SSS5 [R"[S S9 [SSSS 9 SSS5 [R"[SS9 [S //S QQ7SS0SS0D6 SSS5 [R"[S S9 [S //SQQ7SS0SS0D6 SSS5 g!,(df  GN =f!,(df  N=f!,(df  N=f!,(df  N=f!,(df  N=f!,(df  g=f)zGCheck that validate_params works no matter how the arguments are passedz"The 'a' parameter of _func must bervwrongrnr(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)rrZrrerrnrZrfour)rn)rxryrr1r-r3r0test_validate_paramsrsU %I  g  %I  7|!q!  %I  "C>"  %I  a1   %I  a3/3Q33(3  %I  a2)2v2#q2  5                 sG D'D95E $ EE,E=' D69 E E E), E:= F cD[S[/0SS9S5nU"SS5 g)zLCheck that no error is raised when there are parameters without constraints r&Tr*cgrJr-)r&r's r0func1test_validate_params_missing_params..funcrSr3rnrZN)r$r)rs r0#test_validate_params_missing_paramsrs/ cC5\F G  AJr3c~[5"[5n[R"[SS9 U"SSSS9 SSS5 [R"[SS9 [R "[ SS9 U"SSS S9 SSS5 SSS5 g!,(df  N[=f!,(df  N(=f!,(df  g=f) z5Check that validate_params functions can be decoratedzFunction _func is deprecatedrvrnrZrrNrr)r r1rxwarns FutureWarningryr)decorated_functions r0 test_decorate_validated_functionrs#e, m+I J1a1% K m+I J ]] !)N  q!w /  K J K J    K Js/ B B.2 B;B. B B+ 'B.. B<c[R"[SS9 [5R S5 SSS5 [R "[ SS9 [R"[SS9 [5RS5 SSS5 SSS5 g!,(df  Nl=f!,(df  N(=f!,(df  g=f)z-Check that validate_params works with methodsz+The 'a' parameter of _Class._method must bervrNz)Function _deprecated_method is deprecatedz6The 'a' parameter of _Class._deprecated_method must be)rxryrr5r:rrr=r-r3r0test_validate_params_methodrs %R  !  m+V W ]] !J  H ' ' 0   X W      X Ws/BB</B+ B< B(+ B9 5B<< C c[S5n[R"[SS9 UR 5 SSS5 g!,(df  g=f)z9Check that validate_params works with Estimator instancesrz'The 'a' parameter of _Estimator must bervN)rFrxryrrQ)ests r0test_validate_params_estimatorr%s; W C %N      s ? A c[R"[SS9 [1SkSS1S9 SSS5 g!,(df  g=f)z@Check that the deprecated parameter must be a subset of options.z#deprecated options must be a subsetrvr~r&r)r N)rxryrrr-r3r0!test_stroptions_deprecated_subsetr0s, z)N O?Sz: P O Os 2 Ac6[S[[5[/0SS9S5nU"SSSS.5 U"/S Q5 [R "[ S S 9nU"S S 9 SSS5 [WR5nSU;deSU;deg!,(df  N4=f)zECheck that internal constraints are not exposed in the error message.paramTr*cgrJr-rs r0f!test_hidden_constraint..f9s r3rnrZr)r&r'r(rThe 'param' parameterrvbadrNzan instance of 'dict'zan instance of 'list') r$rlistrUrxryrrrrexc_infoerr_msgs r0test_hidden_constraintr6s 6$<&'t  AAAiL %<     (..!G "g -- - "' 11 1  s B  Bc B[S[S15[[S155/0SS9S5nU"S5 U"S5 [R"[ SS9nU"S S 9 S S S 5 [ WR5nSU;deSU;deg !,(df  N4=f) zBCheck that we can have 2 StrOptions constraints, one being hidden.rautowarnTr*cgrJr-rs r0r!test_hidden_stroptions..fQ r3rrvrrN)r$rrrxryrrrrs r0test_hidden_stroptionsrNs :vh' F80D)EFG&*  fIfI %<     (..!G W       s B Bc|[[S5(de[[5RS5(deg)zCheck that the validate_params decorator properly sets the parameter constraints as attribute of the decorated function/method. _skl_parameter_constraintsN)hasattrr1r5r:r-r3r04test_validate_params_set_param_constraints_attributergs4 56 7 77 7 68##%A B BB Br3cr[SS/0SS9S5nU"S5 U"[R"S55 g)zCheck that validate_params raise a deprecation message but still passes validation when using an int for a parameter accepting a boolean. rrTr*cgrJr-rs r0r1test_boolean_constraint_deprecated_int..ftrSr3FN)r$rjbool_rs r0&test_boolean_constraint_deprecated_intros> g {+4P Q dGbhhuor3c[[S/SS.SS9SSj5n[R"[SS9 U"S S 9 SSS5 "S S 5nU"US 9 U"U"5S 9 g!,(df  N*=f)z5Check that validation can be skipped for a parameter.N no_validationparam1param2Tr*cgrJr-rs r0rtest_no_validation..frr3zThe 'param1' parameterrvr)rc\rSrSrSrg)$test_no_validation..SomeTypeir-N)r?r@rArBrDr-r3r0SomeTyper s r3r )rrTr$rrxryr)rr s r0test_no_validationr}sr;/:&*   ,4L M  N  XXZ N Ms A  A.c[R"S5n[5nURUR5(deUR[ R "/SQ55(aeg)z9Add a specific test for checking support for `pandas.NA`.pandasrN)rx importorskiprrNArjarray)pd na_constraints r0$test_pandas_na_constraint_with_pd_narsX   X &B')M  ( ( / // /,,RXXi-@AAA AAr3c[5nUR/SQ5(deUR[S55(deURS5(aeg)zGCheck that a string does not satisfy the _IterableNotString constraint.rrz some stringN)rrrangers r0test_iterable_not_stringrsV$&J  % %i 0 00 0  % %eBi 0 00 0))-888 88r3c4[5nURS5(deUR[55(deURSS/SS/4SS/SS/4/5(deURS5(deURS5(aeg)zQCheck that the _CVObjects constraint accepts all current ways to pass cv objects.rrnrZrrNznot a CV object)rrr rs r0test_cv_objectsrsJ  % %a ( (( (  % %km 4 44 4  % %AA'71a&1a&9I&J K KK K  % %d + ++ +))*;<<< <.ThirdPartyEstimatoric,>Xl[TU] SS9 g)NrrK)r'superrL)r9r'rs r0rL@test_third_party_estimator..ThirdPartyEstimator.__init__sF G q  !r3c$>[TU]X5 grJ)r rQ)r9rOrPrs r0rQ;test_third_party_estimator..ThirdPartyEstimator.fits GK r3r'rT)r?r@rArBrLrQrD __classcell__)rs@r0ThirdPartyEstimatorrs "  r3r&rr$N)rFrQ)r&s r0test_third_party_estimatorr's! j!  "r3c[[SSSS9nURS5(deURS5(aeg)z9Check for the type RealNotInt in the Interval constraint.rrnr_r\rrN)rrrrs r0test_interval_real_not_intr)sC*a6:J  % %c * ** *))!,,, ,,r3c [S[5(de[S[5(ae[[R"S5[5(de[[R"S5[5(aeg)zCheck for the RealNotInt type.rrrnN) isinstancerrjfloat64int64r-r3r0test_real_not_intr.sa c: & && &!Z((( ( bjjmZ 0 00 0"((1+z222 22r3c[S[/0SS9S5n[R"[SS9 U"SS9 S S S 5 [ SS 9 U"SS9 S S S 5 g !,(df  N'=f!,(df  g =f) z@Check that param validation can be skipped using config_context.r&Tr*cgrJr-rKs r0r%test_skip_param_validation..frSr3The 'a' parameterrv1rKNskip_parameter_validation)r$rrxryrrrs r0test_skip_param_validationr6socC5\F G  ,4G H C I $ 7 C 8 7 I H 8 7sA A* A'* A8r+Fch^[S[/0SS9S5m[S[/0US9U4Sj5n[R"[SS9 U"S S 9 S S S 5 U(aU"S S 9 g [R"[S S9 U"S S 9 S S S 5 g !,(df  NE=f!,(df  g =f)z,Check that nested validation can be skipped.r&Tr*cgrJr-rKs r0r&test_skip_nested_validation..frSr3r'c>T"SS9$)Ninvalid_param_valuerKr-)r'rs r0g&test_skip_nested_validation..gs ())r3zThe 'b' parameterrvr;r$Nrnr2r )r+r<rs @r0test_skip_nested_validationr>scC5\F G  se &C* * ,4G H !" I% A ]]08K L FM L I H M Ls BB# B # B1zJskip_parameter_validation, prefer_skip_nested_validation, expected_skipped))TTT)TFT)FTT)FFFc[S[/0US9S5n[US9 U"S5nSSS5 WU:Xdeg!,(df  N=f)z5Check interaction between global skip and local skip.r&r*c[5S$)Nr5)rrKs r0r<9test_skip_nested_validation_and_config_context..g s|788r3r4rnN)r$rr)r5r+expected_skippedr<actual_skippeds r0.test_skip_nested_validation_and_config_contextrDs\ se 4Q99 2K L1 M - -- - M Ls ; A )r)dnumbersrrnumpyrjrx scipy.sparsersklearn._configrr sklearn.baserr sklearn.model_selectionr sklearn.utilsr sklearn.utils._param_validationr rrrrrrrrrrrrrrrrrrr r!r"r#r$sklearn.utils.fixesr%r1r5rFmark parametrizercrhrmrq TypeErrorrr|rrrndarrayrrrrrrrrsfloat32r,callablerrandom RandomStaterprgrrrrrrrrrrrrrrrrrr'r)r.r6r>rD) containers0r0rWs " #64/$4/ &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'( rxx!Q!Q 012 *q!fq!f%567 , + i!Q!Q(89 :+     ..q12  ! " # $ a% & s ' ( ) * + , - . "/ 0 $1 2 !T'"3 4 $5 6 %,'7 8 "&&!9 : )$; < E JO,= > ? #H-I#H- 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..W  s[