a
    oÝEb8  ã                   @  s’   d dl mZ d dlZd dlZd dlmZ d dlZd dlm	Z	m
Z
 ejddd� d dlZd dlmZ e
G d	d
„ d
e	ƒƒZG dd„ deƒZdd„ ZdS )é    )ÚannotationsN)Útype_t)ÚExtensionDtypeÚregister_extension_dtypeÚpyarrowz1.0.1)Z
minversion)ÚArrowExtensionArrayc                   @  s2   e Zd ZejZdZdZejZ	e
ddœdd„ƒZdS )ÚArrowTimestampUSDtypeÚMZarrow_timestamp_usztype_t[ArrowTimestampUSArray])Úreturnc                 C  s   t S )zq
        Return the array type associated with this dtype.

        Returns
        -------
        type
        )ÚArrowTimestampUSArray)Úcls© r   úk/home/ja/django-apps/lartica_env/lib/python3.9/site-packages/pandas/tests/extension/arrow/test_timestamp.pyÚconstruct_array_type   s    	z*ArrowTimestampUSDtype.construct_array_typeN)Ú__name__Ú
__module__Ú__qualname__ÚdatetimeÚtypeÚkindÚnameÚpaZNULLZna_valueÚclassmethodr   r   r   r   r   r      s   r   c                   @  s   e Zd Zdd„ ZdS )r   c                 C  s6   t |tjƒst‚|jt d¡ks$J ‚|| _tƒ | _d S )NÚus)	Ú
isinstancer   ZChunkedArrayÚ
ValueErrorr   Ú	timestampÚ_datar   Z_dtype)ÚselfÚvaluesr   r   r   Ú__init__+   s
    zArrowTimestampUSArray.__init__N)r   r   r   r    r   r   r   r   r   *   s   r   c                   C  s.   t  dt d t ddddddd¡g¡i¡ d S )	Nr   iÚ  é	   é   é   é   é   é   )ÚpdZ	DataFramer   Zfrom_scalarsr   r   r   r   r   Útest_constructor_extensionblock4   s    ÿÿÿr(   )Ú
__future__r   r   ZpytestZpandas._typingr   Zpandasr'   Zpandas.api.extensionsr   r   Zimportorskipr   r   Z#pandas.tests.extension.arrow.arraysr   r   r   r(   r   r   r   r   Ú<module>   s   
