o
    g                     @   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      g | ]}t jg d |dqS ))         Ndtypepdarray.0r    r   f/home/ubuntu/cloudmapper/venv/lib/python3.10/site-packages/pandas/tests/arrays/masked/test_function.py
<listcomp>	   s    r   c                 C   r   ))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 N)param)requestr   r   r   data   s   r   c                 C   s   t | rt}|S | jj}|S r   )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   na_valuer   r   )decimals)roundr
   r   npto_numpyr   tmassert_extension_array_equal)r   r   resultexpectedr   r   r   
test_round   s   r%   )numpyr   pytestpandas.core.dtypes.commonr   pandasr
   pandas._testing_testingr!   ALL_EA_INT_DTYPESarraysFLOAT_EA_DTYPESfixturer   r   r%   r   r   r   r   <module>   s    

	