o
    $jI                     @   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Zkeep)r   listgroupbynlargestr   from_arraystmassert_series_equalabgbre r$   u/var/www/html/Deteccion_Ine/venv/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   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.   r7   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'   r9   zdata, groups)r   r   r   r   )r   r   r   r   dtypemethodr   r8   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)r:   r   r   )namer   )n)r   r<   )r.   arrayr   reversedr   getattrr   
isinstanceintr   r   r   r   r   )datagroupsr:   r;   Zserr5   Zexpidxr6   r$   r$   r%   test_nlargest_and_smallest_noopa   s   
rE   )numpyr.   ZpytestZpandasr   r   r   Zpandas._testingZ_testingr   r&   r7   r9   markZparametrizeZALL_INT_NUMPY_DTYPESrE   r$   r$   r$   r%   <module>   s    -