U
|
ý°dr6 ã @ sr d dl Zd dlmZ d dlZd dlZd dlZd dlZ d dlm
|
Z
|
m Z m Z m Z mZmZ d dlmZ G dd dZdS )é N)Ú combinations)Ú DataFrameÚIndexÚSeriesÚ TimestampÚconcatÚisnac @ sÊ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z e d¡edddgedddgeed e d ¡ ¡e jd ¡ d d e j d!d"d#d$g¡e e d%d&dd'd'¡e d%d&ddd$¡e d%d&dd!d(¡g¡e j d ¡ d ¡ g¡g Zej j d)ed*d+ d,d-d. Z ej j d/e ed0d1d2d+ d,d3d4 Z d5d6 Z!d7d8 Z"d9d: Z#d;d< Z$ej d=d>d?d@dAg¡ej dBd&dCg¡dDdE Z%ej d=d>d?d@dAg¡ej dBd&dCg¡dFdG Z&dHS )IÚ
|
TestAppendc C s* | ¡ }d|d< |jd d
}|jdd
}| |¡}| |¡}| |¡}t |d |d ¡ |d= |j||d} d| ks~t|j||d} d| kst|d d
|dd
¡}t ||¡ |d d
j|dd
|d}
|
|d d
j|dd
|d} t |
|
jddddgd | jddddgd ¡ d S )
|
NÚbarÚfooé ÚA©ÚsortÚBÚCÚD©Úcolumns)ÚcopyÚindexZreindexÚ_appendÚtmZassert_almost_equalÚAssertionErrorÚassert_frame_equal) Úselfr Ú float_frameZ mixed_frameZ begin_indexZ end_indexZ begin_frameZ end_frameÚappendedZpartial_appendedZmixed_appendedZmixed_appended2© r ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/reshape/concat/test_append.pyÚ test_append s*
|
|
|
þzTestAppend.test_appendc C sN t }| |¡}t ||¡ ||k s(t| |¡}t ||¡ ||k sJtd S ©N)r r r r r )r r Úemptyr r r r Útest_append_empty7 s
|
|
z TestAppend.test_append_emptyc C s0 d}t jt|d |j|dd W 5 Q R X d S )Nz Indexes have overlapping values)ÚmatchT)Zverify_integrity)ÚpytestZraisesÚ
|
ValueErrorr )r r Úmsgr r r Útest_append_overlap_raisesB s z%TestAppend.test_append_overlap_raisesc C sr t ddddddd}tddd gd
|
d d gd d}t ddddddddd d id}| |¡}t ||¡ d S )Né é )ÚxÚyé é ©ÚaÚbr é é r0 r1 ÚcÚz©r Úname)r+ r, r5 )r0 r1 r4 )r r r r r )r ÚdfÚrowÚexpectedÚresultr r r Útest_append_new_columnsG s
|
|
ýÿ
|
z"TestAppend.test_append_new_columnsc C sX t dddgd}t ddgddgd}|j||d}t ddgdddgd}t ||¡ d S ) Nr r r r r r) ©r r r ©r r r r )r r r8 Zdf3Zdf5r: r r r Útest_append_length0_frameU s
|
z$TestAppend.test_append_length0_framec C s~ t jddd}ddg|d d
< t jddd}ddd g|d d
< t|}t|}|j|d
|
d }tt ||f¡}t ||¡ d S ) N)r* z i4,f4,a10©Údtype)r) ç @ZHello)r* ç @ZWorld)r- )r- g @r )r g @r
|
)g @g @ZbazT©Ú ignore_index)ÚnpÚzerosr r Z concatenater r )r Zarr1Zarr2Údf1Údf2r; r: r r r Útest_append_records] s z TestAppend.test_append_recordsc C s¢ t ddgddgdddgd}t ddgddgd ddgd
|
}|j||d }t ddd d gddddgd d ddgd ddd gd}|dkr|ddd g }t ||¡ d S )Nr) r* r/ r1 r0 r r- r. )r0 r4 )r r )r1 r0 r4 r4 Fr> )r r rH rI r; r: r r r Útest_append_sortsl s "þz TestAppend.test_append_sortsc C sÄ t tj d¡dktj ddd¡tj d¡ddgd d}|d d
jd d
ddd gf }|dd
jd d
d
|
dd gf }|j||d }t|d
|
dd
¡ s¤t t|d dd
¡ sÀt d S ) Né
|
r r r
|
r )ÚboolsÚintsÚfloatsÚstringsrM rN rO rP r r. )
|
r rF ÚrandomZrandnÚrandintÚlocr r Úallr )r r r8 r0 r1 r r r r Ú test_append_different_columns{ s
|
|
üÿ z(TestAppend.test_append_different_columnsc C sÜ |d d
|dd
|dd
|dd
g}|d |dd
¡}t ||¡ |d ¡ |d< d|d d< |d j |dd
|d }t |jd d
|jf |¡ |d dd
dk ¡ s¼t|d d d
¡ ¡ sØtd S )
|
Nr rL é r r) éÿÿÿÿr
|
r r ) r r r r rS r rT r r )r r r Úchunksr; r r r Útest_append_many s
|
|
|
|
ü zTestAppend.test_append_manyc C sp t dddgd}| dg¡}t dddgdd d
|
gd d d ggdddgd}| dg¡}| |¡}|jjdksltd S )Nr r r r r) r. r3 r* r é r- r2 é )Údatar )r Z set_indexr r r7 r )r rH rI r; r r r Ú test_append_preserve_index_name s *
|
z*TestAppend.test_append_preserve_index_namer- r. r r2 g @g @g @ÚabczA B CzD E FT)Zorderedr3 rZ r[ rL éÝ r) r é r c C s
|
t | jS r! ©ÚtypeÚ__name__©r+ r r r Ú<lambda>» ó zTestAppend.<lambda>)Úidsc C s0 t dddgdddgg|d}|d d
}tdd g|dd
|
}| |¡}t ddd gdddgdd tjggd ddg|d }|jjd jdkst|jjd jdkstt
|
||¡ |}|d d
}t ddgddgg|d}tdd dg|dd
|
}| |¡}t ddtjgddtjgdd dggd ddg|d }t
|
||¡ d S )Nr) r* r- r. r r2 r r3 rZ r6 rC r r= Úir[ ) r r r rF ÚnanZdtypesÚilocÚkindr r r )r r r8 Z ser_indexÚserr; r: r r r Ú test_append_same_columns_typeº s.
|
ÿ
|
ýz(TestAppend.test_append_same_columns_typezdf_columns, series_indexr* )Úrc C s
|
t | jS r! ra rd r r r re Þ rf c C s¸ t dddgdddgg|d}tdd d
|
g|dd }| |¡}|j |¡}t| ¡ |¡}t d d dtj tj tj gdddtj tj tj gtj tj tj dd d
|
ggdddg|d}t
|
||¡ d S )Nr) r* r- r. r r2 r r3 rZ r[ r6 ç ð?rB rC r r= ) r r r r Ú
|
differencer ÚtolistÚappendrF ri r r ) r Z
|
df_columnsZ series_indexr8 rl r; Zidx_diffZcombined_columnsr: r r r Ú#test_append_different_columns_typesÛ s
|
|
ýù z.TestAppend.test_append_different_columns_typesc C sX t ddgt ddddd¡t ddddd¡gdgd}t ddgt ddd dd¡t ddd d
|
d ¡gt ddddd¡t dddd d ¡ggdd gd}tttjtjt ddd d
|
d ¡t dddd d ¡gd dtt ddddd¡t ddddd¡t ddd dd¡t ddddd¡gddgd|d}|j|d|d}|r<|d dg }n |dd g }t ||¡ d S )Nr) r* r_ r Ú
|
start_time)r r\ r r. r r- r2 rL r3 Zend_time)r7 )Zaxisr T©rE r )
|
r ÚdtÚdatetimer r ÚpdÚNaTr r r )r r rH rI r: r; r r r Útest_append_dtype_coerceö sL "ý""þú üù üùöê z#TestAppend.test_append_dtype_coercec C st t dtjddddgddi}t dtjd d
|
d d
|
gtdi}|j|d |d }|d jd ks^t|d jd ksptd S )Nr r) r* r- r. Úi8r@ r TFru Zf8ÚO)r rF ÚarrayÚboolr rA r )r r rH rI r r r r Ú(test_append_missing_column_proper_upcast( s
|
z3TestAppend.test_append_missing_column_proper_upcastc C s t dtj ¡ d}tdd|d}tddgd}|j|d d
|
}ttjtjdd|ggddd d d gd}|d t
|
¡|d< |d t
|
¡|d< t ||¡ ttjtjdd|ggd ddd d d gd}|d t
|
¡|d< |d t
|
¡|d< |j|d d
|
}t ||¡ |j||gd d
|
}t ||¡ d S )Nz2018-10-24 07:30:00)Útzro rB )r0 r1 Údater4 Údr TrD r0 r1 r r* ) r Údateutilr Ztzutcr r r rF ri ÚastypeÚobjectr r )r r rl r8 Zresult_ar: Zresult_br; r r r Ú2test_append_empty_frame_to_series_with_dateutil_tz0 s* ÿ ÿ z=TestAppend.test_append_empty_frame_to_series_with_dateutil_tzc C sâ t dgd d¡}|jdtjidd}t dtjgi t¡}t ||¡ t dgd d¡}tdtjidd}|j|dd}t dtjgi t¡}t ||¡ tdtjidd}|j|dd}t dtjgi t¡}t ||¡ d S ) Nr0 r údatetime64[ns, UTC]TrD údatetime64[ns]r@ zdatetime64[ns, US/Pacific]) r r r rx ry r
r r r )r r8 r; r: Úotherr r r Ú,test_append_empty_tz_frame_with_datetime64nsI s z7TestAppend.test_append_empty_tz_frame_with_datetime64nsÚ dtype_strr r ZInt64Úint64Úvalry c C sP t dgd |¡}t dt |d¡gi}|j|dd}| t¡}t ||¡ d S )Nr0 r ÚnsTrD )r r rF Ú timedelta64r r
r r ©r r r r8 r r; r: r r r Ú.test_append_empty_frame_with_timedelta64ns_nat_ s
|
|
z9TestAppend.test_append_empty_frame_with_timedelta64ns_natc C sl t dtjdg|di}t dt |d¡gi}|j|dd}t d|jd |jd gitd}t ||¡ d S )Nr0 r) r@ r TrD )r r )
|
r rx r} rF r r rj r
r r r r r r Ú(test_append_frame_with_timedelta64ns_natm s
|
z3TestAppend.test_append_frame_with_timedelta64ns_natN)'rc Ú
|
__module__Ú __qualname__r r# r( r< r? rJ rK rU rY r] rx Z
|
RangeIndexr ÚlistZCategoricalIndexÚsplitZ IntervalIndexZ from_breaksZ DatetimeIndexrv rw Z
|
MultiIndexZ from_arraysZindexes_can_appendr% ÚmarkZ parametrizerm r rs rz r r r r r r r r r r sj "
|
ýÿñ ÿ
|
|
ý
|
2
|
ÿ
|
|
ÿr )rw rv Ú itertoolsr r ÚnumpyrF r% Zpandasrx r r r r r r Zpandas._testingZ_testingr r r r r r Ú<module> s
|
