a
    ±<b  ã                   @   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   úf/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/pandas/_testing/compat.pyÚ	get_dtype   s    
r   )Údfc                 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)Ú__doc__Zpandasr   r   r   r   r   r   r   Ú<module>   s   