3
���h�( � @ s� d dl mZ d dlZd dlZd dljjZd dl Z
d dl m Z m Z m
Z
mZmZ d dljZd dlmZ dd� Zdd� Zd d
� Zd d � Zd
d� Zdd� Zdd� Zdd� Z dd� Z dd� Z ej ej j!d�dd� ��Z"d d � Z#ej j$d d d ej%ej%ej%ej%gd!d"d#d"gd#d$d%d$gej%ej%ej%ej%gd"d$d%d$gd%d&d'd&gd'd(d)d*ggfd+dd,d,d"d"gd!d"d#d"gd"d$d$d"gd,d,d$d$gd"d$d%d$gd-d&d.d$gd&d(d'd$ggfg�d/d0� �Z&dS )1� )� OrderedDictN)� DataFrame�Index�Series� Timestamp�concat)�SpecificationErrorc C s� | j dd�}tj|jj| j� | j dd�d }|jj| jd ksDt�| j dd�d }tj|jj| jddg � | j dd�ddg }tj|jj| jddg � d S )N� )�window� � )r r )�rolling�tm�assert_index_equalZ
_selected_obj�columns�name�AssertionError)�frame�r� r �>/tmp/pip-build-5_djhm0z/pandas/pandas/tests/window/test_api.py� test_getitem s r c C sp t ddggddgd�} | jdd�}tjtdd �� |d
g W d Q R X tjtd d �� |dd
g W d Q R X d S ) Nr � �A�B)r r )r
zColumns not found: 'C')�match�Cz^[^A]+$)r r
�pytest�raises�KeyError)�df�gr r r �test_select_bad_cols s r" c C sd t ddggddgd�} | jdd�}tj|jj� |d j� � d}tjt|d �� |j W d Q R X d S )
Nr r r r )r r )r
z%'Rolling' object has no attribute 'F')r )
r r
r Zassert_series_equalr �sumr r �AttributeError�F)r r �msgr r r �test_attribute_access( s r' c C sx t td�tdd�dd��} | jdd�}|ddg j� }t tjtjdd d
gtjtjd d d
gd�td�d�}tj||� d S )Nr �
�foo)r r r r )r
r r � � � � � )r r �AB)r ) r �ranger
r# �np�nan�listr �assert_frame_equal)r r �result�expectedr r r �tests_skip_nuisance2 s $ r7 c C sp t td�tdd�dd��} | jdd�}|j� }t tjtjdddgtjtjd d
d gd �td
�d�}tj||� d S )Nr r( r) )r r r r )r
r* r+ r, r- r. )r r r/ )r ) r r0 r
r# r1 r2 r3 r r4 )r r r5 r6 r r r �test_skip_sum_object_raises> s $ r8 c C s t td�tddd�d��} | jdd�}|d j� }|d j� }|d j� }|d j� }|d j� }|jtjtjg�}t||||gd
d �}t j
j dd gd d
gg�|_ t
j||� |jtjtjd��}t||gd
d �}t
j||dd� |jdd d
gi�}t||gd
d �}t j
jddg�|_ t
j||� |d jd dg�}t||gd
d �}d dg|_ t
j||� d} tjt| d�� |jdd dd�i� W d Q R X tjt| d��" |jd dd�d dd�d�� W d Q R X |jd d
gd d
gd��}t||||gd
d �}ddddg}
t j
j|
�|_ t
j||dd� d S )Nr r r( r )r r r )r
r r r )�axis�mean�stdT)�
check_liker# z nested renamer is not supported)r )r: r# )Zmean2Zsum2)r r: )r r; )r r: )r r; )r r: )r r; )r r0 r
r: r; r# � aggregater1 r �pd�
MultiIndex� from_productr r r4 � from_tuplesr r r ) r r Za_meanZa_std�a_sumZb_meanZb_stdr5 r6 r& Zexp_colsr r r �test_aggI sB
" rC c C s� t td�tddd�d��}|jdd�}|d j� }|jtjd d
� d��}|d jd d
� | d
�}t||gdd�}tj ||dd� d S )Nr r r( r )r r r )r
r c S s t j| dd�S )Nr )�ddof)r1 r; )�xr r r �<lambda>� s z test_agg_apply.<locals>.<lambda>r c S s t j| dd�S )Nr )rD )r1 r; )rE r r r rF � s )�rawr )r9 T)r< )
r r0 r
r# �aggr1 �applyr r r4 )rG r r rB r5 Zrcustomr6 r r r �test_agg_applyy s rJ c C s� t td�tddd�d��} | jdd�}|jtjtjg�j}tj j
t d�d d
gg�}t j
||� |d jtjtjg�j}td d
g�}t j
||� |jd tjtjgi�j}tj jd d
g�}t j
||� d S )Nr r r( r )r r r )r
r/ r# r: r )r r# )r r: )r r0 r
rH r1 r# r: r r> r? r@ r3 r r r rA )r r r5 r6 r r r �test_agg_consistency� s rK c C s. t td�tddd�d��} | jdd�}d}tjt|d ��& |jd
d d gid
d d gid�� W d Q R X t|d
j� |d
j � |d
j� |d
j � gdd�}t
j j ddddg�|_
tjt|d ��. |d
d
g jdd dgidd dgid�� W d Q R X tjt|d ��& |jdd dgidd dgid�� W d Q R X d S )Nr r r( r )r r r )r
z nested renamer is not supported)r r r: r# r )Zr1Zr2r )r9 �rar; �rb)rL r: )rL r; )rM r: )rM r; )r r0 r
r r r r= r r: r; r> r? rA r rH )r r r& r6 r r r �test_agg_nested_dicts� s *22rN c C sv ddddddddd d
d g } t d �t d
�t d �g}tdddgdddgtd�tjd dd�tjdddd�tjd�tjd�tjd�gd d tjgd d tj gd d tj g|tjd�tjd�tjd �gd � | d!�}td d d gd d d gd d d gd d d gd d d gd d d gd d d gd d d gd d d gd d d gd d d gd � | d!�}|j
dd"d#�j � }t j
||� |j
dd"d$�j � }|j� jt�}t j
||� d S )%N�int�float�string�datetime� timedelta�periods�fl_inf�fl_nan�str_nan�dt_nat� periods_natZ20170101Z20170203r r r g @g @g @�abc)rT z1 s�s)rT �freqz2012-01z2012-02z2012-03g �?g @ZaaZbb) rO rP rQ rR rS rT rU rV rW rX rY )r r )r
� min_periods)r] )r r r3 r> Z
date_rangeZtimedelta_rangeZPeriodr1 ZInf�NaNr
�countr r4 ZnotnaZastyperP )�colsZ
dt_nat_colr r6 r5 r r r �test_count_nonnumeric_types� s^
ra z"ignore:can't resolve:ImportWarningc C s� t tjjd��jdddd�} t| jdd�| jdd�gdd�}d d g|_| jd
d � d d � g�}t j
||� d
d� }dd� }t| jdd�| jdd�gdd�}ddg|_| j||g�}t j
||� d S )N�d r( r Zgaussian)r
r] Zwin_type)r; g{�G�z�?)r9 z<lambda>c S s | j dd�S )Nr( )r; )r: )rE r r r rF � s z'test_window_with_args.<locals>.<lambda>c S s | j dd�S )Ng{�G�z�?)r; )r: )rE r r r rF � s c S s | j dd�S )Nr( )r; )r: )rE r r r �a� s z test_window_with_args.<locals>.ac S s | j dd�S )Ng{�G�z�?)r; )r: )rE r r r �b s z test_window_with_args.<locals>.brc rd ) r r1 �randomZrandnr
r r: r r= r r4 )r r6 r5 rc rd r r r �test_window_with_args� s
rf c C sN t tjd�dd�} | jd�j� }| jd�j� }|jdks<t�|jdksJt�d S )Nrb r) )r � � )r r1 Zaranger
r# r r )r[ �s2Zs3r r r �test_preserve_metadata s
rj z func,window_size,expected_valsr
r g .@g 4@g 9@g >@g �A@g D@g N@g T@g @U@�P Z expandingg $@g�RB���:@g I@c
C s� t jdddgdddgdddgdddgdddgdddgdddggd d
d gd �}t|jd �| �}|rj||�}n|� }t jjdddd d d d gd d gd�}t jjd d!d"d#g�}t j|||d�}|jtd
ddgfd ddgff��} tj | |� d S )$Nr r( rh rg �( r rk �Z Zstock�low�high)r r r r r � r r* )�namesr: �max�min)�indexr )r r )r r )r r )r r )r rp )r r )r r* )rn r: )rn rr )ro r: )ro rs )
r> r �getattr�groupbyr? rA rH r r r4 )
�funcZ window_sizeZ
expected_valsr �fr
rt r r6 r5 r r r �test_multiple_agg_funcs s, !
ry )'� collectionsr Znumpyr1 r Z pandas.util._test_decorators�utilZ_test_decorators�tdZpandasr> r r r r r Zpandas._testingZ_testingr Zpandas.core.baser r r" r' r7 r8 rC rJ rK rN ra Zskip_if_no_scipy�mark�filterwarningsrf rj Z parametrizer2 ry r r r r �<module> sP
0C
|