
    MhI                     <   S SK rS SKrS SKJrJrJr  S SKJr	  S r
S rS r\R                  R                  S/ SQ/ SQ4S /S /4/5      \R                  R                  S	S/\	R                  Q5      \R                  R                  S
SS/5      S 5       5       5       rg)    N)
MultiIndexSeries
date_rangec                     [        / SQ5      n [        [        S5      5      nU R                  U5      nUR                  S5      n[        / SQ[        R
                  " [        S5      / SQ/5      S9n[        R                  " X45        [        / SQ5      n U R                  U5      n[        / S	Q[        R
                  " [        S5      / S
Q/5      S9n[        R                  " UR                  SSS9U5        g )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        f/var/www/html/env/lib/python3.13/site-packages/pandas/tests/groupby/methods/test_nlargest_nsmallest.pytest_nlargestr'      s    ./At%&'A	
1B
AA$$d8n6H%IJ	A 1 -.A	
1B$$d8n6H%IJ	A 2;;qv;6:    c                  >   [         R                  R                  S5      n [        SSS9nUSS//n[        R
                  " USS/S	9n[        U R                  S
5      US9nUR                  S5      R                  S5      n[        R                  " US   US   S4US   US   S4US   US   S4US   US   S4US   US   S4US   US   S4US   US   S4US   US   S4US   US   S4US   US   S4/
/ SQS	9n/ SQn[        XvS9n[        R                  " XXSSS9  g )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?gjsտg1Au?gf+Qs?g;?FgMbP?)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       sj   
))


"C
Z
,Cuen%I

!
!)GX3F
GCs""2&c2AYYw((+F$$VSVU#VSVU#VSVU#VSVU#VSVU#VSVU#VSVU#VSVU#VSVU#VSVU#	
 +G J j0H6TJr(   c                     [        / SQ5      n [        [        S5      5      nU R                  U5      nUR                  S5      n[        / SQ[        R
                  " [        S5      / SQ/5      S9n[        R                  " X45        [        / SQ5      n U R                  U5      n[        / S	Q[        R
                  " [        S5      / S
Q/5      S9n[        R                  " UR                  SSS9U5        g )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$$d8n6H%IJ	A 1 -.A	
1B$$d8n6H%IJ	A 2<<<7;r(   zdata, groups)r   r   r   r	   )r   r   r   r   dtypemethodr   rE   c                    Ub  [         R                  " XS9n US:X  a  [        [        U 5      5      n [	        U SS9n[        UR                  U5      U5      " SS9n[        U[        5      (       a  [         R                  " U[        S9OUn[	        U [        R                  " XdR                  /5      SS9n[        R                  " XW5        g )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     xx*HTN#
C
 CS[[(&1A6F,6vt,D,DRXXfC(&Fd*"8"8&))9L"MTWXH6,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!QC!:. 4"B"*A*A"BCJ#<=- > D	-r(   