U
|
®ý°dÒ& ã @ s` d dl Z d dlZd dlZd dlmZmZmZmZm Z m
|
Z
|
m Z m Z m Z d dlmZ d dlmZ dd Zej dddd g¡dd Zd
|
d Zej d d dg¡dd Zdd Zdd Zdd Zdd Z dd Z dd Z ej d dd d igd dgg¡d!d" Z d#d$ Z d%d& Z!d'd( Z"d)d* Z#d+d, Z$d-d. Z%ej d/d d0d1g¡d2d3 Z&d4d5 Z'd6d7 Z(d8d9 Z)dS ):é N) Ú DataFrameÚIndexÚ
|
MultiIndexÚSeriesÚ TimestampÚconcatÚ
|
date_rangeÚisnaÚnotna)Úoffsetsc
|
C sB t ¡ 0 t jddtd | t | ¡ ¡ W 5 Q R £ S Q R X d S )NÚignorez .*(empty slice|0 for slice).*)ÚmessageÚcategory)ÚwarningsÚcatch_warningsÚfilterwarningsÚRuntimeWarningÚnpÚisfiniteÚmean©Úx© r úUd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/window/test_apply.pyÚf s
|
ýr Úbad_rawé c C s: t jtdd" ttd d¡jt| d W 5 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 r r Ú test_rolling_apply_invalid_raw" s r) c C s | \}}t ddddg}| d¡jtj||d}| ¡ ¡ s@t|jdddjtj||d}t ddddgtd }t
|
||¡ d S )
|
Nr é r é é
|
©Úenginer! ©Ú min_periodsé ©Zdtype) r r& r' r Úsumr ÚallÚAssertionErrorÚfloatÚtmÚassert_almost_equal)Úengine_and_rawr. r! ÚvalsÚresultÚexpectedr r r Ú test_rolling_apply_out_of_bounds( s r= Úwindowr* Ú2sc s t tj d¡tjjdddddtddddd fd
|
d } | ¡j|d d } jdd
¡}t
|
||¡ t
|
t ¡ | ¡j|dd W 5 Q R X d S )Né r r, )Úsize©ÚAÚBÚ20130101Ús©ZperiodsÚfreq)Úindexc s$ | j d j d krtjS | jd S )Nr éÿÿÿÿ)rI r ÚnanÚilocr ©Údfr r r @ s z1test_rolling_apply_with_pandas_objects.<locals>.fFr r* T)r r ÚrandomÚrandnÚrandintr r& r' rL Z reindex_liker7 Úassert_frame_equalZexternal_error_raisedÚAttributeError)r> r r; r< r rM r Ú&test_rolling_apply_with_pandas_objects6 s þ rT c C s¼ | \}}t g dd}|jd|djdd ||d}t ||¡ t d d d g}|jdd |d
|
jd d ||d}t d d d gd d |
}t ||¡ |jdd |d
|
jt||d}t ||¡ d S )NÚfloat64r2 r, )Ústepc S s | ¡ S ©N)r r r r r Ú<lambda>R ó z$test_rolling_apply.<locals>.<lambda>r- r* r ©r0 rV c S s t | S rW )r( r r r r rX Y rY ç ð?ç @)r r& r' r7 Úassert_series_equalr( )r9 rV r. r! r< r; rF r r r Útest_rolling_applyM s$ ÿ ÿ r^ c C s~ | \}}t tddddtdd d¡d }|jdd }|jd
|
d }|jd d ||d }|jdd ||d }t ||¡ d S )NrE r@ rF rG rB rC r* r )r> Ú1sc S s dS ©Nr r r r r r rX n rY z test_all_apply.<locals>.<lambda>r- c S s dS r` r r r r r rX o rY )r r r% Z set_indexr& r' r7 rR )r9 r. r! rN ZerÚrr; r< r r r Útest_all_applyb s ÿþýÿ rb c C sê | \}}t dtdi}tdtdtdtdtdg|_dd }|jd
|
d d j|||d }| ¡ }d|d< t ||¡ |jdd d j|||d }| ¡ }d|d< t ||¡ |jdd d j|||d }| ¡ }d|d< t ||¡ d S )NrD r@ z20130101 09:00:00z20130101 09:00:02z20130101 09:00:03z20130101 09:00:05z20130101 09:00:06c S s dS r` r r r r r rX rY z#test_ragged_apply.<locals>.<lambda>r_ r ©r> r0 r- r[ r? Z5s) r r% r rI r& r' Úcopyr7 rR )r9 r. r! rN r r; r< r r r Útest_ragged_applys s* û re c C s> t jtdd& ttd d¡jdd dd W 5 Q R X d S )Nz)engine must be either 'numba' or 'cython'r r c S s | S rW r r r r r rX rY z%test_invalid_engine.<locals>.<lambda>Zfoo)r. ©r" r# r$ r r% r& r' r r r r Útest_invalid_engine s rg c C sD t jtdd, ttd d¡jdd dddid W 5 Q R X d S )
|
Nz+cython engine does not accept engine_kwargsr r c S s | S rW r r r r r rX rY z3test_invalid_engine_kwargs_cython.<locals>.<lambda>ZcythonZnopythonF)r. Z engine_kwargsrf r r r r Ú!test_invalid_engine_kwargs_cython s ÿrh c C s@ t jtdd( ttd d¡jdd ddd W 5 Q R X d S ) Nz.raw must be `True` when using the numba enginer r c S s | S rW r r r r r rX rY z(test_invalid_raw_numba.<locals>.<lambda>FZnumba)r! r. rf r r r r Útest_invalid_raw_numba s
|
ÿri Ú args_kwargsÚparr, )r, c C sÐ dd }t ddgddgd}tddg}t ddgdd gg|d
|
}| d¡j|| d | d d }t ||¡ tjd dgdd gd}tdd g|dd}| d¡d d¡}|j|| d | d d }t
|
||¡ d S )Nc S s t | | ¡S rW )r r3 )r rk r r r Únumpysum¦ s z0test_rolling_apply_args_kwargs.<locals>.numpysumr r* )ÚgrÚarm rn g &@g (@)Úcolumnsr )ÚargsÚkwargs)r r )r r )Únames)rI Úname) r r r& r' r7 rR r Ú from_tuplesr Úgroupbyr] )rj rl rN Úidxr< r; ZmidxZ
|
gb_rollingr r r Ú test_rolling_apply_args_kwargs£ s rw c C sR t tj d¡}tj|d d
< tj|dd
< |jdddjt| d}t |j
|
d t |dd
¡¡ |jdd djt| d}t |j
|
d
|
st t |j
|
d r t t |j
|
d r²t t |j
|
d sÄt t tj d¡}|jdddjt| d}t |j
|
d sþt t|j
|
d st |jdddjt| d}|jdddjt| d}t ||¡ d S )Né2 r, éöÿÿÿé r/ r rJ é é é é iúÿÿÿéûÿÿÿr@ r r+ r r )r r rO rP ÚNaNr& r' r r7 r8 rL r r r5 r
|
)r! Úobjr; Úobj2Zresult0Zresult1r r r Ú test_nansº s" r c C s t tj d¡}tj|d d
< tj|dd
< |jddddjt| d}t|t tjgd gjddd
|
jt| dj d d
j
|
dd }t ||¡ d S ) Nrx r, ry r{ r| T)r0 Úcenterr é r/ )Zdrop) r r rO rP r r& r' r r rL Z reset_indexr7 r] )r! r r; r< r r r Ú test_centerÔ s" ÿ þýüÿr c C sF | d¡jt| d}t|ts"tt |jd t
|
|dd
¡¡ d S )Nrx r rJ éÎÿÿÿ) r& r' r Ú
|
isinstancer r5 r7 r8 rL r r )r! Úseriesr; r r r Ú test_seriesä s r c C sd | d¡jt| d}t|ts"ttj|jdd d
f |jdd
d d
f jt j
|
d| ddd d S ) Nrx r rJ r r )Úaxisr! F©Z check_names) r& r' r r r r5 r7 r] rL r r )r! Úframer; r r r Ú
|
test_frameê s "ýr c C s d}d}|d d d
d¡ ¡ }|j||djt| d}|jd }|dt ¡ }|d d d
||¡}t
|
|d t |¡¡ d S ) Né r, r* rD rc r rJ r~ ) Úresampler r& r' r rI r ÚBDayÚtruncater7 r8 r ) r! r ÚwinÚminpZserZ series_resultÚ last_dateÚ prev_dateZ trunc_seriesr r r Útest_time_rule_seriesô s
|
r c C s d}d}|d d d
d¡ ¡ }|j||djt| d}|jd }|dt ¡ }|d d d
||¡}t j
|
| |¡|jt j| dd d
|
d S ) Nr r, r* rD rc r rJ r~ Fr ) r r r& r' r rI r r r r7 r] Úxsr ) r! r r r ÚfrmZ frame_resultr r Z trunc_framer r r Útest_time_rule_frame s
|
ýr r éc éd c C sv |j t|d ||djt| d}|j t|||djt| d}t|}t |t|¡ | }t || || ¡ d S )Nr rZ r )r& r( r' r r r7 r] r8 )r! r r rV r; r< Znan_maskr r r Útest_min_periods s ÿ ÿr c C sv dd t dD }d}| t|j| ¡jd|djt| d d¡ |j¡}|jd|d d
|
jt| d}t ||¡ d S ) Nc S s g | ]}d |dqS ©r Údr ©Ú.0r r r r Ú
|
<listcomp>! s z.test_center_reindex_series.<locals>.<listcomp>é r, r rc r éôÿÿÿT©r> r0 r )
|
r% ÚreindexÚlistrI r& r' r Úshiftr7 r] )r! r rF r Z series_xpZ series_rsr r r Útest_center_reindex_series s& ÿ þýüÿ ÿr© c C sv dd t dD }d}| t|j| ¡jd|djt| d d¡ |j¡}|jd|d d
|
jt| d}t ||¡ d S ) Nc S s g | ]}d |dqS r r r r r r r¢ 3 s z-test_center_reindex_frame.<locals>.<listcomp>r£ r, r rc r r¤ Tr¥ )
|
r% r¦ r§ rI r& r' r r¨ r7 rR )r! r rF r Zframe_xpZframe_rsr r r Útest_center_reindex_frame1 s ÿ þýüÿrª c C sB t ddg}|jdddjtj| d}t ddg}t ||¡ d S )Nr r* )r> r r r[ r\ )r r& r' r r3 r7 rR )r! rN r; r< r r r Ú
|
test_axis1A s r« )*r Únumpyr r" Zpandasr r r r r r r r r
|
Zpandas._testingZ_testingr7 Zpandas.tseriesr r ÚmarkZ parametrizer) r= rT r^ rb re rg rh ri rw r r r r r r r r© rª r« r r r r Ú<module> s< ,
|
|
|
|
|
|
