a
    PªfI  ã                   @   s(   d dl Zd dlZd dlmZ dd„ ZdS )é    Nc                  C   s´   d} d}d}d}t j d¡ | |f¡}t j|t j d¡ | ¡|k < t |¡}t j d¡jd|| d}| |¡}| 	¡ }dd„ |D ƒ}	tj
|	dd	}
|
j |jj¡|
_t ||
¡ d S )
Niè  é   é   gš™™™™™©?r   )Úsizec                 S   s    g | ]\}}|  ¡  |¡j‘qS © )ÚskewZto_frameÚT)Ú.0ÚiÚgrpr   r   ú_/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/groupby/methods/test_skew.pyÚ
<listcomp>   ó    z1test_groupby_skew_equivalence.<locals>.<listcomp>)Zaxis)ÚnpÚrandomZdefault_rngZstandard_normalÚnanÚpdZ	DataFrameZintegersÚgroupbyr   ÚconcatÚindexZastypeZdtypeÚtmZassert_frame_equal)ZnrowsZngroupsZncolsZnan_fracZarrZdfZgrpsÚgbÚresultZgrpwiseÚexpectedr   r   r   Útest_groupby_skew_equivalence   s    

r   )Únumpyr   Zpandasr   Zpandas._testingZ_testingr   r   r   r   r   r   Ú<module>   s   