o
    5c8                     @  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)
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   a/var/www/html/gps/gps/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dZdS )r   r   Nonec                 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   r   r   *   s    r   c                  C  s2   t d tdddddddg} td| i d S )	Ni  	                  r#   )r   _from_sequencer   pd	DataFrame)arrr   r   r   test_constructor_extensionblock4   s   r3   )
__future__r   r   pytestpandas._typingr   pandasr0   pandas.api.extensionsr   r   importorskipr   r   #pandas.tests.extension.arrow.arraysr   r	   r   r3   r   r   r   r   <module>   s    
