Ë
    äªgI  ã                   ó<  — d dl Zd dlZd dlmZmZmZ d dlmZ	 d„ Z
d„ Zd„ Zej                  j                  dg d¢g d¢fd gd gfg«      ej                  j                  d	dge	j                  ¢«      ej                  j                  d
ddg«      d„ «       «       «       Zy)é    N)Ú
MultiIndexÚSeriesÚ
date_rangec                  óü  — t        g d¢«      } t        t        d«      «      }| j                  |«      }|j                  d«      }t        g d¢t	        j
                  t        d«      g d¢g«      ¬«      }t        j                  ||«       t        g d¢«      } | j                  |«      }t        g d	¢t	        j
                  t        d«      g d
¢g«      ¬«      }t        j                  |j                  dd¬«      |«       y )N©
é   é   é   é   é   é	   r   é   é   é
   Ú
aaaaabbbbbr	   )r   r
   r	   r   r   r   Úaaabbb)r	   r   r   r   r
   é   ©Úindex©
r   r   r	   r   r   r	   r	   r   r   r   )r	   r   r   r	   r	   r   )r   r	   r   r   r
   r   Úlast©Úkeep)r   ÚlistÚgroupbyÚnlargestr   Úfrom_arraysÚtmÚassert_series_equal©ÚaÚbÚgbÚrÚes        úi/var/www/openai/venv/lib/python3.12/site-packages/pandas/tests/groupby/methods/test_nlargest_nsmallest.pyÚtest_nlargestr'      sÊ   € ÜÒ.Ó/€AÜŒtÐ%Ó&Ó'€AØ	
‰1‹€BØ
‰A‹€AÜÚÜ×$Ñ$¤d¨8£nÒ6HÐ%IÓJô	€Aô ×Ñ˜1˜aÔ äÒ-Ó.€AØ	
‰1‹€BÜÚÜ×$Ñ$¤d¨8£nÒ6HÐ%IÓJô	€Aô ×Ñ˜2Ÿ;™; q¨v˜;Ó6¸Õ:ó    c                  ó`  — t         j                  j                  d«      } t        dd¬«      }|ddgg}t	        j
                  |ddg¬	«      }t        | j                  d
«      |¬«      }|j                  d«      j                  d«      }t	        j                  |d   |d   df|d   |d   df|d   |d   df|d   |d   df|d   |d   df|d   |d   df|d   |d   df|d   |d   df|d   |d   df|d   |d   dfg
g d¢¬	«      }g d¢}t        ||¬«      }t        j                  ||dd¬«       y )Nr   Ú20180101r   )ÚperiodsÚoneÚtwoÚfirstÚsecond)Únamesé   r   r   r   r	   r   r
   r   r   r   r   )r.   r.   r/   )
g½. ¶æ2È?go‚¢oÚ¿gò÷™Ëü?g—íÃè?gSL„	[ÿÑ?gˆ‰;Hï?g‚›„jsÕ¿g1÷AuÝ?gf+Qsá?g¦Èå;À?Fgü©ñÒMbP?)Úcheck_exactÚrtol)ÚnpÚrandomÚdefault_rngr   r   Úfrom_productr   Ústandard_normalr   r   Úfrom_tuplesr   r   )	ÚnprÚdtsÚ	iterablesÚidxÚsÚresultÚexp_idxÚ
exp_valuesÚexpecteds	            r&   Útest_nlargest_mi_grouperrC       sn  € ä
)‰)×
Ñ
 Ó
"€Cä
Z¨Ô
,€CØu˜enÐ%€Iä
×
!Ñ
! )°G¸XÐ3FÔ
G€CÜˆs×"Ñ" 2Ó&¨cÔ2€AàY‰YwÓ×(Ñ(¨Ó+€Fä×$Ñ$à‰VS˜‘V˜UÐ#Ø‰VS˜‘V˜UÐ#Ø‰VS˜‘V˜UÐ#Ø‰VS˜‘V˜UÐ#Ø‰VS˜‘V˜UÐ#Ø‰VS˜‘V˜UÐ#Ø‰VS˜‘V˜UÐ#Ø‰VS˜‘V˜UÐ#Ø‰VS˜‘V˜UÐ#Ø‰VS˜‘V˜UÐ#ð	
ò +ô€Gò €Jô j¨Ô0€HÜ×Ñ˜6 8¸ÀTÖJr(   c                  óü  — t        g d¢«      } t        t        d«      «      }| j                  |«      }|j                  d«      }t        g d¢t	        j
                  t        d«      g d¢g«      ¬«      }t        j                  ||«       t        g d¢«      } | j                  |«      }t        g d	¢t	        j
                  t        d«      g d
¢g«      ¬«      }t        j                  |j                  dd¬«      |«       y )Nr   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   r   r   )r   r   r   Ú	nsmallestr   r   r   r   r    s        r&   Útest_nsmallestrF   M   sÊ   € ÜÒ.Ó/€AÜŒtÐ%Ó&Ó'€AØ	
‰1‹€BØ
‰Q‹€AÜÚÜ×$Ñ$¤d¨8£nÒ6HÐ%IÓJô	€Aô ×Ñ˜1˜aÔ äÒ-Ó.€AØ	
‰1‹€BÜÚÜ×$Ñ$¤d¨8£nÒ6HÐ%IÓJô	€Aô ×Ñ˜2Ÿ<™<¨°˜<Ó7¸Õ;r(   zdata, groups)r   r   r   r	   )r   r   r   r   ÚdtypeÚmethodr   rE   c                 ó¦  — |t        j                  | |¬«      } |dk(  rt        t        | «      «      } t	        | d¬«      } t        |j                  |«      |«      d¬«      }t        |t        «      rt        j                  |t        ¬«      n|}t	        | t        j                  ||j                  g«      d¬«      }t        j                  ||«       y )N)rG   r   r!   )Únamer   )Ún)r   rJ   )r4   Úarrayr   Úreversedr   Úgetattrr   Ú
isinstanceÚintr   r   r   r   r   )ÚdataÚgroupsrG   rH   Úserr?   ÚexpidxrB   s           r&   Útest_nlargest_and_smallest_nooprU   a   s¦   € ð ÐÜx‰x˜ EÔ*ˆØÒÜ”H˜T“NÓ#ˆÜ
˜CÔ
 €CØ1ŒWS—[‘[ Ó(¨&Ó1°AÔ6€FÜ,6°v¼tÔ,DŒRX‰Xf¤CÕ(È&€FÜd¤*×"8Ñ"8¸&À#Ç)Á)Ð9LÓ"MÐTWÔX€HÜ×Ñ˜6 8Õ,r(   )Únumpyr4   ÚpytestÚpandasr   r   r   Úpandas._testingÚ_testingr   r'   rC   rF   ÚmarkÚparametrizeÚALL_INT_NUMPY_DTYPESrU   © r(   r&   Ú<module>r_      s®   ðÛ Û ÷ñ õ
 ò;ò(*KòZ<ð( ‡×ÑØÚ’LÐ! Q C¨!¨ :Ð.óð ‡×Ñ˜ 4Ð"B¨"×*AÑ*AÐ"BÓCØ‡×Ñ˜ J°Ð#<Ó=ñ-ó >ó Dó	ñ-r(   