U
    f/e!                     @   sr   d dl m Z  d dlZd dlZd dlmZmZmZmZ ej	dd Z
ej	dd Zdd	 Zej	e d
dd ZdS )    )datetimeN)	DataFrameSeriesbdate_rangenotnac                  C   sB   t jd} t dd}t j| |< t| ttdddddd}|S )zMake mocked series as fixture.d      (        Zperiods)index)nprandomrandnarangeNaNr   r   r   )ZarrZlocsseries r   H/tmp/pip-unpacked-wheel-tiezk1ph/pandas/tests/window/moments/conftest.pyr      s
    
r   c                   C   s.   t tjddttdddddtddS )zMake mocked frame as fixture.r   
   r
   r   r   )r   columns)r   r   r   r   r   r   r   r   r   r   r   frame   s
    r   c                     sD   dd   fdd} dd dd   |   }fd	d
|D S )Nc                   S   s  t tdt tjgt tjtjgt dgt tjdgt dtjgt ddgt ddgt ddgt ddddtjtjtjddtjtjgt tjdddtjtjtjddtjtjgt tjtjddtjtjtjddtjtjgt tjdtjddddtjtjdd	d
ddgt tjdtjddddtjtjdd	d
ddgt ddtjddddtjtjdd	d
ddgt ddtjddddtjtjdd	d
ddgt tdt tdddgS )N)Zdtypeg      @g      ?g       @g      @g      @g      @g      @g      (@g      *@g      ,@g      .@g        g      "@r   r   r   )r   objectr   nanranger   r   r   r   create_series$   s    



"
z/_create_consistency_data.<locals>.create_seriesc                      s~   t  t dgdt ddgdt ddgdt tddt tddt tdddddd	d	gdgd
d   D  S )Na)r   br   )         )r    r    c   dc                 S   s   g | ]}t |qS r   )r   ).0sr   r   r   
<listcomp>   s     zG_create_consistency_data.<locals>.create_dataframes.<locals>.<listcomp>)r   r   r   Zreshaper   )r   r   r   create_dataframes   s    
 z3_create_consistency_data.<locals>.create_dataframesc                 S   s$   | j d}tt|t| dkS )NKr   )valuesZravellensetr   )xr*   r   r   r   is_constant   s    z-_create_consistency_data.<locals>.is_constantc                 S   s   |     S )N)r   all)r-   r   r   r   no_nans   s    z)_create_consistency_data.<locals>.no_nansc                    s   g | ]}| ||fqS r   r   )r%   r-   )r.   r0   r   r   r'      s     z,_create_consistency_data.<locals>.<listcomp>r   )r(   datar   )r   r.   r0   r   _create_consistency_data#   s    xr2   )paramsc                 C   s   | j S )zCreate consistency data)param)requestr   r   r   consistency_data   s    r6   )r   Znumpyr   ZpytestZpandasr   r   r   r   Zfixturer   r   r2   r6   r   r   r   r   <module>   s   
	

 