a
    PfI                     @   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   s   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Zkeep)r   listgroupbynlargestr   from_arraystmassert_series_equalabgbre r#   m/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/groupby/methods/test_nlargest_nsmallest.pytest_nlargest   s     


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
   Z20180101r   )Z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?)Zcheck_exactZrtol)nprandomZdefault_rngr   r   Zfrom_productr   Zstandard_normalr   r   from_tuplesr   r   )	Znprdts	iterablesidxsresultZexp_idxZ
exp_valuesexpectedr#   r#   r$   test_nlargest_mi_grouper    s.    r5   c                  C   s   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 )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   s     


r7   zdata, groups)r   r   r
   r   )r   r   r   r   dtypemethodr   r6   c                 C   s   |d urt j| |d} |dkr*tt| } t| dd}t|||dd}t|trdt j|tdn|}t| t	
||jgdd}t|| d S )N)r8   r   r   )namer
   )n)r   r:   )r,   arrayr   reversedr   getattrr   
isinstanceintr   r   r   r   r   )datagroupsr8   r9   Zserr3   Zexpidxr4   r#   r#   r$   test_nlargest_and_smallest_noopa   s    
rC   )numpyr,   ZpytestZpandasr   r   r   Zpandas._testingZ_testingr   r%   r5   r7   markZparametrizeZALL_INT_NUMPY_DTYPESrC   r#   r#   r#   r$   <module>   s   -