Mpj_ SSKJrJr SSKrSSKrSSKJrJr SSK J r J r SSK J r Jr SSKJr SSKJr SSKJrJrJrJrJrJrJrJrJrJrJrJrJ r J!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\ 5r0\RbReS\\/5S5r3\RbReS\\/5S5r4Sr5\RbReS\"\SSSS9\"\SSSS9/5S5r6\RbReS\SSSS .\7S!4\SS"S#S .\7S$4\SSSS .\8S%4\SSS&S .\8S'4\SS(SS .\8S)4/5S*5r9S+r:S,r;\RbReS-\S2r?\RbReS3\"\SSSS9\"\SSSS9\"\SSS#S9\"1S4k5\"5\"S S59\'"5\"S65\!"5\"5/ 5S75r@\RbReS8\"\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/5SA5rA\RbReS3\"5\ "\B5\"5\#"5\%"5\&"5\"5\"\SSS#S9/5SB5rC\RbReS3\"5\"5\ "\B5\#"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/5SD5rD\RbReSE\"\SSSS9SF4\"\SSS#S9SG4\"1S4k5SH4\"\E\R\R15\R4\HSI4SJSKSS/S9S__static_attributes__r.r4r1r6r61sQTcD6]$G!H!\cD6]$G,H,r4r6cP\rSrSr%SrS\/0r\\S'Sr \ "SS9S S j5r S r g) _Estimator>zs4F$'$=D05 6 r4rG interval_typec[USSSS9nSU;deSU;de[USSSS9nSU;deSU;de[USSSS9nSU;deSU;de[USSSS9nSU;deSU;deg) z.Check the range of values depending on closed.leftclosedrightbothneitherN)rrXintervals r1test_interval_rangerdKs r1V> H   r1W=H X   == r1V> == r1Y?H X   H  r4cH[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. rr[rar]AiN)rfloatrbs r1test_interval_large_integersri_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)paramserrorrxs r1test_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 r1test_stroptionsrsgcU;G  " "3 ' '' '  " "3 ' '' '&&s+++ + W -- -r4c[[SS[R1S1S9nUR S5(deUR [R5(deUR S5(aeS[ U5;deg)z'Sanity check for the Options constraintg?r Gz?z-0.5 (deprecated)N)rrrkrlrrrs r1 test_optionsrsxdT3/TFCG  " "4 ( (( (  " "266 * ** *&&t,,, , #g, .. .r4ztype, expected_type_nameintrhz numpy.ndarraycD[U5n[U5SUS3:Xdeg)z?Check the string representation of the _InstancesOf constraint.zan instance of ''N)rr)rtexpected_type_name constraints r1%test_instances_of_type_human_readablers,d#J z? 01C0DAF FF Fr4c[SS/5n"SS5n"SS5nURU"55(deURU"55(ae[U5S:Xdeg) z Check the HasMethods constraint.r'r(c \rSrSrSrSrSrg)test_hasmethods.._GoodcgrKr.r:s r1r' test_hasmethods.._Good.a r4cgrKr.rs r1r( test_hasmethods.._Good.brr4r.N)r@rArBrCr'r(rEr.r4r1_Goodrs   r4rc\rSrSrSrSrg)test_hasmethods.._BadcgrKr.rs r1r'test_hasmethods.._Bad.arr4r.N)r@rArBrCr'rEr.r4r1_Badrs r4rz"an object implementing 'a' and 'b'N)rrr)rrrs r1test_hasmethodsrskS#J'J  % %eg . .. .))$&111 1 z?B BB Br4rr) numeric_onlyrRcJ[U5nURU5(aeg)z>Check that the value generated does not satisfy the constraintNr"r)r bad_values r1test_generate_invalid_param_valrs("+:6I)))444 44r4zinteger_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 r1+test_generate_invalid_param_val_2_intervalsrsyb+mDI,,Y777 7// ::: :*6FGI,,Y777 7// ::: ::r4c[R"[5 [U5 SSS5 g!,(df  g=f)zeCheck that the function raises NotImplementedError when there's no invalid value for the constraint. N)ryrzNotImplementedErrorr"rs r1)test_generate_invalid_param_val_all_validrWs%" * +":. , + +s 0 >>ror[rcJ[U5nURU5(deg)z;Check that the value generated does satisfy the constraint.N)r#r)rvalues r1test_generate_valid_paramrls%: ! ,E  % %e , ,, ,r4zconstraint_declaration, valuegzG?*r(c US-$)Nror.)xs r1rsQUr4rUz array-likez sparse matrix) random_staterr)rNr)booleanF)verboserorqgimissingrL) cv_objectrcJ[U5nURU5(deg)z+Sanity check for the is_satisfied_by methodN)r$r)constraint_declarationrrs r1test_is_satisfied_byrs'N!!78J  % %e , ,, ,r4z1constraint_declaration, expected_constraint_classoption1option2rrrrc<[U5nURULdeg)zICheck that make_constraint dispatches to the appropriate constraint classN)r$ __class__)rexpected_constraint_classrs r1test_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 berwwrongror)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)rr[rrerror[rfour)ro)ryrzrr2r.r4r1test_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+cgrKr.)r'r(s r1func1test_validate_params_missing_params..funcrTr4ror[N)r%r)rs r1#test_validate_params_missing_paramsrs/ cC5\F G  AJr4c~[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 deprecatedrwror[rrNrr)r r2rywarns FutureWarningrzr)decorated_functions r1 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 berwrNz)Function _deprecated_method is deprecatedz6The 'a' parameter of _Class._deprecated_method must be)ryrzrr6r;rrr>r.r4r1test_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 berwN)rGryrzrrR)ests r1test_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 subsetrwrr'r*r N)ryrzrrr.r4r1!test_stroptions_deprecated_subsetr1s, 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+cgrKr.rs r1f!test_hidden_constraint..f:s r4ror[r)r'r(r)rThe 'param' parameterrwbadrNzan instance of 'dict'zan instance of 'list') r%rlistrVryrzrrrrexc_infoerr_msgs r1test_hidden_constraintr7s 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+cgrKr.rs r1r!test_hidden_stroptions..fR r4rrwrrN)r%rrryrzrrrrs r1test_hidden_stroptionsrOs :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)hasattrr2r6r;r.r4r14test_validate_params_set_param_constraints_attributerhs4 56 7 77 7 68##%A B BB Br4cr[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+cgrKr.rs r1r1test_boolean_constraint_deprecated_int..furTr4FN)r%rkbool_rs r1&test_boolean_constraint_deprecated_intrps> g {+4P Q dGbhhuor4c[[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+cgrKr.rs r1rtest_no_validation..frr4zThe 'param1' parameterrwr)rc\rSrSrSrg)$test_no_validation..SomeTypeir.N)r@rArBrCrEr.r4r1SomeTyper s r4r )rrUr%rryrzr)rr s r1test_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)ry importorskiprrNArkarray)pd na_constraints r1$test_pandas_na_constraint_with_pd_narsX   X &B')M  ( ( / // /,,RXXi-@AAA AAr4c[5nUR/SQ5(deUR[S55(deURS5(aeg)zGCheck that a string does not satisfy the _IterableNotString constraint.rrz some stringN)rrrangers r1test_iterable_not_stringrsV$&J  % %i 0 00 0  % %eBi 0 00 0))-888 88r4c4[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.rror[rrNznot a CV object)rrr rs r1test_cv_objectsrsJ  % %a ( (( (  % %km 4 44 4  % %AA'71a&1a&9I&J K KK K  % %d + ++ +))*;<<< <.ThirdPartyEstimatoric,>Xl[TU] SS9 g)NrrL)r(superrM)r:r(rs r1rM@test_third_party_estimator..ThirdPartyEstimator.__init__sF G q  !r4c$>[TU]X5 grK)r!rR)r:rPrQrs r1rR;test_third_party_estimator..ThirdPartyEstimator.fits GK r4r(rU)r@rArBrCrMrRrE __classcell__)rs@r1ThirdPartyEstimatorrs "  r4r'rr%N)rGrR)r's r1test_third_party_estimatorr(s! j!  "r4c[[SSSS9nURS5(deURS5(aeg)z9Check for the type RealNotInt in the Interval constraint.rror`r]rsN)rrrrs r1test_interval_real_not_intr*sC*a6:J  % %c * ** *))!,,, ,,r4c [S[5(de[S[5(ae[[R"S5[5(de[[R"S5[5(aeg)zCheck for the RealNotInt type.rsroN) isinstancerrkfloat64int64r.r4r1test_real_not_intr/sa c: & && &!Z((( ( bjjmZ 0 00 0"((1+z222 22r4c[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+cgrKr.rLs r1r%test_skip_param_validation..frTr4The 'a' parameterrw1rLNskip_parameter_validation)r%rryrzrrrs r1test_skip_param_validationr7socC5\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+cgrKr.rLs r1r&test_skip_nested_validation..frTr4r(c>T"SS9$)Ninvalid_param_valuerLr.)r(rs r1g&test_skip_nested_validation..gs ())r4zThe 'b' parameterrwr<r%Nror3r)r,r=rs @r1test_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$)Nr6)rrLs r1r=9test_skip_nested_validation_and_config_context..g s|788r4r5roN)r%rr)r6r,expected_skippedr=actual_skippeds r1.test_skip_nested_validation_and_config_contextrEs\ se 4Q99 2K L1 M - -- - M Ls ; A )r)enumbersrrnumpyrkry scipy.sparserrsklearn._configrr sklearn.baser r sklearn.model_selectionr sklearn.utilsr sklearn.utils._param_validationrrrrrrrrrrrrrrrrrrr r!r"r#r$r%sklearn.utils.fixesr&r2r6rGmark parametrizerdrirnrr TypeErrorrr}rrrndarrayrrrrrrrrtfloat32r-callablerrandom RandomStaterqrhrrrrrrrrrrrrrrrrrr(r*r/r7r?rE) containers0r1rXs " .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 )aVaV$456!, + i!Q!Q(89 :+ ! ! ..q12!!" #!$ %!& a'!( s )!* +!, -!. /!0 "1!2 $3!4 !T'"5!6 $7!8 %,'9!: "&&!;!< )$=!> E JO,?!@ A!$J-K$J- 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"[0