U
|
¬ý°dõ ã @ sÖ d dl Zd dlZd dlmZmZmZ d dlmZ d dl
|
m Z d dl m Z dd Zdd Zej d ejgd
|
gfejejgd d
|
gfe ejg¡d
|
gfe ejejg¡d d
|
gfg¡d d Zej dg e g ¡g¡dd Zej deeg¡dd Zdd Zej di dg ig dddg dg dgddgg dg¡dd Zej dd d g¡d d Zd d!d"d#d$d%gZ d&d' e D Z ej d(e d)d* f¡d+d, Z ej d(e ¡d-d. Z d/d0 Z ej dd d g¡d1d2 Z!d3d4 Z"dS )5é N)Ú DataFrameÚ
|
MultiIndexÚSeries)Úframe_transform_kernels)Ú
|
zip_framesc C s, |t k r(| d } |dkr(t d| ¡ | S )zm
|
Helper to ensure we have the right type of object for a test parametrized
|
over frame_or_series.
|
ÚAr z%Test is only for DataFrame with axis=)r ÚpytestÚskip)ÚobjÚklassÚaxis© r ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/apply/test_frame_transform.pyÚ
|
unpack_obj s
|
r c C sR t ||| }tjdd t |¡}W 5 Q R X |jtj| d}|}t ||¡ d S )NÚignore©Úall©r )r ÚnpÚerrstateÚsqrtÚ transformÚtmÚ assert_equal)r Ú float_frameÚframe_or_seriesr
|
Zf_sqrtÚresultÚexpectedr r r Útest_transform_ufunc s r z
|
ops, namesr Úabsolutec s | dkr dnd}t jdd t fdd|D |d}W 5 Q R X | dkr^t j|g¡|_nt j|g¡|_ j|| d}t ||¡ d S ) N> r Úindexé r r r c s g | ] }| qS r r )Ú.0Úop©r r r Ú
|
<listcomp>4 s z+test_transform_listlike.<locals>.<listcomp>r )
|
r r r r Z from_productÚcolumnsr r r Úassert_frame_equal)r r ÚopsÚnamesZ
|
other_axisr r r r$ r Útest_transform_listlike' s $r* r( c C s4 t | |d}tjtdd | |¡ W 5 Q R X d S ©Nr z$No transform functions were provided©Úmatch©r r ÚraisesÚ
|
ValueErrorr ©r r( r r
|
r r r Ú test_transform_empty_listlike= s r2 Úboxc C sl | dkr&|j d }||g tj¡}n |jd }|jdg tj¡}|j||tji| d}t ||¡ d S )N)r r r r )r& r r Úabsr Zilocr r' )r r r3 Úer r r r r Útest_transform_dictlikeE s
|
|
r6 c C sj t ddgddgddgd} | ddgdd¡}t dddgd dd ggtdd
|
gd d gd }t ||¡ d S )Nr! é é )ÚaÚbÚcr r4 )r: r; g ð?g @)r r4 )r r r! )r r! r ©r& )r r r r r' )Údfr r r r r Ú test_transform_dictlike_mixedR s þr> r Zcumsum©r ÚBc C s4 t | |d}tjtdd | |¡ W 5 Q R X d S r+ r. r1 r r r Ú test_transform_empty_dictlike] s rA Ú use_applyTFc s@ t | | } fdd}|j|| d}|d }t ||¡ d S )Nc s t | krt| d S ©Nr! )Ú
|
isinstancer0 ©Úx©r rB r r Úfuncu s z test_transform_udf.<locals>.funcr r! )r r r r )r r rB r r
|
rH r r r rG r Útest_transform_udfo s
|
rI ZffillZbfillZfillnaÚpadZbackfillÚshiftc C s g | ]}|t kr|qS r )Ú wont_fail)r" rF r r r r% s r% r# c C s | d S rC r rE r r r Ú<lambda>
ó rM c C sð | dkr |j tjjtdd¡ tddtg i}t ||¡}t
|
}d ddg¡}tj ||d | | ¡ W 5 Q R X tj ||d | | g¡ W 5 Q R X tj ||d | d| i¡ W 5 Q R X tj ||d | d| gi¡ W 5 Q R X d S )
|
NÚngroupú ngroup not valid for NDFrame©r/ Úreasonr é ú|ú4not supported between instances of 'type' and 'type'úunsupported operand typer, )ÚnodeÚ
|
add_markerr ÚmarkÚxfailr0 r Úobjectr Úget_objÚ TypeErrorÚjoinr/ r )r# r Úrequestr
|
ÚerrorÚmsgr r r Útest_transform_bad_dtype
s( ÿ þÿrb c C sþ |dkr | j tjjtdd¡ tdtg dddgd}t}d d d
|
g¡}tj
|
||d | |g¡ W 5 Q R X tj
|
||d | ||d¡ W 5 Q R X tj
|
||d | |g|gd¡ W 5 Q R X tj
|
||d | |d g|gd¡ W 5 Q R X d S ) NrO rP rQ rS r! r7 r? rT rU rV r, rK ) rW rX r rY rZ r0 r r[ r] r^ r/ r )r_ r# r= r` ra r r r Ú test_transform_failure_typeerror¡ s& ÿþÿ rc c C sØ dd } t dddgdddgd }d
|
}tjt|d | | g¡ W 5 Q R X tjt|d | | | d ¡ W 5 Q R X tjt|d | | g| gd ¡ W 5 Q R X tjt|d | | d g| gd ¡ W 5 Q R X d S ) Nc S s t t | ¡¡dk rt| S ©Né
|
)r Úsumr0 rE r r r r# Ã s z-test_transform_failure_valueerror.<locals>.opr! r7 rS i iô iX r? zTransform function failedr, rK )r r r/ r0 r )r# r= ra r r r Ú!test_transform_failure_valueerrorÁ s rg c s@ ddg ddi fdd}dgj |df d S )Nr! r7 r; rS c s6 t | krt||g ks"t|d ks2t| S )Nr; )rD r0 ÚAssertionError)rF r9 r: r; ©Z expected_argsZexpected_kwargsr rB r r Úfß s
|
z%test_transform_passes_args.<locals>.fr )r )rB r rj r ri r Útest_transform_passes_argsØ s rk c C sP t g ddgd} | dd ¡}t || ¡ | d dd ¡}t || d ¡ d S )NZcol1Zcol2r< c S s | d S rd r rE r r r rM î rN z0test_transform_empty_dataframe.<locals>.<lambda>c S s | d S rd r rE r r r rM ñ rN )r r r r' Zassert_series_equal)r= r r r r Ú test_transform_empty_dataframeë s
|
rl )#Únumpyr r Zpandasr r r Zpandas._testingZ_testingr Zpandas.tests.apply.commonr Zpandas.tests.frame.commonr r r rY Z parametrizer r4 Úarrayr* r2 Údictr6 r> rA rI rL Zframe_kernels_raiserb rc rg rk rl r r r r Ú<module> sX üþ
|
|
|
|
|
úþ
|
|
|
|
|
|
