3
���h� � @ s� d dl Zd dlZd dlmZ d dlmZmZmZm Z d dl
j Z d dl
mZ dd� Zdd� Zejjd d
d d g�d
d� �Zdd� Zdd� Zdd� Zdd� Zejjdejd�jd�jd�e dd dd �e dd dd �jd �d g�ejjd d d!g�d"d#� ��Zd$d%� Z dS )&� N)�UnsupportedFunctionCall)� DataFrame�
DatetimeIndex�Series�
date_range)�ExponentialMovingWindowc C s0 t ddddtjdgi�} | | jdd�j� d S )N�Br � � � g �?)�com)r �np�nan�ewm�mean)�df� r �>/tmp/pip-build-5_djhm0z/pandas/pandas/tests/window/test_ewm.py�test_doc_string s r c 5 C s| | j }|dd� |dd� |dd� |dd� |dd d� |dd d � |dd d
� d }tjt|d �� |ddd
� W d Q R X tjt|d �� |ddd� W d Q R X tjt|d �� |ddd� W d Q R X d}tjt|d �� |dd� W d Q R X d}tjt|d �� |dd� W d Q R X d}tjt|d �� |dd� W d Q R X d}x2dD ]*}tjt|d �� ||d� W d Q R X �qJW d S )Ng �?)r � �?)�span)�alphag �?)�halflife)r r )r r )r r z8comass, span, halflife, and alpha are mutually exclusive)�match)r r )r r )r r z comass must satisfy: comass >= 0z span must satisfy: span >= 1z#halflife must satisfy: halflife > 0r z"alpha must satisfy: 0 < alpha <= 1g �� �)r r )r �pytest�raises�
ValueError)�which�c�msgr r r r �test_constructor s8
r! �methodZstdr �varc C sv t tdddg�dd�}d}tjt|d�� t|| �ddd � W d Q R X tjt|d�� t|| �tjd
� W d Q R X d S ) Nr
r � g �?)r z2numpy operations are not valid with window objects)r r � )Zdtype)r r r r r �getattrr
Zfloat64)r" �er r r r �test_numpy_compat> s r( c
C s< d} t jt| d�� ttd��jtjd�d� W d Q R X d S )Nz%times must be datetime64\[ns\] dtype.)r � )�times)r r r r �ranger r
�arange)r r r r �!test_ewma_times_not_datetime_typeK s r- c
C sB d} t jt| d��& ttd��jtjd�jd�d� W d Q R X d S )Nz,times must be the same length as the object.)r r) r zdatetime64[ns])r* ) r r r r r+ r r
r, �astype)r r r r � test_ewma_times_not_same_lengthQ s r/ c C sD d} t jt| d��( ttd��jdtjd�jd�d� W d Q R X d S )Nz6halflife must be a string or datetime.timedelta object)r r) r zdatetime64[ns])r r* ) r r r r r+ r r
r, r. )r r r r �#test_ewma_halflife_not_correct_typeW s r0 c
C s6 d}t jt|d�� ttd��j| d� W d Q R X d S )NzKhalflife can only be a timedelta convertible argument if times is not None.)r r) )r )r r r r r+ r )�halflife_with_timesr r r r � test_ewma_halflife_without_times] s r2 r* �
z
datetime64[D]zdatetime64[ns]�2000�D)�freq�periods�UTC�time_col� min_periodsr
c C sl | }t jd�}t j|d d d�< t|tdddd�d��}|j|||d�j� }|jd |d
�j� }tj||� d S ) Ng $@r
r4 r5 r3 )r6 r7 )�Ar9 )r r: r* g �?)r r: ) r
r, r r r r r �tm�assert_frame_equal)r1 r* r: r �datar �result�expectedr r r �"test_ewma_with_times_equal_spacingc s
rA c C s^ | }d}t dddg�j|�}tjd�}t|�}|j||d�j� }tddd g�}tj||� d S )
Nz23 daysz
2020-01-01z2020-01-10T00:04:05z2020-02-23T05:00:23r% )r r* g g��E(�?g� *�M��?) r � tz_localizer
r, r r r r< r= )Ztz_aware_fixture�tzr r* r> r r? r@ r r r �%test_ewma_with_times_variable_spacingw s
rD ) Znumpyr
r Z
pandas.errorsr Zpandasr r r r Zpandas._testingZ_testingr< Zpandas.core.windowr r r! �markZ parametrizer( r- r/ r0 r2 r, r. rB rA rD r r r r �<module> s(
,
|