o
    ŀgI                     @   s   d dl Zd dlZd dlmZmZmZ d dlmZ	 dd Z
dd Zdd Zejd	g d
g dfd gd gfgejddge	jejdddgdd ZdS )    N)
MultiIndexSeries
date_rangec                  C      t g d} t td}| |}|d}t g dttdg dgd}t|| t g d} | |}t g d	ttdg d
gd}t|jddd| d S )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e r%   h/var/www/html/myenv/lib/python3.10/site-packages/pandas/tests/groupby/methods/test_nlargest_nsmallest.pytest_nlargest       


r'   c            	      C   s4  t jd} tddd}|ddgg}tj|ddgd	}t| d
|d}|d	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d	}g d}t||d}tj||ddd d S )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expectedr%   r%   r&   test_nlargest_mi_grouper    s.   rB   c                  C   r   )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   r%   r%   r&   test_nsmallestM   r(   rD   zdata, groups)r   r   r   r   )r   r   r   r   dtypemethodr   rC   c                 C   s   |d urt j| |d} |dkrtt| } t| dd}t|||dd}t|tr2t j|tdn|}t| t	
||jgdd}t|| d S )N)rE   r   r    )namer   )n)r   rG   )r3   arrayr   reversedr   getattrr   
isinstanceintr   r   r   r   r   )datagroupsrE   rF   serr>   expidxrA   r%   r%   r&   test_nlargest_and_smallest_noopa   s   
rR   )numpyr3   pytestpandasr   r   r   pandas._testing_testingr   r'   rB   rD   markparametrizeALL_INT_NUMPY_DTYPESrR   r%   r%   r%   r&   <module>   s    -