
    0Ph                     \   d Z ddlZddlZddlZddlZddlZd Zej	        
                    d          ej	        
                    d          ej	                             e             d          dd	ej	        
                    d          id
Zd Zd Z	 ddZd ZddZej	                            d ed                    d             Zej	                            d ed                    d             Zej	                            d ed                    d             ZdS )z&Test loaders for common functionality.    Nc                  2    	 dd l } dS # t          $ r Y dS w xY w)Nr   TF)PILImportError)r   s    b/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/sklearn/datasets/tests/test_common.pyis_pillow_installedr      s7    


t   uus    
z.X is a list and does not have a shape argument)reasonz)fetch_opeml requires a dataset name or idzpillow is not installed)fetch_20newsgroupsfetch_openmlfetch_lfw_peopler
   )
return_X_yas_framec                     	 dd l }t          j        d           d S # t          $ rQ | j        }| d}t          j        t          |          5   | d           d d d            Y d S # 1 swxY w Y   Y d S w xY w)Nr   z-This test requires pandas to not be installedz# with as_frame=True requires pandas)matchTr   )pandaspytestskipr   __name__raises)
fetch_funcr   nameexpected_msgs       r   check_pandas_dependency_messager   )   s    
&CDDDDD & & & "CCC];l;;; 	& 	&J%%%%	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	&&s3    1A7A(A7(A,	,A7/A,	0A76A7c                      |d          }t          |t                    sJ |d         j        | j        j        k    sJ |d         j        | j        j        k    sJ d S )NT)r   r      )
isinstancetupleshapedatatarget)bunchdataset_func	X_y_tuples      r   check_return_X_yr$   7   sj    ---Ii'''''Q<!11111Q<!3333333    c                 P   t          j        d          } |d          }t          |d          sJ t          |j        |j                  sJ t          |j        |j                  sJ |j        j        | j        j        k    sJ |j        j	        dk    rt          |j        |j                  sJ nt          |j        |j
                  sJ |j        j        d         | j        j        d         k    sJ |$t          j        |j        j        |k              sJ |$t          j        |j        j        |k              sJ  |dd          \  }}t          ||j                  sJ |j	        dk    rt          ||j                  sJ d S t          ||j
                  sJ d S )Nr   Tr   framer   r   )r   r   )r   importorskiphasattrr   r'   	DataFramer   r   r    ndimSeriesnpalldtypes)r!   r"   expected_data_dtypeexpected_target_dtypepdframe_bunchframe_Xframe_ys           r   check_as_framer6   >   s    
	X	&	&B,---K;(((((k'66666k&55555!UZ%55555""+,bl;;;;;;+,bi88888#A&%,*<Q*?????&vk&-1DDEEEEE(vk(/3HHIIIII $|TdCCCGWgr|,,,,,|a'2<0000000'29-------r%   c                  J    t           j                            dd          dk    S )NSKLEARN_SKIP_NETWORK_TESTS1)osenvironget r%   r   _skip_network_testsr>   Z   s    :>>6<<CCr%   loadfetchc              #     K   t                               | i           }t          j        t          j                  D ]\  }t          j        |          st          fd|D                       }| t          j        |          j	        v }|r}|r{t          j                                                d          ot                      d          g}|v r|                    |                    t          j        ||          V  d S )Nc                 :    g | ]}                     |          S r=   )
startswith).0tr   s     r   
<listcomp>z3_generate_func_supporting_param.<locals>.<listcomp>d   s%    HHHatq11HHHr%   rA   z.Skip because fetcher requires internet network)	conditionr   )marks)FETCH_PYTEST_MARKERSr<   inspect
getmemberssklearndatasets
isfunctionany	signature
parametersr   markskipifrD   r>   appendparam)rV   dataset_typemarkers_fetchobjis_dataset_typeis_support_paramrI   r   s          @r   _generate_func_supporting_paramr\   ^   s+     (,,UB77M'(899 7 7	c!#&& 	HHHH<HHHII G$5c$:$:$EE 	7/ 	7 """oog66P;N;P;PK #  E }$$]40111,tS666666#7 7r%   zname, dataset_funcr   c                 :     |            }t          ||           d S N)r$   r   r"   r!   s      r   test_common_check_return_X_yr`   t   s%     LNNEUL)))))r%   r   c                 :     |            }t          ||           d S r^   )r6   r_   s      r   test_common_check_as_framerb   |   s%     LNNE5,'''''r%   c                 $    t          |           d S r^   )r   )r   r"   s     r   #test_common_check_pandas_dependencyrd      s     $L11111r%   )NN)r?   )__doc__rK   r:   numpyr-   r   sklearn.datasetsrM   r   rS   xfailrT   rJ   r   r$   r6   r>   r\   parametrizer`   rb   rd   r=   r%   r   <module>rj      s"   , ,  				            %k//C 0 
 
 ))> * 
 
 #K..##%%%.G / 
 

 
 	))> * 
 
  (& & &4 4 4 JN. . . .8D D D7 7 7 7, 99,GG * * *
 99*EE ( ( (
 99*EE 2 2 2 2 2r%   