U
|
ý°dÄ ã @ s
|
d Z ddlZddlmZ ddlZddlZddlZddlZddlm Z ddl
|
m Z m Z m Z ddlZddlmZmZmZmZmZmZmZ ddlmZ ddlmZ ddl m Z G d d
|
d
|
Z G d d d Z G d d dZ G dd dZ!dd Z"dd Z#ej$ %dedgdfedddge&dd d fg¡d d Z'd d! Z(ej$ %d"d ddge ddg¡e d#ddg¡g¡d$d% Z)ej$ %d"d ddge ddg¡e d#ddg¡g¡d&d' Z*ej$ %d"d(d)ge d)g¡e d*d)g¡g¡d+d, Z+ej$ %d"d(d)ge d)g¡e d*d)g¡g¡d-d. Z,dS )/z test fancy indexing & misc é N)Údatetime)Ú IndexingError)Úis_float_dtypeÚis_integer_dtypeÚis_object_dtype)Ú DataFrameÚIndexÚNaTÚSeriesÚ
|
date_rangeÚoffsetsÚtimedelta_range)Ú_mklbl)Úgen_objc @ s< e Zd ZdZdd Zdd Zdd Zdd Zd
|
d Zd d Z dd Z
|
e j dddg¡dd Zdd Zdd Zdd Zdd Zd d Ze j d d d!d"ged#g¡d$d% Zd&d' Zd(d) Zd*d+ Zd,d- Zd.d/ Zd0d1 Zd2d3 Z d4d5 Z d6d7 Z d8d9 Z d:d; Z d<d= Z!d>d? Z"d@dA Z#e j dBe$j%e$j&g¡dCdD Z'dES )FÚ TestFancyz"pure get/set item & fancy indexingc C sò t tt dd¡tjdd}tjdtjd|d< tjdtd|d< d}tj t
|
|d , t d
|
d d d g¡|j |j dd
df< W 5 Q R X t d
|
d d d g¡|j |j dd
df< |j |j dd
df }td
|
d d d gddddgdd}t ||¡ d S )Né é ©Údtype©Úindexé
|
ÚfooÚbarú@Must have equal len keys and value when setting with an iterable©Úmatchy ¤p=
|
×£@y®Gáz®ó?¹?g@ç ð?é é é é é ©r Úname)r r ÚnpÚarangeÚint64ÚzerosÚfloat64ÚcomplexÚpytestÚraisesÚ
|
ValueErrorÚarrayÚlocr r
|
ÚtmÚassert_series_equal)ÚselfÚdfÚmsgÚresultÚexpected© r7 úZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexing/test_indexing.pyÚtest_setitem_ndarray_1d( s 0&
|
|
ÿz!TestFancy.test_setitem_ndarray_1dc C sv t tt dd¡d}tjdtjd|d< tjdtd|d< d}tjt |d t dd
|
¡d |d d
< W 5 Q R X d S )Nr r r r r r r r r r" y ð?r r )
|
r r r% r& r( r) r* r+ r, r- )r2 r3 r4 r7 r7 r8 Útest_setitem_ndarray_1d_2> s z#TestFancy.test_setitem_ndarray_1d_2c C s t ||}||}tjjddd}g }|tkrV|tjtjfkrV| d¡ |rV| d¡ |tksh|tjkr| d¡ |r| d¡ |tj ks|tkr¦|tjkr¦| d¡ |t
|
krÂ|tjkrÂ| d ¡ t |t j râ|tjkrâ| d ¡ t |t jt jt jfr| d
|
¡ t|d ks t |t jr*| d ¡ t|tkr\t |jtjs\| d ¡ | d¡ d |¡} tttf}
|
tj |
|
| d || W 5 Q R X d S )Nr ©r r r ©Úsizez8Wrong number of dimensions. values.ndim > ndim \[3 > 1\]z$Passed array should be 1-dimensionalú;Buffer has wrong number of dimensions \(expected 1, got 3\)z indexer should be 1-dimensionalz&Cannot index with multidimensional keyú Index data must be 1-dimensionalúData must be 1-dimensionalr z%positional indexers are out-of-boundszvalues must be a 1D arrayz only handle 1-dimensional arraysú|r ) r r% ÚrandomÚrandintr
|
r0 ÚsetitemÚilocÚappendr/ r Ú
|
isinstanceÚpdZ IntervalIndexZTimedeltaIndexZ DatetimeIndexZ PeriodIndexÚlenZ
|
MultiIndexÚtyper Ú_valuesÚndarrayÚjoinÚ
|
IndexErrorr- ÚNotImplementedErrorr+ r, ) r2 r Úframe_or_seriesÚ indexer_sliZusing_array_managerÚobjÚidxrÚnd3Zmsgsr4 Zpotential_errorsr7 r7 r8 Útest_getitem_ndarray_3dJ sB
|
|
|
|
|
|
ÿÿ
|
|
|
|
|
|
|
|
|
z!TestFancy.test_getitem_ndarray_3dc C s| t ||}||}tjjddd}|tjkr>t}d|j }nt}d dddd d
|
g¡}t j
|
||d d ||< W 5 Q R X d S ) Nr r; r< z Cannot set values with ndim > rA r> z Cannot set values with ndim > 1r? r@ z,Array conditional must be same shape as selfr r ) r r% rB rC r0 rE r- ÚndimrM r+ r, ) r2 r rP rQ rR rS rT Úerrr4 r7 r7 r8 Útest_setitem_ndarray_3dt s"
|
|
ûÿ
|
z!TestFancy.test_setitem_ndarray_3dc C sf t d¡}tddgddgg}|| }tddgdd}t ||¡ tddg}|| }|dksbtd S )Nr r r r! r" ©r$ )r% r. r r
|
r0 r1 ÚAssertionError)r2 Úkeyr3 r5 r6 Úserr7 r7 r8 Útest_getitem_ndarray_0d s
|
z!TestFancy.test_getitem_ndarray_0dc C sp t dgd}d|jd< d|jd< d|jtj< |jtjdf dksDt|j}tddtjgtjd}t ||¡ d S )Nr ©Úcolumnsr r r! r )
|
r r/ r% ÚinfrZ r r r) r0 Úassert_index_equal©r2 r3 r5 r6 r7 r7 r8 Útest_inf_upcast s
|
|
zTestFancy.test_inf_upcastc C sn t ddidddg}tj|d< |d jtjks4td|jd< t dtjdd ddtjd g}t ||¡ d S )
|
NÚar r! r ©rd ÚbÚcr )r rg )rd rf rg ) r r% Únanr r) rZ r/ r0 Úassert_frame_equal©r2 r3 r6 r7 r7 r8 Útest_setitem_dtype_upcast s
|
|
ÿz#TestFancy.test_setitem_dtype_upcastÚvalg
ëQ¸ @Úwxyzc C s t tjddd dd¡tdddd gd
|
}| ¡ }||jd < t d |dgdd dggtdddd gd
|
}t ||¡ t |d st
|
t |d st
|
d S )Nr r' r r r! Úabr r Úbaz©r r_ ©rd r r r" r ) r r% r& ÚreshapeÚlistÚcopyr/ r0 ri r rZ )r2 rl r3 ÚleftÚrightr7 r7 r8 Útest_setitem_dtype_upcast2¹ s ý
|
ý z$TestFancy.test_setitem_dtype_upcast2c C s t tjddd dd¡d tddd d
|
gd }d |jd < t dd dgdddggtddd d
|
gd }t ||¡ t|d s|t t|d
|
st d S )Nr r' r r r! g $@rn r r ro rp rm rq r gÉ?g333333Ó?gÙ?g à?)
|
r r% r& rr rs r/ r0 ri r rZ )r2 ru rv r7 r7 r8 Útest_setitem_dtype_upcast3Î s ý
|
ý z$TestFancy.test_setitem_dtype_upcast3c C sD t dd¡}dddg|_|ddg j}tdddg}t ||¡ d S )Nr r! rd rf )r0 ZmakeCustomDataframer_ r ra rb r7 r7 r8 Útest_dups_fancy_indexingà s
|
z"TestFancy.test_dups_fancy_indexingc C s t dddddddggtdd }| ¡ t| t dddddddgg}td|_|jd d
d
|
f |jd d
d
|
f t ||¡ d S ) Nr r r ç @ç @r r Zaaaaaaar^ r" )r rs ÚheadÚstrr_ rE r0 ri ©r2 r3 r5 r7 r7 r8 Ú&test_dups_fancy_indexing_across_dtypesé s
|
z0TestFancy.test_dups_fancy_indexing_across_dtypesc C sî t ddddgddddgtddd d d
|
d gd }d d
|
g}t ddgd dgddgd|d }|j| }t ||¡ |jt| }t ||¡ d d
|
dg}tjtdd |j| W 5 Q R X dddd d
|
dg}tjtdd |j| W 5 Q R X d S )Nr é é r ç @r Zabcd)ÚtestZtest1ÚotherÚAÚBÚCr g @Údrg ÚEú not in indexr ÚFÚGÚH) r rs r/ r0 ri r r+ r, ÚKeyError)r2 r3 Úrowsr6 r5 r7 r7 r8 Ú%test_dups_fancy_indexing_not_in_order÷ s&
|
þ ÿ
|
|
z/TestFancy.test_dups_fancy_indexing_not_in_orderc C sJ t tj dd¡tdd}tjtt d¡d |j
|
dg W 5 Q R X d S )Nr r! ZAABCDr z;"None of [Index(['E'], dtype='object')] are in the [index]"r r ) r r% rB Úrandnrs r+ r, r ÚreÚescaper/ )r2 Zdfnur7 r7 r8 Ú+test_dups_fancy_indexing_only_missing_label s ÿþz5TestFancy.test_dups_fancy_indexing_only_missing_labelÚvalsr r r Úabcc C s: t d|i}tjtdd |jdddg W 5 Q R X d S )Nr
r r r é ©r r+ r, r r/ )r2 r r3 r7 r7 r8 Ú&test_dups_fancy_indexing_missing_label s z0TestFancy.test_dups_fancy_indexing_missing_labelc C sN t dddddgiddddgd }tjtd
|
d |jddd g W 5 Q R X d S ) Nr r r r r r
r r r r r r r ©r2 r3 r7 r7 r8 Ú#test_dups_fancy_indexing_non_unique" s z-TestFancy.test_dups_fancy_indexing_non_uniquec C sV t tj dd¡dddddgd}tjtdd |jd d
dddgf W 5 Q R X d S )Nr r
r r^ r r r )r r% rB r r+ r, r r/ r r7 r7 r8 Útest_dups_fancy_indexing2( s z#TestFancy.test_dups_fancy_indexing2c C s¸ t tj dd¡dddddddddg ddgd}|jdd
}|jddg }t ||¡ |}|jd d
ddgf }t ||¡ |jdd
d d
f }|jddgddgf }t ||¡ d S )
|
Nr r r r! rd rf rp r r )r r% rB r rE r/ r0 ri )r2 r3 r6 r5 r7 r7 r8 Útest_dups_fancy_indexing30 s ÿ z#TestFancy.test_dups_fancy_indexing3c C sH t tddddgd}t tdddgd}||dg }t ||¡ d S )Nr! r r r )r
|
Úranger0 r1 )r2 Ú
|
indexer_slr\ r6 r5 r7 r7 r8 Útest_duplicate_int_indexingB s z%TestFancy.test_duplicate_int_indexingc C s| t ddddddddd}|d d
|
d ¡|d < |d d k}|j|d f dd ¡}||j|d f< |jd dksxtd S )NÚaaaÚbbbÚccc)r r r! éo éÞ iM re rd c S s | dkr dS | S )Nr¡ Ú_r7 ©Úxr7 r7 r8 Ú<lambda>P ó z9TestFancy.test_indexing_mixed_frame_bug.<locals>.<lambda>r r¦ c S s | dkr dS | S )Nr¡ ú-----r7 r§ r7 r7 r8 r© T rª )r r r« )r Úapplyr/ rE rZ )r2 r3 ÚidxÚtempr7 r7 r8 Ú test_indexing_mixed_frame_bugI s ÿ z'TestFancy.test_indexing_mixed_frame_bugc C sp t tj d¡dgddddg d}tjtt d¡d |dd
|
d g W 5 Q R X |d jd |jd kslt d S ) N)r r rd é é é é r^ z'[26, -8] not in index'r é iøÿÿÿr )
|
r r% rB r+ r, r r r ÚshaperZ r r7 r7 r8 Ú test_multitype_list_index_accessX s "z*TestFancy.test_multitype_list_index_accessc C sÐ t ddddddddddd d
|
d dd d d d tjtjtjtjdtjtjtjtjtjtjd dddddddddddddd d d d d d d d d d d d d d d d!d d" ¡ }| d#d$d%g¡ ¡ j|jd&}t ||¡ d S )'NZnonQCÚ10Z11Z12Z13Z24Z35Z46Z47Z48Z59)é é é r° r± r² r³ é é r´ é é é é ç r g¨oÓeÑ @gm7Á7M @g,¹Åo @g2 ¥ë?gàà@í ì?gufÌn÷ê?g·µ
ç%é?g æT×Faç?g§ãÖI"ê?g+ew7ê?gy7»ZÓé?gÌ9ê?g¸Gp#¥é?g¼Cåºê?iÖ i× iØ iÁ i )ÚPRuidÚQCÚdataÚyearrÅ r rà r^ ) r r% rh Z reset_indexZ set_indexZreindexr_ r0 ri r~ r7 r7 r8 Útest_set_index_nan` s òòòòÏÿFþÿz TestFancy.test_set_index_nanc
|
C s^ t ddddddgddddddgttdttddd}tj|jd< | ¡ }|j ¡ }d d
|
g}|d }tj|jd < t dtjddddgddddddgt ddd dddgdddtjddgd}|j
|
||f |j
|
||f< t ||¡ t dtjddddgddddddgddd dddgdddtjddgd}| ¡ }|j
|
||f j |j
|
||f< t ||¡ d S )Nrd rf r r r é )ÚFCZPFÚcol1Úcol2©r r rÉ rÊ r )r! r! r" r r r é r° r² )r rs r r% rh rE rt rÈ Zisnar
|
r/ r0 ri Úvalues)r2 r3 Údf2ÚmaskÚcolsZdftr6 r7 r7 r8 Útest_multi_assign® s@
|
üÿ üÿ
|
üÿzTestFancy.test_multi_assignc C s t dddddgdddddgdddddgdddd d
|
gd }| ¡ }|d dk}d D ]}|d |j||f< qR|d |j|d dkd dgf< t ||¡ d S )Nr r r r r r! r" r r r )r
r r ÚDr
©r
r rÒ r )r rt r/ r0 ri )r2 r3 r6 rÏ Úcolr7 r7 r8 Ú"test_multi_assign_broadcasting_rhsÝ s üÿ z,TestFancy.test_multi_assign_broadcasting_rhsc C s` t ddgdgd}dddg|jd< ddg|jd< t ddgdgd}ddg|jd< t ||¡ d S )Nr r rp r r! rË )r rE r0 ri r~ r7 r7 r8 Útest_setitem_listð s zTestFancy.test_setitem_listc C sr t dgtt d¡gtd}|jjs(ttj t
|
dd |d W 5 Q R X tj t
|
dd |j d W 5 Q R X d S )Nr z
|
2011-01-01r ú'2011'r Ú2011©rØ r ) r r rH Ú TimestampÚobjectr Ú _is_all_datesrZ r+ r, r r/ r r7 r7 r8 Útest_string_sliceü s zTestFancy.test_string_slicec C s\ t }|jjrttjtdd |d W 5 Q R X tjtdd |jd W 5 Q R X d S )Nr× r rØ z^0$rÙ )r r rÜ rZ r+ r, r r/ r r7 r7 r8 Útest_string_slice_empty s z!TestFancy.test_string_slice_emptyc C s¢ t dddddddggtdd }| ¡ }|jd d
d
|
d
f tj¡|jd d
d
|
d
f< t d d dddddggtdd }|d t¡|d < |d t¡|d< t ||¡ | ¡ }|j
|
d d
d f tj¡|j
|
d d
d f< t d ddddddggtdd }|d t¡|d < t ||¡ | ¡ }|j
|
d d
ddgf tj¡|j
|
d d
ddgf< t dd dddddggtdd }|d t¡|d< |d t¡|d< t ||¡ d S )NÚ1Ú2Ú3z.4r g @r ZABCDEFGr^ r r r r
r r r! ) r rs rt rE Úastyper% r' rÛ r0 ri r/ )r2 Zdf_origr3 r6 r7 r7 r8 Útest_astype_assignment s< ÿ0 ÿ ( ÿ 0 ÿz TestFancy.test_astype_assignmentc C s t dddddgi}|d tj¡|jd d
df< t dddddgi}t ||¡ t dddddgi}|d tj¡|jd d
df< t ||¡ d S )Nr
r rz r{ r r )r râ r% r' rE r0 ri r/ rj r7 r7 r8 Ú(test_astype_assignment_full_replacements7 s z2TestFancy.test_astype_assignment_full_replacementsÚindexerc C s t tdt tdtdddfD ]®}t|js4t| ¡ }d||d< t|jsVt||d dksjt| ¡ }d||d< |j}d|kr t|j ¡ dg }t
|
|j|¡ | ¡ }d||d< t |js"tq"t tdt d ¡dfD ]}t|jsüt| ¡ }d||d< t|js t||d dks6t| ¡ }d||d< t
|
|j|j¡ | ¡ }d||d< t |jsêtqêd S )
|
Nr r r r r g¹?rÁ Ú0g @)r
|
r r r rZ rt r r Útolistr0 ra r r% r& )r2 rå ÚsÚs2Úexpr7 r7 r8 Útest_index_type_coercionE s6 & z"TestFancy.test_index_type_coercionN)(Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__r9 r: rU rX r] rc rk r+ ÚmarkÚ parametrizerw rx ry r r r rs r r r r r r¯ r¶ rÆ rÑ rÕ rÖ rÝ rÞ rã rä r0 Úgetitemr/ rë r7 r7 r7 r8 r % sB *
|
|
N/ $r c
|
@ s® e Zd Zdd Zdd Zdd Zdd Zej d e
|
d
|
d e d ¡d e d d d ¡g¡dd Zdd Zdd Zdd Zej ddddg¡dd Zd d Zd d Zd d! Zd"S )#ÚTestMiscc C sT t tj d¡tj d¡d}d|d< t |d |d dgd d}t ||¡ d S )Nr )rÁ r rd rÁ r )rÁ r rd )r r% rB Zrandr0 ri rj r7 r7 r8 Útest_float_index_to_mixeds s z"TestMisc.test_float_index_to_mixedc C sh t dddgdddgdddd gd
|
}d|j|jd d
< t dddgdddgd|jd
|
}t ||¡ d S © Nr r r! r" r re r rz r{ r )r r/ r r0 ri rj r7 r7 r8 Ú&test_float_index_non_scalar_assignmentz s $ z/TestMisc.test_float_index_non_scalar_assignmentc C sP t dddgdddgdddd gd
|
}| ¡ }|j|j |j|j< t ||¡ d S rõ )r rt r/ r r0 ri ©r2 r3 rÎ r7 r7 r8 Ú test_loc_setitem_fullindex_views s $z)TestMisc.test_loc_setitem_fullindex_viewsc C sF dd }t d¡ dd¡}dddd g}t||td
|
d d }d |jddd
ddd
f }| ¡ }dddddg|d< dddddg|d< |jdd
dd
f d 9 < | ¡ }|jdd
dd
f d 9 < ||||| ||||fD ],}|d d¡|d< |d dd ¡|d< qâd d d d d!g|d< d"d#d$d%d&g|d< ||||| d S )'Nc S s¨ t ddddgtdd }}}ddgddgtdd }}} | ¡ }
|
||
|
j||f< t |
|
|¡ | ¡ }
|
||
|
j||f< t |
|
|¡ | ¡ }
|
||
|
j|| f< t |
|
|¡ d S )NZbcdr r r! r" ÚjoeÚjolie)rs Úslicert r/ r0 ri rE ) r3 ÚrhsÚ right_locÚ
|
right_ilocZlbl_oneZidx_oneZ slice_oneZlbl_twoZidx_twoZ slice_tworu r7 r7 r8 Ú run_tests s z.TestMisc.test_rhs_alignment.<locals>.run_testsr° r r" Zjimrù rú ZjolineZabcder' )r_ r r éþÿÿÿr! r éÿÿÿÿr r é r r r¸ é r r¹ r) c S s
|
d| S )Nú@r7 r§ r7 r7 r8 r© ¯ rª z-TestMisc.test_rhs_alignment.<locals>.<lambda>r z@-28z@-20z@-12g 1@z@2g :Àg 2Àg $Àz@18) r% r& rr r rs rE rt râ Úmap) r2 rÿ ÚxsrÐ r3 rü rþ rý Úframer7 r7 r8 Útest_rhs_alignment s$ zTestMisc.test_rhs_alignmentr r
r° éd é c C s t j}t|}tt d¡|d}t |||d d d
|dd d
¡ t ||d |d d
|d dd
¡ t |||d |d d
|ddd
¡ t |||d |d d
|d d
¡ d S )Nr° r r r r r r )rH Z
|
IndexSlicer r
|
r% r& r0 Z!assert_indexing_slices_equivalent)r2 r ZSLCr\ r7 r7 r8 Ú)test_str_label_slicing_with_negative_step´ s && ÿz2TestMisc.test_str_label_slicing_with_negative_stepc C sF |t t|¡|d}tjtdd ||d d d
W 5 Q R X d S )Nr zslice step cannot be zeror r )r% r& rI r+ r, r- )r2 r r rP rR r7 r7 r8 Ú test_slice_with_zero_step_raisesà s z)TestMisc.test_slice_with_zero_step_raisesc C s¼ t dddgdd}tdddgddd gd
|
|d }| ¡ }d d d
|
}||jd< d d g|jd< t ||¡ tdddgdddgd
|
|d }||jd< tddd gdddgd
|
|d }t ||¡ d S )Néûÿÿÿr r ÚzrY r r r r )r¨ Úyr r éc rz g @g ÀX@)r r rt r/ r0 ri )r2 r r3 r6 rü r7 r7 r8 Ú8test_loc_setitem_indexing_assignment_dict_already_existsÈ s
|
|
|
zATestMisc.test_loc_setitem_indexing_assignment_dict_already_existsc C sv t dddgdddgd}|jg d d
f }|jd d
df jtjksJtt |jd d
df |jd d
d f ¡ d S )
|
Nr r r! rf Zb2Zb3re rd r ) r rE r/ r r% r' rZ r0 r1 r÷ r7 r7 r8 Ú*test_iloc_getitem_indexing_dtypes_on_empty× s z3TestMisc.test_iloc_getitem_indexing_dtypes_on_emptyr= r i?B i@B c C st t t|tjd}d|jtd< t |jtd t ddgd¡ d|jtd< t |jtd t d ddgd¡ d S )
|
N)r r é* r g E@r r é+ r g E@)r
|
r r% r) r/ r0 r1 )r2 r= rè r7 r7 r8 Ú!test_loc_range_in_series_indexingß s
|
z*TestMisc.test_loc_range_in_series_indexingc C s t t d¡ dd¡tdtdd}t dtdtdd}|| ¡ }t t d d
|
tjgd d tjgtjgd g¡tdtdd}t ||¡ d S ) Ng "@r! r ÚABCrp r rn ZABrÁ r r{ r )
|
r r% r& rr rs Znotnullr. rh r0 ri )r2 r3 Zindex_dfr5 r6 r7 r7 r8 Ú#test_partial_boolean_frame_indexingê s ÿ &ýz,TestMisc.test_partial_boolean_frame_indexingc C sH t ddgddgd}dD ]}t|| qt |¡}~| d ksDtd S )Nr r r r! re )r/ rE ÚatZiat)r ÚgetattrÚweakrefÚrefrZ )r2 r3 r$ Úwrr7 r7 r8 Útest_no_reference_cycleø s
|
z TestMisc.test_no_reference_cyclec C sN t ddd|g}|jdd}|j| }|| }d}||ks>t||ksJtd S )Nr z{1,2}F)Zdropna)r
|
Z value_countsr/ rZ )r2 Z nulls_fixturer3 ZvcZresult1Zresult2r6 r7 r7 r8 Útest_label_indexing_on_nan s
|
z#TestMisc.test_label_indexing_on_nanN)rì rí rî rô rö rø r r+ rð rñ r r% r& Zlinspacer r r r r r r r r7 r7 r7 r8 ró r s" . "ÿ
|
|
|
ró c @ sÎ e Zd Zdddgejddgfdddgejddgfedddedddedddgeedddedddgfdd d
|
gd d d
|
gfgZej
|
d e¡d d Z ej
|
d e¡dd Z ej
|
d e¡dd Z dd Zd S )ÚTestDataframeNoneCoercionr r r! r rz r{ éÐ r r ro Nr6 c C s@ |\}}t d|i}d |jddgf< t d|i}t ||¡ d S ©Nr r ©r r/ r0 ri ©r2 r6 Z
|
start_dataZexpected_resultÚstart_dataframeZexpected_dataframer7 r7 r8 Útest_coercion_with_loc s
|
z0TestDataframeNoneCoercion.test_coercion_with_locc C sH |\}}t d|i}d ||d |d d k< t d|i}t ||¡ d S r! )r r0 ri r# r7 r7 r8 Ú(test_coercion_with_setitem_and_dataframe$ s
|
zBTestDataframeNoneCoercion.test_coercion_with_setitem_and_dataframec C sJ |\}}t d|i}d |j|d |d d k< t d|i}t ||¡ d S r! r" r# r7 r7 r8 Ú$test_none_coercion_loc_and_dataframe. s
|
z>TestDataframeNoneCoercion.test_none_coercion_loc_and_dataframec C s t dddgdddgtdddtdddtdddgdd d
|
gd }d |jd < t tjddgtjddgttdddtdddgd d d
|
gd }t ||¡ d S ) Nr r r! r rz r{ r rd rf rg )rd rf rg r r )r r rE r% rh r r0 ri )r2 r$ rê r7 r7 r8 Ú test_none_coercion_mixed_dtypes8 s üÿ
|
|
|
üÿz9TestDataframeNoneCoercion.test_none_coercion_mixed_dtypes)rì rí rî r% rh r r ZEXPECTED_SINGLE_ROW_RESULTSr+ rð rñ r% r& r' r( r7 r7 r7 r8 r s þö
|
|
|
r c @ sÔ e Zd Zdd Zej deej e
|
j e
|
j e g¡ej dddge dde ddd g¡g¡d
|
d Zej d d e d¡g¡dd Zej deej e
|
j e
|
j e g¡ej dddge dde ddd g¡g¡dd ZdS )ÚTestDatetimelikeCoercionc C s |}t dd|d}t|jdd}|j}d}| |¡ |||d< |d krn|j|jksZt|jj|jks|tn|j|ks|td S )Nú
|
2016-01-01r! ©ÚperiodsÚtzT©Údeepz
|
2018-01-01r )r r
|
rt rK Ú_validate_setitem_valuer rZ Ú_ndarray)r2 Útz_naive_fixturerQ r- Údtir\ rÍ Znewvalr7 r7 r8 Ú test_setitem_dt64_string_scalarO s
|
z8TestDatetimelikeCoercion.test_setitem_dt64_string_scalarÚboxr[ r r r TFc
|
C s¦ |}t |tr"|tjkr"tdd}tdd|d}t|jdd}|j}|dd g} | | ¡ | |||< |d kr|j |j kst
|
|jj |j ks¢t
|
n|j|ks¢t
|
d S )
|
Nr r r* r! r+ Tr. z
|
2019-01-01z
|
2010-01-02) rG rû r0 r/ r r
|
rt rK r0 r rZ r1 )
|
r2 r2 rQ r[ r5 r- r3 r\ rÍ Únewvalsr7 r7 r8 Ú test_setitem_dt64_string_valuesd s
|
|
z8TestDatetimelikeCoercion.test_setitem_dt64_string_valuesÚscalarz3 Daysr" c C sN t ddd}t|jdd}|j}| |¡ |||d< |jj|jksJtd S )Nú1 Dayr! ©r, Tr. r )r r
|
rt rK r0 r1 rZ )r2 rQ r8 Útdir\ rÍ r7 r7 r8 Útest_setitem_td64_scalar s
|
z1TestDatetimelikeCoercion.test_setitem_td64_scalarc C sx t |tr |tjkr tdd}tddd}t|jdd}|j}|dd g}| |¡ ||||< |jj |j kstt
|
d S )
|
Nr r r9 r! r: Tr. z10 Daysz44 hours) rG rû r0 r/ r r
|
rt rK r0 r1 rZ )r2 rQ r[ r5 r; r\ rÍ r6 r7 r7 r8 Ú test_setitem_td64_string_values s
|
|
z8TestDatetimelikeCoercion.test_setitem_td64_string_valuesN)rì rí rî r4 r+ rð rñ rs r% r. rH Z Categoricalr rû r7 r ZHourr< r= r7 r7 r7 r8 r) N s ÿ
|
ÿr) c C s t tjddgddtjddgddddd gd
|
} ttjddgddd d gdd }| jd }t ||¡ | jd }t ||¡ d S )Nr r ÚInt64r r! r" rÓ rd rf r r
r r# r )r rH r. r
|
r/ r0 r1 rE )r3 r6 r5 r7 r7 r8 Ú"test_extension_array_cross_section¡ s þû
|
|
r? c C sì t tjddgddtjddgdddddgd } | jd }tddgdd
|
d gdd }t ||¡ | jd }t ||¡ t tjddgddt ddg¡dddgd } | jd }tddgt d
|
d gdd }t ||¡ | jd }t ||¡ d S )Nr r r> r r' rÓ rd rf r r
r )r r r$ r )
|
r rH r. r% r/ r
|
r0 r1 rE rÛ )r3 r5 r6 r7 r7 r8 Ú+test_extension_array_cross_section_converts² s( þû
|
|
þ
|
|
r@ z ser, keysr )r r r r r! r r )r r c C sn t d }}tj||d || | W 5 Q R X |tjkrDtd }}tj||d d|| |< W 5 Q R X d S )NzToo many indexersr ztoo many indices for arrayr )r r+ r, r0 rE rN )r\ ÚkeysÚ
|
indexer_liZexp_errZexp_msgr7 r7 r8 Ú'test_ser_tup_indexer_exceeds_dimensionsÏ s
|
|
|
rC c C sB t dg}| |ddg }t ddgtddgd}t ||¡ d S )Nr r r )r
|
r r0 r1 )rB r\ Úresrê r7 r7 r8 Ú(test_ser_list_indexer_exceeds_dimensionsá s
|
rE Úvaluerf c C s t ddddgi}d ddg¡}tjt|d | |jd < W 5 Q R X t dddgtjdd
|
d gtd d }tjtdd | |jd < W 5 Q R X d S )Nr
r r r! rA z"Must have equal len keys and valueú(setting an array element with a sequencer ©r r rd rf r rÓ ) r rM r+ r, r- r/ r% r. rÛ )rF r3 r4 r7 r7 r8 Ú%test_scalar_setitem_with_nested_valueë s þÿ$rI c C s t dddg}tjtdd | ||d< W 5 Q R X t dddgtd }| ||d< t| tjrv|jd | k ¡ st
|
n|jd | kst
|
d S )
|
Nr r r! rG r r rd rf r ) r
|
r+ r, r- rÛ rG r% rL r/ ÚallrZ )rF rQ r\ r7 r7 r8 Ú,test_scalar_setitem_series_with_nested_value s rK )rÁ rÁ r c C s® t ddddgi}| |jd< t dddgdtjtjgd}t ||¡ t dddgtjddd gtd
|
d}| |jd< t| tj r|jd | k
|
¡ sªt n|jd | ksªt d S ) Nr
r r r! rH rÁ rÓ rd rf r ) r r/ r% rh r0 ri r. rÛ rG rL rJ rZ )rF r3 r6 r7 r7 r8 Ú-test_scalar_setitem_with_nested_value_length1 s
|
$
|
rL c C sÀ t dddg}t| tjrB| ||d< t dddg}t ||¡ n&tjtdd | ||d< W 5 Q R X t dd d
|
gt d }| ||d< t| tjrª|j
|
d | k ¡ s¼t n|j
|
d | ks¼t d S ) Nr rz r{ r rÁ rG r r rd rf r ) r
|
rG r% rL r0 r1 r+ r, r- rÛ r/ rJ rZ )rF rQ r\ r6 r7 r7 r8 Ú4test_scalar_setitem_series_with_nested_value_length1, s ÿ rM )-rï r. r r r Únumpyr% r+ Z pandas.errorsr Zpandas.core.dtypes.commonr r r ZpandasrH r r r r
|
r r r Zpandas._testingZ_testingr0 Z pandas.tests.indexing.commonr Z!pandas.tests.indexing.test_floatsr r ró r r) r? r@ rð rñ rs rC rE rI rK rL rM r7 r7 r7 r8 Ú<module> sd $ Q BS &þ
|
|
$ÿ
|
$ÿ
|
ÿ
|
ÿ
|
