o
    թZhI                     @   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/html/lang_env/lib/python3.10/site-packages/pandas/tests/groupby/methods/test_skew.py
<listcomp>   s     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arr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    