a
    Pfw                     @   s  d dl Zd dlZd dlZd dlmZmZmZ d dlm	Z
 ejddedgdedgded	gd
edgdedgdedgdedgdedgdejgdejgddgddggdd Zejddedgdgddgdedgdgddgded	gdgddgd
edgdgddgdedgdgddgdedgdgddgdedgdgddgdedgdgddgdeejgdgddgdeejgdgddgdedgdgddgdedgdgddggdd Zejdded	dgdd gded	dgdd gded	dgdd gd
ed	dgdd gded	dgdd gded	dgdd gdeejgdd gdeejgdd gdeejgdd gdeddgdd gdeddgdd ggd!d" Zejdded#d#gd$d%gddgdeddgd$d%gddgded	d	gd$d%gddgd
eddgd$d%gddgdeddgd$d%gddgdeddgd$d%gddgdeddgd$d%gddgded&d&gd$d%gddgdeejejgd$d%gddgdeejejgd$d%gddgdeg d'g d(ddgdeg d'g d(ddggd)d* ZdS )+    N)	DataFrameSeriesarrayzop, expectedsum   prod   min   maxmeang      ?Zmedianvarg      ?Zstdg;f?ZskewZkurtanyTallc                 C   s,   t ddgdd}t||  }t|| d S )Nr
   r   Int64dtype)r   getattrtmassert_equal)opexpectedZserresult r   c/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/arrays/integer/test_reduction.pytest_series_reductions   s    r   ar   )indexr   ZFloat64booleanc                 C   s4   t dtddgddi}t||  }t|| d S )Nr   r
   r   r   r   )r   r   r   r   assert_series_equalr   r   dfr   r   r   r   test_dataframe_reductions$   s    r"   r   c                 C   s^   t g dtg dddd}t|d|  }t |tjddgdd	d
gd}t|| d S )Nr   br$   r
   Nr   r   r   )ABr&   r   r$   )namer'   )r   columns)r   r   r   groupbypdIndexr   Zassert_frame_equalr    r   r   r   test_groupby_reductions;   s    r-      r'   Cg;f?)TTTr&   r'   r/   c                 C   s   | dv r|r| d}tg dg dtg dddd}t|j|  }t||d  | dv rnt||  }nt|| d	d
}t|| d S )N)r   r   boolr#   r%   r   r   r0   r/   T)Znumeric_only)Zastyper   r   r   r/   r   r   r   )r   r   Zusing_infer_stringr!   r   r   r   r   test_mixed_reductionsX   s    
	r2   )numpynpZpytestZpandasr+   r   r   r   Zpandas._testingZ_testingr   markZparametrizeZint64Zfloat64ZNAr   r"   r-   r2   r   r   r   r   <module>   s   


