Ug5_*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*dd l+m,Z,e*egegegegd d d|dd dZ-GddZ.Gdde Z/ej01deegdZ2ej01deegdZ3dZ4ej01deedddeedddgdZ5ej01dedddd e6d!fedd"d#d e6d$fedddd e7d%feddd&d e7d'fedd(dd e7d)fgd*Z8d+Z9d,Z:ej01d-e;d.fed.fed/fej<d0fgd1Z=d2Z>ej01d3eedddeedddeeddd#ehd4eed 5e&ed6e eg d7Z?ej01d8eedd9d&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@dfeedddeedddfeedddeedddfgdAZ@ej01d3eeeAee"e$e%eeeddd#gdBZBej01d3eeeeAe"e$e%ee&eed 5ehd4eehdCeeddd#eedd@d#eeddd#eeddd#eeddd#eedddeeddd&ed6e egdDZCej01dEeeddddFfeeddd#dGfehd4dHfeeDejEejFhejFfeGdIfdJdKddgd9dr?rrBdict__annotations__rFr rJr-r3r0rArA>spFF$'$=D000\555   65   r3rA 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)rrOintervals r0test_interval_ranger[Ks r1V<<>>> H     r1W===H X     ==== r1V<<>>> ==== r1Y???H X     H      r3cbt|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. rrRrXrTAiN)rfloatrYs r0test_interval_large_integersr`_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 == leftrVz(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)paramserrorros r0test_interval_errorsrts@ 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 r0test_stroptionsrzscU;;;G  " "3 ' ''''  " "3 ' ''''&&s+++++ W - - - - - -r3cttddtjhdh}|dsJ|tjsJ|drJdt |vsJdS)z'Sanity check for the Options constraintg?r Gz?z-0.5 (deprecated)N)rrrbrcrvrwrxs r0 test_optionsr~sdT3/TFCCCG  " "4 ( ((((  " "26 * ****&&t,,,,, #g,, . . . . . .r3ztype, expected_type_nameintr_z numpy.ndarraycVt|}t|d|dksJdS)z?Check the string representation of the _InstancesOf constraint.zan instance of ''N)rrw)rkexpected_type_name constraints r0%test_instances_of_type_human_readablers;d##J z??F1CFFF F F F F F Fr3ctddg}Gdd}Gdd}||sJ||rJt|dksJdS) z Check the HasMethods constraint.r&r'ceZdZdZdZdS)test_hasmethods.._GoodcdSrDr-r8s r0r&z test_hasmethods.._Good.a Dr3cdSrDr-rs r0r'z test_hasmethods.._Good.brr3N)r<r=r>r&r'r-r3r0_Goodrs2        r3rceZdZdZdS)test_hasmethods.._BadcdSrDr-rs r0r&ztest_hasmethods.._Bad.arr3N)r<r=r>r&r-r3r0_Badrs#     r3rz"an object implementing 'a' and 'b'N)r rvrw)rrrs r0test_hasmethodsrsS#J''J  % %eegg . ....))$$&&11111 z??B B B B B B Br3r>r&r'r() numeric_onlyrJcRt|}||rJdS)z>Check that the value generated does not satisfy the constraintNr!rv)r bad_values r0test_generate_invalid_param_valrs1"+:66I)))4444444r3zinteger_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 r0+test_generate_invalid_param_val_2_intervalsrsb+mDDDI,,Y77777// :::::*6FGGGI,,Y77777// :::::::r3ctjt5t|ddddS#1swxYwYdS)zmCheck that the function raises NotImplementedError when there's no invalid value for the constraint. N)rprqNotImplementedErrorr!rs r0)test_generate_invalid_param_val_all_validrWs" * + +//":...//////////////////s 7;;>rfrRrcRt|}||sJdS)z;Check that the value generated does satisfy the constraint.N)r"rv)rvalues r0test_generate_valid_paramrls1: ! , ,E  % %e , ,,,,,,r3zconstraint_declaration, valuegzG?*r'c |dzS)Nrfr-)xs r0rs QUr3rLz array-like sparse matrixc4g|]}d|ddgddggfS)rrfrRrrr-).0 containers r0 rsD   ii!Q!Q(899 :   r3) random_staterr)rNr|)booleanF)verboserfrhgimissingrE) cv_objectrcRt|}||sJdS)z+Sanity check for the is_satisfied_by methodN)r#rv)constraint_declarationrrs r0test_is_satisfied_byrs3L!!788J  % %e , ,,,,,,r3z1constraint_declaration, expected_constraint_classoption1option2r}rrrc:t|}|j|usJdS)zICheck that make_constraint dispatches to the appropriate constraint classN)r# __class__)rexpected_constraint_classrs r0test_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 bernwrongrfr(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)rrRrrerrfrRrfour)rf)rprqrr1r-r3r0test_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*cdSrDr-)r&r's r0funcz1test_validate_params_missing_params..funcrKr3rfrRN)r$r)rs r0#test_validate_params_missing_paramsrsG cC5\FFF  GF  DAJJJJJr3ctt}tjtd5|ddddddn #1swxYwYtjtd5tjt d5|ddd dddn #1swxYwYddddS#1swxYwYdS) z5Check that validate_params functions can be decoratedzFunction _func is deprecatedrnrfrRrrNrr)r r1rpwarns FutureWarningrqr)decorated_functions r0 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 bernrNz)Function _deprecated_method is deprecatedz6The 'a' parameter of _Class._deprecated_method must be)rprqrr5r9rrr;r-r3r0test_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 bernN)rArprqrrJ)ests r0test_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 subsetrn>r&r'r(r&r)r N)rprqrrr-r3r0!test_stroptions_deprecated_subsetr/s 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*cdSrDr-rs r0fz!test_hidden_constraint..f8s  r3rfrRr)r&r'r(rThe 'param' parameterrnbadrNzan instance of 'dict'zan instance of 'list') r$rlistrMrprqrrwrrexc_infoerr_msgs r0test_hidden_constraintr5s) 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*cdSrDr-rs r0rz!test_hidden_stroptions..fP r3rrnrrN)r$rrrprqrrwrrs r0test_hidden_stroptionsrMs/ :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)hasattrr1r5r9r-r3r04test_validate_params_set_param_constraints_attributerfs? 56 7 7777 688#%A B BBBBBBr3ctddgidd}|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*cdSrDr-rs r0rz1test_boolean_constraint_deprecated_int..fsrKr3FN)r$rbbool_rs r0&test_boolean_constraint_deprecated_intrns` g {+4PPP  QP AdGGGAbhuoor3c(ttdgddddd}tjtd5|d dddn #1swxYwYGd d }|| || dS)z5Check that validation can be skipped for a parameter.N no_validationparam1param2Tr*cdSrDr-rs r0rztest_no_validation..frr3zThe 'param1' parameterrnr)rceZdZdS)$test_no_validation..SomeTypeN)r<r=r>r-r3r0SomeTypers r3r)rrLr$rrprqr)rrs r0test_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)rp importorskiprrvNArbarray)pd na_constraints r0$test_pandas_na_constraint_with_pd_narsg  X & &B'))M  ( ( / ////,,RXiii-@-@AAAAAAAr3ct}|gdsJ|tdsJ|drJdS)zGCheck that a string does not satisfy the _IterableNotString constraint.rrz some stringN)rrvrangers r0test_iterable_not_stringr sk$&&J  % %iii 0 0000  % %eBii 0 0000))-8888888r3c<t}|dsJ|tsJ|ddgddgfddgddgfgsJ|dsJ|drJdS)zUCheck that the _CVObjects constraint accepts all current ways to pass cv objects.rrfrRrrNznot a CV object)rrvr rs r0test_cv_objectsr sJ  % %a ( ((((  % %kmm 4 4444  % %AA'71a&1a&9I&J K KKKK  % %d + ++++))*;<<<<<<.ThirdPartyEstimatorcZ||_tddS)NrrE)r'superrF)r8r'rs r0rFz@test_third_party_estimator..ThirdPartyEstimator.__init__s+DF GG  q  ! ! ! ! !r3NcLt||dSrD)rrJ)r8rHrIrs r0rJz;test_third_party_estimator..ThirdPartyEstimator.fits! GGKK1     r3rL)r<r=r>rFrJ __classcell__)rs@r0ThirdPartyEstimatorrsV " " " " "          r3rrr'N)rArJ)rs r0test_third_party_estimatorrsT j!  """""r3cttddd}|dsJ|drJdS)z9Check for the type RealNotInt in the Interval constraint.rrfrWrTrjN)rrrvrs r0test_interval_real_not_intrsR*a6:::J  % %c * ****))!,,,,,,,r3ctdtsJtdtrJttjdtsJttjdtrJdS)zCheck for the RealNotInt type.rjrfN) isinstancerrbfloat64int64r-r3r0test_real_not_intrsp c: & &&&&!Z((((( bjmmZ 0 0000"(1++z2222222r3c2tdtgidd}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*cdSrDr-rEs r0rz%test_skip_param_validation..frKr3The 'a' parameterrn1rENskip_parameter_validation)r$rrprqrrrs r0test_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*cdSrDr-rEs r0rz&test_skip_nested_validation..frKr3r'cdS)Ninvalid_param_valuerEr-)r'rs r0gz&test_skip_nested_validation..gs q())))r3zThe 'b' parameterrnr)rNrfr!r)r+r*rs @r0test_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$)rrEs r0r*z9test_skip_nested_validation_and_config_context..gs||788r3r#rfN)r$rr)r$r+expected_skippedr*actual_skippeds r0.test_skip_nested_validation_and_config_contextr0s se 4Q999 2K L L L1 - - - - - - -s A  AA)r)dnumbersrrnumpyrbrp scipy.sparsersklearn._configrr sklearn.baserr sklearn.model_selectionr sklearn.utilsr sklearn.utils._param_validationr rrrrrrrrrrrrrrrrrrr r!r"r#r$sklearn.utils.fixesr%r1r5rAmark parametrizer[r`reri TypeErrorrrtrzr~rndarrayrrrrrrrrkfloat32rcallablerrandom RandomStaterhr_rrrrrrrrrrrrrrrr r rrrr%r+r0r-r3r0rBs]"""""""" ######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   +       ..q112  ! " # $ a% & s ' ( ) * + , - . "/ 0 $1 2 !T'"3 4 $5 6 %%,,'7 8 "&!9 : )$; < E  JJOOO,= > ? ##H--I##H- 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.....r3