U
    *if                     @   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   W/home/mars/bis/venv/lib/python3.8/site-packages/pandas/tests/window/moments/conftest.pycreate_series   s    
r   c                   C   s<   t ddgdt tdddddgdgdd t D  S )Nr   )columns   )r      c   c                 S   s   g | ]}t |qS r   )r   ).0sr   r   r   
<listcomp>   s     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*   



