U
    /e                     @   s   d dl mZ d dlZd dlmZ d dlmZ edZ	d dl
mZmZ d dlmZmZ eedd Zeed	d Zd
d Zdd Zdd ZdS )    DecimalN)	assert_eqZpandas)DecimalArrayDecimalDtype)make_array_nonemptymake_scalarc                 C   s   t jtdtdg| dS )N0NaNZdtype)r   _from_sequencer   r    r   H/tmp/pip-unpacked-wheel-dbjnr7gq/dask/dataframe/tests/test_extensions.py_   s    r   c                 C   s   t dS )N1r   )xr   r   r   r      s    c                  C   s\   t tdgd } t| }t|d}t|| td|i}t|d}t|| d S )N1.0
      A)	r   r   r   pdSeriesddfrom_pandasr   	DataFrame)ZarrserdserZdfZddfr   r   r   test_register_extension_type   s    

r   c                  C   sF   t ttdtdg} t| d}t| jdd|jdd d S )Nr	   r   r   F)Zskipna)	r   r   r   r   r   r   r   r   Zmean)r   r   r   r   r   test_reduction$   s    r   c                  C   s,   t jjtdt d} | tdks(td S )Nr   )Zparent_meta)r   utilsZ	make_metar   r   r   AssertionError)resultr   r   r   test_scalar0   s    r"   )decimalr   ZpytestZdask.dataframeZ	dataframer   Zdask.dataframe.utilsr   Zimportorskipr   Z$pandas.tests.extension.decimal.arrayr   r   Zdask.dataframe.extensionsr   r   registerr   r   r   r"   r   r   r   r   <module>   s   


