U
    ÎZj/^  ã                   @   sÎ  d dl Zd dlZd dlmZ d dlmZmZmZm	Z	m
Z
mZmZ d dlmZ d dlmZ dd„ Zej d¡dd	„ ƒZej d
dde dg¡g¡ej d¡dd„ ƒƒZej dddddddg¡dd„ ƒZej ddejdejjdd�d�g¡dd „ ƒZd!d"„ Zd#d$„ Zd%d&„ Zd'd(„ Z d)d*„ Z!ej d+eddd,gd-d.d/gd0œƒdgd-gd0œd gfddgd-d.gd0œd dgfddd,gd-d.d/gd0œd ddgfgd,feddd,gd-d.d/gd0œƒdgd-gd0œd gfddgd-d.gd0œd dgfddd,gd-d.d/gd0œd ddgfgdfeddd,gd-d.d/gd0œƒdgd-gd0œd gfddgd-d.gd0œd dgfddd,gd-d.d/gd0œd ddgfgdfedgd-gd0œƒg dfeƒ i g fgdfedej"d,gej"d.d/gd0œƒd1gej"gd0œd gfdej"gej"d.gd0œd dgfdej"d,gej"d.d/gd0œd ddgfgd,fedej"d,gej"d.d/gd0œƒd1gej"gd0œd gfdej"gej"d.gd0œd dgfdej"d,gej"d.d/gd0œd ddgfgdfedej"d,gej"d.d/gd0œƒd1gej"gd0œd gfdej"gej"d.gd0œd dgfdej"d,gej"d.d/gd0œd ddgfgdfg¡d2d3„ ƒZ#ej d4e
ddd,gƒdgd gfddgd dgfddd,gd ddgfgd,fe
ddd,gƒdgd gfddgd dgfddd,gd ddgfgdfe
ddd,gƒdgd gfddgd dgfddd,gd ddgfgdfe
ddgƒdgd gfddgd dgfgdfe
ej"dgƒej"gd gfej"dgd dgfgdfe
g d5d6�g dfg¡d7d8„ ƒZ$d9d:„ Z%d;d<„ Z&ej dd=d>g¡d?d@„ ƒZ'ej dAdd,dBdCg¡ej ddddDg¡ej dEdFdGg¡ej dHdFdGg¡ej dIdJdKdLg¡dMdN„ ƒƒƒƒƒZ(dOdP„ Z)dQdR„ Z*dSdT„ Z+dUdV„ Z,dWdX„ Z-dYdZ„ Z.ejjd[dej/fdd\d]„ fdd^d]„ fdd_d]„ fgddddgd`�dadb„ ƒZ0ejjd[dej/fdej1fdej2fdej3fgddddgd`�dcdd„ ƒZ4dedf„ Z5dgdh„ Z6ej didjd]„ dkd]„ g¡dldm„ ƒZ7ej didnd]„ dod]„ dpd]„ dqd]„ drd]„ dsd]„ dtd]„ dud]„ dvd]„ dwd]„ dxd]„ dyd]„ dzd]„ d{d]„ d|d]„ g¡d}d~„ ƒZ8dd€„ Z9d�d‚„ Z:dƒd„„ Z;d…d†„ Z<d‡dˆ„ Z=d‰dŠ„ Z>d‹dŒ„ Z?d�dŽ„ Z@ej d�d�d‘g¡ej d’dFdGg¡d“d”„ ƒƒZAej d•eBeCg¡d–d—„ ƒZDej d�d�d‘g¡ej d’dFdGg¡ej d•eBeCg¡d˜d™„ ƒƒƒZEdS )šé    N)ÚUnsupportedFunctionCall)Ú	DataFrameÚDatetimeIndexÚIndexÚ
MultiIndexÚSeriesÚisnaÚnotna)Ú	Expandingc                  C   s.   t ddddtjdgiƒ} |  |  d¡ ¡  d S )NÚBr   é   é   é   )r   ÚnpÚnanÚ	expandingÚsum©Údf© r   úe/var/www/html/TRUCKING_PROJECT/venv/lib/python3.8/site-packages/pandas/tests/window/test_expanding.pyÚtest_doc_string   s    r   zIignore:The `center` argument on `expanding` will be removed in the futurec                 C   s4   | t dƒƒj}|dd� |ddd� |ddd� d S )Né   r   ©Úmin_periodsT©r   ÚcenterF)Úranger   )Úframe_or_seriesÚcr   r   r   Útest_constructor   s    
r    Úwç       @Úfoor   c              	   C   sd   | t dƒƒj}d}tjt|d�� ||d� W 5 Q R X d}tjt|d�� |d|d� W 5 Q R X d S )Nr   zmin_periods must be an integer©Úmatchr   zcenter must be a booleanr   r   )r   r   ÚpytestÚraisesÚ
ValueError)r   r!   r   Úmsgr   r   r   Útest_constructor_invalid(   s    r*   ÚmethodÚstdÚmeanr   ÚmaxÚminÚvarc              
   C   sº   t tdddgƒƒ}d}d| › �}tjt|d��2 tjt|d�� t|| ƒdddƒ W 5 Q R X W 5 Q R X d	| › �}tjt|d��2 tjt|d�� t|| ƒt	j
d
� W 5 Q R X W 5 Q R X d S )Nr   r   é   z2numpy operations are not valid with window objectsz%Passing additional args to Expanding.r$   r   é   z'Passing additional kwargs to Expanding.©Údtype)r
   r   ÚtmÚassert_produces_warningÚFutureWarningr&   r'   r   Úgetattrr   Úfloat64)r+   ÚeÚ	error_msgZwarn_msgr   r   r   Útest_numpy_compat9   s    
&
r<   Úexpanderr   Zlsz,GH#16425 expanding with offset not supported)Úreason)Zmarksc                 C   sX   t ƒ }t ƒ  | ¡ ¡ }t ||¡ t tg ƒd�}t tg ƒd� | ¡ ¡ }t ||¡ d S )N©Úindex)r   r   r   r5   Úassert_frame_equalr   )r=   ÚexpectedÚresultr   r   r   Útest_empty_df_expandingJ   s    rD   c                  C   s^   t tjgƒ} | jdd� ¡ }t dgƒ}t ||¡ | jdd� ¡ }t tjgƒ}t ||¡ d S )Nr   r   g        r   )r   r   r   r   r   r5   Úassert_series_equal)ÚxrC   rB   r   r   r   Útest_missing_minp_zeroe   s    
rG   c                 C   s„   t t d¡ƒ}| | ¡}|dkr8t dd„ tdƒD ƒƒ}n*t tjgd dd„ td	d
ƒD ƒ gd ƒ}|jd	| d� ¡ }t 	||¡ d S )N)é
   é   r   c                 S   s.   i | ]&}|t jgd  dd„ tddƒD ƒ “qS )r   c                 S   s   g | ]}t |ƒ‘qS r   ©Úfloat)Ú.0Újr   r   r   Ú
<listcomp>z   s     z2test_expanding_axis.<locals>.<dictcomp>.<listcomp>r2   é   )r   r   r   ©rL   Úir   r   r   Ú
<dictcomp>z   s      z'test_expanding_axis.<locals>.<dictcomp>rI   r   c                 S   s   g | ]}t |ƒ‘qS r   rJ   rP   r   r   r   rN   ~   s     z'test_expanding_axis.<locals>.<listcomp>r2   é   rH   ©Úaxis)
r   r   ZonesZ_get_axis_numberr   r   r   r   r5   rA   )Z
axis_framer   rU   rB   rC   r   r   r   Útest_expanding_axiss   s    
ÿ*rV   c                 C   s>   | t dƒƒjdd� ¡ }| tjtjdddgƒ}t ||¡ d S )Nr   r2   r   ç      @ç      @ç      @©r   r   Úcountr   r   r5   Úassert_equal©r   rC   rB   r   r   r   Ú%test_expanding_count_with_min_periods„   s    r^   c                 C   sN   dddt jdddg}ddd	d	d
ddg}| |ƒ ¡  ¡ }| |ƒ}t ||¡ d S )Nr   r   r2   r   r   r1   ç      ð?r"   rW   rX   rY   g      @)r   r   r   r[   r5   r\   )r   ÚvaluesZexpected_countsrC   rB   r   r   r   Ú9test_expanding_count_default_min_periods_with_null_values‹   s
    ra   c                 C   sD   | t dƒƒjdd� ¡ }| tjtjtjtjtjgƒ}t ||¡ d S )Nr   r1   r   rZ   r]   r   r   r   Ú=test_expanding_count_with_min_periods_exceeding_series_length•   s    rb   zdf,expected,min_periodsr2   r   r   r1   )ÚAr   r_   c                 C   s8   dd„ |D ƒ}t ||  |¡ƒD ]\}}t ||¡ qd S )Nc                 S   s   g | ]\}}t ||d �‘qS ©r?   )r   ©rL   r`   r@   r   r   r   rN   Û   s     z1test_iter_expanding_dataframe.<locals>.<listcomp>)Úzipr   r5   rA   )r   rB   r   Úactualr   r   r   Útest_iter_expanding_dataframeœ   s    ?rh   zser,expected,min_periodsZint64r3   c                 C   s8   dd„ |D ƒ}t ||  |¡ƒD ]\}}t ||¡ qd S )Nc                 S   s   g | ]\}}t ||d �‘qS rd   )r   re   r   r   r   rN   î   s     z.test_iter_expanding_series.<locals>.<listcomp>)rf   r   r5   rE   )ÚserrB   r   rg   r   r   r   Útest_iter_expanding_seriesá   s    rj   c               	   C   sl   t ƒ } t t¡� | jdd� W 5 Q R X t t¡� | jdd� W 5 Q R X t d ¡� |  ¡  W 5 Q R X d S )NT)r   F)r   r5   r6   r7   r   r   r   r   r   Útest_center_deprecate_warningô   s    rk   c                 C   sX   | dddgƒ}|  ¡  ¡ }t|tƒr2t|d jƒ}ttjgdgd  ƒ}t 	||¡ d S )Nr   r   r   g»¹øÛž æ?)
r   ZsemÚ
isinstancer   r   r`   r   r   r5   rE   )r   ÚobjrC   rB   r   r   r   Útest_expanding_sem  s    
rn   ÚskewÚkurtc                 C   sL   t tj d¡ƒ}t| d¡| ƒƒ }|d }t| d¡| ƒƒ }t ||¡ d S )NrH   r2   iˆ  )r   r   ÚrandomÚrandr8   r   r5   rE   )r+   ÚsrB   rC   r   r   r   Ú,test_expanding_skew_kurt_numerical_stability  s
    rt   ÚwindowrH   rI   ZaverageÚpctTFÚ	ascendingÚ	test_dataÚdefaultÚ
duplicatesÚnansc           	   	      s°   d}|dkr t tj |¡d�}nN|dkr>t tj d|¡d�}n0|dkrnt tj ddd	tjtjtj g|¡d�}| | ¡ ‡ ‡‡fd
d„¡}| | ¡j	ˆˆˆ d�}t
 ||¡ d S )NrI   ry   )Údatarz   r2   r{   r_   g      Ð?g      è?c                    s   | j ˆˆˆ d�jd S )N©r+   rv   rw   éÿÿÿÿ)ÚrankÚiloc©rF   ©rw   r+   rv   r   r   Ú<lambda>&  ó    ztest_rank.<locals>.<lambda>r}   )r   r   rq   rr   Úchoicer   Úinfr   Úapplyr   r5   rE   )	ru   r+   rv   rw   rx   Úlengthri   rB   rC   r   r‚   r   Ú	test_rank  s     ÿ
ÿr‰   c                 C   sZ   |   ¡ }|tj t|ƒ¡ d d… }| ¡  |¡}|jt|ƒdd� |¡}t 	||¡ d S ©Néûÿÿÿr   ©ru   r   )
Zdropnar   rq   ÚrandnÚlenr   ÚcorrÚrollingr5   Úassert_almost_equal©Úseriesrc   r   rC   Úrolling_resultr   r   r   Útest_expanding_corr-  s
    r•   c                 C   s2   | j dd� ¡ }t || jt| ƒdd� ¡ ¡ d S )Nr   r   rŒ   )r   r[   r5   r‘   r�   rŽ   )r“   rC   r   r   r   Útest_expanding_count8  s
     ÿr–   c                 C   s6   |   ¡  d¡}| jt| ƒdd� d¡}t ||¡ d S )Nç      à?r   rŒ   )r   Úquantiler�   rŽ   r5   r‘   )r“   rC   r”   r   r   r   Útest_expanding_quantile?  s    r™   c                 C   sV   | }|t j t|ƒ¡ d d… }| ¡  |¡}|jt|ƒdd� |¡}t ||¡ d S rŠ   )	r   rq   r�   rŽ   r   Úcovr�   r5   r‘   r’   r   r   r   Útest_expanding_covG  s
    r›   c                 C   s2   |   ¡  ¡ }| jt| ƒdd� ¡ }t ||¡ d S ©Nr   rŒ   )r   rš   r�   rŽ   r5   rA   ©ÚframerC   r”   r   r   r   Útest_expanding_cov_pairwiseR  s    rŸ   c                 C   s2   |   ¡  ¡ }| jt| ƒdd� ¡ }t ||¡ d S rœ   )r   r�   r�   rŽ   r5   rA   r�   r   r   r   Útest_expanding_corr_pairwiseZ  s    r    zfunc,static_compc                 C   s   t j| dd�S ©Nr   rT   ©r   r-   r�   r   r   r   rƒ   e  r„   rƒ   c                 C   s   t j| dd�S r¡   )r   r.   r�   r   r   r   rƒ   f  r„   c                 C   s   t j| dd�S r¡   )r   r/   r�   r   r   r   rƒ   g  r„   )Zidsc                 C   sŠ   |t  ttdƒƒt jgd  ¡ƒ}t|jddd�| ƒƒ }t||ƒsFt‚||d d… ƒ}|t	krpt
 |d |¡ nt
j|jd |dd� d S )NrH   r   r   ©r   rU   rO   F©Zcheck_names)r   ÚarrayÚlistr   r   r8   r   rl   ÚAssertionErrorr   r5   r‘   rE   r€   )ÚfuncÚstatic_compr   r|   rC   rB   r   r   r   Útest_expanding_funca  s    "rª   c                 C   sL  t tj d¡ƒ}t|jddd�| ƒƒ }|d d…  ¡  ¡ s>t‚t	 
|jd ||d d… ƒ¡ t|jddd�| ƒƒ }t|jd ƒs„t‚t|jd	 ƒs–t‚t tj d
¡ƒ}t|jddd�| ƒƒ }t|d ƒsÌt‚t|d ƒsÜt‚t|jddd�| ƒƒ }t|jddd�| ƒƒ }t	 
||¡ t|jddd�| ƒƒ }t	 
|jd ||d d… ƒ¡ d S )Né2   é   r   r£   é   r~   é   é   é   rI   r   r2   r   r   )r   r   rq   r�   r8   r   r   Úallr§   r5   r‘   r€   r	   )r¨   r©   ri   rC   Úser2Úresult0Úresult1r   r   r   Útest_expanding_min_periodsw  s     rµ   c                 C   s¬   | \}}|t  ttdƒƒt jgd  ¡ƒ}|jdd�jdd„ ||d�}t||ƒsTt‚|t	kr€t
 |d t j|d d… d	d
�¡ n(t
j|jd t j|d d… d	d
�dd� d S )NrH   r   r   c                 S   s   |   ¡ S ©N©r-   r�   r   r   r   rƒ   š  r„   z&test_expanding_apply.<locals>.<lambda>©ÚrawÚengineé	   rO   r   rT   Fr¤   )r   r¥   r¦   r   r   r   r‡   rl   r§   r   r5   r‘   r-   rE   r€   )Úengine_and_rawr   rº   r¹   r|   rC   r   r   r   Útest_expanding_apply–  s    "  ÿ$  ÿr½   c                 C   s|  | \}}t tj d¡ƒ}|jdd�jdd„ ||d�}|d d…  ¡  ¡ sLt‚t	 
|jd t |d d… ¡¡ |jd	d�jd
d„ ||d�}t|jd ƒsšt‚t|jd ƒs¬t‚t tj d¡ƒ}|jdd�jdd„ ||d�}t|d ƒsèt‚t|d ƒsøt‚|jdd�jdd„ ||d�}|jdd�jdd„ ||d�}t	 
||¡ |jdd�jdd„ ||d�}t	 
|jd t |d d… ¡¡ d S )Nr«   r¬   r   c                 S   s   |   ¡ S r¶   r·   r�   r   r   r   rƒ   «  r„   z2test_expanding_min_periods_apply.<locals>.<lambda>r¸   r­   r~   r®   c                 S   s   |   ¡ S r¶   r·   r�   r   r   r   rƒ   ²  r„   r¯   r°   rI   r   c                 S   s   |   ¡ S r¶   r·   r�   r   r   r   rƒ   ¹  r„   r2   r   r   c                 S   s   |   ¡ S r¶   r·   r�   r   r   r   rƒ   À  r„   r   c                 S   s   |   ¡ S r¶   r·   r�   r   r   r   rƒ   Ã  r„   c                 S   s   |   ¡ S r¶   r·   r�   r   r   r   rƒ   È  r„   )r   r   rq   r�   r   r‡   r   r±   r§   r5   r‘   r€   r-   r	   )r¼   rº   r¹   ri   rC   r²   r³   r´   r   r   r   Ú test_expanding_min_periods_apply¦  sR      ÿ   ÿ  ÿ  ÿ  ÿ  ÿr¾   Úfc                 C   s   | j dd�j| dd�S ©Nr   r   T©Úpairwise©r   rš   r�   r   r   r   rƒ   Ð  r„   c                 C   s   | j dd�j| dd�S rÀ   ©r   r�   r�   r   r   r   rƒ   Ñ  r„   c                 C   s®   t ƒ }t tdgdd�tg dd�d�}|d  d¡|d< t t |j|jg¡tg ƒd�}t tj|j|jgddgd�tdgdd�dd	�}| |ƒ}t ||¡ | |ƒ}t ||¡ d S )
NÚar#   ©ÚnameÚbar©Úcolumnsr@   r9   ©r@   rÊ   )Únames)r@   rÊ   r4   )	r   r   Úastyper   Zfrom_productr@   rÊ   r5   rA   )r¿   Údf1Údf2Údf1_expectedÚdf2_expectedÚ
df1_resultÚ
df2_resultr   r   r   Ú*test_moment_functions_zero_length_pairwiseÍ  s     	 ÿýrÔ   c                 C   s   |   ¡  ¡ S r¶   )r   r[   r�   r   r   r   rƒ   í  r„   c                 C   s   | j dd�j| dd�S ©Nr   r   FrÁ   rÃ   r�   r   r   r   rƒ   î  r„   c                 C   s   | j dd�j| dd�S rÕ   rÄ   r�   r   r   r   rƒ   ï  r„   c                 C   s   | j dd� ¡ S ©Nr   r   )r   r.   r�   r   r   r   rƒ   ð  r„   c                 C   s   | j dd� ¡ S rÖ   )r   r/   r�   r   r   r   rƒ   ñ  r„   c                 C   s   | j dd� ¡ S rÖ   )r   r   r�   r   r   r   rƒ   ò  r„   c                 C   s   | j dd� ¡ S rÖ   )r   r-   r�   r   r   r   rƒ   ó  r„   c                 C   s   | j dd� ¡ S rÖ   )r   r,   r�   r   r   r   rƒ   ô  r„   c                 C   s   | j dd� ¡ S rÖ   )r   r0   r�   r   r   r   rƒ   õ  r„   c                 C   s   | j dd� ¡ S rÖ   )r   ro   r�   r   r   r   rƒ   ö  r„   c                 C   s   | j dd� ¡ S rÖ   )r   rp   r�   r   r   r   rƒ   ÷  r„   c                 C   s   | j dd� d¡S )Nr   r   r—   )r   r˜   r�   r   r   r   rƒ   ø  r„   c                 C   s   | j dd� ¡ S rÖ   )r   Zmedianr�   r   r   r   rƒ   ù  r„   c                 C   s   | j dd�jtdd�S )Nr   r   F©r¹   ©r   r‡   r   r�   r   r   r   rƒ   ú  r„   c                 C   s   | j dd�jtdd�S )Nr   r   Tr×   rØ   r�   r   r   r   rƒ   û  r„   c           
      C   s|   t tjd�}|}tƒ }|}tdgd�}|d  d¡|d< |}| |ƒ}t ||¡ | |ƒ}t ||¡ | |ƒ}	t |	|¡ d S )Nr3   rÅ   ©rÊ   r9   )r   r   r9   r   rÍ   r5   rE   rA   )
r¿   rs   Z
s_expectedrÎ   rÐ   rÏ   rÑ   Zs_resultrÒ   rÓ   r   r   r   Ú!test_moment_functions_zero_lengthê  s    rÚ   c                 C   s:   | \}}t g tjd�}t || ¡ jdd„ ||d�¡ d S )Nr3   c                 S   s   |   ¡ S r¶   r·   r�   r   r   r   rƒ     r„   z3test_expanding_apply_empty_series.<locals>.<lambda>r¸   )r   r   r9   r5   rE   r   r‡   )r¼   rº   r¹   ri   r   r   r   Ú!test_expanding_apply_empty_series  s     ÿrÛ   c                 C   sP   | \}}t d d d gƒ}|jdd�jdd„ ||d�}t dddgƒ}t ||¡ d S )	Nr   r   c                 S   s   t | ƒS r¶   )rŽ   r�   r   r   r   rƒ     r„   z4test_expanding_apply_min_periods_0.<locals>.<lambda>r¸   r_   r"   rW   )r   r   r‡   r5   rE   )r¼   rº   r¹   rs   rC   rB   r   r   r   Ú"test_expanding_apply_min_periods_0  s
    rÜ   c                  C   sä   t dddgdddgd�} t ddgddgd�}|  ¡  |¡}t d d dgƒ}t ||¡ t dd dgdddgd�}|  ¡  |¡}t ||¡ t ddd	gdddgd�} t dd
d	gdddgd�}|  ¡  |¡}t d d d dgƒ}t ||¡ d S )Nr   r   r2   r   r?   r"   é   é   rH   r»   g      @)r   r   rš   r5   rE   ©Ús1Ús2rC   rB   Zs2ar   r   r   Útest_expanding_cov_diff_index#  s    râ   c                  C   sä   t dddgdddgd�} t ddgddgd�}|  ¡  |¡}t d d dgƒ}t ||¡ t dd dgdddgd�}|  ¡  |¡}t ||¡ t ddd	gdddgd�} t dd
d	gdddgd�}|  ¡  |¡}t d d d dgƒ}t ||¡ d S )Nr   r   r2   r   r?   r_   rÝ   rÞ   rH   r»   )r   r   r�   r5   rE   rß   r   r   r   Útest_expanding_corr_diff_index6  s    rã   c            	      C   sn  t ddgddgddggtddgdd	�d
�} t ddgddggddgtddgdd	�d�}t ddgd d gddggtddgdd	�d
�}t ddgddggddgtddgdd	�d�}|  ¡ j|dd�jd }|  ¡ j|dd�jd }| ¡ j|dd�jd }| ¡ j|dd�jd }t ddgddggtddgdd	�tddgdd	�d�}t ||¡ t ||¡ t ||¡ t ||¡ d S )Nr   r   r2   r   r»   rc   r   r#   rÆ   rÙ   r   rË   r1   ÚXÚYTrÁ   g      Àg      Àg      Àg      $ÀrÉ   )r   r   r   rš   Úlocr5   rA   ©	rÎ   Zdf1arÏ   Zdf2ar´   Zresult2Zresult3Zresult4rB   r   r   r   Ú'test_expanding_cov_pairwise_diff_lengthI  s8    *  ÿ ÿ  ÿýrè   c            	      C   sn  t ddgddgddggddgttdƒdd�d	�} t ddgddggtd
dgdd�ddgd�}t ddgd d gddggddgttdƒdd�d	�}t ddgddggtd
dgdd�ddgd�}|  ¡ j|dd�jd }|  ¡ j|dd�jd }| ¡ j|dd�jd }| ¡ j|dd�jd }t ddgddggddgtddgƒd	�}t ||¡ t ||¡ t ||¡ t ||¡ d S )Nr   r   r2   r   rc   r   rÈ   rÆ   rÉ   r   rË   r   r1   rä   rå   TrÁ   g      ð¿)r   r   r   r   r�   ræ   r5   rA   rç   r   r   r   Ú(test_expanding_corr_pairwise_diff_lengthf  sB      ÿ  ÿý  ÿ  
ÿré   c                 C   s†   dd„ }| \}}t tj dd¡ƒ}| ¡ jtj||d�d }| ¡ j|||dd�}t ||¡ | ¡ j||d	did
�}t ||¡ d S )Nc                 S   s   t  | ¡| S r¶   r¢   )rF   Úconstr   r   r   Ú
mean_w_arg„  s    z4test_expanding_apply_args_kwargs.<locals>.mean_w_argrI   r2   )rº   r¹   g      4@)rI   )rº   r¹   Úargsrê   )r¹   Úkwargs)	r   r   rq   rr   r   r‡   r-   r5   rA   )r¼   rë   rº   r¹   r   rB   rC   r   r   r   Ú test_expanding_apply_args_kwargsƒ  s    rî   c           	      C   s¤   | }t dgdddœƒ}|d  t¡|d< | ¡ }t||d ƒ}|d k	r ||d�}|rZddgndddg}||  |g¡jd	d
� t¡}t|j	ƒ|ks”t
‚t ||¡ d S )Nr   r   r2   ©rÅ   Úbr   r   ©Únumeric_onlyrÅ   rð   T©Zdrop)r   rÍ   Úobjectr   r8   ÚaggÚreset_indexrK   r¦   rÊ   r§   r5   rA   )	Úarithmetic_win_operatorsrò   Úkernelr   r   ÚoprC   rÊ   rB   r   r   r   Útest_numeric_only_frame”  s    
rú   rø   r�   rš   Úuse_argc                 C   s¸   t dddgdddœƒ}|d  t¡|d< |r2|fnd}| ¡ }t|| ƒ}||d|iŽ}|rbdd	gndd	dg}||  t¡}	|r„|	fnd}
|	 ¡ }t|| ƒ}||
d|iŽ}t ||¡ d S )
Nr   r   r2   rï   r   r   rò   rÅ   rð   )r   rÍ   rô   r   r8   rK   r5   rA   )rø   rò   rû   r   Úargr   rù   rC   rÊ   rÏ   Úarg2Ú
expanding2Úop2rB   r   r   r   Ú test_numeric_only_corr_cov_frame¥  s    

r   r4   c           
   	   C   s–   | }t dg|d�}| ¡ }t||ƒ}|rb|tkrbd|› d�}tjt|d�� ||d� W 5 Q R X n0||d�}| |g¡jdd� 	t
¡}	t ||	¡ d S )	Nr   r3   ú
Expanding.ú  does not implement numeric_onlyr$   rñ   Tró   )r   r   r8   rô   r&   r'   ÚNotImplementedErrorrõ   rö   rÍ   rK   r5   rE   )
r÷   rò   r4   rø   ri   r   rù   r)   rC   rB   r   r   r   Útest_numeric_only_series»  s    

r  c              	   C   sÊ   t dddg|d�}|r|fnd}| ¡ }t|| ƒ}|rt|tkrtd| › d�}tjt|d�� ||d	|iŽ W 5 Q R X nR||d	|iŽ}	| t¡}
|r–|
fnd}|
 ¡ }t|| ƒ}||d	|iŽ}t	 
|	|¡ d S )
Nr   r   r2   r3   r   r  r  r$   rò   )r   r   r8   rô   r&   r'   r  rÍ   rK   r5   rE   )rø   rû   rò   r4   ri   rü   r   rù   r)   rC   r²   rý   rþ   rÿ   rB   r   r   r   Ú!test_numeric_only_corr_cov_seriesÌ  s    


r  )FÚnumpyr   r&   Zpandas.errorsr   Zpandasr   r   r   r   r   r   r	   Zpandas._testingZ_testingr5   Zpandas.core.windowr
   r   ÚmarkÚfilterwarningsr    Zparametrizer¥   r*   r<   ÚparamZxfailrD   rG   rV   r^   ra   rb   r   rh   rj   rk   rn   rt   r‰   r•   r–   r™   r›   rŸ   r    r   rª   r-   r.   r/   rµ   r½   r¾   rÔ   rÚ   rÛ   rÜ   râ   rã   rè   ré   rî   rú   r   Úintrô   r  r  r   r   r   r   Ú<module>   sX  $	ÿ
ÿ
ÿþþþ

ýù
ýù
ýù	"ýù
"ýù
"ýùÐþ=
<<<(.úþ


	


ü
ø

"
ý
'þþ
ñþ
	
