rtj_bddlmZmZddlZddlZddlmZmZddl 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 Z0ej12deegdZ3ej12deegdZ4dZ5ej12deedddeedddgdZ6ej12dedddd e7d!fedd"d#d e7d$fedddd e8d%feddd&d e8d'fedd(dd e8d)fgd*Z9d+Z:d,Z;ej12d-ed2Z?ej12d3eedddeedddeeddd#ehd4eed 5e'ed6e!eg d7Z@ej12d8eedd9d&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@dfeedddeedddfeedddeedddfgdAZAej12d3ee eBee#e%e&eeeddd#gdBZCej12d3eee eBe#e%e&ee'eed 5ehd4eehdCeeddd#eedd@d#eeddd#eeddd#eeddd#eedddeeddd&ed6e!egdDZDej12dEeeddddFfeeddd#dGfehd4dHfeeEejFejGhejGfeHdIfdJdKddgd9dr?r@rrCdict__annotations__rGr rKr.r4r1rBrB>spFF$'$=D000\555   65   r4rB interval_typect|ddd}d|vsJd|vsJt|ddd}d|vsJd|vsJt|ddd}d|vsJd|vsJt|ddd}d|vsJd|vsJdS) 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      r4cbt|ddd}ddz|vsJddz|vsJtddz|vsJtddz|vsJt|dddzd}ddz|vsJddz|vsJtddz|vsJtddz|vsJddz|vsJdS) 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 ! ! ! ! B<Expecting 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 leftc|tj||5tdi|ddddS#1swxYwYdS)zNCheck that informative errors are raised for invalid combination of parametersmatchNr.)pytestraisesr)paramserrorrps r1test_interval_errorsrus@ uE * * *6s 155cthddh}|dsJ|dsJ|drJdt|vsJdS)z*Sanity check for the StrOptions constraint>r'r(r)r)r r'r*z'c' (deprecated)N)ris_satisfied_bystroptionss r1test_stroptionsr{scU;;;G  " "3 ' ''' '  " "3 ' ''' '&&s++++ + W - - - - - -r4cttddtjhdh}|dsJ|tjsJ|drJdt |vsJdS)z'Sanity check for the Options constraintg?r Gz?z-0.5 (deprecated)N)rrrcrdrwrxrys r1 test_optionsrsdT3/TFCCCG  " "4 ( ((( (  " "26 * *** *&&t,,,, , #g,, . . . . . .r4ztype, expected_type_nameintr`z numpy.ndarraycVt|}t|d|dksJdS)z?Check the string representation of the _InstancesOf constraint.zan instance of ''N)rrx)rlexpected_type_name constraints r1%test_instances_of_type_human_readablers;d##J z??F1CFFF F F F F F Fr4ctddg}Gdd}Gdd}||sJ||rJt|dksJdS) z Check the HasMethods constraint.r'r(ceZdZdZdZdS)test_hasmethods.._GoodcdSrEr.r9s r1r'z test_hasmethods.._Good.a Dr4cdSrEr.rs r1r(z test_hasmethods.._Good.brr4N)r=r>r?r'r(r.r4r1_Goodrs2        r4rceZdZdZdS)test_hasmethods.._BadcdSrEr.rs r1r'ztest_hasmethods.._Bad.arr4N)r=r>r?r'r.r4r1_Badrs#     r4rz"an object implementing 'a' and 'b'N)rrwrx)rrrs r1test_hasmethodsrsS#J''J  % %eegg . ... .))$$&&1111 1 z??B B B B B B Br4r>r'r(r)) numeric_onlyrKcRt|}||rJdS)z>Check that the value generated does not satisfy the constraintNr"rw)r bad_values r1test_generate_invalid_param_valrs1"+:66I)))4444 444r4zinteger_interval, real_intervali ct|}||rJ||rJt|}||rJ||rJdS)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_intervalsrsb+mDDDI,,Y7777 7// :::: :*6FGGGI,,Y7777 7// :::: :::r4ctjt5t|ddddS#1swxYwYdS)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 7;;>rgrSrcRt|}||sJdS)z;Check that the value generated does satisfy the constraint.N)r#rw)rvalues r1test_generate_valid_paramrls1: ! , ,E  % %e , ,,, ,,,r4zconstraint_declaration, valuegzG?*r(c |dzS)Nrgr.)xs r1rs QUr4rMz array-like sparse matrixc4g|]}d|ddgddggfS)rrgrSrrr.).0 containers r1 rsD   ii!Q!Q(899 :   r4) random_staterr)rNr})booleanF)verbosergrigimissingrF) cv_objectrcRt|}||sJdS)z+Sanity check for the is_satisfied_by methodN)r$rw)constraint_declarationrrs r1test_is_satisfied_byrs3N!!788J  % %e , ,,, ,,,r4z1constraint_declaration, expected_constraint_classoption1option2r~rrrc:t|}|j|usJdS)zICheck that make_constraint dispatches to the appropriate constraint classN)r$ __class__)rexpected_constraint_classrs r1test_make_constraintrs-.!!788J  #< < < < < < > >220111222222222222222222s 9==cBtjtd5tdddddn #1swxYwYtjtd5tddgddidddn #1swxYwYtjtd 5tdiddidddn #1swxYwYtjtd 5tddd dddn #1swxYwYtjtd5td ggd Rddiddidddn #1swxYwYtjtd 5td ggdRddiddiddddS#1swxYwYdS)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_paramsrs %I    g %I   "" 7|!q!!!""""""""""""""" %I   ## ""C>"""############### %I   !! a1    !!!!!!!!!!!!!!! %I   44 a3///333Q33(333444444444444444 %I   33 a2)))222v2#q222333333333333333333sg:>> A;;A?A?!B<<CC"DDD'E  EE0FFFcbtdtgidd}|dddS)zTCheck that no error is raised when there are parameters without constraints r'Tr+cdSrEr.)r'r(s r1funcz1test_validate_params_missing_params..funcrLr4rgrSN)r%r)rs r1#test_validate_params_missing_paramsrsG cC5\FFF  GF  DAJJJJJr4ctt}tjtd5|ddddddn #1swxYwYtjtd5tjt d5|ddd dddn #1swxYwYddddS#1swxYwYdS) z5Check that validate_params functions can be decoratedzFunction _func is deprecatedrorgrSrrNrr)r r2rqwarns FutureWarningrrr)decorated_functions r1 test_decorate_validated_functionrs%e,, m+I J J J&&1a1%%%%&&&&&&&&&&&&&&& m+I J J J00 ] !)N    0 0  q!w / / / / 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0000000000000000000sGAAA9CB0$ C0B4 4C7B4 8CC C ctjtd5tddddn #1swxYwYtjt d5tjtd5tddddn #1swxYwYddddS#1swxYwYdS)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 W W11 ] !J    1 1 HH ' ' 0 0 0  1 1 1 1 1 1 1 1 1 1 1 1 1 1 1111111111111111111sG"A  AA0C "B:. C:B> >CB> CCCctd}tjtd5|ddddS#1swxYwYdS)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     sA  AActjtd5thdddhddddS#1swxYwYdS)z@Check that the deprecated parameter must be a subset of options.z#deprecated options must be a subsetro>r'r(r)r'r*r N)rqrrrrr.r4r1!test_stroptions_deprecated_subsetr1s z)N O O O;;???Sz::::;;;;;;;;;;;;;;;;;;s?AAcftdtttgidd}|dddd|gd t jt d 5}|d dddn #1swxYwYt|j}d|vsJd|vsJdS)zECheck that internal constraints are not exposed in the error message.paramTr+cdSrEr.rs r1fz!test_hidden_constraint..f:s  r4rgrSr)r'r(r)rThe 'param' parameterrobadrNzan instance of 'dict'zan instance of 'list') r%rlistrNrqrrrrxrrexc_infoerr_msgs r1test_hidden_constraintr7s) 6$<<&'t   AAAAAiiiLLL %<      (.!!G "g - - - - "' 1 1 1 1 1 1s- BB  B c ~tdtdhttdhgidd}|d|dtjt d5}|d d d d n #1swxYwYt |j}d|vsJd|vsJd S) zBCheck that we can have 2 StrOptions constraints, one being hidden.rautowarnTr+cdSrEr.rs r1rz!test_hidden_stroptions..fR r4rrorrN)r%rrrqrrrrxrrs r1test_hidden_stroptionsrOs/ :vh'' F80D0D)E)EFG&*   AfIIIAfIII %<      (.!!G W      s9 BBBczttdsJttjdsJdS)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 7 777 7 688#%A B BBB BBBr4ctddgidd}|d|tjddS)zCheck that validate_params raise a deprecation message but still passes validation when using an int for a parameter accepting a boolean. rrTr+cdSrEr.rs r1rz1test_boolean_constraint_deprecated_int..furLr4FN)r%rcbool_rs r1&test_boolean_constraint_deprecated_intrps` g {+4PPP  QP AdGGGAbhuoor4c(ttdgddddd}tjtd5|d dddn #1swxYwYGd d }|| || dS)z5Check that validation can be skipped for a parameter.N no_validationparam1param2Tr+cdSrEr.rs r1rztest_no_validation..frr4zThe 'param1' parameterror)rceZdZdS)$test_no_validation..SomeTypeN)r=r>r?r.r4r1SomeTypers r4r)rrMr%rrqrrr)rrs r1test_no_validationr~s#;/::&*     ,4L M M M         AXAXXZZs AA!Actjd}t}||jsJ|t jgdrJdS)z9Add a specific test for checking support for `pandas.NA`.pandasrN)rq importorskiprrwNArcarray)pd na_constraints r1$test_pandas_na_constraint_with_pd_nar sg  X & &B'))M  ( ( / /// /,,RXiii-@-@AAAA AAAr4ct}|gdsJ|tdsJ|drJdS)zGCheck that a string does not satisfy the _IterableNotString constraint.rrz some stringN)rrwrangers r1test_iterable_not_stringr sk$&&J  % %iii 0 000 0  % %eBii 0 000 0))-8888 888r4c<t}|dsJ|tsJ|ddgddgfddgddgfgsJ|dsJ|drJdS)zUCheck that the _CVObjects constraint accepts all current ways to pass cv objects.rrgrSrrNznot a CV object)rrwr rs r1test_cv_objectsrsJ  % %a ( ((( (  % %kmm 4 444 4  % %AA'71a&1a&9I&J K KKK K  % %d + +++ +))*;<<<< <<.ThirdPartyEstimatorcZ||_tddS)NrrF)r(superrG)r9r(rs r1rGz@test_third_party_estimator..ThirdPartyEstimator.__init__s+DF GG  q  ! ! ! ! !r4NcLt||dSrE)rrK)r9rIrJrs r1rKz;test_third_party_estimator..ThirdPartyEstimator.fits! GGKK1     r4rM)r=r>r?rGrK __classcell__)rs@r1ThirdPartyEstimatorrsV " " " " "          r4rrr(N)rBrK)rs r1test_third_party_estimatorrsT j!  """""r4cttddd}|dsJ|drJdS)z9Check for the type RealNotInt in the Interval constraint.rrgrXrUrkN)rrrwrs r1test_interval_real_not_intrsR*a6:::J  % %c * *** *))!,,,, ,,,r4ctdtsJtdtrJttjdtsJttjdtrJdS)zCheck for the RealNotInt type.rkrgN) isinstancerrcfloat64int64r.r4r1test_real_not_intrsp c: & &&& &!Z(((( ( bjmmZ 0 000 0"(1++z2222 222r4c2tdtgidd}tjtd5|dd d d n #1swxYwYt d 5|dd d d d S#1swxYwYd S) z@Check that param validation can be skipped using config_context.r'Tr+cdSrEr.rFs r1rz%test_skip_param_validation..frLr4The 'a' parameterro1rFNskip_parameter_validation)r%rrqrrrrrs r1test_skip_param_validationr&s5cC5\FFF  GF  ,4G H H H C $ 7 7 7 Cs# AAA2 B  BBr,Fctdtgiddtdtgi|fd}tjtd5|d d d d n #1swxYwY|r|d d Stjtd 5|d d d d d S#1swxYwYd S)z,Check that nested validation can be skipped.r'Tr+cdSrEr.rFs r1rz&test_skip_nested_validation..frLr4r(cdS)Ninvalid_param_valuerFr.)r(rs r1gz&test_skip_nested_validation..gs q())))r4zThe 'b' parameterror*rNrgr"r)r,r+rs @r1test_skip_nested_validationr,scC5\FFF  GF  se &C**** * ,4G H H H##  !""""###############% A ]08K L L L   AFFFF                  s$# A<<BB2 C  CCzJskip_parameter_validation, prefer_skip_nested_validation, expected_skipped))TTT)TFT)FTT)FFFctdtgi|d}t|5|d}dddn #1swxYwY||ksJdS)z5Check interaction between global skip and local skip.r'r+c*tdS)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_contextr1s se 4Q999 2K L L L1 - - - - - - -s A  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`rrrrrrrrrrrrrrr r rrrrr&r,r1r.r4r1rCs"""""""" ........6666666644444444//////$$$$$$4/..... &vTF$88"&:a:::: : , , , , , , , ,          8T*:;;<;&8T*:;;##<;#( # # #  XdAq(((((4tF*S*S*ST"" " sQ& I I  L q3) L L  L ta6 J J  5 q47 K K  7 q2 H H  , +<=< .../// e 5 w _% GGG CCC(tQv...q$v...tT)444 ???##  4((( 5  555 % HXtQw 7 7 7 HZQv 6 6 6 HXtQw 7 7 7 HZQy 9 9 9 HXtQw 7 7 7 HZAf 5 5 5 HXtQw 7 7 7 HZD 8 8 8 HXtQw 7 7 7 HZD ; ; ; HXq$v 6 6 6 HZQv 6 6 6 HXq$v 6 6 6 HZQy 9 9 9 HXq$v 6 6 6 HZAf 5 5 5 HXq$v 6 6 6 HZr& 9 9 9 HXq$v 6 6 6 HZr) < < < HXr1V 4 4 4 HZq 9 9 9 HXr1V 4 4 4 HZD 8 8 8 HXr1V 4 4 4 HZb ; ; ; HXr1V 4 4 4 HZb 9 9 9 HXr1Y 7 7 7 HZBy 9 9 9 HXr1Y 7 7 7 HZBv 6 6 6 HXq$v 6 6 6 HZAf 5 5 5 HXq$v 6 6 6 HZAf 5 5 5 KILLZ ; ;[LLZ ;  T   4i888   //  /   T    4((( ???##)))$$4i8881b3331d955549555q!I...q$v...tQw/// 5 -6--76- # $1V , , ,d3 (AtI 6 6 6; OOO $ $c*  BJ/ 0 0"*= ??# AA'( xrx!Q!Q 0112 **q!fq!f%5667 ))aVaV$4556!  +   ! ! ..q112!!" #!$ %!& a'!( s )!* +!, -!. /!0 "1!2 $3!4 !T'"5!6 $7!8 %%,,'9!: "&!;!< )$=!> E  JJOOO,?!@ A!$$J--K$$J- 7 $1V , , ,h7 Y * + +Z8 d| $ $g. {# /* '  : l I N# D ) ) )=9 E  J' j!   !*==+*= 222 333D    0 0 0 1 1 1 ;;; 2220!!!2CCC   ,BBB999===###&---333   84-HHIH4P.....r4