U
    sVc                     @   s*   d Z ddlmZ dd ZedddZdS )	z4
Helpers for sharing tests between DataFrame/Series
    )	DataFramec                 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-xj8nt62q/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pandasr   r   r   r   r   r   r   <module>   s   