U
|
®ý°d%1 ã @ sÄ d dl Zd dlZd dlmZmZ d dlmZmZm Z m
|
Z
|
m Z m Z m Z mZmZ d dlmZ dd Zdd Zdd Zd
|
d Zd d Zdd Zej ddgddgddidddg¡dd Zdd Z d d Z d d Z d!d" Z d#d$ Z ej 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/d0d1ggfd2dd3d3d)d)gd(d)d*d)gd)d+d+d)gd3d3d+d+gd)d+d,d+gd4d-d5d+gd-d/d.d+ggfg¡d6d7 Z"d8d9 Z#d:d; Z$d<d= Z%d>d? Z&dS )@é N)Ú DataErrorÚSpecificationError) Ú DataFrameÚIndexÚ
|
MultiIndexÚPeriodÚSeriesÚ TimestampÚconcatÚ
|
date_rangeÚtimedelta_rangec C sâ t tj dd¡}|jd| d}t |jj|d d |
j¡ |jd| dd }|jj |d d |
jd ksnt
|
|jd| dd }t |jj|d d |
jddg ¡ |jd| dddg }t |jj|d d |
jddg ¡ d S )Né ©ÚwindowÚstepé )r é r ) r ÚnpÚrandomZrandnÚrollingÚtmÚassert_index_equalZ _selected_objÚcolumnsÚnameÚAssertionError)r ÚframeÚr© r úSd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/window/test_api.pyÚ test_getitem s $r c C sp t ddggddgd} | jdd}tjtdd |d
|
g W 5 Q R X tjtd d |dd
|
g W 5 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}t |j ¡ |d ¡ ¡ d}tjt|d |j W 5 Q R X d S )
|
Nr r r! r" r# r r$ z%'Rolling' object has no attribute 'F'r% )
|
r r r Úassert_series_equalr! Úsumr( r) ÚAttributeErrorÚF)r+ r Úmsgr r r Útest_attribute_access2 s r3 c C s t tdtdddd}|jd| d}|ddg ¡ }t tjtjdd d
|
gtjtjd d d gdtddd d |
}t ||¡ d S )Nr é
|
Úfoo©r! r" r' r r r! r" é é é é é ©r! r" ÚABr# ) r Úranger r/ r ÚnanÚlistr Úassert_frame_equal©r r+ r ÚresultÚexpectedr r r Útests_skip_nuisance; s $þýrE c C sN t tdtdddd}|jd| d}tjtdd | ¡ W 5 Q R X d S ) Nr r4 r5 r6 r r z)Cannot aggregate non-numeric type: objectr% )r r> r r( r) r r/ )r r+ r r r r Útest_sum_object_str_raisesF s rF c C s t tdtdddd}|jd| d}|d ¡ }|d ¡ }|d ¡ }|d ¡ }|d ¡ }| tjtjg¡}t||||gd
|
d } t
|
dd gd d gg¡| _ t || ¡ | tjtjd¡}t||gd
|
d } t j || dd | dd d gi¡}t||gd
|
d } t ddg¡| _ t || ¡ |d d dg¡}t||gd
|
d } d dg| _ t || ¡ d}
|
tjt|
|
d | dd ddi¡ W 5 Q R X tjt|
|
d" | d ddd ddd¡ W 5 Q R X | d d gd d gd¡}t||||gd
|
d } ddddg} t | ¡| _ t j || dd d S )Nr r r4 r r< r r r! r" r ©ÚaxisÚmeanÚstdT©Z
|
check_like©r! rI )r! rJ r/ ú nested renamer is not supportedr% )rI r/ )Zmean2Zsum2)r" rI )r" rJ )r r> r rI rJ r/ Ú aggregater r
|
r Ú from_productr r rA Ú from_tuplesr( r) r ) r r+ r Za_meanZa_stdÚa_sumZb_meanZb_stdrC rD r2 Zexp_colsr r r Útest_aggN sD
|
ÿ rR ÚfuncÚminrI ÚmaxÚbr/ ÚprodZmedian)rV Úcc C sX t dddgdddgdddgd }|jddd
|
}tjtd d | | ¡ W 5 Q R X d S ) Nr r r é r r7 é é )ÚarV rX )r rH z"axis other than 0 is not supportedr% )r r r( r) ÚNotImplementedErrorÚagg)rS r+ r r r r Útest_multi_axis_1_raises~ s "r_ c C s t tdtdddd}|jdd}|d ¡ }| tjd d
|
d¡}|d jd d
|
| d }t||gdd}tj ||dd d S )Nr r r4 r r< r r$ r! c S s t j| ddS ©Nr )Zddof©r rJ ©Úxr r r Ú<lambda> ó z test_agg_apply.<locals>.<lambda>r" c S s t j| ddS r` ra rb r r r rd re )Úrawr rG TrK )
|
r r> r r/ r^ r Úapplyr
|
r rA )rf r+ r rQ rC ZrcustomrD r r r Útest_agg_apply s rh c C sÄ t tdtdddd}|jd| d}| tjtjg¡j}t t
|
dd d
|
gg¡}t ||¡ |d tjtjg¡j}t d d
|
g}t ||¡ | d tjtjgi¡j}t d d g¡}t ||¡ d S )Nr r r4 r r< r r r= r/ rI r! )r! r/ rL )r r> r r^ r r/ rI r r rO r@ r r r rP rB r r r Útest_agg_consistency s ri c C s, t tdtdddd} | jdd}d}tjt|d & | d
|
d d gid d d gid¡ W 5 Q R X t|d
|
¡ |d
|
¡ |d ¡ |d ¡ gdd}t
|
ddddg¡|_ tjt|d . |d
|
d g dd dgidd dgid¡ W 5 Q R X tjt|d & | dd dgidd dgid¡ W 5 Q R X d S )Nr r r4 r r< r r$ rM r% r! rI r/ r" )Zr1Zr2r rG )ÚrarI )rj rJ )ÚrbrI )rk rJ rj rJ rk )r r> r r( r) r rN r
|
rI rJ r rP r r^ )r+ r r2 rD r r r Útest_agg_nested_dicts§ s ** ÿ
|
ÿ2rl c C s~ ddddddddd d
|
d g }t d t d t d g}tdddgdddgtdtd ddtddddtdtdtdgd d tjgd d tjgd d tjg|tdtdtd 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!d d |
}|j dd"| d#
|
¡ }t ||¡ |j dd"| d$
|
¡ }| ¡ t¡d d |
}t ||¡ 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)rr z1 sÚs)rr Úfreqz2012-01z2012-02z2012-03g ð?g @ZaaZbb) rm rn ro rp rq rr rs rt ru rv rw r# r ©r Ú min_periodsr )r| r )r r r@ r r r r ZInfÚNaNr Úcountr rA ZnotnaZastypern )r ÚcolsZ
|
dt_nat_colr+ rD rC r r r Útest_count_nonnumeric_types½ sp õ
|
ý
|
|
|
ýñêõ òñ r c C sN t t d¡dd} | d¡ ¡ }| d¡ ¡ }|jdks<t|jdksJtd S )Néd r5 )r é é )r r Zaranger r/ r r )ry Ús2Zs3r r r Útest_preserve_metadata s
|
r
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 $@gRB°ªª:@g I@c
|
|
C sÖ t dddgdddgdddgdddgdddgdddgdddggd d
|
d gd }t| d ¡| }|rh||}n| }tjd ddddddgd d gd}t ddddg¡}t |||d}| ddgdd gd ¡} t | |¡ d S ) Nr! r4 r r é( r" r éZ ZstockÚlowÚhighr# )r! r )r! r )r! r )r" r )r" rY )r" r )r" r7 )Únames)r rI )r rU )r rI )r rT )Úindexr rI rU rT )r r )r ÚgetattrÚgroupbyr rP r^ r rA )
|
rS Z window_sizeZ expected_valsr+ Úfr r r rD rC r r r Útest_multiple_agg_funcs
|
s2 !ù ö
|
þ
|
ÿr c s` t tdjd||||d fdd jD }t | fdd jD }||ks\td S )Nr )ÚcenterÚclosedr| r c s i | ]}|t |qS r ©r ©Ú.0Úattr©Zroll_objr r Ú
|
<dictcomp>S s z=test_dont_modify_attributes_after_methods.<locals>.<dictcomp>c s i | ]}|t |qS r r r r r r r U s )r r> r Ú _attributesr r )Zarithmetic_win_operatorsr r r| r rD rC r r r Ú)test_dont_modify_attributes_after_methodsL s ÿ r c C sà t t d¡jddd| d ¡ d}tjt|d& t t d¡jddd| d ¡ W 5 Q R X tt d ¡jddd| d ¡ tt d ¡jddd| d ¡ d
|
}tjt|d& tt d ¡jddd | d ¡ W 5 Q R X d S ) Nr4 r Tr )r r rH r z&No axis named 1 for object type Seriesr% r )r4 r4 z)No axis named 2 for object type DataFramer ) r r Úonesr rI r( r) Ú
|
ValueErrorr )r r2 r r r Ú test_centered_axis_validationY s2 * ÿ
|
ÿ
|
ÿÿr c C s t dddddg}|jdd| d ¡ }t t t|¡d d |
}t ||¡ d}tj t
|
|d $ t dddgjdd| d ¡ W 5 Q R X d S )
|
Nr r r rY r r r{ ú!min_periods 5 must be <= window 3r% ) r r rT r r Úlenr r. r( r) r ©r r\ rC rD r2 r r r Ú test_rolling_min_min_periodst s r¡ c C s t dddddgtjd}|jdd| d ¡ }|d d |
}t ||¡ d }tjt |d
|
$ t dddgjdd| d ¡ W 5 Q R X d S ) Nr r r rY r )Zdtyper r{ r r% )
|
r r Úfloat64r rU r Zassert_almost_equalr( r) r r r r r Ú test_rolling_max_min_periods~ s r£ )'Únumpyr r( Z pandas.errorsr r Zpandasr r r r r r r
|
r r Zpandas._testingZ_testingr r r- r3 rE rF rR ÚmarkZ parametrizer_ rh ri rl r r
r? r r r r¡ r£ r r r r Ú<module> sf , 0 ÿ
|
C
|
|
|
|
|
|
ùý
|
|
|
|
|
|
|
ùýòþ
|
#
|
