
    bi                        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y)    )annotationsDecimalN)	assert_eqpandas)DecimalArrayDecimalDtype)make_array_nonemptymake_scalarc                X    t        j                  t        d      t        d      g|       S )N0NaNdtype)r   _from_sequencer   r   s    _/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/dask/dataframe/tests/test_extensions.py_r      s"    &&gen'EUSS    c                    t        d      S )N1r   )xs    r   r   r      s    3<r   c                 (   t        j                  t        d      gdz        } t        j	                  |       }t        j                  |d      }t        ||       t        j                  d|i      }t        j                  |d      }t        ||       y )N1.0
      A)	r   r   r   pdSeriesddfrom_pandasr   	DataFrame)arrserdserdfddfs        r   test_register_extension_typer'      sq    

%
%wu~&6&;
<C
))C.C>>#q!Dc4	sCj	!B
..Q
Cb#r   c                     t         j                  t        j                  t	        d      t	        d      g            } t        j                  | d      }t        | j                  d      |j                  d             y )Nr   r   r   F)skipna)	r   r   r   r   r   r   r    r   mean)r#   r$   s     r   test_reductionr+   &   sX    
))L//ws|0LM
NC>>#q!Dchheh$diiui&=>r   c                     t         j                  j                  t        d      t        j                               } | t        d      k(  sJ y )Nr   )parent_meta)r   utils	make_metar   r   r!   )results    r   test_scalarr1   2   s7    XXBLLN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 F l+T ,T g ?$r   