U
|
ý°d?3 ã
|
@ sÐ d dl Zd dlZd dlmZmZ d dlmZ d dl m
|
Z
|
ej dddddd d
|
didd d didd d d ddiiig¡d d Z dd Zdd Zdd Zdd Zdd Zdd Zdd Zd d Zej d d d gg g¡d d! Zej d d d gg¡d"d# Zej d d
|
d$g¡d%d& Zej d ddgg¡ej d'dd g¡d(d) Zej d'dd g¡d*d+ Zej d'dd g¡d,d- Zej d.d/d0g¡d1d2 Z d3d4 Z ej d5d6dg¡d7d8 Z d9d: Z d;d< Z d=d> Z!dS )?é N)Ú CategoricalÚ DataFrame)Ú get_arrayÚreplace_kwargsé é ©ÚaÚbéÿÿÿÿ©Ú
|
to_replaceÚvaluer r
|
r c C sö t dddgdddgddd gd
|
}| ¡ }|jf |}| r|d |d k ¡ rjt t|d t|d ¡sjtt t|d t|d ¡std |jd< | r°t t|d t|d ¡r°td|d kræt|d}d|jd< t t|d|¡sætt
|
||¡ d S )Nr é é r é é ZfooÚbarZbaz)r r
|
Úcr
|
r r ©r r r r éd )r r ) r ÚcopyÚreplaceÚallÚnpÚ shares_memoryr ÚAssertionErrorÚlocÚtmÚassert_frame_equal)Úusing_copy_on_writer ÚdfÚdf_origZ df_replacedÚarr© r$ úZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/copy_view/test_replace.pyÚ test_replace s "
|
|
|
r& c C s t dddgi}| ¡ }|d d
}t|d}|jddddd | rxt |t|d¡rZt|j d¡sjtt
|
||¡ nt |t|d¡std S ) Nr ÚaaaÚbbbú^a.*$ÚnewT©r r ÚinplaceÚregexr ) r r r r r r r Ú_mgrÚ_has_no_referencer r )r r! r" Úviewr# r$ r$ r% Ú test_replace_regex_inplace_refs2 s
|
r1 c C s t dddgi}t|d}|jddddd | r@|j d¡s@tt |t|d¡sVt| ¡ }|jd ddd
|
}t
|
||¡ t t|dt|d¡rtd S ) Nr r' r( r) r* Tr+ r z^b.*$©r r r- © r r r r. r/ r r r r r r ©r r! r# r" Údf2r$ r$ r% Útest_replace_regex_inplace@ s
|
r6 c C s¼ t dddgi}t|d}|jddddd | r@|j d¡s@tt |t|d¡sVt| ¡ }|jd ddd
|
}t
|
||¡ | rt t|dt|d¡s¸tn t t|dt|d¡r¸td S ) Nr r r z^a.$r* Tr+ r z^x.$r2 r3 r4 r$ r$ r% Ú test_replace_regex_inplace_no_opN s
|
r7 c C sÞ t dddgdddd}| ¡ }|jddd}| rlt t|d t|d ¡sNtt t|d
|
t|d
|
¡r¤tn8t t|d t|d ¡rtt t|d
|
t|d
|
¡r¤td|jd < t ||¡ | rÚt t|d t|d ¡rÚtd S ) Nç ø?r r ç Y@r )r r
|
r Údg ÀK@r r r r ©
|
r r r r r r r r r r ©r r! r" r5 r$ r$ r% Ú(test_replace_mask_all_false_second_block_ s
|
r= c C sÞ t dddgdd}| ¡ }|jddd}| rht t|dt|d¡sJtt t|dt|d¡r¤tn<|s¤t t|dt|d¡stt t|dt|d¡r¤t| rÚd |jd
|
< t ||¡ t t|dt|d¡rÚtd S ) Nr8 r r r9 r r r r
|
ç à?©r r
|
r; )r Zusing_array_managerr! r" r5 r$ r$ r% Ú!test_replace_coerce_single_columnw s
|
r@ c C sÚ t dddgdd}| ¡ }|jddd}| rht t|dt|d¡sJtt t|d t|d ¡s tn8t t|dt|d¡rtt t|d t|d ¡r td
|
|jd < t ||¡ | rÖt t|dt|d¡rÖtd S ) Nr8 r r r9 r Zxxxr r
|
r r> r? r; r< r$ r$ r% Ú#test_replace_to_replace_wrong_dtype s
|
rA c C s¢ t ddddgidd}t|d}|jdgddd t |jt|dj¡sLt| r`|j d¡s`t| ¡ }|jdgdd }t |jt|dj¡rtt
|
||¡ d S )
|
Nr r
|
r Úcategory©ZdtypeT©r r, r ©r ) r r r r r Úcodesr r. r/ r r r r4 r$ r$ r% Ú test_replace_list_categorical s
|
rG c C s t ddddgidd}|d d
}| ¡ }|jdgddd | rnt t|djt|dj¡r`tt ||¡ n t t|djt|dj¡rtd S )Nr r
|
r rB rC TrD )
|
r r r r r r rF r r r )r r! r0 r" r$ r$ r% Ú*test_replace_list_inplace_refs_categorical¯ s
|
|
ÿ
|
|
ÿrH r8 c C sZ t ddddgi}t|d}|jdddd t t|d|¡sBt| rV|j d¡sVtd S © Nr r8 r r g /@T©r r r, r ©r r r r r r r. r/ )r r r! Úarr_ar$ r$ r% Útest_replace_inplaceÀ s
|
rM c C s t ddddgi}t|d}|d d
}|j|ddd | rtt t|d|¡rRt|j d¡sbt|j d¡stnt t|d|¡std S rI rK ©r r r! rL r0 r$ r$ r% Ú test_replace_inplace_referenceË s
|
rO r9 c C sv t ddddgi}t|d}|d d
}|j|ddd t t|d|¡sNt| rr|j d¡rbt|j d¡rrtd S rI rK rN r$ r$ r% Ú$test_replace_inplace_reference_no_opÚ s
|
rP Úvalc C s® t dtdddgi}| ¡ }t|d}|d d
}|j||dd | rt t|dj|j¡rbt|j
|
d¡srt|j
|
d¡stt ||¡ nt t|dj|j¡sªtd S ©Nr r r r TrJ r ) r r r r r r r rF r r. r/ r r )r rQ r r! r" rL r0 r$ r$ r% Ú*test_replace_categorical_inplace_referenceç s
|
rS c C s t dtdddgi}t|d}|jd|dd t t|dj|j¡sJt| r^|j d¡s^tt dt|ddgi}t
|
||¡ d S rR ) r r r r r r rF r r. r/ r r )r rQ r! rL Úexpectedr$ r$ r% Ú test_replace_categorical_inplaceù s
|
rU c C s® t dtdddgi}| ¡ }|jd|d}| rP|j d¡s@t|j d¡sPtt t |dj
|
t |dj
|
¡rptt ||¡ t |dj
|
}d|j d< t t |dj
|
|¡sªtd S ) Nr r r r r r g @)r r )r r r r r. r/ r r r r rF r r Ziloc)r rQ r! r" r5 rL r$ r$ r% Útest_replace_categorical s
|
rV ÚmethodÚwhereÚmaskc C s² t ddddgi}| ¡ }t|d}|d d
}t||}||d dkddd | rt t|d|¡rjt|j d ¡szt|j d ¡stt
|
||¡ nt t|d|¡s®td S )
|
Nr r8 r r gù?r T©r, r ) r r r Úgetattrr r r r. r/ r r )r rW r! r" rL r0 r$ r$ r% Útest_masking_inplace s
|
|
r\ c C s¾ t dddgi}| g g ¡}| rNt t|dt|d¡s<t|j d¡rjtn t t|dt|d¡rjtt|d}| g g ¡ | rºt t|d|¡st|j d¡rªt|j d¡rºtd S )Nr r r r )r r r r r r r. r/ )r r! r5 rL r$ r$ r% Útest_replace_empty_list+ s
|
r] r r: c C s t ddddgi}t|d}|jdg|dd | s:|d krRt |t|d¡shtnt |t|d¡rht| r||j d¡s|td S )Nr r
|
r TrZ r rK )r r r! r# r$ r$ r% Ú test_replace_object_list_inplace= s
|
r^ c C sv t ddddgi}t|d}|jddgddd | r\t |t|d¡rJt|j d¡srtnt |t|d¡srtd S ) Nr r r r r TrZ r rK )r r! r# r$ r$ r% Ú+test_replace_list_multiple_elements_inplaceK s
|
r_ c C sV t ddddgi}| ¡ }|jdgd d}t ||¡ t t|dt|d¡rRtd S )Nr r
|
r rE ) r r r r r r r r r r< r$ r$ r% Útest_replace_list_noneW s
|
r` c C s t ddddgi}t|d}| ¡ }|d d
}|jdgd dd | rz|j d¡sVtt |t|d¡rltt
|
||¡ nt |t|d¡std S )Nr r
|
r TrD r ) r r r r r. r/ r r r r r )r r! r# r" r0 r$ r$ r% Ú#test_replace_list_none_inplace_refsa s
|
ra )"Únumpyr ZpytestZpandasr r Zpandas._testingZ_testingr Zpandas.tests.copy_view.utilr ÚmarkZ parametrizer& r1 r6 r7 r= r@ rA rG rH rM rO rP rS rU rV r\ r] r^ r_ r` ra r$ r$ r$ r% Ú<module> sT ûþ
|
|
|
|
|
|
|
|
|
|
