U
    sVc]                     @   s   d dl Zd dlZd dlmZ d dlZd dlmZ	 dd e	j
D Zedd e	jD 7 Zejedd eD ddd	 Ze d
d Zdd ZdS )    N)is_integer_dtypec                 C   s"   g | ]}t jd dddg|dqS )         Ndtypepdarray.0r    r   L/tmp/pip-unpacked-wheel-xj8nt62q/pandas/tests/arrays/masked/test_function.py
<listcomp>	   s     r   c                 C   s"   g | ]}t jd dddg|dqS )g?5^I?gx&ѿgGz@Nr   r   r   r   r   r   r   
   s    c                 C   s   g | ]}|j jqS r   )r   name)r   ar   r   r   r      s     )paramsidsc                 C   s   | j S )zi
    Fixture returning parametrized 'data' array with different integer and
    floating point types
    )param)requestr   r   r   data   s    r   c                 C   s   t | rt}n| jj}|S )z@
    Fixture returning numpy dtype from 'data' input array.
    )r   floatr   type)r   numpy_dtyper   r   r   r      s    r   c                 C   st   |   }tjt | j|d d| jd}t|| | j dd}tjtj | j|d ddd| jd}t|| d S )N)r   Zna_valuer   r   )Zdecimals)roundr	   r
   npZto_numpyr   tmZassert_extension_array_equal)r   r   resultexpectedr   r   r   
test_round%   s     r   )Znumpyr   ZpytestZpandas.core.dtypes.commonr   Zpandasr	   Zpandas._testingZ_testingr   ZALL_INT_EA_DTYPESZarraysZFLOAT_EA_DTYPESZfixturer   r   r   r   r   r   r   <module>   s   

