§
    tŠtjz  ã                   ót   — d Z ddlmZ ddlZddlZddlZddlm	Z
 ej        d„ ¦   «         Z G d„ d¦  «        ZdS )z�
Note: for naming purposes, most tests are title with as e.g. "test_nlargest_foo"
but are implicitly also testing nsmallest_foo.
é    )Úascii_lowercaseNc                  óP  — t          j        g d¢g d¢g d¢t          d¦  «        t          j        t          d¦  «        ¦  «                             d¦  «        g d¢t          j        dd¬	¦  «        t          j        ddd
¬¦  «        t          j        ddd¬¦  «        dœ	g d¢¬¦  «        S )N)é   r   é   ©r   r   é   )g      @g      @g      @ÚabcÚcategory)é   é   é	   Ú20130101r   )Úperiodsz
US/Eastern)r   Útzz1 sÚs)r   Úfreq)	ÚgroupÚintÚfloatÚstringÚcategory_stringÚcategory_intÚdatetimeÚ
datetimetzÚ	timedelta©Úcolumns)ÚpdÚ	DataFrameÚlistÚSeriesÚastypeÚ
date_rangeÚtimedelta_range© ó    úf/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/pandas/tests/frame/methods/test_nlargest.pyÚdf_main_dtypesr(      sµ   € åŒ<à�Y�YØ�9�9Ø$�_�_Ý˜5‘k”kÝ!œy­¨e©¬Ñ5Ô5×<Ò<¸ZÑHÔHØ%˜I˜IÝœ j¸!Ð<Ñ<Ô<Ýœ-¨
¸AÀ,ÐOÑOÔOÝÔ+¨E¸1À3ÐGÑGÔGð
	
ð 
	
ð

ð 

ð 

ðñ ô ð r&   c                   ó,  — e Zd Zej                             ddgdgddgddgddgddgg d¢g d¢g d¢g d¢g d	¢g d
¢g¦  «        ej                             d edd¦  «        ¦  «        d„ ¦   «         ¦   «         Zej                             dddgddgg¦  «        d„ ¦   «         Zd„ Z	d„ Z
d„ Zej                             dg d¢g d¢dgdgddgddgg¦  «        ej                             d edd¦  «        ¦  «        d„ ¦   «         ¦   «         Zd„ Zd„ Zd„ Zd„ ZdS )ÚTestNLargestNSmallestÚorderÚaÚcÚb©r,   r.   r-   )r-   r,   r.   )r-   r.   r,   )r.   r-   r,   )r.   r,   r-   )r.   r-   r-   Únr   é   c                 ó  — t          j        t          j                             d¦  «                             d¦  «        t          t          d d…         ¦  «        t          j                             d¦  «                             d¦  «                             d¦  «        dœ¦  «        }d|v rVd|› d�}t          j
        t          |¬¦  «        5   t          ||¦  «        ||¦  «         d d d ¦  «         d S # 1 swxY w Y   d S |d	k    } t          ||¦  «        ||¦  «        }t          j        t          |j        ¦  «        ¦  «        |_        |                     ||¬
¦  «                             |¦  «        }t#          j        ||¦  «         d S )Nr   é
   Úfloat64r/   r.   z6Column 'b' has dtype (object|str), cannot use method 'ú' with this dtype©ÚmatchÚ	nsmallest©Ú	ascending)r   r   ÚnpÚrandomÚdefault_rngÚpermutationr    r   r"   ÚpytestÚraisesÚ	TypeErrorÚgetattrÚIndexÚindexÚsort_valuesÚheadÚtmÚassert_frame_equal)	ÚselfÚnselect_methodr0   r+   ÚdfÚ	error_msgr:   ÚresultÚexpecteds	            r'   Útest_nlargest_nz%TestNLargestNSmallest.test_nlargest_n.   sÃ  € õ* Œ\å”Y×*Ò*¨1Ñ-Ô-×9Ò9¸"Ñ=Ô=Ý�/¨#¨2¨#Ô.Ñ/Ô/Ý”Y×*Ò*¨1Ñ-Ô-×9Ò9¸"Ñ=Ô=×DÒDÀYÑOÔOðð ñ
ô 
ˆð �%ˆ<ˆ<ðHØ&4ðHð Hð Hð õ ”�y°	Ð:Ñ:Ô:ð 6ð 6Ø+•˜˜NÑ+Ô+¨A¨uÑ5Ô5Ð5ð6ð 6ð 6ñ 6ô 6ð 6ð 6ð 6ð 6ð 6ð 6ð 6øøøð 6ð 6ð 6ð 6ð 6ð 6ð '¨+Ò5ˆIØ0•W˜R Ñ0Ô0°°EÑ:Ô:ˆFÝœ8¥D¨¬Ñ$6Ô$6Ñ7Ô7ˆFŒLØ—~’~ e°y�~ÑAÔA×FÒFÀqÑIÔIˆHÝÔ! &¨(Ñ3Ô3Ð3Ð3Ð3s   ÃC3Ã3C7Ã:C7r   r   r   r   c                 óˆ  — |}|d         }d|› d||         j         › d|› d�}|                     dd¦  «                             dd	¦  «                             d
d¦  «                             dd¦  «        }t          j        t          |¬¦  «        5   t          ||¦  «        d|¦  «         d d d ¦  «         d S # 1 swxY w Y   d S )Nr   zColumn 'z' has dtype z, cannot use method 'r5   ú(z\(ú)z\)ú[z\[ú]z\]r6   r   )ÚdtypeÚreplacer?   r@   rA   rB   )rI   r(   rJ   r   rK   ÚcolrL   s          r'   Útest_nlargest_errorz)TestNLargestNSmallest.test_nlargest_errorX   s2  € ð ˆØ�aŒjˆðD�sð Dð D¨¨3¬¬ð Dð DØ"0ðDð Dð Dð 	ð ×Ò˜c 5Ñ)Ô)ßŠW�S˜%Ñ Ô ßŠW�S˜%Ñ Ô ßŠW�S˜%Ñ Ô ð	 	õ Œ]�9¨IÐ6Ñ6Ô6ð 	4ð 	4Ø'�G�B˜Ñ'Ô'¨¨7Ñ3Ô3Ð3ð	4ð 	4ð 	4ñ 	4ô 	4ð 	4ð 	4ð 	4ð 	4ð 	4ð 	4ð 	4øøøð 	4ð 	4ð 	4ð 	4ð 	4ð 	4s   ÂB7Â7B;Â>B;c                 óÞ   — |}|                      dt          t          |¦  «        ddhz
  ¦  «        ¦  «         |                     dt          t          |¦  «        ddhz
  ¦  «        ¦  «         d S )Nr   r   r   )r8   r    ÚsetÚnlargest)rI   r(   rK   s      r'   Útest_nlargest_all_dtypesz.TestNLargestNSmallest.test_nlargest_all_dtypesl   sg   € ØˆØ
�Š�Q��S ™WœWÐ(9¸8Ð'DÑDÑEÔEÑFÔFÐFØ
�Š�A•t�C ™GœGÐ'8¸(Ð&CÑCÑDÔDÑEÔEÐEÐEÐEr&   c                 óp  — t          j        g d¢g d¢dœ¦  «        }|                     dddg¬¦  «        }t          j        g d¢g d	¢dœg d
¢¬¦  «        }t          j        ||¦  «         |                     dddg¬¦  «        }t          j        g d¢g d¢dœg d¢¬¦  «        }t          j        ||¦  «         d S )N)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   r   r   r   ©rD   )r   r   r   r   )r   r   r   r   )é   r_   r   r   )r   r   r[   rG   rH   r8   ©rI   rK   rM   rN   s       r'   Ú+test_nlargest_duplicates_on_starter_columnszATestNLargestNSmallest.test_nlargest_duplicates_on_starter_columnsq   sé   € õ Œ\Ð 2Ð 2Ð 2Ð9KÐ9KÐ9KÐLÐLÑMÔMˆà—’˜Q¨¨c¨
�Ñ3Ô3ˆÝ”<Ø�,�, \ \ \Ð2Ð2¸,¸,¸,ð
ñ 
ô 
ˆõ 	Ô˜f hÑ/Ô/Ð/à—’˜a¨#¨s¨�Ñ4Ô4ˆÝ”<Ø�,�, \ \ \Ð2Ð2¸,¸,¸,ð
ñ 
ô 
ˆõ 	Ô˜f hÑ/Ô/Ð/Ð/Ð/r&   c                 ó~  — t          j        dgdz  g d¢dœ¦  «        }|                     dd¦  «        }t          j        dgdz  g d¢dœt          d¦  «        ¬¦  «        }t	          j        ||¦  «         |                     dd¦  «        }t          j        dgdz  g d¢dœ¦  «        }t	          j        ||¦  «         d S )	Nr   ra   ©r   r   r   r_   ra   r^   r   r,   r   r`   )r   r   r[   ÚrangerG   rH   r8   rb   s       r'   Ú test_nlargest_n_identical_valuesz6TestNLargestNSmallest.test_nlargest_n_identical_values‚   sÂ   € åŒ\   a¡¨o¨o¨oÐ>Ð>Ñ?Ô?ˆà—’˜Q Ñ$Ô$ˆÝ”< q c¨A¡g°I°I°IÐ >Ð >ÅeÈAÁhÄhÐOÑOÔOˆÝ
Ô˜f hÑ/Ô/Ð/à—’˜a Ñ%Ô%ˆÝ”< q c¨A¡g°I°I°IÐ >Ð >Ñ?Ô?ˆÝ
Ô˜f hÑ/Ô/Ð/Ð/Ð/r&   é   c                 ó:  — t          j        g d¢g d¢g d¢dœg d¢¬¦  «        }|                     ||¦  «        }|                     |d¬¦  «                             |¦  «        }t          j        ||¦  «         |                     ||¦  «        }|                     |d	d¬
¦  «                             |¦  «        }|dgk    r|dv s|ddgk    r:|dk    r4|                     t          j
                             dd	¬¦  «        ¦  «         t          j        ||¦  «         d S )N)r   r   r   r_   r_   )r   r   r   r   r   )r   r   r   ra   r_   r/   )r   r   r   r   r   r`   Ústable)ÚkindF)r:   rk   r,   )r   r   r   r_   r.   ra   zYpandas default unstable sorting of duplicatesissue with numpy>=1.25 with AVX instructions)ÚreasonÚstrict)r   r   r8   rE   rF   rG   rH   r[   Úapplymarkerr?   ÚmarkÚxfail)rI   r0   r+   ÚrequestrK   rM   rN   s          r'   Útest_nlargest_n_duplicate_indexz5TestNLargestNSmallest.test_nlargest_n_duplicate_indexŽ   sD  € õ Œ\Ø!�/�/¨¨¨¸o¸o¸oÐNÐNØ!�/�/ð
ñ 
ô 
ˆð —’˜a Ñ'Ô'ˆØ—>’> %¨h�>Ñ7Ô7×<Ò<¸QÑ?Ô?ˆÝ
Ô˜f hÑ/Ô/Ð/à—’˜Q Ñ&Ô&ˆØ—>’> %°5¸x�>ÑHÔH×MÒMÈaÑPÔPˆØ�c�UŠNˆN˜q LÐ0Ð0°uÀÀcÀ
Ò7JÐ7JÐPQÐUVÒPVÐPVØ×ÒÝ”×!Ò!ðGð !ð "ñ ô ñô ð õ 	Ô˜f hÑ/Ô/Ð/Ð/Ð/r&   c                 ó„  — t          j        g d¢g d¢dœ¦  «        }|                     ddd¬¦  «        }t          j        g d¢g d	¢dœg d
¢¬¦  «        }t          j        ||¦  «         |                     ddd¬¦  «        }t          j        g d¢g d¢dœt          dd¦  «        ¬¦  «        }t          j        ||¦  «         d S )N)ra   r_   r_   r   r   r   r   r   )r3   r   r   r   ra   é2   r3   é   r^   r_   r,   Úall)Úkeep)ra   r_   r_   r   r   r   r   )r3   r   r   ra   rt   r3   ru   )r   r   r   r_   ra   rh   r   r`   r   )r   r   r   r   r   )r   ra   rt   r3   ru   r   r   )r   r   r[   rG   rH   r8   rf   rb   s       r'   Ú%test_nlargest_duplicate_keep_all_tiesz;TestNLargestNSmallest.test_nlargest_duplicate_keep_all_ties¬   s  € åŒ\Ø*Ð*Ð*Ð1MÐ1MÐ1MÐNÐNñ
ô 
ˆð —’˜Q ¨%�Ñ0Ô0ˆÝ”<à*Ð*Ð*Ø.Ð.Ð.ðð ð (Ð'Ð'ð
ñ 
ô 
ˆõ 	Ô˜f hÑ/Ô/Ð/à—’˜a ¨5�Ñ1Ô1ˆÝ”<à$�_�_Ø'Ð'Ð'ðð õ ˜˜1‘+”+ð
ñ 
ô 
ˆõ 	Ô˜f hÑ/Ô/Ð/Ð/Ð/r&   c                 óv  — t          j        t           j                             dgddgg¦  «        ddgddgddgd	d
gg¬¦  «        }|                     dd¦  «        }|j        g d¢         }t          j        ||¦  «         |                     dd¦  «        }|j        g d¢         }t          j        ||¦  «         d S )NÚxr,   r.   g…ëQ¸Õ?g¤p=
×£À?g…ëQ¸…ë?g      Ð?gffffffæ?g333333ë?g…ëQ¸í?)r   Údatar   )rz   r,   )r   r   r   )rz   r.   )r   r   r   )	r   r   Ú
MultiIndexÚfrom_productr8   ÚilocrG   rH   r[   rb   s       r'   Ú&test_nlargest_multiindex_column_lookupz<TestNLargestNSmallest.test_nlargest_multiindex_column_lookupÅ   sÊ   € õ Œ\Ý”M×.Ò.°°¸¸S°zÐ/BÑCÔCØ˜�,  t ¨t°T¨l¸TÀ4¸LÐIð
ñ 
ô 
ˆð —’˜a Ñ,Ô,ˆØ”7˜9˜9˜9Ô%ˆÝ
Ô˜f hÑ/Ô/Ð/ð —’˜Q 
Ñ+Ô+ˆØ”7˜9˜9˜9Ô%ˆÝ
Ô˜f hÑ/Ô/Ð/Ð/Ð/r&   c                 ó  — t          j        t          j        t          j        ddddg¦  «        }|                     dd¦  «        }|                     dd¬¦  «                             d¦  «        }t          j        ||¦  «         d S )Nr   r   r   r   ra   Fr9   )	r   r   r;   Únanr[   rE   rF   rG   rH   rb   s       r'   Útest_nlargest_nanz'TestNLargestNSmallest.test_nlargest_nan×   sq   € åŒ\�2œ6¥2¤6¨1¨a°°AÐ6Ñ7Ô7ˆØ—’˜Q Ñ"Ô"ˆØ—>’> !¨u�>Ñ5Ô5×:Ò:¸1Ñ=Ô=ˆÝ
Ô˜f hÑ/Ô/Ð/Ð/Ð/r&   c                 ó@  — t          j        g d¢g d¢g d¢dœt          d¦  «        ¬¦  «        }|                     ddd	g¬
¦  «        }t          j        g d¢g d¢g d¢dœt          d¦  «        ¬¦  «                             ddi¦  «        }t          j        ||¦  «         d S )N)r   r   r   r_   ra   Nr   )r   rh   ra   r_   r   r   r   )r   r   r   r   r   r   r   r/   r   r`   ra   r,   r.   r   re   )r   rh   ra   r_   r   )r   r   r   r   r   r   )r   r   rf   r8   r"   rG   rH   rb   s       r'   Ú"test_nsmallest_nan_after_n_elementz8TestNLargestNSmallest.test_nsmallest_nan_after_n_elementÞ   sÔ   € åŒ\à-Ð-Ð-Ø*Ð*Ð*Ø*Ð*Ð*ðð õ
 ˜‘(”(ð
ñ 
ô 
ˆð —’˜a¨#¨s¨�Ñ4Ô4ˆÝ”<à$�_�_Ø$�_�_Ø$�_�_ðð õ
 ˜‘(”(ð
ñ 
ô 
÷ Š&�#�w�Ñ
 Ô
 ð 	õ 	Ô˜f hÑ/Ô/Ð/Ð/Ð/r&   N)Ú__name__Ú
__module__Ú__qualname__r?   ro   Úparametrizerf   rO   rX   r\   rc   rg   rr   rx   r   r‚   r„   r%   r&   r'   r*   r*   +   sö  € € € € € ð „[×ÒØàˆEØˆEØ�#ˆJØ�#ˆJØ�#ˆJØ�#ˆJØˆOˆOØˆOˆOØˆOˆOØˆOˆOØˆOˆOàˆOˆOð	
ñô ð$ „[×Ò˜S % %¨¨2¡,¤,Ñ/Ô/ð4ð 4ñ 0Ô/ñ%ô ð&4ð. „[×ÒØ�WÐ/Ð0°7¸HÐ2EÐFñô ð4ð 4ñô ð4ð"Fð Fð Fð
0ð 0ð 0ð"
0ð 
0ð 
0ð „[×ÒØØ	ˆˆ˜/˜/˜/¨C¨5°3°%¸#¸s¸ÀcÈ3ÀZÐPñô ð „[×Ò˜S % %¨¨1¡+¤+Ñ.Ô.ð0ð 0ñ /Ô.ñ	ô ð
0ð20ð 0ð 0ð20ð 0ð 0ð$0ð 0ð 0ð0ð 0ð 0ð 0ð 0r&   r*   )Ú__doc__r   r   Únumpyr;   r?   Úpandasr   Úpandas._testingÚ_testingrG   Úfixturer(   r*   r%   r&   r'   ú<module>r�      s¯   ððð ð
 #Ð "Ð "Ð "Ð "Ð "à Ð Ð Ð Ø €€€à Ð Ð Ð Ø Ð Ð Ð Ð Ð ð „ðð ñ „ðð6F0ð F0ð F0ð F0ð F0ñ F0ô F0ð F0ð F0ð F0r&   