o
    g9                     @  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0.13.0)
minversion)ArrowExtensionArrayc                   @  s.   e Zd ZejZdZdZejZ	e
dddZdS )	ArrowTimestampUSDtypeMarrow_timestamp_usreturntype_t[ArrowTimestampUSArray]c                 C  s   t S )zq
        Return the array type associated with this dtype.

        Returns
        -------
        type
        )ArrowTimestampUSArray)cls r   i/home/ubuntu/cloudmapper/venv/lib/python3.10/site-packages/pandas/tests/extension/arrow/test_timestamp.pyconstruct_array_type   s   	z*ArrowTimestampUSDtype.construct_array_typeN)r   r   )__name__
__module____qualname__datetimetypekindnamepaNULL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   ChunkedArray
ValueErrorr   	timestamp_datar	   _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  	                  )pd	DataFramer   from_scalarsr   r   r   r   r   test_constructor_extensionblock4   s   r1   )
__future__r   r   pytestpandas._typingr   pandasr.   pandas.api.extensionsr   r   importorskipr   r   #pandas.tests.extension.arrow.arraysr   r	   r   r1   r   r   r   r   <module>   s    
