U
    h6                     @   s<   d Z ddlmZ ddlmZ edddZeddd	Zd
S )z4
Helpers for sharing tests between DataFrame/Series
    )DtypeObj)	DataFrame)returnc                 C   s    t | tr| jjd S | jS d S )Nr   )
isinstancer   ZdtypesZiatZdtype)obj r   :/tmp/pip-unpacked-wheel-sh7bfnb4/pandas/_testing/compat.py	get_dtype	   s    
r	   )dfc                 C   s   |t k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)__doc__Zpandas._typingr   Zpandasr   r	   r   r   r   r   r   <module>   s   