3
���h� � @ s� d dl Zd dlZd dlmZ d dljjZd dl m
Z
m Z m Z m
Z
mZmZ d dljZejjdddd g�dd� �Zdd � Zejjd
d d g�d
d� �Zdd� Zdd� Zdd� Zdd� Zdd� Z dd� Z ej d�d d � �Z ejjd dd d igd#dgg�d!d"� �Z dS )$� N)�NumbaUtilError)� DataFrame�Index�
MultiIndex�Series� Timestamp�
date_range�bad_raw� c C s: t jtdd��" ttd��jd�jt| d� W d Q R X d S )Nz'raw parameter must be `True` or `False`)�match� r
)�raw)�pytest�raises�
ValueErrorr �range�rolling�apply�len)r � r �@/tmp/pip-build-5_djhm0z/pandas/pandas/tests/window/test_apply.py� test_rolling_apply_invalid_raw s r c C s� | \}}t ddddg�}|jd�jtj||d�}|j� j� s@t�|jddd�jtj||d�}t ddddgtd �}t j
||� d S )
Nr
� r � �
)�enginer
)� min_periods� )�dtype) r r r �np�sumZisna�all�AssertionError�float�tmZassert_almost_equal)�engine_and_rawr r
�vals�result�expectedr r r � test_rolling_apply_out_of_bounds s r) �windowr �2sc s� t tjjd�tjjdddd�d�tdddd�d �� � fd
d �}� j| �j|d d
�}� jdd � j � �}t
j ||� t j
t�� � j| �j|dd
� W d Q R X d S )N� r r )�size)�A�B�20130101�s)�periods�freq)�indexc s$ | j d � j d krtjS | jd S )Nr r
���)r4 r �nan�iloc)�x)�dfr r �f) s z1test_rolling_apply_with_pandas_objects.<locals>.fF)r
r T)r r �randomZrandn�randintr r r r7 Z reindex_liker$ �assert_frame_equalr r �AttributeError)r* r: r' r( r )r9 r �&test_rolling_apply_with_pandas_objects s r? c C s� | \}}t g dd�}|jd�jdd� ||d�}tj||� t d d d g�}|jddd �jd
d� ||d�}t d d d g�}tj||� |jddd �jt||d�}tj||� d S )
NZfloat64)r r c S s | j � S )N)Zmean)r8 r r r �<lambda>: s z$test_rolling_apply.<locals>.<lambda>)r r
r r )r c S s t | �S )N)r )r8 r r r r@ ? s g �?g @)r r r r$ �assert_series_equalr )r% r r
r( r' r1 r r r �test_rolling_apply6 s rB c C s~ | \}}t tdddd�td�d��jd�d }|jdd �}|jd
d �}|jd d � ||d
�}|jdd � ||d
�}tj||� d S )Nr0 r, r1 )r2 r3 )r. r/ r. r r
)r* �1sc S s dS )Nr
r )r8 r r r r@ S s z test_all_apply.<locals>.<lambda>)r r
c S s dS )Nr
r )r8 r r r r@ T s )r r r Z set_indexr r r$ r= )r% r r
r9 Zer�rr' r( r r r �test_all_applyG s rE c C s� | \}}t dtd�i�}td�td�td�td�td�g|_dd � }|jd
d d �j|||d
�}|j� }d|d< tj||� |jdd d �j|||d
�}|j� }d|d< tj||� |jdd d �j|||d
�}|j� }d|d< tj||� d S )Nr/ r, z20130101 09:00:00z20130101 09:00:02z20130101 09:00:03z20130101 09:00:05z20130101 09:00:06c S s dS )Nr
r )r8 r r r r@ d s z#test_ragged_apply.<locals>.<lambda>rC r
)r* r )r r
g �?r+ Z5s) r r r r4 r r �copyr$ r= )r% r r
r9 r: r' r( r r r �test_ragged_applyX s( rG c C s> t jtdd��& ttd��jd�jdd� dd� W d Q R X d S )Nz)engine must be either 'numba' or 'cython')r r
c S s | S )Nr )r8 r r r r@ w s z%test_invalid_engine.<locals>.<lambda>�foo)r )r r r r r r r r r r r �test_invalid_engineu s rI c C sD t jtdd��, ttd��jd�jdd� dddid � W d Q R X d S )
Nz+cython engine does not accept engine_kwargs)r r
c S s | S )Nr )r8 r r r r@ } s z3test_invalid_engine_kwargs_cython.<locals>.<lambda>ZcythonZnopythonF)r Z
engine_kwargs)r r r r r r r r r r r �!test_invalid_engine_kwargs_cythonz s rJ c C s@ t jtdd��( ttd��jd�jdd� ddd� W d Q R X d S ) Nz.raw must be `True` when using the numba engine)r r
c S s | S )Nr )r8 r r r r@ � s z(test_invalid_raw_numba.<locals>.<lambda>F�numba)r
r )r r r r r r r r r r r �test_invalid_raw_numba� s rL rK c
C sF t jtdd��. ttd��jd�jdd� ddiddd � W d Q R X d S )
Nz"numba does not support kwargs with)r r
c S s | S )Nr )r8 r r r r@ � s z.test_invalid_kwargs_nopython.<locals>.<lambda>�arK T)�kwargsr r
)r r r r r r r r r r r � test_invalid_kwargs_nopython� s rO � args_kwargs�parr c C s� dd� }t ddgddgd��}tddg�}t ddgdd gg|d
�}|jd�j|| d | d d �}tj||� |jd�j|dd�}tjddgdd gd�}tdd g|dd�}|j d�d jd�}|j|| d | d d �}tj
||� d S )Nc S s t j| | �S )N)r r )r8 rQ r r r rH � s z+test_rolling_apply_args_kwargs.<locals>.foor
r )�grrM rR rM g &@g (@)�columnsr )�argsrN r )rT )�names)r4 �name)r )r
r )r
r
) r r r r r$ r= r � from_tuplesr �groupbyrA )rP rH r9 �idxr( r' ZmidxZ
gb_rollingr r r � test_rolling_apply_args_kwargs� s rZ )r )!Znumpyr r Z
pandas.errorsr Z pandas.util._test_decorators�utilZ_test_decorators�tdZpandasr r r r r r Zpandas._testingZ_testingr$ �markZ parametrizer r) r? rB rE rG rI rJ rL Z
skip_if_norO rZ r r r r �<module> s
|