t]e\zdZddlmZmZddlZddlZddlmZddl m Z ddl m Z ddl mZddlmZmZdd lmZdd lmZmZmZmZmZmZmZmZmZmZmZm 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{dddZ.GddZ/Gddee Z0ej12deegdZ3dZ4ej12deedddeedddgdZ5ej12dedd dd!e6d"fedd#d$d!e6d%fedddd!e7d&feddd'd!e7d(fedd)dd!e7d*fgd+Z8d,Z9d-Z:ej12d.e;d/fed/fed0fej<d1fgd2Z=d3Z>ej12d4eedddeedddeeddd$ehd5eed 6e'ed7e"e g d8Z?ej12d9eedd:d'eed;dd$feed;dd$feed;dd'feed;d= 1.2. )IntegralRealN) csr_matrix) BaseEstimator) LeaveOneOut deprecated)config_context get_config)_ParamsValidationMixin) HasMethodsHiddenIntervalInvalidParameterError MissingValuesOptions RealNotInt StrOptions _ArrayLikes _Booleans _Callables _CVObjects _InstancesOf_IterablesNotString_NoneConstraint_PandasNAConstraint _RandomStates_SparseMatrices_VerboseHelpergenerate_invalid_param_valgenerate_valid_parammake_constraintvalidate_params) _fit_context)abcdTprefer_skip_nested_validation)r(cdS)z/A function to test the validation of functions.N)r%r&r'r(argskwargss Jlib/python3.11/site-packages/imblearn/utils/tests/test_param_validation.py_funcr0,ceZdZdZedegiddZeedegiddZdS)_ClasszJA class to test the _InstancesOf constraint and the validation of methods.r%Tr)cdS)zA validated methodNr,selfr%s r/_methodz_Class._method7r1r2cdS)zA deprecated validated methodNr,r6s r/_deprecated_methodz_Class._deprecated_method;r1r2N) __name__ __module__ __qualname____doc__r#rr8r r:r,r2r/r4r44sTT_cD6]$GGG!!HG!Z\\_cD6]$GGG,,HG\,,,r2r4cZeZdZUdZdegiZeed<dZe dd dZ dS) _EstimatorzrrAdict__annotations__rEr$rIr,r2r/r@r@AspFF$'$=D000\555   65   r2r@ interval_typect|ddd}d|vrd|vsJt|ddd}d|vrd|vsJt|ddd}d|vrd|vsJt|ddd}d|vrd|vsJdS) z.Check the range of values depending on closed.leftclosedrightbothneitherN)r)rNintervals r/test_interval_rangerYNs r1V<<>ax//// r1W===H X  !x---- r1V<<>a8mmmm r1Y???H X  !8"3"3"3"3"3"3r2cttddd}tj|vsJttddd}tj |vsJttddd}tj|vsJtj |vsJdS)zjCheck that inf is included iff a bound is closed and set to None. Only valid for real intervals. rNrUrSrRrW)rrnpinfrXs r/test_interval_inf_in_boundsr^^s ag666H 6X    dAf555H F7h    dD;;;H 6 ! ! ! ! F7( " " " " " "r2rXrRrSrVc&tj|vsJdS)z)Check that np.nan is not in any interval.N)r[nanr]s r/test_nan_not_in_intervalrbns 6 ! ! ! ! ! !r2zparams, error, match?rQ)typerRrUrTz=Expecting left to be an int for an interval over the integersg@rWz>Expecting right to be an int for an interval over the integersz&left can't be None when closed == leftrUz(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)paramserrorrhs r/test_interval_errorsrmws@ uE * * *6s 155cthddh}|dsJ|dsJ|drJdt|vsJdS)z*Sanity check for the StrOptions constraint>r%r&r'r'rr%r(z'c' (deprecated)N)ris_satisfied_bystroptionss r/test_stroptionsrsscU;;;G  " "3 ' ''''  " "3 ' ''''&&s+++++ W - - - - - -r2cttddtjhdh}|dsJ|tjsJ|drJdt |vsJdS)z'Sanity check for the Options constraintg?rGz?z-0.5 (deprecated)N)rrr[r\rorprqs r/ test_optionsrwsdT3/TFCCCG  " "4 ( ((((  " "26 * ****&&t,,,,, #g,, . . . . . .r2ztype, expected_type_nameintfloatz numpy.ndarraycVt|}t|d|dksJdS)z?Check the string representation of the _InstancesOf constraint.zan instance of ''N)rrp)rdexpected_type_name constraints r/%test_instances_of_type_human_readabler~s;d##J z??F1CFFF F F F F F Fr2ctddg}Gdd}Gdd}||sJ||rJt|dksJdS) z Check the HasMethods constraint.r%r&ceZdZdZdZdS)test_hasmethods.._GoodcdSrCr,r7s r/r%z test_hasmethods.._Good.a Dr2cdSrCr,rs r/r&z test_hasmethods.._Good.brr2N)r;r<r=r%r&r,r2r/_Goodrs2        r2rceZdZdZdS)test_hasmethods.._BadcdSrCr,rs r/r%ztest_hasmethods.._Bad.arr2N)r;r<r=r%r,r2r/_Badrs#     r2rz"an object implementing 'a' and 'b'N)r rorp)r}rrs r/test_hasmethodsrsS#J''J  % %eegg . ....))$$&&11111 z??B B B B B B Br2r}>r%r&r') numeric_onlyrIcRt|}||rJdS)z>Check that the value generated does not satisfy the constraintNr ro)r} bad_values r/test_generate_invalid_param_valrs1"+:66I)))4444444r2zinteger_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. r}Nr)integer_interval real_intervalrs r/+test_generate_invalid_param_val_2_intervalsrsb+mDDDI,,Y77777// :::::*6FGGGI,,Y77777// :::::::r2ctjt5t|ddddS#1swxYwYdS)zmCheck that the function raises NotImplementedError when there's no invalid value for the constraint. N)rirjNotImplementedErrorr rs r/)test_generate_invalid_param_val_all_validrAs" * + +//":...//////////////////s 7;;>r_rQrcRt|}||sJdS)z;Check that the value generated does satisfy the constraint.N)r!ro)r}values r/test_generate_valid_paramrVs1: ! , ,E  % %e , ,,,,,,r2zconstraint_declaration, valuegzG?*r&c |dzS)Nr_r,)xs r/r~s QUr2rKz array-likez sparse matrix) random_staterr)rNru)booleanF)verboser_gramissingrD) cv_objectrcRt|}||sJdS)z+Sanity check for the is_satisfied_by methodN)r"ro)constraint_declarationrr}s r/test_is_satisfied_byrws3@!!788J  % %e , ,,,,,,r2z1constraint_declaration, expected_constraint_classoption1option2rvrrrc:t|}|j|usJdS)zHCheck that make_constraint dispaches to the appropriate constraint classN)r" __class__)rexpected_constraint_classr}s r/test_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 bergwrongr_r'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)rrQrrerr_rQrfour)r_)rirjrr0r,r2r/test_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)cdSrCr,)r%r&s r/funcz1test_validate_params_missing_params..funcrJr2r_rQN)r#rx)rs r/#test_validate_params_missing_paramsrsG cC5\FFF  GF  DAJJJJJr2ctt}tjtd5|ddddddn #1swxYwYtjtd5tjt d5|ddd dddn #1swxYwYddddS#1swxYwYdS) z5Check that validate_params functions can be decoratedzFunction _func is deprecatedrgr_rQrrNrr)r r0riwarns FutureWarningrjr)decorated_functions r/ 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 bergrNz)Function _deprecated_method is deprecatedz6The 'a' parameter of _Class._deprecated_method must be)rirjrr4r8rrr:r,r2r/test_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 bergN)r@rirjrrI)ests r/test_validate_params_estimatorrs W  C %N     sA  AActjtd5thdddhddddS#1swxYwYdS)z@Check that the deprecated parameter must be a subset of options.z#deprecated options must be a subsetrg>r%r&r'r%r(rN)rirjrrr,r2r/!test_stroptions_deprecated_subsetrs 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)cdSrCr,rs r/fz!test_hidden_constraint..fs  r2r_rQr)r%r&r'rThe 'param' parameterrgbadrNzan instance of 'dict'zan instance of 'list') r#rlistrLrirjrrprrexc_infoerr_msgs r/test_hidden_constraintrs) 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)cdSrCr,rs r/rz!test_hidden_stroptions..f3 r2rrgrrN)r#rrrirjrrprrs r/test_hidden_stroptionsr0s/ :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)hasattrr0r4r8r,r2r/4test_validate_params_set_param_constraints_attributerIs? 56 7 7777 688#%A B BBBBBBr2ctddgidd}|d|tjdtjt d5|d d d d d S#1swxYwYd S) zCheck that validate_params raise a deprecation message but still passes validation when using an int for a parameter accepting a boolean. rrTr)cdSrCr,rs r/rz1test_boolean_constraint_deprecated_int..fVrJr2Fz4Passing an int for a boolean parameter is deprecatedrgr_N)r#r[bool_rirrrs r/&test_boolean_constraint_deprecated_intrQs g {+4PPP  QP AdGGGAbhuoo S      !                  s! A::A>A>c(ttdgddddd}tjtd5|d dddn #1swxYwYGd d }|| || dS)z5Check that validation can be skipped for a parameter.N no_validationparam1param2Tr)cdSrCr,rs r/rztest_no_validation..fhrr2zThe 'param1' parameterrgr)rceZdZdS)$test_no_validation..SomeTypeN)r;r<r=r,r2r/SomeTyperts r2r)rrKr#rxrirjr)rrs r/test_no_validationres#;/::&*     ,4L M M M         AXAXXZZs AA!Actjd}t}||jsJ|t jgdrJdS)z9Add a specific test for checking support for `pandas.NA`.pandasrN)ri importorskiprroNAr[array)pd na_constraints r/$test_pandas_na_constraint_with_pd_nar{sg  X & &B'))M  ( ( / ////,,RXiii-@-@AAAAAAAr2ct}|gdsJ|tdsJ|drJdS)zGCheck that a string does not satisfy the _IterableNotString constraint.rrz some stringN)rrorangers r/test_iterable_not_stringrsk$&&J  % %iii 0 0000  % %eBii 0 0000))-8888888r2c<t}|dsJ|tsJ|ddgddgfddgddgfgsJ|dsJ|drJdS)zUCheck that the _CVObjects constraint accepts all current ways to pass cv objects.rr_rQrrNznot a CV object)rrorrs r/test_cv_objectsrsJ  % %a ( ((((  % %kmm 4 4444  % %AA'71a&1a&9I&J K KKKK  % %d + ++++))*;<<<<<<.ThirdPartyEstimatorcZ||_tddS)NrrD)r&superrE)r7r&rs r/rEz@test_third_party_estimator..ThirdPartyEstimator.__init__s+DF GG  q  ! ! ! ! !r2NcLt||dSrC)rrI)r7rGrHrs r/rIz;test_third_party_estimator..ThirdPartyEstimator.fits! GGKK1     r2rK)r;r<r=rErI __classcell__)rs@r/ThirdPartyEstimatorrsV " " " " "          r2r rr&N)r@rI)r s r/test_third_party_estimatorr sT j!  """""r2cttddd}|dsJ|drJdS)z9Check for the type RealNotInt in the Interval constraint.rr_rVrSrcN)rrrors r/test_interval_real_not_intrsR*a6:::J  % %c * ****))!,,,,,,,r2ctdtsJtdtrJttjdtsJttjdtrJdS)zCheck for the RealNotInt type.rcr_N) isinstancerr[float64int64r,r2r/test_real_not_intrsp c: & &&&&!Z((((( bjmmZ 0 0000"(1++z2222222r2c2tdtgidd}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)cdSrCr,rDs r/rz%test_skip_param_validation..frJr2The 'a' parameterrg1rDNskip_parameter_validation)r#rxrirjrr rs r/test_skip_param_validationrs5cC5\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)cdSrCr,rDs r/rz&test_skip_nested_validation..frJr2r&cdS)Ninvalid_param_valuerDr,)r&rs r/gz&test_skip_nested_validation..gs q())))r2zThe 'b' parameterrgrr Nr_rr)r*rrs @r/test_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)r rDs r/rz9test_skip_nested_validation_and_config_context..gs||788r2rr_N)r#rxr )rr*expected_skippedractual_skippeds r/.test_skip_nested_validation_and_config_contextr%s se 4Q999 2K L L L1 - - - - - - -s A  AA)r)dr>numbersrrnumpyr[ri scipy.sparser sklearn.basersklearn.model_selectionr sklearn.utilsr imblearn._configr r imblearn.baser imblearn.utils._param_validationr rrrrrrrrrrrrrrrrrrr r!r"r#imblearn.utils.fixesr$r0r4r@mark parametrizerYr^rb TypeErrorrrmrsrwrxndarrayr~rrrrrrrdfloat32rcallablerrandom RandomStateryrarrrrrrrrrrrrrrrrrr rrrr r%r,r2r/r8sv#""""""" ######&&&&&&//////$$$$$$777777770000002.----- &vTF$88"&:a:::: : , , , , , , , ,      '    8T*:;; 4 4<; 4 # # #  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  " $ $ %%,,' "&! )$ E  JJOOO,3<--=<- 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.....r2