
    >ie                        d dl mZ d dlmZ d dlZd dlmZ d dlm	Z	  ej
        d          Zd dlmZmZ d dlmZmZ  ej        e          d             Z ej        e          d	             Zd
 Zd Zd ZdS )    )annotationsDecimalN)	assert_eqpandas)DecimalArrayDecimalDtype)make_array_nonemptymake_scalarc                f    t          j        t          d          t          d          g|           S )N0NaNdtype)r   _from_sequencer   r   s    Dlib/python3.11/site-packages/dask/dataframe/tests/test_extensions.py_r      s)    &genn'EUSSSS    c                     t          d          S )N1r   )xs    r   r   r      s    3<<r   c                 P   t          j        t          d          gdz            } t                              |           }t          j        |d          }t          ||           t                              d|i          }t          j        |d          }t          ||           d S )N1.0
      A)	r   r   r   pdSeriesddfrom_pandasr   	DataFrame)arrserdserdfddfs        r   test_register_extension_typer'      s    

%wu~~&6&;
<
<C
))C..C>#q!!Dc4	sCj	!	!B
.Q

Cb#r   c                 0   t                               t          j        t	          d          t	          d          g                    } t          j        | d          }t          |                     d          |                    d                     d S )Nr   r   r   F)skipna)	r   r   r   r   r   r   r    r   mean)r#   r$   s     r   test_reductionr+   &   sr    
))L/ws||0LMM
N
NC>#q!!Dchheh$$diiui&=&=>>>>>r   c                     t           j                            t          d          t                                                    } | t          d          k    sJ d S )Nr   )parent_meta)r   utils	make_metar   r   r!   )results    r   test_scalarr1   2   sG    XBLLNNKKFWU^^######r   )
__future__r   decimalr   pytestdask.dataframe	dataframer   dask.dataframe.utilsr   importorskipr   $pandas.tests.extension.decimal.arrayr   r	   dask.dataframe.extensionsr
   r   registerr   r'   r+   r1    r   r   <module>r=      s%   " " " " " "              * * * * * *V"" K K K K K K K K F F F F F F F F l++T T ,+T g    ? ? ?$ $ $ $ $r   