o
    թZh                     @  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   ZdtypesZiatZdtype)obj r	   M/var/www/html/lang_env/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      )Zaxis)r   Z_ixs)r   klassr	   r	   r
   get_obj   s   r   N)r   r   )r   r   )__doc__
__future__r   typingr   Zpandasr   Zpandas._typingr   r   r   r	   r	   r	   r
   <module>   s    
