
    Q3jW5                        d Z ddlZddlZddlZddlZddlmZ ddlmZ ddl	Z	ddl
mZ ddlZddlmZ ddlmZ ddlmZ dd	lmZmZ dd
lmZ ddlmZmZ ddlmZmZmZmZ ddl m!Z! ddl"m#Z#m$Z$m%Z%m&Z& ddl'm(Z(m)Z) ddl*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8 ddl9m:Z: e	jv                  jx                  d        Z=dDdZ> G d de      Z?e	jv                  j                  d ee>d      dfe>df ee>d      df ed      df eddd d!"      d#f e?       d$fg      d%        ZAe	jv                  j                   e8 eC e%             e$&      d'               ZDe	jv                  j                  d(      e	jv                  jx                  d)               ZFd* ZGe	jv                  jx                  d+        ZHd, ZId- ZJ eC e e%        e ed            g eJ                   ZKe	jv                  j                  d.eKe#/      d0        ZLd1d2gZMd3 ZN e%d4      D  cg c]  }  eN|       r|  c} ZOe	jv                  j                  d4eOe#/      d5        ZP e%       D  cg c]  }  eQ| d6      s|  c} ZRe	jv                  j                  d7eRe#/      d8        ZSe	jv                  j                  d7 eC e%             e#/      d9        ZT eC e e%d4       e e        e              ed:;       ed<=      g            ZUe	jv                  j                  d.eUe#/      d>        ZVe	jv                  j                  d.eUe#/      e	jv                  j                  d?e4e/e5e0g      d@               ZWe	jv                  j                  d7 e%       e#/      dA        ZX e%       D  cg c]  }  e:| dB      s|  c} ZYe	jv                  j                  d7eYe#/      dC        ZZyc c} w c c} w c c} w )Ez.
General tests for all estimators in sklearn.
    N)partial)chain)LinAlgWarning)BaseEstimator)ColumnTransformer)ConvergenceWarning)enable_halving_search_cvenable_iterative_imputer)LogisticRegression)FeatureUnionmake_pipeline)FunctionTransformerMinMaxScalerOneHotEncoderStandardScaler)all_estimators)_get_check_estimator_ids_get_expected_failed_checks_tested_estimators_yield_instances_for_check)SkipTestignore_warnings)#check_all_zero_sample_weights_error(check_dataframe_column_names_consistencycheck_estimator!check_get_feature_names_out_error$check_global_output_transform_pandas(check_global_set_output_transform_polarscheck_inplace_ensure_writeablecheck_param_validationcheck_set_output_transform!check_set_output_transform_pandas!check_set_output_transform_polars'check_transformer_get_feature_names_out.check_transformer_get_feature_names_out_pandasparametrize_with_checks)has_fit_parameterc                      t               D ]<  \  } }dj                  |       }| j                         j                  d      s7J |        y )NzDBase estimators such as {0} should not be included in all_estimatorsbase)r   formatlower
startswith)name	Estimatormsgs      F/DATA/.local/lib/python3.12/site-packages/sklearn/tests/test_common.py test_all_estimator_no_base_classr1   B   sG     *+iR
&, 	 ::<**627C72	 ,       c                      y N )xys     r0   _sample_funcr9   L   s    r2   c                       e Zd ZdZd Zy)CallableEstimatorzqDummy development stub for an estimator.

    This is to make sure a callable estimator passes common tests.
    c                      y r5   r6   )selfs    r0   __call__zCallableEstimator.__call__V   s    r2   N)__name__
__module____qualname____doc__r>   r6   r2   r0   r;   r;   P   s    
r2   r;   zval, expected)r8   z_sample_func(y=1)r9   worldg       @)CzLogisticRegression(C=2.0)z	newton-cgbalancedT)random_statesolverclass_weight
warm_startz]LogisticRegression(class_weight='balanced',random_state=1,solver='newton-cg',warm_start=True)zCallableEstimator()c                 $    t        |       |k(  sJ y r5   )r   )valexpecteds     r0   test_get_check_estimator_idsrM   Z   s    . $C(H444r2   )expected_failed_checksc                 ~    t        t        t        t        t        f      5   ||        d d d        y # 1 sw Y   y xY wNcategory)r   FutureWarningr   UserWarningr   )	estimatorcheckrequests      r0   test_estimatorsrX   t   s3     
!3[-P
 	i
 
 
s   	3<zZignore:Since version 1.0, it is not needed to import enable_hist_gradient_boosting anymorec                  ~   t         j                  j                  t        j                        g} t        j                  | dd       }|D cg c]  \  }}}|
 }}}|dgz   D ]R  }d|v sd|v rd|v rt        |d	      }t        |d
d      D ]%  }t        ||      rJ dj                  ||              T y c c}}w )Nsklearn.c                      y r5   r6   _s    r0   <lambda>z-test_import_all_consistency.<locals>.<lambda>   s    r2   )pathprefixonerrorsklearnz.tests.zsklearn.externalszsklearn._build_utilsdummy)fromlist__all__r6   z#Module '{0}' has no attribute '{1}')osr_   dirnamerb   __file__pkgutilwalk_packages
__import__getattrhasattrr*   )sklearn_pathpkgsr]   modnamesubmodspackager-   s          r0   test_import_all_consistencyrs      s     GGOOG$4$456L   *nD -11D=1gqwDG1i[(#6'#A!W,Ww7GY3D7D) +P+W+W, ) 4 ) 2s   B9c                     t         j                  j                  t        j                        g} d}t        j                  | d       D ]4  \  }}}d|v s|j                  d      s||v r!|t        j                  v r4J  y )N)utilstestsr)   conftestc                      y r5   r6   r\   s    r0   r^   z3test_root_import_all_completeness.<locals>.<lambda>   s    Tr2   )r_   ra   .r]   )	rf   r_   rg   rb   rh   ri   rj   r,   re   )rn   
EXCEPTIONSr]   rp   s       r0   !test_root_import_all_completenessr{      st    GGOOG$4$456L7J ..>7A '>W//4:8M'//)))r2   c                     t        j                  d      } h d}t        j                  j	                  t
        j                        g}t        j                  |d      D ci c]	  \  }}}|| }}}}|j                         D cg c]%  \  }}|r||vr| j                  |      s	|dz   |vr|' }}}|g k(  sJ dj                  |             y c c}}}w c c}}w )NzK(?x)
        \.externals(\.|$)|
        \.tests(\.|$)|
        \._
        >   sklearn.datasets.datasklearn.datasets.descrsklearn.datasets.imagesrZ   )r`   z.testszn{0} do not have `tests` subpackages. Perhaps they require __init__.py or a meson.build in the parent directory)recompilerf   r_   rg   rb   rh   ri   rj   itemssearchr*   )HAS_TESTS_EXCEPTIONSresource_modulesrn   r]   r-   ispkglookupmissing_testss           r0   test_all_tests_are_importabler      s    
 ::	
 GGOOG$4$456L &33LTTNAtU 	eT   "<<>)KD%(($++D18O6) 	)   B 	 F=)s   #C*Cc                     d} t        j                  t        |       5  t        t               d d d        t        j                  t        |       5  t        t        g       d d d        y # 1 sw Y   >xY w# 1 sw Y   y xY w)Nz8Passing a class was deprecated.* isn't supported anymore)match)pytestraises	TypeErrorr   r   r&   r/   s    r0   test_class_support_removedr      s\     EC	y	,*+ 
- 
y	,!3 45 
-	, 
-	, 
-	,s   A+A7+A47B c               #     K   t               D ]  } t        | j                               }d|v r| j                  dd       6d|v rh| j                  dd      }|j                  j
                  dv r8t        j                  |t        j                  j                  d	      
       | d|v s| j                  d        y w)N	oob_scoreT)r   	bootstrapearly_stoppingr3   )r   n_iter_no_change>   MLPRegressorMLPClassifierzMLP still validates in fitr   )marksr   )r   )
r   set
get_params
set_params	__class__r?   r   parammarkxfail)rU   
est_paramsests      r0   _estimators_that_predict_in_fitr      s     ')	--/0
*$&&&FF+&&dQ&OC}}%%)JJllv{{005Q0R  	:-&&&:: *s   B1C4Cestimator_orig)idsc                    t        | t              rt        j                  d       dt	        |       v rt        j                  d       t        t        |       D ]{  }t        t              5  t        j                  d      5 }t        |j                  j                  |       d d d        D ]  }dt        |j                        vrJ  	 d d d        } y # 1 sw Y   5xY w# 1 sw Y   xY w)Nz$ColumnTransformer is not tested herer   zCEstimator does not support check_dataframe_column_names_consistencyrQ   T)recordz was fitted without feature names)
isinstancer   r   skipr   r   r   r   rS   warningscatch_warningsr   r?   strmessage)r   rU   r   warnings       r0   #test_pandas_column_name_consistencyr      s     ."34:;15P6  	Q	
 00.	 }6((58''00) 6 "9W__AUUUU " 76 65 76s*   +C#!C#$C#C#C C##C,	ensemblekernel_approximationc                 h    t        | d      ry| j                  j                  d      d   }|t        vS )Nget_feature_names_outTry   r3   )rm   r@   split"GET_FEATURES_OUT_MODULES_TO_IGNORE)transformermodules     r0   '_include_in_get_feature_names_out_checkr     s7    {34##))#.q1F;;;r2   r   c                     t        t              5  t        | j                  j                  |        t        | j                  j                  |        d d d        y # 1 sw Y   y xY wrP   )r   rS   r$   r   r?   r%   )r   s    r0   'test_transformers_get_feature_names_outr   $  sL     
=	2/!!**K	
 	7!!**K	
	 
3	2	2s   AAA$r   rU   c                 H    | j                   j                  }t        ||        y r5   )r   r?   r   )rU   estimator_names     r0   +test_estimators_get_feature_names_out_errorr   6  s     ((11N%ni@r2   c                     t        | t              rt        j                  d       | j                  j
                  }t        ||        y )NzFeatureUnion is not tested here)r   r   r   r   r   r?   r    rU   r-   s     r0   test_check_param_validationr   >  s6     )\*56''D4+r2   F)sparse_outputz
one-to-one)feature_names_outc                 .   | j                   j                  }t        | d      st        j                  d| d       t        t        |       D ];  }t        t              5  t        |j                   j                  |       d d d        = y # 1 sw Y   HxY w)N
set_outputz(Skipping check_set_output_transform for z!: Does not support set_output APIrQ   )	r   r?   rm   r   r   r   r!   r   rS   )r   r-   rU   s      r0   test_set_output_transformr   T  s     ##,,D><06tf = 	
 0"N	 }6&y':':'C'CYO 76 76s   !BB	
check_funcc                 :   | j                   j                  }t        | d      s&t        j                  d|j                   d| d       t        ||       D ]8  }t        t              5   ||j                   j                  |       d d d        : y # 1 sw Y   ExY w)Nr   z	Skipping z for z%: Does not support set_output API yetrQ   )r   r?   rm   r   r   r   r   rS   )r   r   r-   rU   s       r0   $test_set_output_transform_configuredr   e  s     ##,,D><0
++,E$ 8" "	
 0
NK	}6y**33Y? 76 L66s   (BB	c                 l   | j                   j                  }t        | d      r| j                  d       n-t        | d      r| j                  d       nt	        | d      |dk(  r| j                  dd	
       |dk(  r| j                  d       |dk(  r| j                  d       t        ||        y )NcopyF)r   copy_X)r   z! doesn't require writeable input.HDBSCANprecomputedbrute)metric	algorithmPCAfull)
svd_solver	KernelPCA)kernel)r   r?   rm   r   r   r   r   s     r0   #test_check_inplace_ensure_writeabler   }  s     ''Dy&!%(	H	%E*$@ABB yMWEu}/{M2"43r2   sample_weightc                 H    | j                   j                  }t        ||        y r5   )r   r?   r   r   s     r0   (test_check_all_zero_sample_weights_errorr     s     ''D'i8r2   )r3   )[rB   rf   ri   r   r   	functoolsr   	itertoolsr   r   scipy.linalgr   rb   sklearn.baser   sklearn.composer   sklearn.exceptionsr   sklearn.experimentalr	   r
   sklearn.linear_modelr   sklearn.pipeliner   r   sklearn.preprocessingr   r   r   r   sklearn.utilsr   -sklearn.utils._test_common.instance_generatorr   r   r   r   sklearn.utils._testingr   r   sklearn.utils.estimator_checksr   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   sklearn.utils.validationr'   r   thread_unsafer1   r9   r;   parametrizerM   no_check_spmatrixlistrX   filterwarningsrs   r{   r   r   r   column_name_estimatorsr   r   r   GET_FEATURES_OUT_ESTIMATORSr   rm   %ESTIMATORS_WITH_GET_FEATURE_NAMES_OUTr   r   SET_OUTPUT_ESTIMATORSr   r   r   #ESTIMATORS_ACCEPTING_SAMPLE_WEIGHTSr   )r   s   0r0   <module>r      s   
  	     &  & - 1 4 8  )      7 8 8	  		#%89	~&	w	'8	c	"$?@"'	6	
 
	34#,5-,5 		7R  ,  	
** # #L	6;* 		)A.	/0')  ,2J  VV0 & "< "-00.s3 0  .4L  

 &')'C738O+PC') %
 6<T  AA
 (*+1I  ,, 	=).*LN;.,?	
	  +1I  PP +1I   ),)0		@	@ #%+C  440 &'''C+<S/+RC'' #
 4:R  99E()L's   OO%OO"O