Q3j_6ddlmZmZddlZddlZddlmZmZddl m Z m Z ddl m Z mZddlmZddlmZddlmZmZmZmZmZmZmZmZmZmZmZmZm Z m!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{dd dZ.GddZ/Gdde Z0ejbjedeegdZ3ejbjedeegdZ4dZ5ejbjedeedddeedddgdZ6ejbjededddd e7d!fedd"d#d e7d$fedddd e8d%feddd&d e8d'fedd(dd e8d)fgd*Z9d+Z:d,Z;ejbjed-ed2Z?ejbjed3eedddeedddeeddd#ehd4eed 5e'ed6e!eg d7Z@ejbjed8eedd9d&eed:d;dfeedd9d&eed:d;d#feedd9d&eedd=d#feed:d;deed>d=d&feed:d;d#eed?d@d#feed:d;d#eed?d@dfeedddeedddfeedddeedddfgdAZAejbjed3ee eBee#e%e&eeeddd#gdBZCejbjed3eee eBe#e%e&ee'eed 5ehd4eehdCeeddd#eedd@d#eeddd#eeddd#eeddd#eedddeeddd&ed6e!egdDZDejbjedEeeddddFfeeddd#dGfehd4dHfeeEejejhejfeHdIfdJdKddgd9dr?r@rrCdict__annotations__rGr rKr.r4r1rBrB>s4F$'$=D05 6 r4rB 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)rrPintervals r1test_interval_ranger\Ks r1V> H   r1W=H X   == r1V> == r1Y?H X   H  r4c0t|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. rrSrYrUAiN)rfloatrZs r1test_interval_large_integersra_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 == leftrWz(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)paramserrorrps r1test_interval_errorsrus*@ uE *6 + * *s -6cthddh}|jdsJ|jdsJ|jdrJdt|vsJy)z*Sanity check for the StrOptions constraintr'r(r)r)r r'r*z'c' (deprecated)N)ris_satisfied_bystroptionss r1test_stroptionsr|s^cU;G  " "3 '' '  " "3 '' '&&s++ + W -- -r4cttddtjhdh}|j dsJ|j tjsJ|j drJdt |vsJy)z'Sanity check for the Options constraintg?r Gz?z-0.5 (deprecated)N)rrrcrdrxryrzs r1 test_optionsrsodT3/TFCG  " "4 (( (  " "266 ** *&&t,, , #g, .. .r4ztype, expected_type_nameintr`z numpy.ndarraycBt|}t|d|dk(sJy)z?Check the string representation of the _InstancesOf constraint.zan instance of ''N)rry)rlexpected_type_name constraints r1%test_instances_of_type_human_readablers,d#J z? 01C0DAF FF Fr4ctddg}Gdd}Gdd}|j|sJ|j|rJt|dk(sJy) z Check the HasMethods constraint.r'r(ceZdZdZdZy)test_hasmethods.._GoodcyrEr.r9s r1r'z test_hasmethods.._Good.a r4cyrEr.rs r1r(z test_hasmethods.._Good.brr4N)r=r>r?r'r(r.r4r1_Goodrs   r4rceZdZdZy)test_hasmethods.._BadcyrEr.rs r1r'ztest_hasmethods.._Bad.arr4N)r=r>r?r'r.r4r1_Badrs r4rz"an object implementing 'a' and 'b'N)rrxry)rrrs r1test_hasmethodsrseS#J'J  % %eg .. .))$&11 1 z?B BB Br4rrw) numeric_onlyrKc@t|}|j|rJy)z>Check that the value generated does not satisfy the constraintNr"rx)r bad_values r1test_generate_invalid_param_valrs%"+:6I)))44 44r4zinteger_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 r1+test_generate_invalid_param_val_2_intervalsrsmb+mDI,,Y77 7// :: :*6FGI,,Y77 7// :: ::r4cvtjt5t|dddy#1swYyxYw)zmCheck that the function raises NotImplementedError when there's no invalid value for the constraint. N)rqrrNotImplementedErrorr"rs r1)test_generate_invalid_param_val_all_validrWs%" * +":. , + +s /8>rgrSrc@t|}|j|sJy)z;Check that the value generated does satisfy the constraint.N)r#rx)rvalues r1test_generate_valid_paramrls": ! ,E  % %e ,, ,r4zconstraint_declaration, valuegzG?*r(c |dzS)Nrgr.)xs r1rsQUr4rMz array-likez sparse matrix) random_staterr)rNr~)booleanF)verbosergrigimissingrF) cv_objectrc@t|}|j|sJy)z+Sanity check for the is_satisfied_by methodN)r$rx)constraint_declarationrrs r1test_is_satisfied_byrs$N!!78J  % %e ,, ,r4z1constraint_declaration, expected_constraint_classoption1option2rrrrc:t|}|j|usJy)zICheck that make_constraint dispatches to the appropriate constraint classN)r$ __class__)rexpected_constraint_classrs r1test_make_constraintrs$.!!78J   #< << 01 ? > >s 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 berowrongrgr)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)rrSrrerrgrSrfour)rg)rqrrrr2r.r4r1test_validate_paramsrsU %I  g  %I  7|!q!  %I  "C>"  %I  a1   %I  a3/3Q33(3  %I  a2)2v2#q2  5                 sGD9 E>E/E!E*E69EEEE'*E36E?cJtdtgidd}|ddy)zTCheck that no error is raised when there are parameters without constraints r'Tr+cyrEr.)r'r(s r1funcz1test_validate_params_missing_params..funcrLr4rgrSN)r%r)rs r1#test_validate_params_missing_paramsrs/ cC5\F G  AJr4cxtt}tjtd5|ddddddtjtd5tj t d5|ddd ddddddy#1swY]xYw#1swYxYw#1swYyxYw) z5Check that validate_params functions can be decoratedzFunction _func is deprecatedrorgrSrrNrr)r r2rqwarns FutureWarningrrr)decorated_functions r1 test_decorate_validated_functionrs%e, m+I J1a1% K m+I J ]] !)N  q!w /  K J K J    K Js/ 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 berorNz)Function _deprecated_method is deprecatedz6The 'a' parameter of _Class._deprecated_method must be)rqrrrr6r:rrr<r.r4r1test_validate_params_methodrs %R  !  m+V W ]] !J  H ' ' 0   X W      X Ws/B B85B,B8 B),B5 1B88Cctd}tjtd5|j dddy#1swYyxYw)z9Check that validate_params works with Estimator instancesrz'The 'a' parameter of _Estimator must beroN)rBrqrrrrK)ests r1test_validate_params_estimatorr&s; 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 subsetrorwr'r*r N)rqrrrrr.r4r1!test_stroptions_deprecated_subsetr1s, z)N O?Sz: P O Os 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+cyrEr.rs r1fz!test_hidden_constraint..f:s r4rgrSr)r'r(r)rThe 'param' parameterrobadrNzan instance of 'dict'zan instance of 'list') r%rlistrNrqrrrryrrexc_infoerr_msgs r1test_hidden_constraintr7s 6$<&'t  AAAiL %<     (..!G "g -- - "' 11 1  s  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+cyrEr.rs r1rz!test_hidden_stroptions..fR r4rrorrN)r%rrrqrrrryrrs r1test_hidden_stroptionsrOs :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)hasattrr2r6r:r.r4r14test_validate_params_set_param_constraints_attributerhs. 56 77 7 68##%A BB Br4cvtddgidd}|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+cyrEr.rs r1rz1test_boolean_constraint_deprecated_int..furLr4FN)r%rcbool_rs r1&test_boolean_constraint_deprecated_intrps> g {+4P Q dGbhhuor4cttdgddddd}tjtd5|d dddGd d }|| || y#1swY+xYw)z5Check that validation can be skipped for a parameter.N no_validationparam1param2Tr+cyrEr.rs r1rztest_no_validation..frr4zThe 'param1' parameterror)rc eZdZy)$test_no_validation..SomeTypeN)r=r>r?r.r4r1SomeTypers r4r)rrMr%rrqrrr)rrs r1test_no_validationr~sr;/:&*   ,4L M  N  XXZ N Ms A--A6ctjd}t}|j|jsJ|jt j gdrJy)z9Add a specific test for checking support for `pandas.NA`.pandasrN)rq importorskiprrxNArcarray)pd na_constraints r1$test_pandas_na_constraint_with_pd_narsR   X &B')M  ( ( // /,,RXXi-@AA AAr4ct}|jgdsJ|jtdsJ|jdrJy)zGCheck that a string does not satisfy the _IterableNotString constraint.rrz some stringN)rrxrangers r1test_iterable_not_stringrsM$&J  % %i 00 0  % %eBi 00 0))-88 88r4ct}|jdsJ|jtsJ|jddgddgfddgddgfgsJ|jdsJ|jdrJy)zUCheck that the _CVObjects constraint accepts all current ways to pass cv objects.rrgrSrrNznot a CV object)rrxr rs r1test_cv_objectsr sJ  % %a (( (  % %km 44 4  % %AA'71a&1a&9I&J KK K  % %d ++ +))*;<< <.ThirdPartyEstimatorc4||_t| dy)NrrF)r(superrG)r9r(rs r1rGz@test_third_party_estimator..ThirdPartyEstimator.__init__sDF G q  !r4c&t|||yrE)rrK)r9rIrJrs r1rKz;test_third_party_estimator..ThirdPartyEstimator.fits GK1 r4rM)r=r>r?rGrK __classcell__)rs@r1ThirdPartyEstimatorr s "  r4rrr(N)rBrK)rs r1test_third_party_estimatorrs! j!  "r4cvttddd}|jdsJ|jdrJy)z9Check for the type RealNotInt in the Interval constraint.rrgrXrUrkN)rrrxrs r1test_interval_real_not_intrs=*a6:J  % %c ** *))!,, ,,r4ctdtsJtdtrJttjdtsJttjdtrJy)zCheck for the RealNotInt type.rkrgN) isinstancerrcfloat64int64r.r4r1test_real_not_intrsU c: && &!Z(( ( bjjmZ 00 0"((1+z22 22r4ctdtgidd}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+cyrEr.rFs r1rz%test_skip_param_validation..frLr4The 'a' parameterro1rFNskip_parameter_validation)r%rrqrrrrrs r1test_skip_param_validationr"socC5\F G  ,4G H C I $ 7 C 8 7 I H 8 7s 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+cyrEr.rFs r1rz&test_skip_nested_validation..frLr4r(cdS)Ninvalid_param_valuerFr.)r(rs r1gz&test_skip_nested_validation..gs ())r4zThe 'b' parameterror&rNrgrr)r,r'rs @r1test_skip_nested_validationr(scC5\F G  se &C* * ,4G H !" I% A ]]08K L FM L I H M Ls 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!)rrFs r1r'z9test_skip_nested_validation_and_config_context..g s|788r4r rgN)r%rr)r!r,expected_skippedr'actual_skippeds r1.test_skip_nested_validation_and_config_contextr-s\ se 4Q99 2K L1 M - -- - M Ls AA )r)enumbersrrnumpyrcrq scipy.sparserrsklearn._configrr sklearn.baser r sklearn.model_selectionr sklearn.utilsr sklearn.utils._param_validationrrrrrrrrrrrrrrrrrrr r!r"r#r$r%sklearn.utils.fixesr&r2r6rBmark parametrizer\rarfrj TypeErrorrrur|rrndarrayrrrrrrrrlfloat32rcallablerrandom RandomStaterir`rrrrrrrrrrrrrrrrr rrrr"r(r-) containers0r1r@s " .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'( xrxx!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^