U
|
¬ý°dª, ã @ s, d dl mZ d dlZd dlZd dlZd dlmZ d dlm Z m
|
Z
|
m Z m Z m Z d dlmZ ej dddg¡dd Zd
|
d Zd d Zdd Zdd Zej dddg¡dd Zdd Zdd Zej dd d d g¡ej d d d!d"iid d!d"giig¡d#d$ Zej d%dd&d'gid&d'gd"d(gd)g¡d*d+ Z d,d- Z ej dd d g¡d.d/ Z ej dd d d g¡ej d d!d"id!d"gig¡d0d1 Z d2d3 Z ej d4d5d6dd7d8gffd9g¡d:d; Z!d<d= Z"d>d? Z#ej d@e $e
|
d7d8gd8d7ggdAe%gg¡¡dBdC Z&ej dDee $e dE '¡ d(e%fdFe%fdGe%fdHe%fdIe%fdAe%fg¡¡dJdK Z(dLdM Z)dNdO Z*ej d dPdQ dRdQ dSdQ g¡dTdU Z+dVdW Z,ej dXdYd'gdZfej-d"d)d[fg¡d\d] Z.ej d d'd&gd'dYgg¡d^d_ Z/ej d d&d'gdYd'gg¡d`da Z0ej dbdcdQ dddQ dedQ dfdQ g¡dgdh Z1dS )ié )ÚchainN)ÚSpecificationError)Ú CategoricalÚ DataFrameÚSeriesÚ
|
date_rangeÚnotnaÚ result_typeÚfooé c C s: |}d}t jt|d |jdd d| d W 5 Q R X d S )NzUinvalid value for result_type, must be one of {None, 'reduce', 'broadcast', 'expand'}©Úmatchc S s
|
dddgS )Nr é é © ©Úxr r úZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/apply/test_invalid_arg.pyÚ<lambda>% ó z(test_result_type_error.<locals>.<lambda>r ©Úaxisr ©ÚpytestÚraisesÚ
|
ValueErrorÚapply)r Úint_frame_const_colÚdfÚmsgr r r Útest_result_type_error s
|
ÿr c C s\ t dddgdddgddd ggd
|
d
|
d gd } d }tjt|d | dd d¡ W 5 Q R X d S )Nr r r é é é é é é ÚaÚc)Úindexz)No axis named 2 for object type DataFramer c S s | S ©Nr r r r r r , r z/test_apply_invalid_axis_value.<locals>.<lambda>©r r r r r ©r r r r r Ú test_apply_invalid_axis_value( s *r- c C s0 t jtdd | jdd dd W 5 Q R X d S )Nz na_action must be .*Got 'abc'r c S s t t| S r* )ÚlenÚstrr r r r r 2 r z1test_applymap_invalid_na_action.<locals>.<lambda>Úabc©Z na_action)r r r Zapplymap)Ú float_framer r r Ú test_applymap_invalid_na_action/ s r3 c C s@ t ddgddgd} d}tjt|d | ¡ W 5 Q R X d S )Nr r r ©ÚAÚBz Must provider )r r r Ú TypeErrorÚaggr, r r r Útest_agg_raises5 s r9 c C sB t dddg} d}tjt|d | jdd dd W 5 Q R X d S )
|
Nr r r z)na_action must either be 'ignore' or Noner c S s | S r* r r r r r r C r z8test_map_with_invalid_na_action_raises.<locals>.<lambda>Ú____r1 ©r r r r Úmap)Úsr r r r Ú&test_map_with_invalid_na_action_raises> s r> Úinput_na_actionr: Tc C sJ t dddg}d| d}tjt|d |jddi| d W 5 Q R X d S )Nr r r z+na_action must either be 'ignore' or None, z was passedr r1 r; )r? r= r r r r Ú2test_map_arg_is_dict_with_invalid_na_action_raisesF s r@ c C sZ t tdtddd} t| dtdd}tjttjd |jd d
|
d d W 5 Q R X d S ) NZABBABCDZDCBAT)Ú
|
categoriesZorderedÚXXZabcdefg)Únamer) r c S s | S r* r r r r r r S r z0test_map_categorical_na_action.<locals>.<lambda>Úignorer1 ) r Úlistr r r ÚNotImplementedErrorÚtmÚEMPTY_STRING_PATTERNr< ©Úvaluesr= r r r Ú test_map_categorical_na_actionO s rK c C sR t dddd d¡} t| dd}tjttjd |jd d
|
d d W 5 Q R X d S ) Nz
|
2011-01-01z
|
2011-01-02ÚH)Úfreqz
|
Asia/TokyorB )rC r c S s | S r* r r r r r r Z r z/test_map_datetimetz_na_action.<locals>.<lambda>rD r1 ) r Z tz_localizer r r rF rG rH r< rI r r r Ú test_map_datetimetz_na_actionV s rN Úmethodr r8 Ú transformÚfuncr5 r6 Úsumc C s> | ddgi}d}t jt|d t||| W 5 Q R X d S )Nr5 r ú nested renamer is not supportedr )r r r Úgetattr)Úframe_or_seriesrO rQ Úobjr r r r Útest_nested_renamer] s rW ÚrenamerÚminÚmaxÚmean)r
|
Úbarc C s> t tdddd}d}tjt|d | | ¡ W 5 Q R X d S )Nr# Úint64Úseries)ZdtyperC rS r )r Úranger r r r8 )rX r= r r r r Útest_series_nested_renamerg s r` c C s\ t tj dd¡dddgtdddd} d }tjt|d
|
| j d d d gi¡ W 5 Q R X d S )Né
|
r r5 r6 ÚCz1/1/2000)Zperiods)Úcolumnsr) rS r r
|
rR r[ )
|
r ÚnpÚrandomÚrandnr r r r r5 r8 )Ztsdfr r r r Útest_apply_dict_deprr s
|
ýrg c C sP t tddd}d}tjt|d" t|| ddiddid W 5 Q R X d S ) Nr" r4 rS r r
|
rY r\ rZ )r r_ r r r rT )rO r r r r r Ú test_dict_nested_renaming_depr} s rh c C sD t ddgi}t d¡}tjt|d t|| | W 5 Q R X d S )Nr5 r z Column(s) ['B'] do not existr )r ÚreÚescaper r ÚKeyErrorrT )rO rQ rV r r r r Útest_missing_column s
|
rl c C sB t ddgi} d}tjt|d | tttd¡ W 5 Q R X d S )Nr' Ú1z#Column\(s\) \[1, 'b'\] do not existr )r' r Úb)r r r rk rP Úintr/ r, r r r Ú'test_transform_mixed_column_name_dtypes s rp z how, args)Z
|
pct_changer Ú nsmallestr' rn )Útailr c C sP t ddgddgd}d| d}tjt|d |j| d|d W 5 Q R X d S )
|
Nr r r r! )r' rn z
|
Operation z does not support axis=1r )r Úargsr+ )Úhowrs r r r r r Ú test_apply_str_axis_1_raises s ru c C s6 d} t jt| d tdgjddd W 5 Q R X d S )Nz&No axis named 1 for object type Seriesr r rR ©r )r r r r rP )r r r r Ú test_transform_axis_1_raises¤ s rw c C s¼ t dddddddddddg dddddddddddg dddddddddddg tj d¡tj d¡tj d¡d} tj| jd < d
|
d }d d }d}tjt|d | j |dd W 5 Q R X d S )Nr
|
r\ ZoneZtwoZdullZshinyé )r5 r6 rb ÚDÚEÚF)r! rb c S s& | d d¡r"| d dkr"d| d< | S ©Nrb Zshinr5 r
|
r$ ry )Ú
|
startswith©Úrowr r r rP Ý s z.test_apply_modify_traceback.<locals>.transformc S s2 t | d r.| d d¡r.| d dkr.d| d< | S r| )r r} r~ r r r Ú
|
transform2â s &z/test_apply_modify_traceback.<locals>.transform2z,'float' object has no attribute 'startswith'r r rv )
|
r rd re rf ÚnanÚlocr r ÚAttributeErrorr )ÚdatarP r r r r r Útest_apply_modify_traceback« s` õõõ
|
|
|
Öÿ/ r
zdf, func, expectedZcumprodc C s0 d}t j||d | j||d W 5 Q R X d S )Nz0can't multiply sequence by non-int of type 'str'r rv ©r r r8 )r rQ Úexpectedr r r r r Ú"test_agg_cython_table_raises_frameì s r zseries, func, expectedza b cÚprodZstdÚvarZmedianc C s, d}t j||d | |¡ W 5 Q R X d S )Nz?[Cc]ould not convert|can't multiply sequence by non-int of typer r )r^ rQ r r r r r Ú#test_agg_cython_table_raises_seriesù s r c C sH t dd gi} t d¡}tjt|d | ddd i¡ W 5 Q R X d S )Nr' z int() argument must be a stringr c S s t | jd S ©Nr ©ro Zilocr r r r r r z'test_agg_none_to_type.<locals>.<lambda>)r ri rj r r r7 r8 r, r r r Útest_agg_none_to_type s
|
r c C sB t dd gi} d}tjt|d | ddd i¡ W 5 Q R X d S )Nr' zargument must be ar c S s t | jd S r r r r r r r r z-test_transform_none_to_type.<locals>.<lambda>)r r r r7 rP r, r r r Útest_transform_none_to_type s r c C s t ddg¡ dd¡S )Nr r éÿÿÿÿ)rd ÚarrayZreshaper r r r r $ r r c C s ddgS ©Nr r r r r r r r % r c C s t ddgS r )r r r r r r & r c C s6 | }d}t jt|d |j|ddd W 5 Q R X d S )Nz2too many dims to broadcast|cannot broadcast resultr r Ú broadcastr r )r rQ r r r r r Útest_apply_broadcast_error! s r c
|
C sL d}t jt|d0 tjdd |jddg| d W 5 Q R X W 5 Q R X d S )Nz3cannot combine transform and aggregation operationsr rD ©ÚallrZ Úsqrtrv ©r r r rd Zerrstater8 )r r2 r r r r Ú test_transform_and_agg_err_agg2 s r z func, msgr z(cannot combine transform and aggregationz2cannot perform both aggregation and transformationc
|
C s@ t jt|d( tjdd | |¡ W 5 Q R X W 5 Q R X d S )Nr rD r r )Ú string_seriesrQ r r r r Ú!test_transform_and_agg_err_series: s r c C s0 d}t jt|d |j|| d W 5 Q R X d S )NúFunction did not transformr rv )r r r rP )r r2 rQ r r r r Ú test_transform_wont_agg_frameK s r c
|
C s` d}|d dkrt nd }d}tjt|d, tj||dd | |¡ W 5 Q R X W 5 Q R X d S )Nr r r z!invalid value encountered in sqrtr F)r Zcheck_stacklevel)ÚRuntimeWarningr r r rG Zassert_produces_warningrP )r rQ r ÚwarnZwarn_msgr r r Ú test_transform_wont_agg_seriesT s r Ú
|
op_wrapperc C s | S r* r r r r r r b r c C s | gS r* r r r r r r b r c C s d| iS ©Nr5 r r r r r r b r c C s
|
d| giS r¢ r r r r r r b r c C sR || }t ddddgi}t ||¡}d}tjt|d | |¡ W 5 Q R X d S )Nr5 r r r r r )r rG Úget_objr r r rP )Zall_reductionsrU r¡ ÚoprV r r r r Ú test_transform_reducer_raisesa s r¥ )2Ú itertoolsr ri Únumpyrd r Z pandas.errorsr Zpandasr r r r r Zpandas._testingZ_testingrG ÚmarkZ parametrizer r- r3 r9 r> r@ rK rN rW r` rg rh rl rp ru rw r
Zget_cython_table_paramsr7 r Úsplitr r r r r r r r r r¥ r r r r Ú<module> s¶
|
|
" þ
|
|
ÿ
|
A ÿþ
|
|
úþÿþ
|
ýþ
|
|
|
þþþ
|
|
|
|
ÿ
|
