U
|
ý°d±Z ã @ s d dl mZ d dlZd dlZd dlmZmZ d dl m
|
m Z d dl Zd dl mZmZmZmZmZ d dlmZ dZd ddZG d d
|
d
|
ZG d d d ZdS )é )Ú ascii_lettersN)ÚSettingWithCopyErrorÚSettingWithCopyWarning)Ú DataFrameÚSeriesÚ TimestampÚ
|
date_rangeÚoption_contextzAA value is trying to be set on a copy of a slice from a DataFrameéd c C s@ t jjtt| dfd}|jdd dd |D }t|dgdS ) Né )Úsizeé )Zaxisc S s g | ]}t |d |d
qS )r r )Úletters)Ú.0Úx© r úfd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexing/test_chaining_and_caching.pyÚ
|
<listcomp> s z random_text.<locals>.<listcomp>r ©Úcolumns)ÚnpÚrandomÚrandintÚlenr Úsortr )ZnobsZidxsÚstringsr r r Ú random_text s r c @ s> e Zd Zdd Zej dddg¡dd Zdd Zd
|
d Z d S ) Ú TestCachingc
|
C s¸ t dd ¤ tt d¡dgd d}d|d< |d t| |rft ¡ d|d jd < W 5 Q R X nd|d jd < | ¡ |st |d d d¡ nt |d d d¡ W 5 Q R X d S )
|
NÚchained_assignmenté ç@©ZaaÚbbç Úccr" gÃõ(\ÂÅ?r )
|
r r r ÚarangeÚreprÚtmÚ raises_chained_assignment_errorÚilocZ_clear_item_cacheZassert_almost_equal©ÚselfÚusing_copy_on_writeÚdfr r r Ú,test_slice_consolidate_invalidate_item_cache# s
|
z8TestCaching.test_slice_consolidate_invalidate_item_cacheÚdo_refTFc C s| dddddddg}t ||dd
|d d
t d ¡d
|
}|rJ|jd d |jd < |jd dksft|jd dksxtd S )NÚoneÚtwoÚthreeZfourZfiveÚsixZsevené é ©ÚaÚbÚc)r r9 r )r5 r9 r# g ð?)r r r% ÚlocÚAssertionError)r+ r/ Zcontr- r r r Útest_setitem_cache_updating? s *
|
|
z'TestCaching.test_setitem_cache_updatingc
|
C s t ddddgitddd}t ddddgitddd}t dddgddd gd
|
}td}td}| ¡ D ]8\}}|j||
|d f |d |j||
|d f< qjt ||¡ t |d |d ¡ t ddddgitddd}| ¡ } | ¡ D ]h\}}||d ||
|d }
|
|rDt ¡ |
|
||d ||
< W 5 Q R X qð|
|
||d ||
< qð|st ||¡ t |d |d ¡ n t || ¡ t |d | d ¡ t ddddgitddd}| ¡ D ],\}}|j||
|d f |d 7 < qÆt ||¡ t |d |d ¡ d S ) NÚAiX z5/7/2014z5/9/2014)Úindexr r
|
éÈ i, )ÚCÚDr@ rA )
|
r r r Ziterrowsr: r' Úassert_frame_equalÚassert_series_equalÚcopyr( ) r+ r, ÚexpectedÚoutr- r3 ZeixÚixÚrowZ out_originalÚvr r r Ú"test_setitem_cache_updating_slicesP s< ÿ 2
|
& z.TestCaching.test_setitem_cache_updating_slicesc C s t ddgddggddgddgd }|d }|r>d|jksLtnd|jksLtd
|
|d < t|dksdtd|jksrt|d |k stt|d dkstd S ) Nr r r4 é r7 r8 r= ÚB)r> r r r9 )r Z _item_cacher; r )r+ r, r- Úserr r r Ú(test_altering_series_clears_parent_cache| s "z4TestCaching.test_altering_series_clears_parent_cacheN)
|
Ú__name__Ú
|
__module__Ú __qualname__r. ÚpytestÚmarkÚ parametrizer< rJ rN r r r r r " s
|
|
,r c
|
@ st e Zd Zdd Zejjdd Zejjdd Zejjdd Z ejjd d
|
Z
|
ejjd d Z ejjd d Z ejjdd Z ejjdd Zejjdd Zejjdd Zejjdd Zejjdd Zejjdd Zejjd d Zejjd d Zd!d" Zd#d$ Zd%d& Zej d'd(ed)d*d+d(d,gig¡d-d. Ze j d/d0 Z d1d2 Z d3d4 Z d5d6 Z!d7d8 Z"d9S ):Ú TestChainingc
|
C s¢ dddddddg}dddddddg}t dt |¡i}|jdk}|rxt ¡ d|j|< W 5 Q R X t |t d|i¡ n d|j|< t |t d|i¡ tjj|gdgd}t |}|jdk}|rôt ¡ d|j|< W 5 Q R X t |t d|i¡ n d|j|< t |t d|i¡ t ||d}| ¡ }|jdk}|rdt ¡ d|j|< W 5 Q R X t ||¡ n d|j|< t |t ||d¡ t dtj
|
d d
|
d d gi}t dt d d d
|
d d g¡i}|rút ¡ tj
|
|d j d < W 5 Q R X t dd d d
|
d d gi}n(tj
|
|d j d < t dtj
|
d d
|
d d gi}| ¡ } t | |¡ t dt d d d
|
d d g¡i}|r|t ¡ tj
|
|j j d < W 5 Q R X ntj
|
|j j d < | ¡ } t | |¡ d S ) NÚrightÚleftÚtimeoutÚnoneÚresponse)Únames)rZ Z response1r= ÚbarZbahÚfoor )r r ÚarrayrZ r' r( rB ZrecZ
|
fromarraysrD Únanr) Úheadr= )
|
r+ r, ÚdataZmdatar- ÚmaskZrecarrayÚ df_originalrE Úresultr r r Ú test_setitem_chained_setfault sZ
|
|
|
|
|
|
|
|
|
|
|
z*TestChaining.test_setitem_chained_setfaultc
|
C sâ t ddÎ tddgddggtdd}tt d ¡ d
|
d
|
¡tdd d }| ¡ }|jd ks^t|r°t
|
¡ d|d d< W 5 Q R X t
|
¡ d|d d< W 5 Q R X t ||¡ n$d|d d< d|d d< t ||¡ W 5 Q R X d S )Nr Úraiseéûÿÿÿr éúÿÿÿr4 ÚABr rK r Úint64)r Údtyper= r ) r r Úlistr r% ÚreshaperD Ú_is_copyr; r' r( rB )r+ r, rE r- rc r r r Ú test_detect_chained_assignmentË s$ ÿ
|
|
z+TestChaining.test_detect_chained_assignmentc C sR t ttdddtjt dd¡tjdd}| ¡ }|jd ksDt |rt
|
¡ d|d d< W 5 Q R X t
|
¡ d |d d
|
< W 5 Q R X t
|
||¡ n¸|süt jttd d|d d< W 5 Q R X t jttd tj|d d
|
< W 5 Q R X |d jd ksút nRd|d d< d |d d
|
< t ddgd d ggtd d}|d d¡|d< t
|
||¡ d S )Nr rj ©rk rK ©r= rL rg r= r rh r ©Úmatchr4 ri r rL Úfloat64)r r Úranger r^ r% rt rD rn r; r' r( rB rR Úraisesr Úmsgr_ rl Zastype)r+ Úusing_array_managerr, r- rc rE r r r Ú%test_detect_chained_assignment_raisesá s0 þÿ
|
|
z2TestChaining.test_detect_chained_assignment_raisesc C s t ttdddtjt dd¡tjdd}|rVt ¡ d|j d d< W 5 Q R X n(t
|
j t t d d|j d d< W 5 Q R X d S )
|
Nr rj rp rK rq rg r r= rr )r r ru r r^ r% rt r' r( r: rR rv r rw r* r r r Ú$test_detect_chained_assignment_fails s þÿ
|
z1TestChaining.test_detect_chained_assignment_failsc C s¢ t dddddddgttdddd}|jd ks6t|rj|jj d ¡}t ¡ d
|
|| d < W 5 Q R X n4t
|
j t t d |jj d ¡}d
|
|| d < W 5 Q R X d S ) Nr0 r1 r2 r3 r5 rj rp )r7 r9 Úoé* r9 rr )r r ru rn r; r7 ÚstrÚ
|
startswithr' r( rR rv r rw ©r+ r, r- Úindexerr r r Ú*test_detect_chained_assignment_doc_example s þÿ
|
z7TestChaining.test_detect_chained_assignment_doc_examplec C sø t dddgdddgd}t dddgdddgd}| ¡ }|shtjttd d|jd
|
d < W 5 Q R X |rt ¡ d|d d
|
< W 5 Q R X t ||¡ nZ|sÜtjttd d|d d
|
< W 5 Q R X d|jd < t ||¡ nd|d d
|
< t ||¡ d S ) Néo ÚbbbÚcccr r r4 rq Úaaarr r r= )r r= )
|
r rD rR rv r rw r: r' r( rB )r+ rx r, rE r- rc r r r Ú+test_detect_chained_assignment_object_dtype+ s"
|
|
z8TestChaining.test_detect_chained_assignment_object_dtypec C sd t dddgi}|jd ks tt d¡2}| |¡ t |¡}|d |d< |d |d< W 5 Q R X d S )Nr= r r Z __tmp__picklerL )r rn r; r' Z ensure_cleanZ to_pickleÚpdZ read_pickle)r+ r- ÚpathÚdf2r r r Ú-test_detect_chained_assignment_is_copy_pickleG s
|
|
z:TestChaining.test_detect_chained_assignment_is_copy_picklec C s t d}|jdddg }|jd k s&t|jddddg }|jd k sFt|j dd ¡}|j| ¡ }|jd ksrt|d tj ¡|d< d S ) Né r r r rK c S s t | dkS ©Né
|
©r ©r r r r Ú<lambda>a ó zSTestChaining.test_detect_chained_assignment_setting_entire_column.<locals>.<lambda>r )
|
r r) rn r; r Úapplyr: rD r} Úlower)r+ r- r r r r r Ú4test_detect_chained_assignment_setting_entire_columnS s zATestChaining.test_detect_chained_assignment_setting_entire_columnc C sH t d}|j dd ¡}|j| }|jd k s0t|d tj¡|d< d S )Nr c S s t | dkS r r r r r r r k r zKTestChaining.test_detect_chained_assignment_implicit_take.<locals>.<lambda>r )r r r r: rn r; r} r ©r+ r- r r r r Ú,test_detect_chained_assignment_implicit_takeg s
|
|
z9TestChaining.test_detect_chained_assignment_implicit_takec C s |rt d¡ td}|j dd ¡}|j| }|jd k s>t|d tj ¡|jd d
df< |jd k sjt|d tj ¡|d< |jd kstd S )Nz"_is_copy is not always set for CoWr c S s t | dkS r r r r r r r w r zLTestChaining.test_detect_chained_assignment_implicit_take2.<locals>.<lambda>r )
|
rR Úskipr r r r: rn r; r} r r r r r Ú-test_detect_chained_assignment_implicit_take2q s
|
|
z:TestChaining.test_detect_chained_assignment_implicit_take2c C s< t d}|j dd ¡}|j|df tj¡|j|df< d S )Nr c S s t | dkS r r r r r r r r zATestChaining.test_detect_chained_assignment_str.<locals>.<lambda>r )r r r r: r} r r r r r Ú"test_detect_chained_assignment_str s z/TestChaining.test_detect_chained_assignment_strc C s4 t ddgi ¡ }|jd ks t|d d7 < d S )Nr7 r )r Zdropnarn r; ©r+ r- r r r Ú&test_detect_chained_assignment_is_copy s z3TestChaining.test_detect_chained_assignment_is_copyc C s^ t tj dd¡}|jd d
df ¡ }t ||jd d
df ¡ ¡ t ||d ¡ ¡ d S )Nr rK r )r r r Úrandnr) Z sort_valuesr' rC )r+ r- rM r r r Ú&test_detect_chained_assignment_sorting s z3TestChaining.test_detect_chained_assignment_sortingc C sn t dddgdddgd}t| |d d |d< t| ||d dk }t| |d d |d< t| d S )
|
Nr7 rK é é )Úcolumn1Úcolumn2r r8 r¡ r9 )r r} r r r r Ú.test_detect_chained_assignment_false_positives s z;TestChaining.test_detect_chained_assignment_false_positivesc C s t t dd¡dgd}d|d< | ¡ }|r^t ¡ d|jdd
d< W 5 Q R X t ||¡ n,tj t
|
t d d|jdd
d< W 5 Q R X d S )
|
Nr r Úcountr r8 Úgroupr7 r rr ) r r r% rD r' r( r) rB rR rv r rw )r+ r, r- rc r r r Ú/test_detect_chained_assignment_undefined_column¨ s
|
z<TestChaining.test_detect_chained_assignment_undefined_columnc C sZ t tdddtj d¡tjddddddd d
|
gd }| ¡ }|r²t ¡ d |j d d< W 5 Q R X t ¡ d |j d d< W 5 Q R X t ¡ d |d d < W 5 Q R X t
|
||¡ |sVt j t td d |j d d< W 5 Q R X t j t td d |j d d< W 5 Q R X |s6t j t td d |d d < W 5 Q R X n d |d d < |j d d ksVtd S )NZ20130101r )Zperiodsrj rp r7 r8 r9 ÚdÚe)r= rL r@ rA r] r rA r@ rr )r r@ )r r r r r r% rD r' r( r: rB rR rv r rw r; )r+ rx r, r- rc r r r Ú-test_detect_chained_assignment_changing_dtype¸ s4
|
|
üÿ
|
|
|
z:TestChaining.test_detect_chained_assignment_changing_dtypec C s® t ttdtdddtjdgd}| ¡ }t |j¡}|rxt
|
¡ |dg | |dg |< W 5 Q R X t ||¡ n2t j ttd |dg | |dg |< W 5 Q R X d S ) NrK zab..r7 r8 r¦ r6 r9 rr )r rl ru r r_ rD r Zisnar9 r' r( rB rR rv r rw )r+ r, r- rc rb r r r Útest_setting_with_copy_bugà s ÿ
|
"z'TestChaining.test_setting_with_copy_bugc C s> t tdddgtdddgd}|dg }d d
|
d g|d < d S ) Nr7 r8 r9 r¦ r§ Úf)r Úyr ÚgÚhÚir« )r r )r+ Zdf1r r r r Ú%test_setting_with_copy_bug_no_warningð s "
|
z2TestChaining.test_setting_with_copy_bug_no_warningc
|
C sÀ t dddgdddgd}|rDt ¡ d|jd d
|
< W 5 Q R X d S td d * t t¡ d|jd d
|
< W 5 Q R X W 5 Q R X td d . tjt t
|
d d|jd d
|
< W 5 Q R X W 5 Q R X d S )Nr
r r r r r4 rq r r r= r Úwarnrf rr ) r r' r( r: r Úassert_produces_warningr rR rv r rw r* r r r Ú.test_detect_chained_assignment_warnings_errorsù s
|
" z;TestChaining.test_detect_chained_assignment_warnings_errorsÚrhsr4 r r r rK c
|
C s¦ t t d¡ dd¡}| ¡ }|jd d
}tddf |snt t ¡ }||d< |d j
|
t ksbt W 5 Q R X n*t d ¡ ||d< W 5 Q R X t ||¡ W 5 Q R X d S )Né r r4 r r° r r )r r r% rm rD r: r r' r± r ÚfilenameÚ__file__r; rB )r+ r³ r, r- rc ZchainedÚtr r r Ú1test_detect_chained_assignment_warning_stacklevel s z>TestChaining.test_detect_chained_assignment_warning_stacklevelc C s t dt d¡g dt d¡g d}|d jd }|jd }t ||¡ |jd d }t ||¡ |d jd }t ||¡ |d jd }t ||¡ d S )Nr r4 rq r= r )r r= )r r ÚzerosZonesr) r: r' Zassert_numpy_array_equal)r+ r- rE rd Zresult2Zresult3Zresult4r r r Ú test_chained_getitem_with_lists s &
|
z,TestChaining.test_chained_getitem_with_listsc C sH t ¡ }|d |jd |jd< d|d jks2td|d jksDtd S )Nr= r z Hello FriendrL )r' Z makeDataFramer) r: r> r; r r r r Útest_cache_updating- s
|
z TestChaining.test_cache_updatingc C sî t tjddddddddgtd d
|
}d |d < | ¡ }|rptjtd d d|jj d< W 5 Q R X t
|
||¡ d S d|jj d< d|jj d< t tjddddddddd gtd d
|
}d|j d< t
|
||¡ t d d d dd gd d}t
|
|j|¡ d S )N)r r rj rp r7 r8 r9 r¦ r§ r )r r> r rª z read-onlyrr r r4 r )r é )r4 rª ©Úname)r r r¹ ru rD rR rv Ú
|
ValueErrorrª Úvaluesr' rB Úatr rC )r+ r, r- Zdf_origrE r r r Útest_cache_updating26 s. ý ý
|
z!TestChaining.test_cache_updating2c
|
C sð t dd Ü ttddgd d}d|d< dgt| }|rbt ¡ d|d jd
|
< W 5 Q R X nd|d jd
|
< |j| |r¢t ¡ d |d jd
|
< W 5 Q R X nd |d jd
|
< |sÌ|d jd
|
d ksâtn|d jd
|
dksâtW 5 Q R X d S ) Nr r r r! r# r$ Tg¤p=
|
×£À?r" r g333333Ã?)r r ru r r' r( r) r; )r+ r, r- Zckr r r Ú$test_iloc_setitem_chained_assignmentR s
|
|
|
z1TestChaining.test_iloc_setitem_chained_assignmentc C s t ddddgi}|r:t ¡ d|d jd< W 5 Q R X nd|d jd< t |t ddddgi¡ t |d tdddgdd¡ d S )Nr7 r é é é( rK r½ )r r' r( r: rB rC r r* r r r Ú'test_getitem_loc_assignment_slice_staten s
|
z4TestChaining.test_getitem_loc_assignment_slice_stateN)#rO rP rQ re rR rS Zarm_slowro ry rz r r r r r r r r r r¢ r¥ r¨ r© r¯ r² rT r r¸ ÚtdZ&skip_array_manager_not_yet_implementedrº r» r rà rÇ r r r r rU sT ;
|
|
$
|
|
|
|
|
|
|
|
|
|
|
|
|
'
|
|
rU )r
|
)Ústringr r Únumpyr rR Z pandas.errorsr r Z pandas.util._test_decoratorsÚutilZ_test_decoratorsrÈ Zpandasr r r r r r Zpandas._testingZ_testingr' rw r r rU r r r r Ú<module> s
|
m
|
