a
    oÝEbÙ  ã                   @   sÚ   d dl Z d dlZd dlZd dlmZmZmZ dd„ Zdd„ Z	dd„ Z
ejd	d
„ e  eƒ e	ƒ ¡D ƒdd�dd„ ƒZejeƒ d�dd„ ƒZeje  eƒ e	ƒ ¡d�dd„ ƒZejddgd�dd„ ƒZejd dgd�dd„ ƒZdS )é    N)Ú	DataFrameÚSeriesÚnotnac                   C   s~   t tjdd�t tjgd ƒt dgd ƒt tdddƒƒt tdƒƒt tjdtjddgƒt tjdtjddgƒt tjdtjddgƒgS )	NÚa)ZdtypeÚnameé   g      ð?r   éÿÿÿÿg       @g      @)r   ÚnpZfloat64ÚnanÚrange© r   r   úd/home/ja/django-apps/lartica_env/lib/python3.9/site-packages/pandas/tests/window/moments/conftest.pyÚcreate_series   s    
ør   c                   C   s:   t ddgd�t t d¡ d¡g d¢d�gdd„ tƒ D ƒ S )Nr   )Úcolumnsé   )r   é   )r   r   éc   c                 S   s   g | ]}t |ƒ‘qS r   )r   )Ú.0Úsr   r   r   Ú
<listcomp>   ó    z%create_dataframes.<locals>.<listcomp>)r   r	   ZarangeZreshaper   r   r   r   r   Úcreate_dataframes   s
    þýr   c                 C   s$   | j  d¡}tt|t|ƒ ƒƒdkS )NÚKé   )ÚvaluesZravelÚlenÚsetr   )Úxr   r   r   r   Úis_constant!   s    r   c                 c   s   | ]}t |ƒr|V  qd S ©N)r   )r   Úobjr   r   r   Ú	<genexpr>'   s   þr!   Úmodule)ÚparamsÚscopec                 C   s   | j S r   ©Úparam©Úrequestr   r   r   Úconsistent_data&   s    	r)   )r#   c                 C   s   | j S r   r%   r'   r   r   r   Úseries_data2   s    r*   c                 C   s   | j S )a0  
    Test:
        - Empty Series / DataFrame
        - All NaN
        - All consistent value
        - Monotonically decreasing
        - Monotonically increasing
        - Monotonically consistent with NaNs
        - Monotonically increasing with NaNs
        - Monotonically decreasing with NaNs
    r%   r'   r   r   r   Úall_data7   s    r+   )r   r   )r   r   c                 C   s   | j S )zwindow, min_periodsr%   r'   r   r   r   Úrolling_consistency_casesG   s    r,   é   c                 C   s   | j S r   r%   r'   r   r   r   Úmin_periodsM   s    r.   )Ú	itertoolsÚnumpyr	   ZpytestZpandasr   r   r   r   r   r   ZfixtureÚchainr)   r*   r+   r,   r.   r   r   r   r   Ú<module>   s*   þú



