U
|
ý°d|_ ã @ sz d Z ddlZddlZddlZddlmZmZmZm Z m
|
Z
|
m Z m Z ddl mZ G dd dZG dd dZG dd d ZdS )
|
zx
|
test setting *parts* of objects both positionally and label based
|
|
TODO: these should be split among the indexer tests
|
é N)Ú DataFrameÚIndexÚPeriodÚSeriesÚ TimestampÚ
|
date_rangeÚ period_rangec @ s| e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd Zdd Zd S ) Ú TestEmptyFrameSetitemExpansionc C sd t i tjdddd}tdtjdddd}||d< t ddgd itjdddd}t ||¡ d S ) Nr Zdf_index©Úname©Úindexç®Gáz®ó?é Ú series_indexÚseries)r ÚpdÚ
|
RangeIndexr ÚtmÚassert_frame_equal©ÚselfÚdfr Úexpected© r úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexing/test_partial.pyÚ,test_empty_frame_setitem_index_name_retained s ÿzKTestEmptyFrameSetitemExpansion.test_empty_frame_setitem_index_name_retainedc C sT t }tdtjdddd}||d< t ddgd itjdddd}t ||¡ d S )Nr r r r
|
r r )r r r r r r r r r r Ú-test_empty_frame_setitem_index_name_inherited$ s ÿzLTestEmptyFrameSetitemExpansion.test_empty_frame_setitem_index_name_inheritedc C s t ddgd}tdtdd|jd< t ddgdgtjd}t ||¡ t ddgd}tddgd|jd< t tjdggddgdgd d}t ||¡ d S )
|
NÚAÚB©Úcolumnsé r r r )r! r ÚdtypeÚfloat64) r r ÚrangeÚlocÚnpr$ r r Únan)r r r Úexpr r r Ú-test_loc_setitem_zerolen_series_columns_align. s zLTestEmptyFrameSetitemExpansion.test_loc_setitem_zerolen_series_columns_alignc C s t ddgd}d}tjt|d dddg|jd < W 5 Q R X t ddgd}d
|
d g|jd< t d
|
d ggdgddgtjd }t ||¡ d S ) Nr r r z(cannot set a row with mismatched columns©Úmatchr" é é r é é )r r! r# ) r ÚpytestÚraisesÚ
|
ValueErrorr& r' Úint64r r )r r Úmsgr) r r r Ú7test_loc_setitem_zerolen_list_length_must_match_columns< s zVTestEmptyFrameSetitemExpansion.test_loc_setitem_zerolen_list_length_must_match_columnsc C s t }d}tjt|d d|jd< W 5 Q R X tjt|d tdgdgd|jd< W 5 Q R X d}tjt|d d|jd d
df< W 5 Q R X d S )Nz*cannot set a frame with no defined columnsr+ r" Úfoor z5cannot set a frame with no defined index and a scalar)r r1 r2 r3 r& r )r r r5 r r r Ú test_partial_set_empty_frameJ s z;TestEmptyFrameSetitemExpansion.test_partial_set_empty_framec C s t dgtg ddd}t tg ddd}tg dd|d< t ||¡ t tg d}t|j|d< t ||¡ t tg d}|j|d< t ||¡ d S )Nr7 Úobject©r# ©r! r r )r r r r r r ©r r r r r r Ú test_partial_set_empty_frame2[ s
|
z<TestEmptyFrameSetitemExpansion.test_partial_set_empty_frame2c C s t dgtg ddd}|d d¡|d< t tg ddd}g |d< t ||¡ t tg ddd}tt t|¡dd|d< t ||¡ d S )Nr7 r4 r: r; r$ r ) r r Úastyper r r r' ÚarangeÚlenr< r r r Ú test_partial_set_empty_frame3p s z<TestEmptyFrameSetitemExpansion.test_partial_set_empty_frame3c C sZ t tg ddd}tt||d< t dgtg ddd}|d d¡|d< t ||¡ d S )Nr4 r: r r7 r; )r r r% r@ r> r r ©r r r r r r Ú test_partial_set_empty_frame4~ s
|
z<TestEmptyFrameSetitemExpansion.test_partial_set_empty_frame4c C s t }t |jt d¡¡ t }tdgdgd|d< tdgdgd|jd d
df< t |t dggdgdgd¡ t ||¡ d S )Nr r" r7 r ©r r! ) r r Úassert_index_equalr! r r r r& r )r r Zdf2r r r Ú test_partial_set_empty_frame5 s z<TestEmptyFrameSetitemExpansion.test_partial_set_empty_frame5c C s¨ t dtdtddidddgd}t ddgd}tdtdd|d< |j t| t ||¡ t ddgd}tdtdd|jd d
df< |j t| t ||¡ d S )Nr r" r r r r r )r r r% ZdtypesÚstrr r r& r< r r r Ú%test_partial_set_empty_frame_no_index s " zDTestEmptyFrameSetitemExpansion.test_partial_set_empty_frame_no_indexc C s~ t dddgtg ddd}|d d¡|d< |d d¡|d< |d d¡|d< t dd d
|
gd d d gd}||jdk }tj|d< t ||¡ t ddddgd}|d d¡|d< t dddgd}d
|
|d< t ||¡ t |d t dt
|
d¡ t dd d
|
gd d d gd}||jdk }|j dddgd}t dddgd}|d d¡|d< |d d¡|d< |d d¡|d< t ||¡ d S )Nr r ZNewr4 r: r; r$ r" r- r. g333333ó?gÍÌÌÌÌÌ@gÍÌÌÌÌÌ@©r r é ÚaÚbzc cÚdr ©r r# ÚC) r r r> r r' r( r r Úassert_series_equalr r9 Úreindex)r r r ÚyÚresultr r r Ú test_partial_set_empty_frame_row s,
|
z?TestEmptyFrameSetitemExpansion.test_partial_set_empty_frame_rowc C s^ t ttd}t dttdi}t ||¡ t tdtd}t dttdi}t ||¡ d S )Nr: r r7 rN )r r r9 r r rB r r r Ú'test_partial_set_empty_frame_set_series¾ s zFTestEmptyFrameSetitemExpansion.test_partial_set_empty_frame_set_seriesc C s> t dgd}| ¡ }d|d< t ddgdgd}t ||¡ d S )Nr r rK rD )r Úcopyr r rB r r r Ú2test_partial_set_empty_frame_empty_copy_assignmentÉ s
|
zQTestEmptyFrameSetitemExpansion.test_partial_set_empty_frame_empty_copy_assignmentc C sÈ t ddgd}ddg|d< t ddgtjtjgd}tj||dd t ddgd}d d
|
g|d< t d d
|
gtjtjgdtd }t ||¡ t ddgd}d|jd < t dgtjgd}tj||dd d S ) NÚxrR r r" r- )rX rR F)Z check_dtypeÚ1Ú2r: )r rX )r r' r( r r r9 r& rB r r r Ú0test_partial_set_empty_frame_empty_consistenciesÒ s
|
zOTestEmptyFrameSetitemExpansion.test_partial_set_empty_frame_empty_consistenciesN)Ú__name__Ú
|
__module__Ú __qualname__r r r* r6 r8 r= rA rC rF rH rT rU rW r[ r r r r r s
|
r c @ s® e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Ze j
|
d ddg¡dd Z dd Z e j
|
dedddddddgeddd eddd eddd gfedddddddgedededgfejd dd d d d!ge d"¡e d#¡e d$¡gfg¡d%d& Ze j
|
d'edddddd(gfedddddd(gfejd dd d)d*gfg¡d+d, Ze j
|
d-eddddd d gd.feddddd d gd.fejd dd ddgd/fg¡d0d1 Zd2S )3ÚTestPartialSettingc C sN t dddg}| ¡ }d|d< t ddddgddddgd}t ||¡ | ¡ }d|jd< t ddddgddddgd}t ||¡ | ¡ }d|d< t ddddgddddgd}t ||¡ | ¡ }d|jd< t ddddgddddgd}t ||¡ | ¡ }d}tjt|d d|jd< W 5 Q R X d
|
}tjt|d d|j d< W 5 Q R X d S ) Nr" r- r. rJ r r ç @ú%iloc cannot enlarge its target objectr+ z/index 3 is out of bounds for axis 0 with size 3)
|
r rV r rP r& r1 r2 Ú
|
IndexErrorÚilocÚiat)r Zs_origÚsr r5 r r r Útest_partial_settingæ s0
|
|
z'TestPartialSetting.test_partial_settingc C s t t d¡ dd¡ddgdd}| ¡ }d}tjt|d d
|
|jd < W 5 Q R X d }|r\d }tjt|d d
|
|j d < W 5 Q R X t t
|
dddgdddgd}| ¡ }|jd |jd< t ||¡ t t
|
dddgdddgd}| ¡ }|j d |j d< t ||¡ t t
|
ddddgddddgd}| ¡ }|j d |j d< t ||¡ t t
|
dddgdddgd}| ¡ }|j d d
df |j d d
df< t ||¡ t t
|
dddgtdddgd}| ¡ }|d tj¡|d< |j d d
df |j d d
df< t ||¡ | ¡ }|d |d< | ¡ }|j d d
df |j d d
df< t ||¡ | ¡ }|d |d< | ¡ }|j d d
df |j d d
df< t ||¡ d S )Nr/ r. r- r r r4 )r! r# ra r+ r` )r r- z/index 2 is out of bounds for axis 0 with size 2zlist index out of ranger r r" rJ rI g g @g @rO )r r' r? ZreshaperV r1 r2 rb rc rd Údictr r r& r r> r$ )r Zusing_array_managerÚdf_origr r5 r r r r Útest_partial_setting_frame sZ ÿ " " z-TestPartialSetting.test_partial_setting_framec C s< t ddd}ttj dd¡|ddddgd }tj|tdd
|
i|d d
|j d gd d}| ¡ }d
|
|j |d |j df< t
|
||¡ | ¡ }d
|
|j |d |j df< t
|
||¡ tdd
|
i|d d
|j d }tj||gdd}| ¡ }d
|
|j |d |j df< t
|
||¡ | ¡ }d
|
|j |d |j df< t
|
||¡ d S )Nz1/1/2000é )Úperiodsr r r rO ÚDrD r0 éÿÿÿÿr T)Úsortr r" ©Zaxis) r r r' ÚrandomZrandnr ÚconcatÚfreqrV r& r r Úat)r Údatesrh r r Z exp_otherr r r Útest_partial_setting2K s0
|
ÿ ÿ z(TestPartialSetting.test_partial_setting2c C sh t ddgddggddgd}|jd ¡ }d|_t |t |j ¡ g¡}|jd |jd< t ||¡ d S )NTr" Fr- ZfemaleZfitnessr )
|
r r& rV r r rq ÚTÚ infer_objectsr r )r r re r r r r Ú test_partial_setting_mixed_dtypef s z3TestPartialSetting.test_partial_setting_mixed_dtypec C sò t ddgddgd}t tjdtjgdddgd}tjtdd |jdddg W 5 Q R X | dddg¡}tj ||d d
|
t tjdtjtjgdddd gd}tjtdd |jdddd g W 5 Q R X | dddd g¡}tj ||d d
|
t dddgdddgd}|jdddg }tj ||d d
|
t ddtjdgddd dgd}tjtdd |jddd dg W 5 Q R X | ddd dg¡}tj ||d d
|
d t
|
¡ j d }tjt|d |jdddg W 5 Q R X t ddtjgdddgd}tjtdd |jdddg W 5 Q R X | dddg¡}tj ||d d
|
t dddgdddgd}t dtjtjgdddgd}tjtdd |jdddg W 5 Q R X | dddg¡}tj ||d d
|
t ddddgddddgd}t tjddgdddgd}tjtdd |jdddg W 5 Q R X | dddg¡}tj ||d d
|
t ddddgddddgd}t tjddgdddgd}tjtdd |jdddg W 5 Q R X | dddg¡}tj ||d d
|
t ddddgddddgd}t dtjtjgdddgd}tjtdd |jdddg W 5 Q R X | dddg¡}tj ||d d
|
t ddddgddddgd}t dtjtjgdddgd}tjtdd |jdddg W 5 Q R X | dddg¡}tj ||d d
|
t ddddgddddgd}|j ddddg }tj ||d d
|
d S )Nç¹?çÉ?r" r- r r. ú not in indexr+ T©Zcheck_index_typerX ú'\"None of \[Index\(\[3, 3, 3\], dtype='z '\)\] are in the \[index\]\"ç333333Ó?r çÙ?rJ r/ r0 r ) r r' r( r1 r2 ÚKeyErrorr& rQ r rP Úint_r# rc )r Úserr rS r5 re r r r Útest_series_partial_setr sz " ÿ z*TestPartialSetting.test_series_partial_setc C s< t ddgddd}tddg|dd }tjtd
|
d |jd dd g W 5 Q R X tjtd d |jd dd dg W 5 Q R X t dddgddd}tdddg|dd }|jdddg }tj||dd tjtdd |jddddg W 5 Q R X dt ¡ j
|
d}tjt|d |jd d d g W 5 Q R X tjtd d |jddd g W 5 Q R X t ddd gddd}tjtd d & tdddg|dd jd ddg W 5 Q R X t ddd dgddd}tjtd d ( tddddg|dd jdd d g W 5 Q R X t ddd dgddd}tjtd d ( tddddg|dd jdddg W 5 Q R X t ddddgddd}tjtd d ( tddddg|dd jdddg W 5 Q R X t ddd dgddd}tjtd d ( tddddg|dd jdddg W 5 Q R X t ddddgddd}tddddg|dd }|j ddddg }tj||dd d S )Nr" r- r4 Úidx©r# r ry rz re )r r z\[3\] not in indexr+ r. r{ rX Tr| z\['x'\] not in indexr} z(', name='idx'\)\] are in the \[index\]\"r~ r r rJ r/ r0 r ) r r r1 r2 r r& r rP r' r r# rc )r r r Zexp_idxr rS r5 r r r Ú!test_series_partial_set_with_nameÍ sJ ÿ*,,,,z4TestPartialSetting.test_series_partial_set_with_nameÚkeyéd g Y@c C s t ¡ }| ¡ }|jd |j|d d
f< tt|j|g t|jj d}t
|
j |j |jdg j gdd}t |||jd}t ||¡ d S )Nr r
ro rD )r ÚmakeTimeDataFramerV rc r& r Úlistr r9 r r' Z concatenateÚvaluesr r! r )r r Úorigr Zex_indexZex_datar r r r Ú6test_setitem_with_expansion_numeric_into_datetimeindex s zITestPartialSetting.test_setitem_with_expansion_numeric_into_datetimeindexc C s t ¡ }| ¡ }|jd |jdd d
f< t|jd dd}t |t|j
|
¡ g¡}t ||¡ t |j t|j ¡ dg ¡ |j jdkstd S )Nr rK r
|
r9 )r r rV rc r& r r rq r rv rw r rE r r Útolistr# ÚAssertionError)r r r r r) r r r Útest_partial_set_invalid s z+TestPartialSetting.test_partial_set_invalidzidx,labels,expected_idxÚ2000é rl )Ústartrk rr z
|
2000-01-04z
|
2000-01-08z
|
2000-01-12)rr z1 day)r rk Z4DZ8DZ12Dz4 dayz8 dayz12 dayc C sR |t d|d}dddg}|||}t ||j| ¡ |tkrNt ||| ¡ d S )Nr r r. r0 é )r% r Z assert_equalr& r rP )r r ÚlabelsZ expected_idxZframe_or_seriesÚobjZexpected_valuer r r r Ú4test_loc_with_list_of_strings_representing_datetimes# s
|
|
zGTestPartialSetting.test_loc_with_list_of_strings_representing_datetimesz
|
idx,labelsz
|
2000-01-30z3 dayz30 dayc C s t td|d}ttd|d}d}tjt|d |j| W 5 Q R X tjt|d || W 5 Q R X tjt|d |j| W 5 Q R X d S )Nr r r{ r+ ©r r% r r1 r2 r r& )r r r r r r5 r r r ÚBtest_loc_with_list_of_strings_representing_datetimes_missing_valueL s zUTestPartialSetting.test_loc_with_list_of_strings_representing_datetimes_missing_valuezidx,labels,msgzHNone of \[Index\(\['4D', '8D'\], dtype='object'\)\] are in the \[index\]zXNone of \[Index\(\['2000-01-04', '2000-01-08'\], dtype='object'\)\] are in the \[index\]c C s t td|d}ttd|d}tjt|d |j| W 5 Q R X tjt|d || W 5 Q R X tjt|d |j| W 5 Q R X d S )Nr r r+ r )r r r r5 r r r r r ÚEtest_loc_with_list_of_strings_representing_datetimes_not_matched_typei s !zXTestPartialSetting.test_loc_with_list_of_strings_representing_datetimes_not_matched_typeN)r\ r] r^ rf ri ru rx r r r1 ÚmarkZ parametrizer r r r r r r Ztimedelta_rangeZ Timedeltar r r r r r r r_ å sz %@ [:
|
|
|
|
ýý
|
ýý
|
ýíþ
|
þ þ÷þ
|
ü ü üïþ r_ c @ s e Zd Zdd ZdS )ÚTestStringSlicingc C sf t dddd g¡}ttt||d}ttt|d d
|d d
d}|dd
}t ||¡ d S )Nz
|
2012-01-01z
|
2012-01-02z
|
2012-01-03r r. z
|
2012-01-04)r Ú to_datetimer r% r@ r r )r r r r rS r r r Ú,test_slice_irregular_datetime_index_with_nan s
|
$ z>TestStringSlicing.test_slice_irregular_datetime_index_with_nanN)r\ r] r^ r r r r r r s r )Ú__doc__Únumpyr' r1 Zpandasr r r r r r r r Zpandas._testingZ_testingr r r_ r r r r r Ú<module> s $ O 3
|
