§
    rŠtjæc  ã                   óÚ  — d dl Z d dlmZ d dlm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mZ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 d dlmZ d dlmZm Z m!Z!m"Z"m#Z#m$Z$ d dl%m&Z&m'Z'  ej(        d¦  «         )                    d¦  «        Z*d„ Z+ej,         -                    dddd ej.        d¦  «        dfg d¢dfg d¢dfdd e/dd¦  «        df e/d d¦  «        df ej0        g d¢ej1        ¬¦  «        df ej0        g d¢ej2        ¬¦  «        df ej0        g d¢ej3        ¬¦  «        dfddgdfd  ej0        ddg¦  «        dfd!g d"¢dfd# e/d$d%¦  «        df ej0        g d"¢¦  «        df ej0        g d"¢e4¬¦  «        dfg¦  «        d&„ ¦   «         Z5d'„ Z6d(„ Z7e$ej,         -                    d) e¦   «         ¦  «        d*„ ¦   «         ¦   «         Z8e$ej,         -                    d) e¦   «         ¦  «        ej,         -                    d+d d,d-d.g ej0        d-d.g¦  «         e/d-d¦  «        g d/¢ ej9        g d0¢¦  «        f¦  «        ej,         -                    d1d d-g¦  «        d2„ ¦   «         ¦   «         ¦   «         ¦   «         Z:ej,         -                    d3g d4¢¦  «        ej,         -                    d5g d6¢¦  «        d7„ ¦   «         ¦   «         Z;ej,         -                    d3g d8¢¦  «        ej,         -                    d5g d6¢¦  «        d9„ ¦   «         ¦   «         Z<ej,         -                    d3g d:¢¦  «        ej,         -                    d5g d6¢¦  «        ej,         -                    d;d-dgd<d=gg¦  «        d>„ ¦   «         ¦   «         ¦   «         Z=ej,         -                    d?ddg¦  «        ej,         -                    d@ddg¦  «        ej,         -                    d3g d:¢¦  «        ej,         -                    d5dAdBg¦  «        ej,         -                    dCd g dD¢g dE¢gfd-dd.gdFdGgdHdggfg¦  «        dI„ ¦   «         ¦   «         ¦   «         ¦   «         ¦   «         Z>ej,         -                    d3g d8¢¦  «        ej,         -                    d5g dJ¢¦  «        dK„ ¦   «         ¦   «         Z?ej,         -                    d3g d:¢¦  «        ej,         -                    d5g dJ¢¦  «        ej,         -                    dLd g dD¢g dE¢gfd-dd.gdFdGgdHdggfg¦  «        dM„ ¦   «         ¦   «         ¦   «         Z@ej,         -                    dNg dO¢¦  «        dP„ ¦   «         ZAej,         -                    d3g d8¢¦  «        dQ„ ¦   «         ZBej,         -                    dNg dR¢¦  «        ej,         -                    d;dd=g¦  «        dS„ ¦   «         ¦   «         ZCej,         -                    d3g dT¢¦  «        dU„ ¦   «         ZDdV„ ZEej,         -                    d1dd.g¦  «        dW„ ¦   «         ZFej,         -                    dXg dY¢¦  «        dZ„ ¦   «         ZGd[„ ZHd\„ ZIej,         -                    d;d d d-g e/d d¦  «         ej0        d d-g¦  «        g¦  «        d]„ ¦   «         ZJej,         -                    d3g d^¢¦  «        d_„ ¦   «         ZKej,         -                    d`dadb e4¦   «         dcfg¦  «        dd„ ¦   «         ZLej,         -                    dedfgdggdfdggdfdhgdgdhgg¦  «        di„ ¦   «         ZMej,         -                    djg dk¢¦  «        dl„ ¦   «         ZNdm„ ZOdn„ ZPdo„ ZQdp„ ZRdq„ ZSdr„ ZTej,         -                    dse'¦  «        dt„ ¦   «         ZUdu„ ZVej,         -                    dve&¦  «        dw„ ¦   «         ZWdS )xé    N)Úcopy)ÚSkipTest)ÚDuplicateError)Úkstest)Úparse)Ú_safe_indexingÚresampleÚshuffle©Údevice)Úmove_toÚ)yield_namespace_device_dtype_combinations)Ú_determine_key_typeÚ_get_column_indicesÚ_safe_assign)ÚMockDataFrame)Ú_array_api_for_testsÚ_convert_containerÚassert_allcloseÚassert_allclose_dense_sparseÚassert_array_equalÚ'skip_if_array_api_compat_not_configured)ÚCSC_CONTAINERSÚCSR_CONTAINERSé	   )é   r   c                  óP  — t          j        dd¬¦  «        } |                      g d¢g d¢g d¢dœd¬	¦  «        }d
dlm} dgddgg d¢dgdgg}|D ]&}t          ||d¬¦  «        } |||         |¦  «         Œ'g d¢ddgfg d¢dgfg}|D ]-\  }}t          ||d¬¦  «        } ||dd…|f         |¦  «         Œ.d
dgddgfdgdgfg}	|	D ]-\  }
}t          ||
d¬¦  «        } ||dd…|f         |¦  «         Œ.d
dgddgddgg}|D ]&}t          ||d
¬¦  «        } |||         |¦  «         Œ'dS )z,Check _safe_indexing for polars as expected.Úpolarsz0.18.2)Ú
minversion)é   é   r   é   )r"   é   é   é   )r    r"   r    é
   ©ÚaÚbÚcÚrow)Úorientr   )Úassert_frame_equalr)   r(   )r)   r(   r*   r*   r    ©Úaxis©TFT)FFTNr!   r   )ÚpytestÚimportorskipÚ	DataFrameÚpolars.testingr-   r   )ÚplÚdfr-   Ústr_keysÚkeyÚoutÚ	bool_keysÚbool_keyÚstr_keyÚint_keysÚint_keyÚaxis_0_keyss               ú_/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/sklearn/utils/tests/test_indexing.pyÚtest_polars_indexingrA   (   sò  € å	Ô	˜X°(Ð	;Ñ	;Ô	;€BØ	�ŠØˆlˆl   °M°M°MÐBÐBÈ5ð 
ñ 
ô 
€Bð 2Ð1Ð1Ð1Ð1Ð1à�˜˜S�z ? ? ?°S°E¸C¸5ÐA€Hàð )ð )ˆÝ˜R ¨1Ð-Ñ-Ô-ˆØÐ˜2˜cœ7 CÑ(Ô(Ð(Ð(à%Ð%Ð%¨¨S zÐ2Ð5IÐ5IÐ5IÈCÈ5Ð4QÐR€Ià&ð 0ð 0Ñˆ�'Ý˜R °Ð2Ñ2Ô2ˆØÐ˜2˜a˜a˜a ˜jœ>¨3Ñ/Ô/Ð/Ð/à�Q�˜#˜s˜Ð$¨ s¨S¨E lÐ3€Hà$ð 0ð 0Ñˆ�Ý˜R ¨qÐ1Ñ1Ô1ˆØÐ˜2˜a˜a˜a ˜jœ>¨3Ñ/Ô/Ð/Ð/à�q�6˜A˜q˜6 A q 6Ð*€KØð )ð )ˆÝ˜R ¨1Ð-Ñ-Ô-ˆØÐ˜2˜cœ7 CÑ(Ô(Ð(Ð(ð)ð )ó    z
key, dtype)r   Úint)Ú0Ústr)TÚboolTrF   ©r   r    r!   rC   ©rD   Ú1Ú2rE   )rG   rC   )rH   rE   r!   ©ÚdtypeF))TFrF   )Úcol_0rE   ©rM   Úcol_1Úcol_2)rN   rE   ÚbeginÚendc                 ó0   — t          | ¦  «        |k    sJ ‚d S ©N)r   )r8   rL   s     r@   Útest_determine_key_typerU   I   s#   € õ8 ˜sÑ#Ô# uÒ,Ð,Ð,Ð,Ð,Ð,rB   c                  óŒ   — t          j        t          d¬¦  «        5  t          d¦  «         d d d ¦  «         d S # 1 swxY w Y   d S )NúNo valid specification of the©Úmatchç      ð?)r1   ÚraisesÚ
ValueErrorr   © rB   r@   Útest_determine_key_type_errorr^   h   sŠ   € Ý	Œ•zÐ)HÐ	IÑ	IÔ	Ið !ð !Ý˜CÑ Ô Ð ð!ð !ð !ñ !ô !ð !ð !ð !ð !ð !ð !ð !øøøð !ð !ð !ð !ð !ð !s   œ9¹=Á =c                  ó®   — t          j        t          d¬¦  «        5  t          t	          ddd¦  «        d¬¦  «         d d d ¦  «         d S # 1 swxY w Y   d S )NzOnly array-like or scalar arerX   r   r!   r    F)Úaccept_slice)r1   r[   Ú	TypeErrorr   Úslicer]   rB   r@   Ú#test_determine_key_type_slice_errorrc   m   s­   € Ý	Œ•yÐ(GÐ	HÑ	HÔ	Hð @ð @Ý�E ! Q¨™NœN¸Ð?Ñ?Ô?Ð?ð@ð @ð @ñ @ô @ð @ð @ð @ð @ð @ð @ð @øøøð @ð @ð @ð @ð @ð @s   œ!A
Á
AÁAz(array_namespace, device_name, dtype_namec                 ó$  — t          | ||¦  «        \  }}t          j        d¬¦  «        5  |                     g d¢|¬¦  «        }t	          |¦  «        dk    sJ ‚|                     g d¢|¬¦  «        }t	          |¦  «        dk    sJ ‚	 |                     g d¢|¬¦  «        }n# t
          $ r d }Y nw xY w|�Bt          j        t          d	¬
¦  «        5  t	          |¦  «         d d d ¦  «         n# 1 swxY w Y   d d d ¦  «         d S # 1 swxY w Y   d S )NT©Úarray_api_dispatch©r    r!   r   r   rC   r0   rF   )y      ð?      ð?y       @       @y      @      @rW   rX   )	r   ÚsklearnÚconfig_contextÚasarrayr   ra   r1   r[   r\   )Úarray_namespaceÚdevice_nameÚ
dtype_nameÚxpr   Úint_array_keyÚbool_array_keyÚcomplex_array_keys           r@   Ú!test_determine_key_type_array_apirr   r   sÆ  € õ & o°{ÀJÑOÔO�J€Bˆå	Ô	°4Ð	8Ñ	8Ô	8ð 7ð 7ØŸ
š
 9 9 9°V˜
Ñ<Ô<ˆÝ" =Ñ1Ô1°UÒ:Ð:Ð:Ð:àŸšÐ$7Ð$7Ð$7À˜ÑGÔGˆÝ" >Ñ2Ô2°fÒ<Ð<Ð<Ð<ð	%Ø "§
¢
Ð+CÐ+CÐ+CÈF 
Ñ SÔ SÐÐøÝð 	%ð 	%ð 	%à $ÐÐÐð	%øøøð Ð(Ý”�zÐ1PÐQÑQÔQð 7ð 7Ý#Ð$5Ñ6Ô6Ð6ð7ð 7ð 7ñ 7ô 7ð 7ð 7ð 7ð 7ð 7ð 7øøøð 7ð 7ð 7ð 7ð7ð 7ð 7ñ 7ô 7ð 7ð 7ð 7ð 7ð 7ð 7ð 7øøøð 7ð 7ð 7ð 7ð 7ð 7sZ   ªADÂB"Â!DÂ"B1Â.DÂ0B1Â1 DÃC-Ã!DÃ-C1	Ã1DÃ4C1	Ã5DÄD	ÄD	Úindexing_keyéÿÿÿÿr    r   )TFTT)FFFFr/   c                 óø  — t          | ||¦  «        \  }}t          j        d¦  «                             dd¦  «        }t	          |||¬¦  «        }t          |||¬¦  «        }	t          j        d¬¦  «        5  t	          |	||¬¦  «        }
t          |
¦  «        t          |	¦  «        k    sJ ‚|
j	        |	j	        k    sJ ‚	 d d d ¦  «         n# 1 swxY w Y   t          t          |
t          d¬¦  «        |¦  «         d S )Né   r"   r.   )rn   r   Tre   Úcpu)r   ÚnpÚarangeÚreshaper   r   rh   ri   Úarray_api_devicerL   r   )rk   rl   rm   rs   r/   rn   r   Úarray_to_index_npÚexpected_resultÚarray_to_index_xpÚindexed_array_xps              r@   Ú$test_safe_indexing_array_api_supportr€   Œ   sZ  € õ* & o°{ÀJÑOÔO�J€Bˆåœ	 "™œ×-Ò-¨a°Ñ3Ô3ÐÝ$Ð%6¸È4ÐPÑPÔP€OÝÐ 1°bÀÐHÑHÔHÐå	Ô	°4Ð	8Ñ	8Ô	8ð Að AÝ)Ð*;¸\ÐPTÐUÑUÔUÐÝÐ 0Ñ1Ô1Õ5EÐFWÑ5XÔ5XÒXÐXÐXÐXØÔ%Ð):Ô)@Ò@Ð@Ð@Ð@Ð@ðAð Að Añ Aô Að Að Að Að Að Að Aøøøð Að Að Að Aõ
 •GÐ,µ¸EÐBÑBÔBÀOÑTÔTÐTÐTÐTs   Á6AC
Ã
CÃCÚ
array_type)ÚlistÚarrayÚsparseÚpandasr   ÚpyarrowÚindices_type)r‚   Útuplerƒ   Úseriesrb   c                 ó.  — ddg}|dk    r+t          |d         t          ¦  «        r|dxx         dz  cc<   t          g d¢g d¢g d¢g| ¦  «        }t          ||¦  «        }t          ||d¬¦  «        }t	          |t          g d¢g d¢g| ¦  «        ¦  «         d S )	Nr    r!   rb   rg   ©r"   r#   r$   ©é   r%   r   r   r.   ©Ú
isinstancerC   r   r   r   ©r�   r‡   Úindicesrƒ   Úsubsets        r@   Ú&test_safe_indexing_2d_container_axis_0r“   ¯   s¾   € ð
 �!ˆf€GØ�wÒÐ¥:¨g°a¬j½#Ñ#>Ô#>ÐØ�ˆ
ˆ
Œ
�a‰ˆ
ˆ
‰
Ý 	 	 	¨9¨9¨9°i°i°iÐ@À*ÑMÔM€EÝ  ¨,Ñ7Ô7€GÝ˜E 7°Ð3Ñ3Ô3€FÝ ØÕ" I I I¨y¨y¨yÐ#9¸:ÑFÔFñô ð ð ð rB   )r‚   rƒ   r‰   Úpolars_seriesÚpyarrow_arrayc                 ó  — ddg}|dk    r+t          |d         t          ¦  «        r|dxx         dz  cc<   t          g d¢| ¦  «        }t          ||¦  «        }t          ||d¬¦  «        }t	          |t          ddg| ¦  «        ¦  «         d S )Nr    r!   rb   ©	r    r!   r   r"   r#   r$   r�   r%   r   r   r.   r   rŽ   r�   s        r@   Útest_safe_indexing_1d_containerr˜   ¿   sž   € ð
 �!ˆf€GØ�wÒÐ¥:¨g°a¬j½#Ñ#>Ô#>ÐØ�ˆ
ˆ
Œ
�a‰ˆ
ˆ
‰
ÝÐ:Ð:Ð:¸JÑGÔG€EÝ  ¨,Ñ7Ô7€GÝ˜E 7°Ð3Ñ3Ô3€FÝ  Õ);¸QÀ¸FÀJÑ)OÔ)OÑPÔPÐPÐPÐPrB   )rƒ   r„   r…   r   r†   r‘   rO   rP   c           	      ó&  — t          |¦  «        }|dk    r+t          |d         t          ¦  «        r|dxx         dz  cc<   g d¢}t          g d¢g d¢g d¢g| |¦  «        }t          ||¦  «        }t          |d         t          ¦  «        rN| dv rJd	}t          j        t          |¬
¦  «        5  t          ||d¬¦  «         d d d ¦  «         d S # 1 swxY w Y   d S t          ||d¬¦  «        }t          |t          ddgddgddgg| ¦  «        ¦  «         d S )Nrb   r    rN   rg   r‹   rŒ   r   ©rƒ   r„   úESpecifying the columns using strings is only supported for dataframesrX   r.   r!   r   r#   r$   r%   r   )
r   r�   rC   r   rE   r1   r[   r\   r   r   )r�   r‡   r‘   Úindices_convertedÚcolumn_namesrƒ   Úerr_msgr’   s           r@   Ú&test_safe_indexing_2d_container_axis_1rŸ   Í   s¡  € õ ˜W™œÐØ�wÒÐ¥:¨g°a¬j½#Ñ#>Ô#>ÐØ˜!ÐÐÔ Ñ!ÐÐÑà.Ð.Ð.€LÝØ	ˆˆ�I�I�I˜y˜y˜yÐ)¨:°|ñô €Eõ +Ð+<¸lÑKÔKÐå�'˜!”*�cÑ"Ô"ð 

 zÐ5HÐ'HÐ'HàSð 	õ Œ]�:¨WÐ5Ñ5Ô5ð 	=ð 	=Ý˜5Ð"3¸!Ð<Ñ<Ô<Ð<ð	=ð 	=ð 	=ñ 	=ô 	=ð 	=ð 	=ð 	=ð 	=ð 	=ð 	=ð 	=øøøð 	=ð 	=ð 	=ð 	=ð 	=ð 	=õ   Ð'8¸qÐAÑAÔAˆÝ$ØÕ&¨¨A¨°°A°¸¸A¸Ð'?ÀÑLÔLñ	
ô 	
ð 	
ð 	
ð 	
s   Â+CÃCÃCÚarray_read_onlyÚindices_read_onlyrƒ   r‰   zaxis, expected_arrayr‹   rŒ   r#   r$   r%   c                 ól  — t          j        g d¢g d¢g d¢g¦  «        }| r|                     d¬¦  «         t          ||¦  «        }t          j        ddg¦  «        }|r|                     d¬¦  «         t          ||¦  «        }t	          |||¬¦  «        }t          |t          ||¦  «        ¦  «         d S )	Nrg   r‹   rŒ   F)Úwriter    r!   r.   )rx   rƒ   Úsetflagsr   r   r   )	r    r¡   r�   r‡   r/   Úexpected_arrayrƒ   r‘   r’   s	            r@   Ú&test_safe_indexing_2d_read_only_axis_1r¦   ì   sÇ   € õ ŒH�i�i�i   ¨I¨I¨IÐ6Ñ7Ô7€EØð $Ø�Š˜UˆÑ#Ô#Ð#Ý˜u jÑ1Ô1€EÝŒh˜˜1�vÑÔ€GØð &Ø×Ò˜uÐÑ%Ô%Ð%Ý  ¨,Ñ7Ô7€GÝ˜E 7°Ð6Ñ6Ô6€FÝ  Õ);¸NÈJÑ)WÔ)WÑXÔXÐXÐXÐXrB   )r‚   rˆ   rƒ   r‰   c                 óÐ   — dgdgdz  z   dgdz  z   }t          g d¢| ¦  «        }t          ||¦  «        }t          ||d¬¦  «        }t          |t          ddg| ¦  «        ¦  «         d S )	NFTr!   r$   r—   r   r.   r   ©r   r   r   r�   s        r@   Ú$test_safe_indexing_1d_container_maskr©     s   € ð
 ˆg˜˜ ™
Ñ" e W¨q¡[Ñ0€GÝÐ:Ð:Ð:¸JÑGÔG€EÝ  ¨,Ñ7Ô7€GÝ˜E 7°Ð3Ñ3Ô3€FÝ  Õ);¸QÀ¸FÀJÑ)OÔ)OÑPÔPÐPÐPÐPrB   zaxis, expected_subsetc                 óÊ   — g d¢}t          g d¢g d¢g d¢g| |¦  «        }g d¢}t          ||¦  «        }t          |||¬¦  «        }t          |t          || ¦  «        ¦  «         d S )NrN   rg   r‹   rŒ   )FTTr.   r¨   )r�   r‡   r/   Úexpected_subsetr�   rƒ   r‘   r’   s           r@   Útest_safe_indexing_2d_maskr¬     s”   € ð /Ð.Ð.€LÝØ	ˆˆ�I�I�I˜y˜y˜yÐ)¨:°|ñô €Eð "Ð!Ð!€GÝ  ¨,Ñ7Ô7€Gå˜E 7°Ð6Ñ6Ô6€FÝ ØÕ" ?°JÑ?Ô?ñô ð ð ð rB   z array_type, expected_output_type))r‚   r‚   ©rƒ   rƒ   ©r„   r„   ©r…   r‰   ©r   r”   ©r†   r•   c                 ó¤   — t          g d¢g d¢g d¢g| ¦  «        }d}t          ||d¬¦  «        }t          g d¢|¦  «        }t          ||¦  «         d S )Nrg   r‹   rŒ   r!   r   r.   r¨   )r�   Úexpected_output_typerƒ   r‘   r’   r¥   s         r@   Ú#test_safe_indexing_2d_scalar_axis_0r´   &  si   € õ  	 	 	¨9¨9¨9°i°i°iÐ@À*ÑMÔM€EØ€GÝ˜E 7°Ð3Ñ3Ô3€FÝ'¨	¨	¨	Ð3GÑHÔH€NÝ  ¨Ñ8Ô8Ð8Ð8Ð8rB   c                 ób   — t          g d¢| ¦  «        }d}t          ||d¬¦  «        }|dk    sJ ‚d S )Nr—   r!   r   r.   r   )r   r   )r�   rƒ   r‘   r’   s       r@   Útest_safe_indexing_1d_scalarr¶   9  sD   € õ Ð:Ð:Ð:¸JÑGÔG€EØ€GÝ˜E 7°Ð3Ñ3Ô3€FØ�QŠ;ˆ;ˆ;ˆ;ˆ;ˆ;rB   )r­   r®   r¯   r°   r±   c                 ó�  — g d¢}t          g d¢g d¢g d¢g| |¦  «        }t          |t          ¦  «        rN| dv rJd}t          j        t
          |¬¦  «        5  t          ||d¬	¦  «         d d d ¦  «         d S # 1 swxY w Y   d S t          ||d¬	¦  «        }g d
¢}|dk    rdgdgdgg}t          ||¦  «        }t          ||¦  «         d S )NrN   rg   r‹   rŒ   rš   r›   rX   r    r.   )r   r$   r   r„   r   r$   r   )r   r�   rE   r1   r[   r\   r   r   )	r�   r³   r‘   r�   rƒ   rž   r’   Úexpected_outputr¥   s	            r@   Ú#test_safe_indexing_2d_scalar_axis_1r¹   C  sP  € ð /Ð.Ð.€LÝØ	ˆˆ�I�I�I˜y˜y˜yÐ)¨:°|ñô €Eõ �'�3ÑÔð = JÐ2EÐ$EÐ$EàSð 	õ Œ]�:¨WÐ5Ñ5Ô5ð 	3ð 	3Ý˜5 '°Ð2Ñ2Ô2Ð2ð	3ð 	3ð 	3ñ 	3ô 	3ð 	3ð 	3ð 	3ð 	3ð 	3ð 	3ð 	3øøøð 	3ð 	3ð 	3ð 	3ð 	3ð 	3õ    w°QÐ7Ñ7Ô7ˆØ#˜)˜)ˆØ 8Ò+Ð+à !˜s Q C¨!¨˜oˆOÝ+¨OÐ=QÑRÔRˆÝ$ V¨^Ñ<Ô<Ð<Ð<Ð<s   ÁA5Á5A9Á<A9)r‚   rƒ   r„   c                 ó|   — t          g d¢g d¢g d¢g| ¦  «        }t          |d d¬¦  «        }t          ||¦  «         d S )Nrg   r‹   rŒ   r   r.   r¨   )r�   ÚXÚX_subsets      r@   Útest_safe_indexing_None_axis_0r½   d  sN   € å˜I˜I˜I y y y°)°)°)Ð<¸jÑIÔI€AÝ˜a ¨AÐ.Ñ.Ô.€HÝ  ¨1Ñ-Ô-Ð-Ð-Ð-rB   c                  óô   — t          j        d¦  «        } d}|                      t          ¦  «        }t          j        t
          |¬¦  «        5  t          |dgd¬¦  «         d d d ¦  «         d S # 1 swxY w Y   d S )Nr…   z&No valid specification of the columns.rX   rZ   r    r.   )r1   r2   r3   ÚX_toyr[   r\   r   )Úpdrž   r»   s      r@   Ú0test_safe_indexing_pandas_no_matching_cols_errorrÁ   k  s¸   € Ý	Ô	˜XÑ	&Ô	&€BØ6€GØ
�Š•UÑÔ€AÝ	Œ•z¨Ð	1Ñ	1Ô	1ð )ð )Ý�q˜3˜% aÐ(Ñ(Ô(Ð(ð)ð )ð )ñ )ô )ð )ð )ð )ð )ð )ð )ð )øøøð )ð )ð )ð )ð )ð )s   ÁA-Á-A1Á4A1c                 ó    — t          j        t          d¬¦  «        5  t          t          ddg| ¬¦  «         d d d ¦  «         d S # 1 swxY w Y   d S )Nz'axis' should be either 0rX   r   r    r.   )r1   r[   r\   r   r¿   r.   s    r@   Útest_safe_indexing_error_axisrÃ   s  s•   € å	Œ•zÐ)DÐ	EÑ	EÔ	Eð 1ð 1Ý•u˜q !˜f¨4Ð0Ñ0Ô0Ð0ð1ð 1ð 1ñ 1ô 1ð 1ð 1ð 1ð 1ð 1ð 1ð 1øøøð 1ð 1ð 1ð 1ð 1ð 1s   œAÁAÁ
AÚX_constructor)rƒ   r‰   r”   r•   c                 ó(  — t          t          d¦  «        ¦  «        }| dk    rt          j        |¦  «        } n�| dk    r*t	          j        d¦  «        }|                     |¦  «        } n`| dk    r+t	          j        d¦  «        }|                     |¬¦  «        } n/| dk    r)t	          j        d	¦  «        }|                     |¦  «        } d
}t	          j        t          |¬¦  «        5  t          | ddgd¬¦  «         d d d ¦  «         d S # 1 swxY w Y   d S )Nr#   rƒ   r‰   r…   r”   r   )Úvaluesr•   r†   z='X' should be a 2D NumPy array, 2D sparse matrix or dataframerX   r   r    r.   )r‚   Úrangerx   rj   r1   r2   ÚSeriesrƒ   r[   r\   r   )rÄ   r»   rÀ   r5   Úparž   s         r@   Ú!test_safe_indexing_1d_array_errorrÊ   y  sL  € õ 	�U�1‰XŒX‰Œ€AØ˜ÒÐÝœ
 1™œˆˆØ	˜(Ò	"Ð	"ÝÔ  Ñ*Ô*ˆØŸ	š	 !™œˆˆØ	˜/Ò	)Ð	)ÝÔ  Ñ*Ô*ˆØŸ	š	¨˜	Ñ+Ô+ˆˆØ	˜/Ò	)Ð	)ÝÔ  Ñ+Ô+ˆØŸš ™œˆàM€GÝ	Œ•z¨Ð	1Ñ	1Ô	1ð 6ð 6Ý�} q¨! f°1Ð5Ñ5Ô5Ð5ð6ð 6ð 6ñ 6ô 6ð 6ð 6ð 6ð 6ð 6ð 6ð 6øøøð 6ð 6ð 6ð 6ð 6ð 6s   Ã%DÄDÄDc                  ó´   — ddg} g d¢g d¢g d¢g}d}t          j        t          |¬¦  «        5  t          || d¬	¦  «         d d d ¦  «         d S # 1 swxY w Y   d S )
NrO   rP   rg   r‹   rŒ   z/String indexing.*is not supported with 'axis=0'rX   r   r.   ©r1   r[   r\   r   )r‘   rƒ   rž   s      r@   Ú4test_safe_indexing_container_axis_0_unsupported_typerÍ   ‘  s·   € Ø˜Ð €GØˆYˆY˜	˜	˜	 9 9 9Ð-€EØ@€GÝ	Œ•z¨Ð	1Ñ	1Ô	1ð /ð /Ý�u˜g¨AÐ.Ñ.Ô.Ð.ð/ð /ð /ñ /ô /ð /ð /ð /ð /ð /ð /ð /øøøð /ð /ð /ð /ð /ð /s   ­AÁAÁAc                  óB  — t          j        d¦  «        } t          | j        ¦  «        }t          |j        ¦  «        }|t          d¦  «        k    rt          d¦  «        ‚|                      g d¢g d¢dœ¦  «        }t          |ddgd¬	¦  «        }t          | j	        d
¦  «        r| j	        j
        }n| j        j        j
        }t          j        ¦   «         5  t          j        d|¦  «         d|j        d<   d d d ¦  «         n# 1 swxY w Y   |j        d         dk    sJ ‚d S )Nr…   Ú3z;SettingWithCopyWarning has been removed in pandas 3.0.0.devrg   )r   r"   r#   )r(   r)   r   r    r.   ÚSettingWithCopyWarningÚerrorr&   )r   r   )r1   r2   Úparse_versionÚ__version__Úbase_versionr   r3   r   ÚhasattrÚerrorsrÐ   ÚcoreÚcommonÚwarningsÚcatch_warningsÚsimplefilterÚiloc)rÀ   Ú
pd_versionÚpd_base_versionr»   r’   rÐ   s         r@   Ú4test_safe_indexing_pandas_no_settingwithcopy_warningrß   ™  sd  € õ 
Ô	˜XÑ	&Ô	&€Bå˜rœ~Ñ.Ô.€JÝ# JÔ$;Ñ<Ô<€Oà�-¨Ñ,Ô,Ò,Ð,ÝÐTÑUÔUÐUà
�Š˜9˜9˜9¨9¨9¨9Ð5Ð5Ñ6Ô6€AÝ˜A  1˜v¨AÐ.Ñ.Ô.€FÝˆrŒyÐ2Ñ3Ô3ð GØ!#¤Ô!AÐÐð "$¤¤Ô!FÐÝ	Ô	 Ñ	"Ô	"ð ð ÝÔ˜gÐ'=Ñ>Ô>Ð>ØˆŒ�DÑðð ð ñ ô ð ð ð ð ð ð øøøð ð ð ð ð Œ6�$Œ<˜1ÒÐÐÐÐÐs   Ã DÄDÄDc                 ó¬   — ddgddgddgg}d}t          j        t          |¬¦  «        5  t          || d¬	¦  «         d
d
d
¦  «         d
S # 1 swxY w Y   d
S )z@Check that we raise a ValueError when axis=1 with input as list.r    r!   r"   r#   r�   r%   z!axis=1 is not supported for listsrX   r.   NrÌ   )r‘   r»   rž   s      r@   Ú*test_safe_indexing_list_axis_1_unsupportedrá   ²  s¯   € ð ˆQˆ�!�Q�˜!˜Q˜Ð €AØ1€GÝ	Œ•z¨Ð	1Ñ	1Ô	1ð +ð +Ý�q˜'¨Ð*Ñ*Ô*Ð*ð+ð +ð +ñ +ô +ð +ð +ð +ð +ð +ð +ð +øøøð +ð +ð +ð +ð +ð +s   ©A	Á	AÁA)rƒ   r„   r…   c                 ó(  — t           j                             d¦  «        }|                     dd¦  «        }ddg}|                     t	          |¦  «        |j        d         ¦  «        }t          || ¦  «        }t          |||¬¦  «         t          ||d¬¦  «        }t          |t          || ¦  «        ¦  «         ddg}|                     |j        d         t	          |¦  «        ¦  «        }t          || ¦  «        }t          |||¬¦  «         t          ||d¬¦  «        }t          |t          || ¦  «        ¦  «         d	\  }} |j        |j        Ž }t          || ¦  «        }t          |||¬¦  «         t          |t          || ¦  «        ¦  «         d
S )z,Check that `_safe_assign` works as expected.r   r&   r#   r    r!   )Úrow_indexerr.   )Úcolumn_indexer)NNN)
rx   ÚrandomÚRandomStateÚrandnÚlenÚshaper   r   r   r   )r�   ÚrngÚX_arrayrã   rÆ   r»   Úassigned_portionrä   s           r@   Útest_safe_assignrí   »  s›  € õ Œ)×
Ò
 Ñ
"Ô
"€CØ�iŠi˜˜AÑÔ€Gà�a�&€KØ�YŠY•s˜;Ñ'Ô'¨¬°qÔ)9Ñ:Ô:€FÝ˜7 JÑ/Ô/€AÝ��F¨Ð4Ñ4Ô4Ð4å% a¨¸1Ð=Ñ=Ô=ÐÝ ØÕ,¨V°ZÑ@Ô@ñô ð ð ˜�V€NØ�YŠY�w”} QÔ'­¨^Ñ)<Ô)<Ñ=Ô=€FÝ˜7 JÑ/Ô/€AÝ��F¨>Ð:Ñ:Ô:Ð:å% a¨¸aÐ@Ñ@Ô@ÐÝ ØÕ,¨V°ZÑ@Ô@ñô ð ð #-Ñ€K�ØˆSŒY˜œÐ €FÝ˜7 JÑ/Ô/€AÝ��F¨>Ð:Ñ:Ô:Ð:å  Õ$6°v¸zÑ$JÔ$JÑKÔKÐKÐKÐKrB   zkey, err_msg)r&   z all features must be in \[0, 2\])ÚwhateverzBSome column names are not columns of the dataframe: \{'whatever'\}z%No valid specification of the columnsc                 óò   — t          j        d¦  «        }|                     t          g d¢¬¦  «        }t          j        t
          |¬¦  «        5  t          || ¦  «         d d d ¦  «         d S # 1 swxY w Y   d S )Nr…   rN   ©ÚcolumnsrX   )r1   r2   r3   r¿   r[   r\   r   )r8   rž   rÀ   ÚX_dfs       r@   Útest_get_column_indices_errorró   Ý  s¹   € õ 
Ô	˜XÑ	&Ô	&€BØ�<Š<�Ð'BÐ'BÐ'Bˆ<ÑCÔC€Då	Œ•z¨Ð	1Ñ	1Ô	1ð 'ð 'Ý˜D #Ñ&Ô&Ð&ð'ð 'ð 'ñ 'ô 'ð 'ð 'ð 'ð 'ð 'ð 'ð 'øøøð 'ð 'ð 'ð 'ð 'ð 's   ÁA,Á,A0Á3A0r8   Úcol1Úcol2Úcol3c                 ó&  — t          j        d¦  «        }t          j        dt          ¬¦  «        }g d¢}|                     ||¬¦  «        }d}t          j        t          |¬¦  «        5  t          || ¦  «         d d d ¦  «         d S # 1 swxY w Y   d S )Nr…   )r    r#   rK   )rô   rô   rõ   rö   rõ   rð   z#Expected unique column names, got.*rX   )	r1   r2   rx   ÚzerosrC   r3   r[   r   r   )r8   rÀ   Útoyrñ   r»   rž   s         r@   Ú6test_get_column_indices_pandas_nonunique_columns_errorrú   ð  sÖ   € õ 
Ô	˜XÑ	&Ô	&€BÝ
Œ(�6¥Ð
%Ñ
%Ô
%€CØ6Ð6Ð6€GØ
�Š�S 'ˆÑ*Ô*€Aà3€GÝ	Œ•~¨WÐ	5Ñ	5Ô	5ð $ð $Ý˜A˜sÑ#Ô#Ð#ð$ð $ð $ñ $ô $ð $ð $ð $ð $ð $ð $ð $øøøð $ð $ð $ð $ð $ð $s   Á(BÂB
ÂB
Úconstructor_name)rƒ   r„   r…   r†   r   c                 ój  — t          g d¢g d¢g| g d¢¬¦  «        }t          dd¦  «        ddgdft          dd¦  «        d	dgdft          dd¦  «        dgdfd
dgddgdft          dd
¦  «        d	dgdft          dd¦  «        g d¢dft          dd¦  «        d	gdfddgdd	gdfg g dfg	}d}|D ]f\  }}}| dv rF|rDt          j        t          |¬¦  «        5  t          ||¦  «         ddd¦  «         n# 1 swxY w Y   ŒPt          ||¦  «        |k    sJ ‚Œg| dvr�d}t          j        t          |¬¦  «        5  t          |dg¦  «         ddd¦  «         n# 1 swxY w Y   d}t          j        t          |¬¦  «        5  t          |t          ddd¦  «        ¦  «         ddd¦  «         dS # 1 swxY w Y   dS dS )z9Check _get_column_indices for edge cases with 2d input X.rg   r‹   r'   )r�   r    Nr!   Fr   r)   r*   Tr(   rG   r›   rš   rX   zFSome column names are not columns of the dataframe: \{'not_a_column'\}Únot_a_columnzkey.step must be 1 or None)r   rb   r1   r[   r\   r   ÚNotImplementedError)rû   r6   Úkey_resultsÚmsgr8   ÚresultÚ	use_namess          r@   Ú"test_get_column_indices_dataframesr  þ  sè  € õ
 
Ø	ˆˆ�I�I�IÐÐ 0¸¸¸ð
ñ 
ô 
€Bõ
 
ˆq�$‰Œ˜!˜Q˜ Ð'Ý	ˆt�Q‰Œ˜!˜Q˜ Ð'Ý	ˆq�!‰Œ�q�c˜5Ð!Ø
ˆsˆ�a˜�V˜TÐ"Ý	ˆs�C‰Œ˜1˜a˜& $Ð'Ý	ˆs�DÑ	Ô	˜9˜9˜9 dÐ+Ý	ˆt�SÑ	Ô	˜A˜3 Ð%Ø
ˆsˆ�a˜�V˜TÐ"Ø	ˆR�ˆð
€Kð R€CØ"-ð :ð :ÑˆˆV�YØÐ2Ð2Ð2°yÐ2Ý”�z°Ð5Ñ5Ô5ð -ð -Ý# B¨Ñ,Ô,Ð,ð-ð -ð -ñ -ô -ð -ð -ð -ð -ð -ð -øøøð -ð -ð -ð -øõ ' r¨3Ñ/Ô/°6Ò9Ð9Ð9Ð9Ð9àÐ2Ð2Ð2ØWˆÝŒ]�:¨SÐ1Ñ1Ô1ð 	6ð 	6Ý  ^Ð$4Ñ5Ô5Ð5ð	6ð 	6ð 	6ñ 	6ô 	6ð 	6ð 	6ð 	6ð 	6ð 	6ð 	6øøøð 	6ð 	6ð 	6ð 	6ð +ˆÝŒ]Õ.°cÐ:Ñ:Ô:ð 	9ð 	9Ý ¥E¨#¨t°QÑ$7Ô$7Ñ8Ô8Ð8ð	9ð 	9ð 	9ñ 	9ô 	9ð 	9ð 	9ð 	9ð 	9ð 	9ð 	9ð 	9øøøð 	9ð 	9ð 	9ð 	9ð 	9ð 	9ð 3Ð2s6   ÃC/Ã/C3	Ã6C3	Ä3EÅEÅEÅ9 F&Æ&F*Æ-F*c                  óŠ  — t          ¦   «         �J ‚t          j        t          ¦  «        5  t          dgddg¦  «         d d d ¦  «         n# 1 swxY w Y   t          j        t          ¦  «        5  t          ddgddgdd¬¦  «         d d d ¦  «         n# 1 swxY w Y   t	          t          ddgd¬¦  «        ¦  «        dk    sJ ‚d S )	Nr   r    Fr   )ÚreplaceÚ	n_samplesr!   r#   )r  )r	   r1   r[   r\   rè   r]   rB   r@   Útest_resampler  $  sE  € å‰:Œ:ÐÐÐõ 
Œ•zÑ	"Ô	"ð ð Ý�!��q˜!�fÑÔÐðð ð ñ ô ð ð ð ð ð ð øøøð ð ð ð å	Œ•zÑ	"Ô	"ð =ð =Ý�!�Q�˜!˜Q˜¨¸!Ð<Ñ<Ô<Ð<ð=ð =ð =ñ =ô =ð =ð =ð =ð =ð =ð =øøøð =ð =ð =ð =õ �x˜˜A˜¨!Ð,Ñ,Ô,Ñ-Ô-°Ò2Ð2Ð2Ð2Ð2Ð2s#   ªA
Á
AÁAÁ.BÂBÂBc                  ó|  — t          j        g d¢¦  «        } t          j        g d¢¦  «        }g }g }t          d¦  «        D ]œ}|                     t          |                      |¦  «        d|| j        d         ¬¦  «                             ¦   «         ¦  «         |                     t          | |d|| j        d         ¬¦  «                             ¦   «         ¦  «         Œ�t          j        |¦  «        }t          j        |¦  «        }t          ||¦  «        }t          j
        |dk    ¦  «        sJ ‚|j        dk    sJ ‚d S )	N)rt   r   r    )r   éd   r    r	  Tr   )r  Úrandom_stater  )Úsample_weightr  r
  r  gš™™™™™©?)rx   rƒ   rj   rÇ   Úappendr	   Úrepeatré   Úmeanr   ÚallÚpvalue)Údatar  Úmean_repeatedÚmean_reweightedÚseedÚtest_results         r@   Útest_resample_weightedr  2  sS  € õ Œ8�J�J�JÑÔ€DÝ”J˜{˜{˜{Ñ+Ô+€Mà€MØ€Oå�c‘
”
ð 
ð 
ˆØ×ÒÝØ—’˜MÑ*Ô*ØØ!Øœ* Qœ-ð	ñ ô ÷
 Šd‰fŒfñ	
ô 	
ð 	
ð 	×ÒÝØØ+ØØ!Øœ* Qœ-ðñ ô ÷ Šd‰fŒfñ	
ô 	
ð 	
ð 	
õ ”J˜}Ñ-Ô-€MÝ”j Ñ1Ô1€Oå˜¨Ñ8Ô8€KåŒ6�/ QÒ&Ñ'Ô'Ð'Ð'Ð'ð Ô Ò$Ð$Ð$Ð$Ð$Ð$rB   c                  ó¤  — t           j                             d¦  «        } d}d}|                      |df¬¦  «        }|                      d||¬¦  «        }t          ||ddd ¬¦  «        \  }}t          j        |dk    ¦  «        sJ ‚t          ||dd|¬¦  «        \  }}t          j        |dk    ¦  «        rJ ‚t          j        |¦  «        dk    sJ ‚d S )	Nr   r	  gÍÌÌÌÌÌì?r    ©Úsizer&   ©r  r
  Ústratifyr   )rx   rå   ræ   ÚnormalÚbinomialr	   r  Úsum)rê   r  Úpr»   ÚyÚ_Úy_not_stratifiedÚy_stratifieds           r@   Útest_resample_stratifiedr$  Z  sâ   € å
Œ)×
Ò
 Ñ
"Ô
"€CØ€IØ€AØ�
Š
˜ A˜ˆ
Ñ'Ô'€AØ�Š�Q˜ 	ˆÑ*Ô*€Aå" 1 a°2ÀAÐPTÐUÑUÔUÑ€AÐÝŒ6Ð" aÒ'Ñ(Ô(Ð(Ð(Ð(å˜q !¨rÀÈAÐNÑNÔN�O€A€|ÝŒv�l aÒ'Ñ(Ô(Ð(Ð(Ð(ÝŒ6�,ÑÔ 1Ò$Ð$Ð$Ð$Ð$Ð$rB   c                  ó<  — t           j                             d¦  «        } d}|                      |df¬¦  «        }|                      dd|¬¦  «        }t          ||dd| |¬¦  «        \  }}t          ||d	d| |¬¦  «        \  }}t          j        |¦  «        j        d         dk     sJ ‚t          j        |¦  «        j        d         dk    sJ ‚t          ||dd
| |¬¦  «        \  }}|j        d         d
k    sJ ‚t          j        |¦  «        j        d         dk    sJ ‚d S )Nr   r	  r    r  r!   Té2   )r  r  r
  r  Fiè  )rx   rå   ræ   r  Úrandintr	   Úuniqueré   )rê   r  r»   r   Ú	X_replacer!  ÚX_no_replaces          r@   Ú test_resample_stratified_replacer+  j  sH  € å
Œ)×
Ò
 Ñ
"Ô
"€CØ€IØ�
Š
˜ A˜ˆ
Ñ'Ô'€AØ�Š�A�q˜yˆÑ)Ô)€AåØ	ˆ1�d b°sÀQðñ ô �L€Iˆqõ Ø	ˆ1�e r¸Àaðñ ô �O€L�!õ Œ9�YÑÔÔ% aÔ(¨2Ò-Ð-Ð-Ð-ÝŒ9�\Ñ"Ô"Ô(¨Ô+¨rÒ1Ð1Ð1Ð1õ Ø	ˆ1�d d¸Àqðñ ô �L€Iˆqð Œ?˜1Ô Ò%Ð%Ð%Ð%ÝŒ9�YÑÔÔ% aÔ(¨CÒ/Ð/Ð/Ð/Ð/Ð/rB   c                  óô   — t           j                             d¦  «        } d}|                      |df¬¦  «        }|                      dd|df¬¦  «        }t          ||d| |¬¦  «        \  }}|j        dk    sJ ‚d S )Nr   r	  r    r  r!   r&  r  )rx   rå   ræ   r  r'  r	   Úndim)rê   r  r»   r   s       r@   Útest_resample_stratify_2dyr.  ƒ  s}   € å
Œ)×
Ò
 Ñ
"Ô
"€CØ€IØ�
Š
˜ A˜ˆ
Ñ'Ô'€AØ�Š�A�q 	¨1˜~ˆÑ.Ô.€AÝ�A�q B°SÀ1ÐEÑEÔE�D€A€qØŒ6�QŠ;ˆ;ˆ;ˆ;ˆ;ˆ;rB   c                  ó<  — t          j        t          d¬¦  «        5  t          ddgddgddgd¬¦  «         d d d ¦  «         n# 1 swxY w Y   t          j        t          d¬¦  «        5  t          ddgddgddgddg¬¦  «         d d d ¦  «         d S # 1 swxY w Y   d S )	NzCResampling with sample_weight is only implemented for replace=True.rX   r   r    F)r  r  zCResampling with sample_weight is only implemented for stratify=None)r  r  )r1   r[   rþ   r	   r]   rB   r@   Útest_notimplementederrorr0  �  sj  € Ý	ŒÝØSð
ñ 
ô 
ð Fð Fõ 	�!�Q�˜!˜Q˜°°1¨v¸uÐEÑEÔEÐEð	Fð Fð Fñ Fô Fð Fð Fð Fð Fð Fð Føøøð Fð Fð Fð Fõ 
ŒÝØSð
ñ 
ô 
ð Hð Hõ 	�!�Q�˜!˜Q˜°°1¨vÀÀAÀÐGÑGÔGÐGð	Hð Hð Hñ Hô Hð Hð Hð Hð Hð Hð Hð Høøøð Hð Hð Hð Hð Hð Hs#   œAÁAÁ	AÁ(BÂBÂBÚcsr_containerc                 ó|  — t           j                             d¦  «        }d}|                     |df¬¦  «        }|                     dd|¬¦  «        } | |                     dd¦  «        ¦  «        }t          j        t          d¬¦  «        5  t          ||d	||¬
¦  «        \  }}d d d ¦  «         d S # 1 swxY w Y   d S )Nr   r	  r!   r  rt   r    zSparse data was passedrX   r&  r  )
rx   rå   ræ   r  r'  rz   r1   r[   ra   r	   )r1  rê   r  r»   r   r  s         r@   Ú#test_resample_stratify_sparse_errorr3  ›  s  € õ Œ)×
Ò
 Ñ
"Ô
"€CØ€IØ�
Š
˜ A˜ˆ
Ñ'Ô'€AØ�Š�A�q˜yˆÑ)Ô)€AØˆ}˜QŸYšY r¨1Ñ-Ô-Ñ.Ô.€HÝ	Œ•yÐ(@Ð	AÑ	AÔ	Að Qð QÝ˜˜1¨¸ÀxÐPÑPÔP‰ˆˆ1ðQð Qð Qñ Qô Qð Qð Qð Qð Qð Qð Qð Qøøøð Qð Qð Qð Qð Qð Qs   ÂB1Â1B5Â8B5c                  óÚ   — d„ } t          j        ddgddggddgdd	ggg¦  «        }t           | |¦  «        ¦  «        }t          |¦  «         t           | |¦  «        ¦  «        |k    sJ ‚d S )
Nc                 ó4   — t          d„ | D ¦   «         ¦  «        S )Nc              3   óH   K  — | ]}t          d „ |D ¦   «         ¦  «        V — ŒdS )c              3   ó4   K  — | ]}t          |¦  «        V — Œd S rT   ©rˆ   )Ú.0ÚCs     r@   ú	<genexpr>zPtest_shuffle_on_ndim_equals_three.<locals>.to_tuple.<locals>.<genexpr>.<genexpr>©  s(   è è € Ð/Ð/¨�5 ™8œ8Ð/Ð/Ð/Ð/Ð/Ð/rB   Nr8  )r9  ÚBs     r@   r;  zFtest_shuffle_on_ndim_equals_three.<locals>.to_tuple.<locals>.<genexpr>©  s9   è è € Ð;Ð;°A•UÐ/Ð/¨QÐ/Ñ/Ô/Ñ/Ô/Ð;Ð;Ð;Ð;Ð;Ð;rB   r8  )ÚAs    r@   Úto_tuplez3test_shuffle_on_ndim_equals_three.<locals>.to_tuple¨  s   € ÝÐ;Ð;¸Ð;Ñ;Ô;Ñ;Ô;Ð;rB   r    r!   r   r"   r#   r$   r�   r%   )rx   rƒ   Úsetr
   )r>  r=  ÚSs      r@   Ú!test_shuffle_on_ndim_equals_threerA  §  sˆ   € ð<ð <ð <õ 	Œ�A�q�6˜A˜q˜6Ð" a¨ V¨a°¨VÐ$4Ð5Ñ6Ô6€AÝˆHˆH�Q‰KŒKÑÔ€AÝˆA�J„J€JÝˆxˆx˜‰{Œ{ÑÔ˜qÒ Ð Ð Ð Ð Ð rB   Úcsc_containerc           	      ó4  — g d¢}t          j        g d¢t          ¬¦  «        }g d¢}t          t          j        ddgddgdd	ggt          ¬¦  «        ¦  «        } | t          j        d
¦  «                             dd	¦  «        ¦  «        }t          |||||d¬¦  «        \  }}}}	}
|g d¢k    sJ ‚t          |¦  «        t          k    sJ ‚t          |g d¢¦  «         |j
        t          k    sJ ‚|g d¢k    sJ ‚t          |¦  «        t          k    sJ ‚t          |	t          j        dd	gddgddggt          ¬¦  «        ¦  «         t          |	¦  «        t          k    sJ ‚t          |
                     ¦   «         t          j        ddgd	dgddgg¦  «        ¦  «         d S )Nr'   rK   rg   r(   r   r)   r    r*   r!   r$   r   )r
  )r*   r)   r(   )r   r!   r    r"   r#   )rx   rƒ   Úobjectr   ry   rz   r
   Útyper‚   r   rL   Útoarray)rB  r(   r)   r*   ÚdÚeÚa_sÚb_sÚc_sÚd_sÚe_ss              r@   Ú"test_shuffle_dont_convert_to_arrayrN  ±  s¬  € ð 	ˆˆ€AÝ
Œ���­Ð/Ñ/Ô/€AØˆ	ˆ	€AÝ•b”h  a ¨3°¨(°S¸!°HÐ=ÅVÐLÑLÔLÑMÔM€AØˆ•b”i ‘l”l×*Ò*¨1¨aÑ0Ô0Ñ1Ô1€AÝ% a¨¨A¨q°!À!ÐDÑDÔDÑ€Cˆˆc�3˜à�/�/�/Ò!Ð!Ð!Ð!Ý�‰9Œ9�ÒÐÐÐå�s˜O˜O˜OÑ,Ô,Ð,ØŒ9�ÒÐÐÐà�)�)�)ÒÐÐÐÝ�‰9Œ9�ÒÐÐÐå�s�BœH s¨A h°°a°¸3À¸(Ð%CÍ6ÐRÑRÔRÑSÔSÐSÝ�‰9Œ9�Ò%Ð%Ð%Ð%å�s—{’{‘}”}¥b¤h°°A°¸¸A¸ÀÀAÀÐ/GÑ&HÔ&HÑIÔIÐIÐIÐIrB   )XrÙ   r   Úunittestr   Únumpyrx   r1   Únarwhals.exceptionsr   Úscipy.statsr   rh   Ú$sklearn.externals._packaging.versionr   rÒ   Úsklearn.utilsr   r	   r
   Úsklearn.utils._array_apir   r{   r   r   Úsklearn.utils._indexingr   r   r   Úsklearn.utils._mockingr   Úsklearn.utils._testingr   r   r   r   r   r   Úsklearn.utils.fixesr   r   ry   rz   r¿   rA   ÚmarkÚparametrizeÚbool_rb   rƒ   Úint32Úint64Úuint8rD  rU   r^   rc   rr   rj   r€   r“   r˜   rŸ   r¦   r©   r¬   r´   r¶   r¹   r½   rÁ   rÃ   rÊ   rÍ   rß   rá   rí   ró   rú   r  r  r  r$  r+  r.  r0  r3  rA  rN  r]   rB   r@   ú<module>r`     s  ðØ €€€Ø Ð Ð Ð Ð Ð Ø Ð Ð Ð Ð Ð à Ð Ð Ð Ø €€€Ø .Ð .Ð .Ð .Ð .Ð .Ø Ð Ð Ð Ð Ð à €€€Ø GÐ GÐ GÐ GÐ GÐ GØ ;Ð ;Ð ;Ð ;Ð ;Ð ;Ð ;Ð ;Ð ;Ð ;ðð ð ð ð ð ðð ð ð ð ð ð ð ðð ð ð ð ð ð ð ð ð ð
 1Ð 0Ð 0Ð 0Ð 0Ð 0ðð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ?Ð >Ð >Ð >Ð >Ð >Ð >Ð >ð 	ˆŒ	�!‰Œ×Ò˜VÑ$Ô$€ð)ð )ð )ðB „×ÒØàØØØ	ˆŒ�$‰Œ˜Ð Ø	ˆˆ�EÐØ	ˆˆ˜%Ð ØØ Ø	ˆˆt�TÑ	Ô	˜DÐ!Ø	ˆˆq�!‰Œ�eÐØ	ˆŒ�)�)�) 2¤8Ð	,Ñ	,Ô	,¨eÐ4Ø	ˆŒ�)�)�) 2¤8Ð	,Ñ	,Ô	,¨eÐ4Ø	ˆŒ�)�)�) 2¤8Ð	,Ñ	,Ô	,¨eÐ4Ø
�ˆ˜ÐØØ	ˆŒ�4˜�-Ñ	 Ô	  &Ð)ØØ	$Ð	$Ð	$ eÐ,Ø,Ø	ˆˆw˜Ñ	Ô	 Ð&Ø	ˆŒÐ-Ð-Ð-Ñ	.Ô	.°Ð6Ø	ˆŒÐ-Ð-Ð-°VÐ	<Ñ	<Ô	<¸eÐDð-ñô ð6-ð -ñ7ô ð6-ð!ð !ð !ð
@ð @ð @ð
 )Ø„×ÒØ.Ø-Ð-Ñ/Ô/ñô ð7ð 7ñ	ô ñ )Ô(ð
7ð* )Ø„×ÒØ.Ø-Ð-Ñ/Ô/ñô ð „×ÒØà	Ø
Ø	
ˆAˆØˆŒ�!�Q�ÑÔØˆˆa�‰ŒØ!Ð!Ð!ØˆŒ
Ð/Ð/Ð/Ñ0Ô0ðñô ð „×Ò˜ ! Q Ñ(Ô(ðUð Uñ )Ô(ñô ñ	ô ñ )Ô(ð$Uð" „×ÒØÐLÐLÐLñô ð „×Ò˜Ð)VÐ)VÐ)VÑWÔWð	ð 	ñ XÔWñô ð	ð „×ÒØÐOÐOÐOñô ð „×Ò˜Ð)VÐ)VÐ)VÑWÔWðQð Qñ XÔWñô ðQð „×ÒØÐDÐDÐDñô ð „×Ò˜Ð)VÐ)VÐ)VÑWÔWØ„×Ò˜ a¨ V¨g°wÐ-?Ð$@ÑAÔAð
ð 
ñ BÔAñ XÔWñô ð

ð4 „×ÒÐ*¨T°5¨MÑ:Ô:Ø„×ÒÐ,¨t°U¨mÑ<Ô<Ø„×ÒØÐDÐDÐDñô ð „×Ò˜¨'°8Ð)<Ñ=Ô=Ø„×ÒØ˜a ) ) )¨Y¨Y¨YÐ!7Ð8¸1ÀÀ1¸vÈÈ1ÀvÐPQÐSTÈvÐ>VÐ:WÐXñô ðYð Yñô ñ >Ô=ñô ñ =Ô<ñ ;Ô:ðYð „×ÒØÐOÐOÐOñô ð „×Ò˜Ð)MÐ)MÐ)MÑNÔNðQð Qñ OÔNñô ðQð „×ÒØÐDÐDÐDñô ð „×Ò˜Ð)MÐ)MÐ)MÑNÔNØ„×ÒØØˆ)ˆ)ˆ)�Y�Y�YÐ	Ð  1¨¨1 v°°1¨v¸¸1°vÐ&>Ð"?Ð@ñô ðð ñ	ô ñ OÔNñô ðð „×ÒØ&ðð ð ñ
ô 
ð9ð 9ñ
ô 
ð9ð „×ÒØÐOÐOÐOñô ðð ñô ðð „×ÒØ&ðð ð ñ	ô 	ð „×Ò˜ Q¨ LÑ1Ô1ð=ð =ñ 2Ô1ñ	ô 	ð=ð, „×Ò˜Ð'BÐ'BÐ'BÑCÔCð.ð .ñ DÔCð.ð)ð )ð )ð „×Ò˜ $¨ Ñ+Ô+ð1ð 1ñ ,Ô+ð1ð
 „×ÒØÐJÐJÐJñô ð6ð 6ñô ð6ð*/ð /ð /ðð ð ð2 „×Ò˜ Q¨¨A¨°°°a¸±´¸X¸R¼XÀqÈ!ÀfÑ=MÔ=MÐ$NÑOÔOð+ð +ñ PÔOð+ð „×Ò˜Ð'DÐ'DÐ'DÑEÔEðLð Lñ FÔEðLðB „×ÒØà1ð	
ð 
ˆ‰ŒÐ:Ð;ðñ
ô 
ð'ð 'ñ
ô 
ð'ð „×ÒØ	ˆVˆH�v�h ¨Ð 0°6¸6Ð2BÀVÈVÐDTÐUñô ð$ð $ñô ð$ð „×ÒØÐJÐJÐJñô ð 9ð  9ñô ð 9ðF3ð 3ð 3ð%%ð %%ð %%ðP%ð %ð %ð 0ð 0ð 0ð2ð ð ðHð Hð Hð „×Ò˜¨.Ñ9Ô9ðQð Qñ :Ô9ðQð!ð !ð !ð „×Ò˜¨.Ñ9Ô9ðJð Jñ :Ô9ðJð Jð JrB   