U
    ¾mœd‡  ã                   @   sÂ   d dl Zd dlZd dlZejdd„ ƒZej ddej	gdddej	fdej	gddej	dfej	gddej	ej	fg ddddfddgddddfddgddddfg¡dd„ ƒZ
ej d	ddg¡d
d„ ƒZdS )é    Nc                   C   s:   t jddgd tjg ddgd  tjg ddg dd�S )z?Fixture returning boolean array, with valid and missing values.TFé   é,   Úboolean©Zdtype)ÚpdÚarrayÚnpÚnan© r
   r
   úc/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/pandas/tests/arrays/boolean/test_reduction.pyÚdata   s    .þr   z8values, exp_any, exp_all, exp_any_noskip, exp_all_noskipTFc                 C   sþ   |t jkrt jnt |¡}|t jkr*t jnt |¡}|t jkrDt jnt |¡}|t jkr^t jnt |¡}t jt jfD ]„}|| dd�}| ¡ |ks”t‚| ¡ |ks¤t‚|jdd�|ks¸t‚|jdd�|ksÌt‚t | ¡ ¡|ksât‚t | ¡ ¡|kstt‚qtd S )Nr   r   F)Zskipna)	r   ÚNAr   Úbool_r   ÚSeriesÚanyÚAssertionErrorÚall)ÚvaluesZexp_anyZexp_allZexp_any_noskipZexp_all_noskipÚconÚar
   r
   r   Útest_any_all   s    r   Údropnac                 C   sœ   |}t  |¡}| r| ¡ }|dkr<tt||ƒƒ tjƒs˜t‚n\|dkr^tt||ƒƒ tjƒs˜t‚n:|dkr€tt||ƒƒ tj	ƒs˜t‚ntt||ƒƒ tj
ƒs˜t‚d S )N)ÚsumÚprodÚcount)ÚminÚmax)r   r   r   Ú
isinstanceÚgetattrr   Úint_r   Úintegerr   Zfloat64)r   r   Zall_numeric_reductionsÚopÚsr
   r
   r   Útest_reductions_return_types.   s    
r#   )Únumpyr   ZpytestZpandasr   Zfixturer   ÚmarkZparametrizer   r   r#   r
   r
   r
   r   Ú<module>   s"   
ùþ
