Q(h5_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/ej`jcdeegdZ2ej`jcdeegdZ3dZ4ej`jcdeedddeedddgdZ5ej`jcdedddd e6d!fedd"d#d e6d$fedddd e7d%feddd&d e7d'fedd(dd e7d)fgd*Z8d+Z9d,Z:ej`jcd-e;d.fed.fed/fejxd0fgd1Z=d2Z>ej`jcd3eedddeedddeeddd#ehd4eed 5e&ed6e eg d7Z?ej`jcd8eedd9d&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@ej`jcd3eeeAee"e$e%eeeddd#gdBZBej`jcd3eeeeAe"e$e%ee&eed 5ehd4eehdCeeddd#eedd@d#eeddd#eeddd#eeddd#eedddeeddd&ed6e egdDZCej`jcdEeeddddFfeeddd#dGfehd4dHfeeDejejhejfeGdIfdJdKddgd9dr?rrBdict__annotations__rFr rJr-r3r0rArA>s4F$'$=D05 6 r3rA 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)rrOintervals r0test_interval_ranger[Ks r1V> H   r1W=H X   == r1V> == r1Y?H X   H  r3c0t|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. rrRrXrTAiN)rfloatrYs r0test_interval_large_integersr`_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 == leftrVz(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)paramserrorros r0test_interval_errorsrts3@ uE *6s -6cthddh}|jdsJ|jdsJ|jdrJdt|vsJy)z*Sanity check for the StrOptions constraint>r&r'r(r(r r&r)z'c' (deprecated)N)ris_satisfied_bystroptionss r0test_stroptionsrzs^cU;G  " "3 '' '  " "3 '' '&&s++ + W -- -r3cttddtjhdh}|j dsJ|j tjsJ|j drJdt |vsJy)z'Sanity check for the Options constraintg?r Gz?z-0.5 (deprecated)N)rrrbrcrvrwrxs r0 test_optionsr~sodT3/TFCG  " "4 (( (  " "266 ** *&&t,, , #g, .. .r3ztype, expected_type_nameintr_z numpy.ndarraycBt|}t|d|dk(sJy)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? 01C0DAF FF Fr3ctddg}Gdd}Gdd}|j|sJ|j|rJt|dk(sJy) z Check the HasMethods constraint.r&r'ceZdZdZdZy)test_hasmethods.._GoodcyrDr-r8s r0r&z test_hasmethods.._Good.a r3cyrDr-rs r0r'z test_hasmethods.._Good.brr3N)r<r=r>r&r'r-r3r0_Goodrs   r3rceZdZdZy)test_hasmethods.._BadcyrDr-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_hasmethodsrseS#J'J  % %eg .. .))$&11 1 z?B BB Br3r>r&r'r() numeric_onlyrJc@t|}|j|rJy)z>Check that the value generated does not satisfy the constraintNr!rv)r bad_values r0test_generate_invalid_param_valrs%"+:6I)))44 44r3zinteger_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 r0+test_generate_invalid_param_val_2_intervalsrsmb+mDI,,Y77 7// :: :*6FGI,,Y77 7// :: ::r3cvtjt5t|dddy#1swYyxYw)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 /8>rfrRrc@t|}|j|sJy)z;Check that the value generated does satisfy the constraint.N)r"rv)rvalues r0test_generate_valid_paramrls": ! ,E  % %e ,, ,r3zconstraint_declaration, valuegzG?*r'c |dzS)Nrfr-)xs r0rs QUr3rLz array-likez sparse matrix) random_staterr)rNr|)booleanF)verboserfrhgimissingrE) cv_objectrc@t|}|j|sJy)z+Sanity check for the is_satisfied_by methodN)r#rv)constraint_declarationrrs r0test_is_satisfied_byrs$L!!78J  % %e ,, ,r3z1constraint_declaration, expected_constraint_classoption1option2r}rrrc:t|}|j|usJy)zICheck that make_constraint dispatches to the appropriate constraint classN)r# __class__)rexpected_constraint_classrs r0test_make_constraintrs$,!!78J   #< << 201222s 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 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_paramsrss %I  g %I " 7|!q!" %I # "C>"# %I ! a1 ! %I 4 a3/3Q33(34 %I 3 a2)2v2#q2335 "" ## !! 44 33sGD9 E>E/E!E*E69EEEE'*E36E?cJtdtgidd}|ddy)zTCheck that no error is raised when there are parameters without constraints r&Tr*cyrDr-)r&r's r0funcz1test_validate_params_missing_params..funcrKr3rfrRN)r$r)rs r0#test_validate_params_missing_paramsrs/ cC5\F G  AJr3cxtt}tjtd5|ddddddtjtd5tj t d5|ddd ddddddy#1swY]xYw#1swYxYw#1swYyxYw) 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&1a1%& m+I J0 ]] !)N  0 q!w / 000 &&  0 000s/ 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 bernrNz)Function _deprecated_method is deprecatedz6The 'a' parameter of _Class._deprecated_method must be)rprqrr5r9rrr;r-r3r0test_validate_params_methodrs %R " !" m+V W1 ]] !J  1 H ' ' 0  111 "" 1 111s/B B85B,B8 B),B5 1B88Cctd}tjtd5|j dddy#1swYyxYw)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   s AA ctjtd5thdddhdddy#1swYyxYw)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/s4 z)N O;?Sz:;;;s 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*cyrDr-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  AAAiL %<    (..!G "g -- - "' 11 1s  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*cyrDr-rs r0rz!test_hidden_stroptions..fP r3rrnrrN)r$rrrprqrrwrrs r0test_hidden_stroptionsrMs :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)hasattrr1r5r9r-r3r04test_validate_params_set_param_constraints_attributerfs. 56 77 7 68##%A BB Br3cvtddgidd}|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*cyrDr-rs r0rz1test_boolean_constraint_deprecated_int..fsrKr3FN)r$rbbool_rs r0&test_boolean_constraint_deprecated_intrns> g {+4P Q dGbhhuor3cttdgddddd}tjtd5|d dddGd d }|| || y#1swY+xYw)z5Check that validation can be skipped for a parameter.N no_validationparam1param2Tr*cyrDr-rs r0rztest_no_validation..frr3zThe 'param1' parameterrnr)rc eZdZy)$test_no_validation..SomeTypeN)r<r=r>r-r3r0SomeTypers r3r)rrLr$rrprqr)rrs r0test_no_validationr|sw;/:&*   ,4L M   XXZs A--A6ctjd}t}|j|jsJ|jt j gdrJy)z9Add a specific test for checking support for `pandas.NA`.pandasrN)rp importorskiprrvNArbarray)pd na_constraints r0$test_pandas_na_constraint_with_pd_narsR   X &B')M  ( ( // /,,RXXi-@AA AAr3ct}|jgdsJ|jtdsJ|jdrJy)zGCheck that a string does not satisfy the _IterableNotString constraint.rrz some stringN)rrvrangers r0test_iterable_not_stringrsM$&J  % %i 00 0  % %eBi 00 0))-88 88r3ct}|jdsJ|jtsJ|jddgddgfddgddgfgsJ|jdsJ|jdrJy)zUCheck that the _CVObjects constraint accepts all current ways to pass cv objects.rrfrRrrNznot a CV object)rrvr rs r0test_cv_objectsrsJ  % %a (( (  % %km 44 4  % %AA'71a&1a&9I&J KK K  % %d ++ +))*;<< <.ThirdPartyEstimatorc4||_t| dy)NrrE)r'superrF)r8r'rs r0rFz@test_third_party_estimator..ThirdPartyEstimator.__init__sDF G q  !r3c&t|||yrD)r rJ)r8rHrIrs r0rJz;test_third_party_estimator..ThirdPartyEstimator.fits GK1 r3rL)r<r=r>rFrJ __classcell__)rs@r0ThirdPartyEstimatorr s "  r3rrr'N)rArJ)rs r0test_third_party_estimatorrs! j!  "r3cvttddd}|jdsJ|jdrJy)z9Check for the type RealNotInt in the Interval constraint.rrfrWrTrjN)rrrvrs r0test_interval_real_not_intrs=*a6:J  % %c ** *))!,, ,,r3ctdtsJtdtrJttjdtsJttjdtrJy)zCheck for the RealNotInt type.rjrfN) isinstancerrbfloat64int64r-r3r0test_real_not_intrsU c: && &!Z(( ( bjjmZ 00 0"((1+z22 22r3ctdtgidd}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*cyrDr-rEs r0rz%test_skip_param_validation..frKr3The 'a' parameterrn1rENskip_parameter_validation)r$rrprqrrrs r0test_skip_param_validationr s}cC5\F G  ,4G H C $ 7 C s 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*cyrDr-rEs r0rz&test_skip_nested_validation..frKr3r'cdS)Ninvalid_param_valuerEr-)r'rs r0gz&test_skip_nested_validation..gs ())r3zThe 'b' parameterrnr$rNrfrr)r+r%rs @r0test_skip_nested_validationr&scC5\F G  se &C* * ,4G H# !"#% A ]]08K L  F   ##   s 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)rrEs r0r%z9test_skip_nested_validation_and_config_context..gs|788r3rrfN)r$rr)rr+expected_skippedr%actual_skippeds r0.test_skip_nested_validation_and_config_contextr+sa se 4Q99 2K L1 - -- -s 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_rrrrrrrrrrrrrrrrrrrrr r&r+) containers0r0r>ss " #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 , 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..U  s1]8