U
|
®ý°d4 ã @ s d dl mZ d dlZd dlZd dlmZmZmZm Z m
|
Z
|
d dl m Z d dlmZ dd Zej ddd d
|
d d g¡d d Zej ddd d
|
d d g¡dd Zej ddd d
|
d d g¡dd Zej ddd d
|
d d g¡dd Zej ddd d
|
d d g¡dd Zej dd ddg¡ej ddd d
|
d d g¡dd Zej ddd d
|
d d g¡d d Zej ddd d
|
d d g¡d d Zej ddd d
|
d d g¡d d! ZdS )"é )ÚpartialN)Ú DataFrameÚSeriesÚconcatÚisnaÚnotna)Úoffsetsc C s t j| dd}t|d |jd d }||jd d krD|d }nX||jd d }|d |jd d }|| }||d }||| || || }|S )Nr )Úaxisç ð?é éÿÿÿÿ)ÚnpÚsortÚintÚshape) ÚaÚperÚvaluesÚidxÚretvalZqlowZqhigZvlowZvhig© r ú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/window/test_rolling_quantile.pyÚscoreatpercentile s
|
r Úqg g¹?g à?gÍÌÌÌÌÌì?r
|
c C sp t t|d}| jd|d |¡}t|ts.ttdt| |p>dd d }t
|
|j d || |d |
¡ d S )N©r é2 ©Ústepr r r ) r r ÚrollingÚquantileÚ
|
isinstancer ÚAssertionErrorÚrangeÚlenÚtmÚassert_almost_equalÚiloc)Úseriesr r Ú compare_funcÚresultÚendr r r Ú test_series$ s
|
r+ c C s t t|d}|jd|d |¡}t|ts.ttdt||p>dd d }t j
|
|j dd d
f |j |d |
d d
f j |d| ddd d S )
|
Nr r r r r r )r ÚrawF©Z check_names) r r r r r r r! r" r# r$ Úassert_series_equalr& Úapply)r, Úframer r r( r) r* r r r Ú
|
test_frame- s $ýr1 c C s t t|d}d}| d d d
d¡ ¡ }|j|dd |¡}|jd }|dt ¡ }| d d d
||¡}t
|
|d ||¡ d S ) Nr é é ÚBé
|
©ÚwindowÚ min_periodsr é ) r r ÚresampleÚmeanr r Úindexr ÚBDayÚtruncater$ r% ) r' r r( ÚwinZserZ series_resultÚ last_dateÚ prev_dateZ trunc_seriesr r r Útest_time_rule_series: s
|
rB c
|
C s t t|d}d}|d d d
d¡ ¡ }|j|dd |¡}|jd }|dt ¡ }|d d d
||¡} t
|
j | |¡| j || d d
|
d d S ) Nr r2 r3 r4 r5 r6 r r9 )r, Fr- )r r r: r; r r r< r r= r> r$ r. Úxsr/ )
|
r, r0 r r( r? ÚfrmZ frame_resultr@ rA Z trunc_framer r r Útest_time_rule_frameG s
|
ýrE c C sH t t| d}ttj d¡}tj|d d
< tj|dd
< |jddd | ¡}t
|
|j d ||dd
¡ |jdd d | ¡}t |j d
|
st t |j d r¢t t |j d r´t t |j d sÆt ttj d¡}|jddd | ¡}t |j d süt t|j d st |jddd | ¡}|jddd | ¡}t
|
||¡ d S )Nr r r5 éöÿÿÿé )r8 r é é é r9 iúÿÿÿéûÿÿÿé é é r r )r r r r ÚrandomÚrandnÚNaNr r r$ r% r& r r! r )r r( Úobjr) Úobj2Zresult0Zresult1r r r Ú test_nansX s$ rT Úminpéc éd c C sn | j t| d ||d |¡}| j t| ||d |¡}t|}t |t|¡ | }t || || ¡ d S )Nr )r8 r )r r# r r r$ r. r% )r' rU r r r) ÚexpectedZnan_maskr r r Útest_min_periodst s rY c C s t tj d¡}tj|d d
< tj|dd
< |jddd | ¡}t|t tjgd g d¡ | ¡jdd
j dd}t
|
||¡ d S ) Nr r5 rF rH T)Úcenteré )Zdrop) r r rO rP rQ r r r r& Z reset_indexr$ r. )r rR r) rX r r r Ú test_center s ÿþýüÿr\ c C sf dd t dD }| t| j| ¡jdd |¡ d¡ | j¡}| jddd |¡}t ||¡ d S ) Nc S s g | ]}d |dqS ©ÚxÚdr ©Ú.0r^ r r r Ú
|
<listcomp> s z.test_center_reindex_series.<locals>.<listcomp>é r2 ©r7 éôÿÿÿT©r7 rZ ) r" ÚreindexÚlistr< r r Úshiftr$ r. )r' r ÚsZ series_xpZ series_rsr r r Útest_center_reindex_series s ÿþýüÿrk c C sf dd t dD }| t| j| ¡jdd |¡ d¡ | j¡}| jddd |¡}t ||¡ d S ) Nc S s g | ]}d |dqS r] r r` r r r rb ¥ s z-test_center_reindex_frame.<locals>.<listcomp>rc r2 rd re Trf ) r" rg rh r< r r ri r$ Zassert_frame_equal)r0 r rj Zframe_xpZframe_rsr r r Útest_center_reindex_frame¢ s ÿþýüÿrl ) Ú functoolsr Únumpyr ZpytestZpandasr r r r r Zpandas._testingZ_testingr$ Zpandas.tseriesr r ÚmarkZ parametrizer+ r1 rB rE rT rY r\ rk rl r r r r Ú<module> s2
|
|
|
|
|
|
|
|
|
