U
    sVÞc†  ã                   @   sr   d dl Zd dlZd dlZd dlmZmZ dd„ Zej	 
deƒ dfeƒ dfg¡dd	„ ƒZd
d„ Zdd„ Zdd„ ZdS )é    N)ÚFloat32DtypeÚFloat64Dtypec                 C   s$   t  | j¡jdk | jd k	s t‚d S )NÚf)ÚnpÚdtypeÚtypeÚkindÚnameÚAssertionError)r   © r   úJ/tmp/pip-unpacked-wheel-xj8nt62q/pandas/tests/arrays/floating/test_repr.pyÚtest_dtypes   s    r   zdtype, expectedzFloat32Dtype()zFloat64Dtype()c                 C   s   t | ƒ|kst‚d S )N)Úreprr
   )r   Úexpectedr   r   r   Útest_repr_dtype   s    r   c                  C   s(   t t dd dg¡ƒ} d}| |ks$t‚d S )Nç      ð?g      @z:<FloatingArray>
[1.0, <NA>, 3.0]
Length: 3, dtype: Float64)r   ÚpdÚarrayr
   )Úresultr   r   r   r   Útest_repr_array   s    r   c                  C   s0   t  ddd gd ¡} d}t| ƒ}||ks,t‚d S )Nr   g       @iè  z«<FloatingArray>
[ 1.0,  2.0, <NA>,  1.0,  2.0, <NA>,  1.0,  2.0, <NA>,  1.0,
 ...
 <NA>,  1.0,  2.0, <NA>,  1.0,  2.0, <NA>,  1.0,  2.0, <NA>]
Length: 3000, dtype: Float64)r   r   r   r
   )Údatar   r   r   r   r   Útest_repr_array_long    s    r   c                 C   s*   t  d| i¡}t|ƒ}d}||ks&t‚d S )NÚAz      A
0  <NA>
1   0.1)r   Z	DataFramer   r
   )Zdata_missingZdfr   r   r   r   r   Útest_frame_repr+   s    r   )Znumpyr   ZpytestZpandasr   Zpandas.core.arrays.floatingr   r   r   ÚmarkZparametrizer   r   r   r   r   r   r   r   Ú<module>   s   þ
