U
    ÃmœdÖ  ã                   @   s®   d dl mZmZmZ d dlZd dlZd dlZej	 
dddg¡ej	 
dddd	d
g¡dd„ ƒƒZej	 
dd	d
g¡dd„ ƒZej	je dd�dd„ ƒZej	je dd�dd„ ƒZdS )é    )Ú	PD_LT_1_4Úis_float_indexÚis_int_indexNÚint_typeÚuÚiÚint_sizeé   é   é   é   c                 C   s8   t jt d¡| › |› �d�}t|ƒs(t‚t|ƒr4t‚d S )Néd   ©Zdtype)ÚpdÚIndexÚnpÚaranger   ÚAssertionErrorr   )r   r   Úindex© r   ú]/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/statsmodels/compat/tests/test_pandas.pyÚtest_is_int_index   s    r   Ú
float_sizec                 C   s6   t jt d¡d| › �d�}t|ƒs&t‚t|ƒr2t‚d S )Ng      Y@Úfr   )r   r   r   r   r   r   r   )r   r   r   r   r   Útest_is_float_index   s    r   zRequires U/Int64Index)Úreasonc                  C   s`   ddl m} m} | t d¡ƒ}t|ƒs*t‚t|ƒr6t‚|t d¡ƒ}t|ƒsPt‚t|ƒr\t‚d S )Nr   )Ú
Int64IndexÚUInt64Indexr   )Úpandasr   r   r   r   r   r   r   )r   r   r   r   r   r   Útest_legacy_int_index   s    r   zRequires Float64Indexc                  C   s6   ddl m}  | t d¡ƒ}t|ƒr&t‚t|ƒs2t‚d S )Nr   )ÚFloat64Indexr   )r   r    r   r   r   r   r   )r    r   r   r   r   Útest_legacy_float_index$   s    r!   )Zstatsmodels.compat.pandasr   r   r   Únumpyr   r   r   ZpytestÚmarkZparametrizer   r   Zskipifr   r!   r   r   r   r   Ú<module>   s   

