o
    <n‘j’  ã                   @  sP   d Z ddlmZ ddlmZ ddlmZ erddlmZ ddd	„Z	ddd„Z
dS )z4
Helpers for sharing tests between DataFrame/Series
é    )Úannotations)ÚTYPE_CHECKING)Ú	DataFrame)ÚDtypeObjÚreturnr   c                 C  s   t | tƒr| jjd S | jS )Nr   )Ú
isinstancer   ÚdtypesÚiatÚdtype)Úobj© r   úT/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/pandas/_testing/compat.pyÚ	get_dtype   s   
r   Údfr   c                 C  s   |t u r| S | jdddS )z…
    For sharing tests using frame_or_series, either return the DataFrame
    unchanged or return it's first column as a Series.
    r   é   )Úaxis)r   Ú_ixs)r   Úklassr   r   r   Úget_obj   s   r   N)r   r   )r   r   )Ú__doc__Ú
__future__r   Útypingr   Úpandasr   Úpandas._typingr   r   r   r   r   r   r   Ú<module>   s    
