U
|
ý°dué ã @ s¨ d dl mZ d dlmZ d dlZd dlZd dlZd dlZd dlm Z m
|
Z
|
m Z m Z m Z d dlmZ ejddddZejd dd
|
d ZG d d d ZG dd dZdS )é )Ú annotations)ÚdatetimeN)Ú DataFrameÚIndexÚSeriesÚ TimestampÚ
|
date_rangezdict[str, list[int | str]])Úreturnc C s t tdt ddS )Né úab..©ÚaÚb)ÚlistÚrange© r r ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_replace.pyÚmix_ab s r z dict[str, list[float | str]]c C s" t tdt dddtjdgdS )Nr
|
r r r Úd©r r Úc)r r ÚnpÚnanr r r r Úmix_abc s r c @ sj e Zd Zdd Zej dddgejdgddejejgdgd d
|
g d
|
dd d gd fddgddgddddgdddd
|
gd
|
dd d gd fddgddgddddgdddd
|
gd
|
dd d gd fg¡ej dddg¡ej d ddg¡d d Z d d Z
|
d!d" Z d#d$ Z d%d& Z d'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Zd3d4 Zej d5ejd6d7d6ejd8g¡d9d: Zej d;d<d=d>d?d@g¡dAdB Zej dCdDdEgdFdGdHdIdJgfdGd6d6gdKejidGejejgfg¡dLdM ZdNdO ZdPdQ ZdRdS Z dTdU Z dVdW Z dXdY Z dZd[ Z d\d] Z!d^d_ Z"d`da Z#dbdc Z$ej dddedfdgg¡dhdi Z%djdk Z&dldm Z'dndo Z(dpdq Z)drds Z*dtdu Z+dvdw Z,ej dxe-dydzd{d gidzd|e-dyd|d{d gife-dydzd{d giej.d}dzd|e-dyd|d{d giej.d}fe-dydzd{d giej/d}dzd|e-dyd|d{d giej/d}fe-d~dddgidde-d~dddgife-ddddgidd|e-ddddgife-de0e1d
dde1d
dde1d
ddgie1d
dde1d
dde-de0e1d
ddgd ife-e1dddgdgddde-e1dddgdgdfe-de1dddzgie1dddze1dddze-de1dddzgife-e2dd ddd|ejd{gde3ddde3ddde-e3ddde3ddde3dddgd|ejd{gdfe-dzdgd{dggdde-ddgd{dggfe-dzdgd{dggdzde-ddgd{dggfe-dzdgd{dggdde-ddgd{dggfe-dzdgd{dggdzde-ddgd{dggfg ¡dd Z4d d¡ Z5d¢d£ Z6d¤d¥ Z7d¦d§ Z8d¨d© Z9dªd« Z:d¬d Z;d®d¯ Z<d°d± Z=d²d³ Z>d´dµ Z?d¶d· Z@d¸d¹ ZAdºd» ZBd¼d½ ZCd¾d¿ ZDdÀdÁ ZEdÂdà ZFdÄdÅ ZGdÆdÇ ZHej dÈd|dÉdzdzd{gdejdgdddFgdÊfejdÉd|dzd{gddËdËgdddFgdÊfdGdÌd|dzd{gdejdgdddFgdÊfd|d{gdÉdzdzd{gdejdgdddFgdÊfdzd{gdÍd|d|d|gdejdgdddFgdÊfdÎdÉd|d{d{gdejdgdddFgdÊfddFgdÌd|dzd{gdejdgdddgdÊfg¡dÏdÐ ZIej dÑdzdzdHd d gd{d{ggfdzd{dHd dzgd{d ggfg¡dÒdÓ ZJej dÔdzd{d gdÕddÖgd×ddÖddØdzd{d gddÖdgd×fdzd{d gdÙdÚdÛgd×dÚdÛdÜdÝdzd{d gdÚdÛdÜgd×fg¡dÞdß ZKej dàe3dáe Ldz¡e /dz¡e Mdz¡e Ndz¡g¡dâdã ZOdädå ZPdædç ZQej dddèdédêdfdeg¡ej dëejejg¡dìdí ZRej dîejdg¡dïdð ZSej dëe Tdñ¡e Ud|d¡g¡dòdó ZVdôdõ ZWdöd÷ ZXdødù ZYdúdû ZZdüdý Z[dþdÿ Z\ej d dzgdgd|gd|gfdzgdzgd|gd|gfdgdgd|gdgfdgdzgd|gdgfg¡ej de]e^ej_g¡dd Z`ej dd{ejdg¡dd ZadS ( ÚTestDataFrameReplacec C s t j|j|jd d
df< t j|j|jdd
df< | ¡ }|jt jddd}|d ksZtt || d¡¡ |}t j|j
|
dd
|j d¡f< t j|j
|
d d
|j d¡f< | t jd¡}|j dd
|
}t ||¡ | ¡ }|jt jgdgdd}|d ksøtt || d¡¡ d S ) Né ÚAéûÿÿÿr T©Úinplaceé Úfooéöÿÿÿ©Úvalue) r r ÚlocÚindexÚcopyÚreplaceÚAssertionErrorÚtmÚassert_frame_equalÚfillnaÚilocÚcolumnsÚget_loc)ÚselfÚdatetime_frameÚfloat_string_frameÚtsframeÚ return_valueÚmfÚresultÚexpectedr r r Útest_replace_inplace s z)TestDataFrameReplace.test_replace_inplacezto_replace,values,expectedú\s*\.\s*ze|f|gÚcrapr r é ÚhÚlÚor ú
|
\s*(\.)\s*z(e|f|g)ú\1\1ú\1_crapú..Ze_crapZf_crapZg_crapÚeÚfÚgr TFÚuse_value_regex_argsc C sp t tdtdtdd}|r2|j|||d}n|j||d|d}|rX|d ksTt|}t |}t ||¡ d S )Nr ÚefghÚhelor )r$ Úregexr T©rI r )r r r( r) r* r+ )r0 Ú
|
to_replaceÚvaluesr7 r rF Údfr6 r r r Útest_regex_replace_list_obj6 s * z0TestDataFrameReplace.test_regex_replace_list_objc C s^ t |}ddg}tjdg}ttdtdtdd}t |}|j||dd }t |d dd
|
tjtjgd dd d gd}t ||¡ ddg}ddg}|j||dd }t |d ddddgd}t ||¡ dddg}dddg}|j||dd }t |d ddddgd}t ||¡ dddg}dddg}|j||d}t |d ddddgd}t ||¡ d S )Nr9 r r: r
|
r Zhalor T©rI r r< r= r> r? ú(a|b)r@ rA Úa_crapÚb_craprB r ú(b)©rI r$ )r r r r r r( r* r+ ) r0 r ÚdfmixÚto_replace_resrL Zmix2Zdfmix2ÚresÚexpecr r r Ú test_regex_replace_list_mixedn s8
|
|
ýÿ
|
|
|
|
z2TestDataFrameReplace.test_regex_replace_list_mixedc C s t |}ddg}tjdg}| ¡ }|j||ddd}|d ks@tt |d ddtjtjgd}t ||¡ dd g}d
|
d g}| ¡ }|j||ddd}|d ks tt |d d d ddgd}t ||¡ dddg}d
|
dd g}| ¡ }|j||ddd}|d kstt |d dd ddgd}t ||¡ dddg}d
|
dd g}| ¡ }|j||dd}|d ksbtt |d dd ddgd}t ||¡ d S )Nr9 r r: T©r rI r r r? rP r@ rA rQ rR rB rS ©rI r$ r )r r r r' r( r) r* r+ )r0 r rU rV rL rW r4 rX r r r Ú%test_regex_replace_list_mixed_inplace s:
|
|
|
|
|
z:TestDataFrameReplace.test_regex_replace_list_mixed_inplacec C s t |}|jddidtjidd}| ¡ }|jddidtjiddd}|d ksRtt |d ddtjtjg|d d}t ||¡ t ||¡ |jdd idd
|
idd}| ¡ }|jdd idd
|
iddd}|d ksÔtt |d ddd d g|d d}t ||¡ t ||¡ |jdd idd
|
id }| ¡ }|jdd idd
|
idd }|d ksPtt |d ddd d g|d d}t ||¡ t ||¡ t |d tjdddg|d d}|jddtjidd}| ¡ }|jddtjiddd}|d ksîtt ||¡ t ||¡ |jddtjidd}| ¡ }|jddtjidd }|d ksHtt |d tjdddg|d d}t ||¡ t ||¡ d S )Nr r9 TrO rZ r r r r? z\1tyz.tyrT r[ Ú.rJ ©r r( r r r' r) r* r+ )r0 r rU rW Úres2r4 rX r r r Ú test_regex_replace_dict_mixed¿ sx ÿ ÿ ÿ ÿ ÿÿ ÿ ÿ z2TestDataFrameReplace.test_regex_replace_dict_mixedc C sè t |}|jddtjiidd}| ¡ }| ¡ }|jddtjiiddd}|d ksVt|jddtjiid}|jddtjiidd}|d kstt |d ddtjtjg|d d }t ||¡ t ||¡ t ||¡ t ||¡ d S )
|
Nr r9 TrO rZ rJ r r r r^ ) r0 r rU rW r_ Zres4r4 Úres3rX r r r Ú test_regex_replace_dict_nested s( ÿ ÿ z3TestDataFrameReplace.test_regex_replace_dict_nestedc C sR |}t ddddgi|d}t ddddgi|d}|jd d
|
id d }t ||¡ d S ) NÚfirstÚabcZbcaZcab©Údtypez.bczbc.zc.br r] TrO ©r r( r* r+ )r0 Úany_string_dtyperf rM r7 r6 r r r Ú2test_regex_replace_dict_nested_non_first_character s
|
zGTestDataFrameReplace.test_regex_replace_dict_nested_non_first_characterc C sT t dddddgdd}t dddddgdd}| ddddi¡}t ||¡ d S ) NÚQÚTé )ÚTypeÚtmpr é rm )rj rk rg ©r0 rM r7 r6 r r r Ú%test_regex_replace_dict_nested_gh4115! s z:TestDataFrameReplace.test_regex_replace_dict_nested_gh4115c C sÌ t |}t |d t tjgd ¡tjtjtjdgd}|jddgtjdd}| ¡ }| ¡ }|jddgtjddd }|d kst|jddgtjdd
|
}|d ks¤tt ||¡ t ||¡ t ||¡ d S ) Nr r
|
r r r9 za|bTrO rJ r[ ) r r Úarrayr r( r' r) r* r+ )r0 r rM rX rW r_ ra r4 r r r Ú!test_regex_replace_list_to_scalar' s4 ýÿ ÿ ÿ z6TestDataFrameReplace.test_regex_replace_list_to_scalarc C sª t |}|jdddd}| ¡ }|jddddd}|d ks>t| ¡ }|jdddd}|d ksbtt |d ddddg|d d
|
}t ||¡ t ||¡ t ||¡ d S ) Nr9 r TrO rZ r[ r r r r )r r( r' r) r* r+ ©r0 r rM rW r_ r4 ra rX r r r Ú!test_regex_replace_str_to_numeric? s z6TestDataFrameReplace.test_regex_replace_str_to_numericc C s¼ t |}|jddgddd}| ¡ }|jddgdddd}|d ksFt| ¡ }|jddgddd}|d ksntt |d ddddgddtjd gd
|
}t ||¡ t ||¡ t ||¡ d S ) Nr9 r r TrO rJ r[ r r r )r r( r' r) r r r* r+ rt r r r Ú(test_regex_replace_regex_list_to_numericN s ÿ z=TestDataFrameReplace.test_regex_replace_regex_list_to_numericc
|
C sÈ t |}tddi}tdtji}|j||dd}| ¡ }|j||ddd}|d ksXt| ¡ }|j||dd}|d ks|tt |d ddtjtjg|d d } t || ¡ t || ¡ t || ¡ d S )
|
Nr r9 TrO rZ r[ r r r ) r r r r r( r' r) r* r+ )
|
r0 r rM Ús1Ús2rW r_ r4 ra rX r r r Ú$test_regex_replace_series_of_regexes^ s ÿ z9TestDataFrameReplace.test_regex_replace_series_of_regexesc C sV t |}t ddddg|d |d d}| dd¡}t ||¡ |jjtjksRtd S ) Nr ro rl r; r r r r ) r r( r* r+ r rf r Zobject_r) )r0 r rM rX rW r r r Ú/test_regex_replace_numeric_to_object_conversionp s
|
zDTestDataFrameReplace.test_regex_replace_numeric_to_object_conversionrK Ú )r{ ú,)r| r{ c C sb t dddgdddgdddgd}|j|d}t d ddgdtjdgdddgd}t ||¡ d S )
|
Nz1,000r Ú3r{ r r ©Úcol1Úcol2Úcol3rO Z1000©r r( r r r* r+ )r0 rK rM r6 r7 r r r Ú+test_joint_simple_replace_and_regex_replacew s ýÿ
|
ýÿz@TestDataFrameReplace.test_joint_simple_replace_and_regex_replaceÚmetacharz[]z()z\dz\wz\sc C sB t d|dgi}| d|dii¡}t dddgi}t ||¡ d S )Nr ÚelseZparenrg )r0 r rM r6 r7 r r r Útest_replace_regex_metachar s z0TestDataFrameReplace.test_replace_regex_metacharzdata,to_replace,expectedZxaxZxbxr r r ZxcxZxdxz^\s*$c C s: |}|||d}|j |dd}|||d}t ||¡ d S )Nre TrO ©r( r* Ú assert_equal) r0 ÚdatarK r7 Úframe_or_seriesrh rf Úobjr6 r r r Ú test_regex_replace_string_types s
|
z4TestDataFrameReplace.test_regex_replace_string_typesc C s8 t j|j|jd d
df< t j|j|jdd
df< | t jd¡}t || d¡¡ t | dt j¡|¡ t j|j|jd d
df< t j|j|jdd
df< d|j|jd d
df< tddgd}t || dd ¡¡ td
|
t
|
d ¡fdt
|
d ¡fg}| d
|
t j¡}tt jt
|
d ¡fdt
|
d ¡fg}t ||¡ d S ) Nr r r ç ×ÁÚBr r )r& é ú-Z20150101Z20150102) r r r% r& r( r* r+ r, r ÚpdÚ to_datetime)r0 r1 Ú zero_filledrM Údf1Z expected_dfr r r Ú test_replace¦ s$ ÿ ÿz!TestDataFrameReplace.test_replacec C sÊ t dt dt dd}t|}ddg}tjdg}| ||¡}tdd tjtjgdd
|
d d gd dd dgd}t ||¡ dd
|
g}ddg}| ||¡}tdd ddgddd d gd dd dgd}t ||¡ d S )Nr rG rH r r] rC r: r r rD rE r< r= r> rB )r r r r r( r* r+ )r0 r ZdfobjrV rL rW rX r r r Útest_replace_listÁ s,
|
|
|
ýÿ
|
|
|
ýÿz&TestDataFrameReplace.test_replace_listc C s¾ t ddgg tjdgg}td|i}t ||¡}|}| g tj¡}t ||¡ d}tj t
|
|j ddd | tjg i¡ W 5 Q R X tj t
|
|j d dd | tjd
|
d gi¡ W 5 Q R X d S ) Nr r ro ÚcolzwNumPy boolean array indexing assignment cannot assign {size} input values to the 1 output values where the mask is truer )Úsize©Úmatchrl ÚdummyZalt) r r r r r* Úget_objr( r ÚpytestÚraisesÚ
|
ValueErrorÚformat)r0 r Zserr r7 r6 Úmsgr r r Ú test_replace_with_empty_listà s ÿz1TestDataFrameReplace.test_replace_with_empty_listc C sî t ddddddd}| dddd¡}t ddddddd}t ||¡ | d| ¡ ¡}t ||¡ t ddddddd}tddd}| |ddd¡}t ddddddd}t ||¡ | || ¡ ¡}t ||¡ d S ) Nç ro r ç @r )ZzeroÚoneç à?ç ð?)r r( r* r+ Úmeanr )r0 rM r6 r7 Úsr r r Útest_replace_series_dictó s z-TestDataFrameReplace.test_replace_series_dictc C sT t dddgdddgg}dddd}| |¡}ttjgd }|j}t ||¡ d S )Nr! ÚbarÚbahro rl r; )r! r« r¬ )r r( r r Úint64Údtypesr* Úassert_series_equal)r0 rM ÚmÚreprX rW r r r Útest_replace_convert s
|
z)TestDataFrameReplace.test_replace_convertc C s¸ |}t j|jdd
|j d¡f< t j|jdd
|j d¡f< | t jd¡}|jdd}t ||¡ t | dt j¡|¡ | t jd¡}|jdd}t ||¡ t | dt j¡|¡ d S ) Nr r r! r" r iîÿÿÿr# r ) r r r- r. r/ r( r, r* r+ )r0 r2 r5 r6 r7 r r r Útest_replace_mixed s z'TestDataFrameReplace.test_replace_mixedc C s t tddgddtddgddd}t tddgddtd dgddd}| dd ¡}t ||¡ |jdd d
|
d }|d kstt ||¡ d S ) Nr§ r¤ Úfloat64re r ro r ©r r r¦ Tr )r r r( r* r+ r) )r0 rM r7 r6 r4 r r r Ú&test_replace_mixed_int_block_upcasting s þÿþÿ z;TestDataFrameReplace.test_replace_mixed_int_block_upcastingc C s t tddgddtddgddtddgddd }t tddgddtd
|
dgddtddgddd }| dd
|
¡}t ||¡ d S ) Nr§ r¤ r´ re r ro r rl ©r r ÚCr¦ ©r r r( r* r+ rp r r r Ú&test_replace_mixed_int_block_splitting4 s ýÿýÿ z;TestDataFrameReplace.test_replace_mixed_int_block_splittingc C s® t tddgddtddgddd}t tdd gd
|
dtddgddd}| d d ¡}t ||¡ t td d gd
|
dtdd gd
|
dd}| dd gd d g¡}t ||¡ d S ) Nr§ r¤ r´ re r ro r rµ r! Úobjectrl r« r¹ rp r r r Útest_replace_mixed2G s& þÿþÿ þÿz(TestDataFrameReplace.test_replace_mixed2c C s| t tddgddtddgddd}| d| ¡ ¡ ¡}| ¡ d¡}| ¡ }|d |jd< |d |jd < t ||¡ d S )
|
Nr; r r re rµ r´ ©r r ro )ro ro )
|
r r r( r¨ Úto_dictr' Úastyper- r* r+ )r0 rM r6 r7 r° r r r Útest_replace_mixed3a s ÿz(TestDataFrameReplace.test_replace_mixed3c C sX t dddtjgddddgd}|d d
|
¡|d < | d tj¡}t |d |d ¡ d S ) Nro rl r; ZsomeÚstringsÚhereÚher r ÚInt64r{ )r r r r¿ r( r* r¯ ©r0 rM rW r r r Ú1test_replace_nullable_int_with_string_doesnt_castm s zFTestDataFrameReplace.test_replace_nullable_int_with_string_doesnt_castrf ÚbooleanrÄ ZFloat64c C sV t dddg|d}tdddg|d}| dd¡}tdddg|d}t ||¡ d S ) Nro r re r r Úxrµ ÚX©r r r( r* r+ )r0 rf Z nullable_serrM r6 r7 r r r Ú!test_replace_with_nullable_columnu s
|
z6TestDataFrameReplace.test_replace_with_nullable_columnc C sf t dtddi}t dddddgi}| ddddi¡}t ||¡ | ddd¡}t ||¡ d S ) Nr ro r r rl r; r )ro r
|
©r r r( r* r+ rp r r r Ú test_replace_simple_nested_dict~ s z4TestDataFrameReplace.test_replace_simple_nested_dictc C sj t dtddi}t dddddgi}| dddd ¡}t ||¡ | ddddd i¡}t ||¡ d S )
|
Nr ro r r rl r; r r )éÿÿÿÿro r
|
rÌ rp r r r Ú6test_replace_simple_nested_dict_with_nonexistent_value s zKTestDataFrameReplace.test_replace_simple_nested_dict_with_nonexistent_valuec C sN t ddd gi ddi¡}| tjd i¡}t ddd gitd}t ||¡ d S )Nr$ é* rÄ re )r r¿ r( r ÚNAr» r* r+ ©r0 rM r6 r7 r r r Útest_replace_NA_with_None s z.TestDataFrameReplace.test_replace_NA_with_Nonec C sB t tjtjg}| tjd tjd i¡}t d d g}t ||¡ d S ©N)r r ÚNaTr( r ÚNaNr* r+ rÒ r r r Útest_replace_NAT_with_None s z/TestDataFrameReplace.test_replace_NAT_with_Nonec C sn t ddddgdd}tt ddddgd d|d
|
}| dd i¡}tt d d d d gdd|d
|
}t ||¡ d S )Nr r Úcategoryre r r
|
r; rl r´ )Úidr ç @g @r¤ r» rÊ )r0 Z
|
cat_seriesrM r6 r7 r r r Ú(test_replace_with_None_keeps_categorical¡ s þÿþÿz=TestDataFrameReplace.test_replace_with_None_keeps_categoricalc C s¸ |j d }|j d }tj|j d< d|j d< |jtjdid}|jjtjdidj}t ||¡ |jtjdddid}| ¡ }d|j d< d|j d< |}t ||¡ ||j d< ||j d< d S )Nr½ ©ro r ro r ©rK r )r- r r r( rk r* r+ r' )r0 r1 Z
|
orig_valueZorig2r6 r7 r3 r r r Útest_replace_value_is_none´ s
|
|
|
|
|
|
z/TestDataFrameReplace.test_replace_value_is_nonec C s | ¡ tj¡}tj|j|jd d
df< tj|j|jdd
df< | tjd¡}t ||
|
d¡¡ t | dtj¡|¡ tj|j|jd d
df< tj|j|jdd
df< d|j|jd d
df< |d }tj||dk< ||d< |j
|
dd}t ||j
|
dd¡ d S )Nr r r r r Úbfill)Úmethod) r' r¿ r Úfloat32r r% r& r( r* r+ r, )r0 r1 r3 r r r6 r r r Útest_replace_for_new_dtypesÈ s z0TestDataFrameReplace.test_replace_for_new_dtypesz"frame, to_replace, value, expectedÚintsro rl r re ZboolsÚcomplexy ð?y @y @y Z
|
datetime64iâ r é r iÉ é r r! )ÚdtÚstrr« rç ih é
|
iä Ú20130101ú
|
US/Eastern©ZperiodsÚtzrµ Ú20130102©rí Ú20130104Ú20130103r§ r¤ rÚ c C s | ||¡}t ||¡ d S rÔ )r( r* r+ )r0 ÚframerK r$ r7 r6 r r r Útest_replace_dtypesÜ s j z(TestDataFrameReplace.test_replace_dtypesc sÄ t jddd ddddtt jdt jgdddgddd gd}| ¡} fd
|
d | ¡ D }t |t|¡ | dddgdddg¡}tt jdt jgdddgddd gd}t ||¡ ddddtt jdt jgdddgddd gd}| t j¡}fd d | ¡ D }t |t|¡ t jddg d ddg| ¡}| ¡ }t D ](\}}|j||dd}|d ksXt
|
qXt ||¡ d}t j t |d | dd
¡ W 5 Q R X d S )Nr r{ r· rÎ Úmissingrl r ÚasdfÚfdc s& i | ] \}}|| | | ¡qS r ©r( ©Ú.0ÚkÚv©Úto_reprL r r Ú
|
<dictcomp>Q s zLTestDataFrameReplace.test_replace_input_formats_listlike.<locals>.<dictcomp>c s$ i | ] \}}|| tj | ¡qS r )r( r r rø )rL r r rþ ` s éþÿÿÿTr z:Replacement lists must match in length\. Expecting 3 got 2r ro )r r r Úinfr( Úitemsr* r+ r' Úzipr) r r r ) r0 rM Úfilledr7 r6 r± r$ r4 r¡ r rü r Ú#test_replace_input_formats_listlikeI s> ÿ ÿ ÿ
|
z8TestDataFrameReplace.test_replace_input_formats_listlikec sî t tjdtjgdddgdddgd}tjddd | d¡} fdd | ¡ D }t |t |¡ d
|
}tj t
|
|d | tjddg¡ W 5 Q R X tjddg | d ¡}| ¡ } D ] }|j|d d d}|d ks¼t q¼t ||¡ d S )Nr rl r r{ rõ rö r· c s" i | ]\}}|| | d ¡qS )r r÷ rø ©rý r r rþ y s zJTestDataFrameReplace.test_replace_input_formats_scalar.<locals>.<dictcomp>z.value argument must be scalar, dict, or Seriesr rÎ Tr ) r r r r r( r r* r+ r r Ú TypeErrorr' r) )r0 rM r r7 r¡ r6 r± r4 r r r Ú!test_replace_input_formats_scalarq s" ÿ z6TestDataFrameReplace.test_replace_input_formats_scalarc C s d S rÔ r )r0 r r r Útest_replace_limit s z'TestDataFrameReplace.test_replace_limitc C sR t dddddd}ddd d
|
d d }t d
|
dd dd d}| |¡}t ||¡ d S © NúStrongly AgreeÚAgreeÚNeutralÚDisagreeúStrongly Disagree)r ro rl r; r
|
r
|
rl r; r ro )r r r r
|
r ©r r( r* r¯ ©r0 ZanswerÚweightsr7 r6 r r r Útest_replace_dict_no_regex s" ûÿ
|
û
|
z/TestDataFrameReplace.test_replace_dict_no_regexc C sV t dddddd}t ddd d
|
d d }t d
|
dd dd d}| |¡}t ||¡ d S r r r r r r Ú test_replace_series_no_regex¢ s& ûÿ ûÿ
|
z1TestDataFrameReplace.test_replace_series_no_regexc C s t dtjdgi}|jtjdddid}|jdtjfddgd}|jdtjgddgd}t dddgi}t ||¡ t ||¡ t ||¡ d S )Nr ro r r rÝ ©rK r$ ©r r r r( r* r+ )r0 rM Zres1r_ ra r7 r r r Ú6test_replace_dict_tuple_list_ordering_remains_the_same¹ s zKTestDataFrameReplace.test_replace_dict_tuple_list_ordering_remains_the_samec C sP t ddddgddddgddddgddddgd}| d di¡}t ||¡ d S )
|
Nro rl r; Ú0ZvrZbtZvo)ZfolZT_oppZT_DirZT_Enhz\Drg rÅ r r r Ú)test_replace_doesnt_replace_without_regexÄ s
|
|
|
|
üÿz>TestDataFrameReplace.test_replace_doesnt_replace_without_regexc C sF t ddgtdd}| dd¡}t ddg|jd}t ||¡ d S )NTFÚabr r )r r r( r r* r+ rÒ r r r Ú test_replace_bool_with_stringÐ s z2TestDataFrameReplace.test_replace_bool_with_stringc C s2 t tj dd¡dk}| dd¡}t ||¡ d S )Nrl r¦ rõ Zfdsa)r r ÚrandomÚrandr( r* r+ ©r0 rM r6 r r r Ú(test_replace_pure_bool_with_string_no_opÖ s z=TestDataFrameReplace.test_replace_pure_bool_with_string_no_opc C sD t tj dd¡dk}| dd¡}t tjdtd}t ||¡ d S )Nrl r¦ FT)rl rl re ) r r r r r( ÚonesÚboolr* r+ rÒ r r r Útest_replace_bool_with_boolÛ s z0TestDataFrameReplace.test_replace_bool_with_boolc C sL t ddgddgd}| ddd¡}t ddgddgd}t ||¡ d S )NTF©r ro ZasdbÚyes)rõ Trg rÒ r r r Ú%test_replace_with_dict_with_bool_keysá s z:TestDataFrameReplace.test_replace_with_dict_with_bool_keysc C sV t ddgddgd}| ddi¡}t ||¡ |d ddi¡}t ||d ¡ d S ) Nro rl r; r
|
)ÚY0ZY1Zreplace_stringÚtestr% )r r( r* r+ r¯ r r r r Ú!test_replace_dict_strings_vs_intsç s
|
z6TestDataFrameReplace.test_replace_dict_strings_vs_intsc C s< t dddgi}| tjtj gtj¡}|}t ||¡ d S )Nr T)r r( r r r r* r+ )r0 rM ÚrrC r r r Útest_replace_truthyñ s z(TestDataFrameReplace.test_replace_truthyc C sj t dttddi}| dtttddtddi¡}| tttddtdd¡}t ||¡ d S )Nr ro r rl é )r r r r( Údictr r* r+ rÒ r r r Ú-test_nested_dict_overlapping_keys_replace_int÷ s $ zBTestDataFrameReplace.test_nested_dict_overlapping_keys_replace_intc C sp t dd¡}| t¡}t dd¡ t¡}td|i}| tt||¡}| dtt||i¡}t ||¡ d S )Nro r rl r* r )
|
r Zaranger¿ rè r r( r+ r r* r+ )r0 r ZastrZbstrrM r6 r7 r r r Ú-test_nested_dict_overlapping_keys_replace_str s
|
zBTestDataFrameReplace.test_nested_dict_overlapping_keys_replace_strc C s t ddddgi}| ddddi¡}t ddddgi}t ||¡ t ddddgi}| dddd i¡}t ddddgi}t ||¡ d S )
|
Nr TFÚYÚN)TFr ro r" rg )r0 rM rW Úexpectr r r Útest_replace_swapping_bug
|
s z.TestDataFrameReplace.test_replace_swapping_bugc sò dt jddddt jddddt jddddt jdd ddt jddddt jddddt jd
|
d ddd i td ddddddgdgd}t|jjt d ¡ ks¨ttd fdd|jjD i}|jd dksØt| ¡}t
|
||¡ d S )NÚfnameiÛ é ÚM)ÚyearÚmonthÚfreqro iÜ r r
|
iÝ é ©úout_augmented_AUG_2011.jsonúout_augmented_JAN_2011.jsonúout_augmented_MAY_2012.jsonú out_augmented_SUBSIDY_WEEK.jsonúout_augmented_AUG_2012.jsonúout_augmented_MAY_2011.jsonúout_augmented_SEP_2013.jsonr> r@ r= r< r? r: r; ©r. c s g | ]} d | qS ©r2 r ©rù rú ©r r r Ú
|
<listcomp>2 s z<TestDataFrameReplace.test_replace_period.<locals>.<listcomp>r z Period[M]) r ÚPeriodr Úsetr2 rL Úkeysr) r® r( r* r+ rp r rD r Útest_replace_period s< ÿ÷ÿù ö
|
z(TestDataFrameReplace.test_replace_periodc s¨ dt dt dt dt dt dt dt dd i td
|
d d d dddgdgd}t|jjt d ¡ kspttd fdd|jjD i}| ¡}t ||¡ d S )Nr2 z2011-08z2011-01z2012-05z2011-04z2012-08z2011-05z2013-09r9 r> r@ r= r< r? r: r; rA c s g | ]} d | qS rB r rC rD r r rE Q s z>TestDataFrameReplace.test_replace_datetime.<locals>.<listcomp>)
|
r r rG r2 rL rH r) r( r* r+ rp r rD r Útest_replace_datetime7 s2 ùÿ ù ö
|
z*TestDataFrameReplace.test_replace_datetimec C s& t tdddddtjdgd}| tjd¡}t tddddtdddgd d
|
d}t ||¡ | d¡}t ||¡ | dtj¡}t tddddtjtjdgd}t ||¡ | t d dd t d dd ¡}t t ddd t d dd t ddd gdtjdgd}t ||¡ |
|
¡ }tj|j d< | dt j it d dd ¡}t ||¡ |
|
¡ }tj|j d< | dt j it d dd ¡}t t ddd t d dd d¡t ddd gdtjdgd}t ||¡ |
|
¡ }tj|j d< | dtjit d ¡}t t ddd t d t ddd gdtjdgd}t ||¡ d S )Nrê r; rë rì r rl rµ ro r´ re rî rï rð rñ rÜ r z
|
US/Pacific)r r r r r( r r* r+ r, r r' r- r rÕ Z
|
tz_convertrÒ r r r Útest_replace_datetimetzU s|
|
þÿ þÿ
|
þÿ
|
|
þ
|
|
|
ý
|
úÿ
|
|
|
ý
|
úÿ
|
|
|
ý
|
úÿ
|
z,TestDataFrameReplace.test_replace_datetimetzc C sh t |}t || i ¡¡ t || tg td¡¡ t || di i¡¡ t || tdi i¡¡ d S )Nre r )r r* r+ r( r r» )r0 r rM r r r Ú test_replace_with_empty_dictlike§ s
|
z5TestDataFrameReplace.test_replace_with_empty_dictlikez to_replace, method, expectedrß r· g @ZffillÚpad)ro rl c C sL t dddgdtjdgdddgd }|j|d |d
|
}t |}t ||¡ d S ) Nr ro rl r r r r r r· )rK r$ rà r )r0 rK rà r7 rM r6 r r r Útest_replace_method° s "$z(TestDataFrameReplace.test_replace_methodzreplace_dict, final_datac C sø t ddgddggddgdd}t |¡}tj|d d
df ddgd }|d dkrZddgnddg}tj|d d
df |d }t ||d
|
}| |d¡}t ||¡ d } tj t
|
| d t ||¡ W 5 Q R X |j|dd d}
|
|
|
d ksèt
|
t ||¡ d S )Nro rl r r rØ )r. rf r r; )Ú
|
categoriesr zHAttributes of DataFrame.iloc\[:, 0\] \(column name=\"a\"\) are differentr Tr ) r r rr r Z Categoricalr( r* r+ r r r) ) r0 Z replace_dictZ
|
final_datarM r Zex_catr r7 r6 r¡ r4 r r r Ú"test_categorical_replace_with_dictØ s
|
ÿ z7TestDataFrameReplace.test_categorical_replace_with_dictzdf, to_replace, expr
|
r* )r r )r
|
r r* Ú4Ú5Ú6Ú7)rQ rR rS c C s* t |}t |}| |¡}t ||¡ d S rÔ rg )r0 rM rK Úexpr7 r6 r r r Útest_replace_commutativeõ s
|
z-TestDataFrameReplace.test_replace_commutativeÚreplacerZ20170827c C s4 t dg}| ||d¡}t |g}t ||¡ d S )Nr r rg )r0 ÚrequestrW rM r6 r7 r r r Útest_replace_replacer_dtype s
|
|
z0TestDataFrameReplace.test_replace_replacer_dtypec C sP t ddddddgidd}| dd ¡}t dd ddddgidd}t ||¡ d S )
|
NÚgrpro rl r; r
|
r rÄ re ré rg rÒ r r r Ú!test_replace_after_convert_dtypes s z6TestDataFrameReplace.test_replace_after_convert_dtypesc C sJ t dddgdddgd}d}tjt|d | d
|
d ¡ W 5 Q R X d S ) Nr zb r zd ze zf )r¥ ÚtwozZExpecting 'to_replace' to be either a scalar, array-like, dict or None, got invalid type.*r c S s | ¡ S rÔ )Ústrip)rÈ r r r Ú<lambda>0 ó zFTestDataFrameReplace.test_replace_invalid_to_replace.<locals>.<lambda>)r r r r r( )r0 rM r¡ r r r Ú test_replace_invalid_to_replace' s
|
ÿz4TestDataFrameReplace.test_replace_invalid_to_replaceÚfloatr´ r r$ c C s< t t d¡|d}|jd tj tjg|d}t ||¡ d S )Nrl re r )r r Zeyer( r r* r+ )r0 rf r$ rM r6 r r r Ú"test_replace_no_replacement_dtypes2 s z7TestDataFrameReplace.test_replace_no_replacement_dtypesÚ replacementc C s| t dddgdddgddd gd
|
}td |_t dddgdddg|dd gd
|
}td |_|d d|¡|d < t ||¡ d S ) Nro rl r; r
|
r r* r r3 r8 )r ÚA1r ZAABr )r r r. r( r* r+ )r0 rc r6 r7 r r r Ú#test_replace_with_duplicate_columns: s "
|
ÿ
|
z8TestDataFrameReplace.test_replace_with_duplicate_columnsz2020-01c C sB t d|gd i}t ||¡}| ¡ }| dd¡}t ||¡ d S )NZPerr; r§ r£ )r r* r r' r( r )r0 r r$ r r7 r6 r r r Ú test_replace_ea_ignore_floatI s
|
z1TestDataFrameReplace.test_replace_ea_ignore_floatc C sP ddddgddddgd d
|
d d gd dddgddddgd}t |d ddd¡}|d jjddddgdd|d< |d jjd dddgdd|d < ddddgdddd gd d
|
d d gd d ddgd dddgd}t |d ddd¡}|d jjdddd gdd|d< |d jjd d ddgdd|d < | dd ¡}| dd ¡}| dd ¡}t ||¡ d S )!zt
|
Test for #23305: to ensure category dtypes are maintained
|
after replace with direct values
|
ro rl r; r
|
r r r r g ø?g @g @g @Úcat1Zcat2Zcat3Zcat4Úobj1Úobj2Zobj3Zobj4)r r r Úcol4Zcol5©r rØ )r rj r T)Zorderedrj ÚzÚcatXÚobj9N)r r¿ ÚcatZreorder_categoriesr( r* r+ ©r0 Z
|
input_dictZinput_dfÚ expected_dictr7 r6 r r r Ú test_replace_value_category_typeS sL
|
|
|
|
|
û
|
ÿ
|
|
ÿ
|
|
|
ÿ
|
|
|
|
|
|
û
|
ÿ
|
|
ÿ
|
|
|
ÿ
|
z5TestDataFrameReplace.test_replace_value_category_typec C sv dgdgdgd}t |d dddd¡}dgdgd gd}t |d dddd¡}| ddd d
|
¡}t ||¡ d S ) zf
|
Test to ensure category dtypes are maintained
|
after replace with dict values
|
r rh rg r~ rk rØ rl rn rm )r rh rg N)r r¿ r( r* r+ rp r r r Ú test_replace_dict_category_type s
|
|
ÿ
|
|
ÿz4TestDataFrameReplace.test_replace_dict_category_typec C sH t dddg}t d¡}|j|didd}t dddg}t ||¡ d S )Nr r r z^a$rl TrO )r ÚreÚcompiler( r* r+ )r0 rM rI r6 r7 r r r Ú test_replace_with_compiled_regex s
|
|
z5TestDataFrameReplace.test_replace_with_compiled_regexc C sZ t dt dd¡t dd¡gi}| dt dd¡dii¡}t dddgi}t ||¡ d S )Nr r ro rÈ )r r ÚIntervalr( r* r+ rÒ r r r Útest_replace_intervals¨ s z+TestDataFrameReplace.test_replace_intervalsc C sN dddddi}t dt d¡i}| |¡}t dt d¡i}t ||¡ d S )NZpositivero r )u æ£é¢u ä¸ç«u è´é¢r; )r r r r( r* r+ )r0 Zcolumns_values_mapr r6 r7 r r r Útest_replace_unicode¯ s
|
|
z)TestDataFrameReplace.test_replace_unicodec C s8 |dg d¡}| ¡ }| d tji¡}t ||¡ d S )Nr> z|S)r¿ r' r( r r r* r )r0 r r r7 r r r Útest_replace_bytes· s z'TestDataFrameReplace.test_replace_bytesz!data, to_replace, value, expectedr£ Úboxc C s0 ||}||}| |||¡}t ||¡ d S rÔ r ) r0 r rK r$ r7 r{ r r r6 r r r Ú!test_replace_list_with_mixed_type¾ s z6TestDataFrameReplace.test_replace_list_with_mixed_typeÚvalc C sj t dd|gi}| |d ¡}t ddd gitd}t ||¡ t dd|gi}| |d i¡}t ||¡ d S )Nr ro re )r r( r» r* r+ )r0 r} rM r6 r7 r r r Ú%test_replace_value_none_dtype_numericÑ s z:TestDataFrameReplace.test_replace_value_none_dtype_numericN)bÚ__name__Ú
|
__module__Ú __qualname__r8 r ÚmarkÚ parametrizer r rN rY r\ r` rb ri rq rs ru rv ry rz r r r rÑ r r r r¢ rª r² r³ r¶ rº r¼ rÀ rÆ rË rÍ rÏ rÓ r× rÛ rÞ râ r Úint32Úint16r r r r ró r r r r r r r r r r! r$ r' r) r, r- r1 rI rJ rK rL rN rP rV Zint8rá r´ rY r[ r` rb re rF rw rf rr rs rv rx ry rz r Útuplerr r| r~ r r r r r s
|
ýý
|
|
|
ýý
|
|
|
ýýèþ%)(E ÿ
|
|
þþ
|
|
|
&üüüü
|
|
|
ýÿÿÿ
|
|
òü
|
|
ü
|
þÿ
|
|
|
|
|
ý
|
úÿ÷üüüü¡þi
|
(
|
|
" R $ ý$ ý ý ý ýéþ
|
2þ
|
|
ý
|
ýúþ
|
ûþ
|
|
|
|
3üþ r c @ s¾ e Zd Zej dededdedeeddg¡ej ddej fdg¡ej d d
|
d g¡ej d d
|
d g¡ej d d
|
d g¡dd Z
|
ej dd d
|
g¡dd Z dd Z dd Z dS )ÚTestDataFrameReplaceRegexr r rG r r
|
zto_replace,valuer9 )r? z\1\1\1Ú compile_regexTFÚ regex_kwargr c C s t |}| ¡ }|r t |¡}|r,|} d }nd} |j|||| d}
|
|rV|
|
d ksRt|}
|
|tjkrhtj} nd} | |j|d dkdf< t
|
|
|
|¡ d S )NTrZ z...r r] ) r r' rt ru r( r) r r r% r* r+ ) r0 r rK r$ r r r rM r7 rI r6 Zexpected_replace_valr r r Útest_regex_replace_scalarß s"
|
|
z3TestDataFrameReplaceRegex.test_regex_replace_scalarrI c C s t dgdgd}t dgdgd}|jdd|d}t ||¡ t dgdgd}t dgdgd}|jdd|d}t ||¡ d S )Nr rµ ro )rK r$ rI Ú1rg )r0 rI r Z expected_df1Z
|
result_df1Zdf2Z expected_df2Z
|
result_df2r r r Ú test_replace_regex_dtype_frame s z8TestDataFrameReplaceRegex.test_replace_regex_dtype_framec C sZ t dddgdddgd}| dtjd¡}t dtjdgtjddgd}t ||¡ d S )Nr ro rl rµ r" r rÒ r r r Ú+test_replace_with_value_also_being_replaced s zETestDataFrameReplaceRegex.test_replace_with_value_also_being_replacedc C sP t ddd dgdd ddgddd}| ¡ }|jddgd d gd
|
}t ||¡ d S ) Nr¥ r\ Zthreer rØ re r] ÚdefÚ_r )r r' r( r* r+ rp r r r Ú'test_replace_categorical_no_replacement s
|
|
þûzATestDataFrameReplaceRegex.test_replace_categorical_no_replacementN)r r r r r r r r r r r r r r r r r r r Þ s$ þþ ÿ
|
r )Ú
|
__future__r r rt Únumpyr r Zpandasr r r r r r Zpandas._testingZ_testingr* Zfixturer r r r r r r r Ú<module> s0 K
|
