U
|
ý°dÓK ã @ s d dl m Z d dlZd dlZd dlmZ d dlmZ d dlZ d dlm
|
Z
|
m Z m Z m Z d dlmZ G dd dZej de d d
|
d
|
e d d
|
d fe d d ¡e dd ¡fe d¡e d¡fe jd d
|
de jd dddfg¡dd Zdd Zdd Zdd Zd d Zd d Z d!d" Z dS )#é )ÚdatetimeN)Úfind_common_type)Úis_dtype_equal)Ú DataFrameÚIndexÚ
|
MultiIndexÚSeriesc @ sè e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Ze j
|
d e ddde ddde dddge je je jge ddde ddde dddgfe je je jge ddde ddde dddge ddde ddde dddgfe ddde je jge ddde ddde dddge ddde ddde dddgfe ddde ddde dddge ddde je jge ddde ddde dddgff¡dd Zdd Zdd Zdd Zdd Zd d Ze j
|
d dd g¡d d! Zd"d# Zd$S )%ÚTestDataFrameCombineFirstc C s° t ddgtdd}t tdtdd}t||d}t ddgtddd}t tdtddd}t||d}ttdd d
|
d d
|
gdd d
|
dd gd}| |¡}t ||¡ d S ) NÚaÚbé ©Úindex©ÚAÚBé é Zababr é é )r Úranger ÚlistÚ combine_firstÚtmÚassert_frame_equal)Úselfr
|
r ÚfÚgÚexpÚcombined© r údd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_combine_first.pyÚtest_combine_first_mixed s &
|
z2TestDataFrameCombineFirst.test_combine_first_mixedc C sd |d d
|dd
}}| |¡}| |j¡}t ||¡ t |j|j¡sPtt |d |d ¡ | ¡ }d|d< |d= | ¡ }d|d< |d= | |¡}|d dk
|
¡ s®tt |d |d ¡ t |d |d ¡ t |d |d ¡ |d d
¡ | }}d|d< | |¡}|d d d
dk
|
¡ s0td|j d d
|j d¡f< | |¡}|d d d
dk
|
¡ srt|d d
}|dd
} | | ¡}t |d |j¡|d ¡ t |d | j¡| d ¡ | t ¡}
|
t |
|
|¡ t |¡}
|
t |
|
|¡ | t d d
|
gd ¡}
|
d |
|
jks tt d dgitd ddgd } t dgd} | | ¡} d| ks`td S )Nr r r ÚCr r ÚDé
|
ZfazZboor r
|
iÜ r ©Úcolumns)r Zreindexr r r Z equalContentsr' ÚAssertionErrorÚassert_series_equalÚcopyÚallÚilocZget_locr r )r Z float_frameÚheadÚtailr Zreordered_frameZfcopyZfcopy2r r ÚcombÚdfÚdf2Úresultr r r! Útest_combine_first! sN
|
|
|
|
|
|
z,TestDataFrameCombineFirst.test_combine_firstc C sØ t ddddg}tddddg|d }tddddg|d }td
|
d d d g|d }t|||d}t ddddg}tddddg|d }tddddg|d }td
|
d d d g|d }t|||d}| |¡}t|jd ksÔtd S )Nr
|
r ÚcÚeç @g "Àç @g Y@r é é r éa )Zcol0Úcol2Zcol3r )Zcol1r; Zcol5)r r r r Úlenr' r( )r ÚidxZser1Zser2Zser3Zframe1Zframe2r r r r! Ú test_combine_first_mixed_bug^ s
|
z6TestDataFrameCombineFirst.test_combine_first_mixed_bugc C s t ddddgddddggddd d
|
gd }t d d ggd gddgd}| |¡}t ||¡ tj|jd< | |¡}d |jd< t ||¡ d S )Nç ð?ç @FTr7 r6 r r Zbool1Zbool2r& é- r )r r' )r r )r r r r ÚnpÚnanÚloc)r r0 Úotherr2 r r r! Ú$test_combine_first_same_as_in_updatep s
|
þ
|
|
|
z>TestDataFrameCombineFirst.test_combine_first_same_as_in_updatec C s t dtjddtjgtjddtjdgd}t dddtjddgtjtjdddd gd}| |¡}t d
|
d d d d dgtjd d dddgd}t ||¡ d S )Nr? ç @r6 r@ ç @r r7 ç @g @r r é r r9 r é )r rB rC r r r )r Údf1r1 r2 Úexpectedr r r! Ú test_combine_first_doc_example s $ÿþÿ
|
(z8TestDataFrameCombineFirst.test_combine_first_doc_examplec C s t tjddgdtjdgtjddgg}t dtjdgdddggd d
|
gd }tdddgd
|
td }| |¡d
|
}t ||¡ | |¡d
|
}t ||¡ d S ) NrG TgffffffÀrI FgÍÌÌÌÌLEÀg Àgù?r r r )ÚnameÚdtype)r rB rC r Úboolr r r) )r rL r1 rM Ú result_12Ú result_21r r r! Ú-test_combine_first_return_obj_type_with_bools s ÿ" zGTestDataFrameCombineFirst.test_combine_first_return_obj_type_with_boolszdata1, data2, data_expectediÐ r r rJ c C s@ t d|it d|i }}| |¡}t d|i}t ||¡ d S )Nr
|
©r r r r )r Údata1Údata2Z data_expectedrL r1 r2 rM r r r! Ú-test_combine_first_convert_datatime_correctly¡ s
|
zGTestDataFrameCombineFirst.test_combine_first_convert_datatime_correctlyc C s$ t t d¡dggddgd}t dgdggdgd}|d jdksDt|d jd ksVt| |¡}t t d¡tjgddgd
|
ddgd}t ||¡ |d jdks¤t|d jd ks¶t|j d d
|¡}t t
|
j t
|
j gddgd
|
ddgd}t ||¡ |d jd ks t|d jd ks td S ) Nú
|
2011-01-01r r
|
r r& r9 r zdatetime64[ns]Úint64©r
|
r r Úfloat64) r ÚpdÚ TimestamprP r( r ÚNaTr r r, rB rC )r ZdfaZdfbÚresr r r r! Ú test_combine_first_align_nanÄ s"
|
þ " z6TestDataFrameCombineFirst.test_combine_first_align_nanc
|
C sÚ t d¡ d¡}tddg|t jdddd}t d ¡ d¡}tdd
|
g|t jd ddd}|dg |¡}tt jd dd t jddd gt jddd t jgdddgt jddddd}|d jdksÆt |d jdksØt t
|
||¡ t jdddd }td|i}t jdddd }td|i}| |¡}t
|
||¡ |d jdksFt t j ddd d gd d }td|idd d d!gd"}t j d#d$d%gd d }td|idd&d gd"}| |¡}t j dd#dd$d d gd d } td| iddd d&d d!gd"}t
|
||¡ t jddd d }td|i}t dd¡}td|i}| |¡}t
|
||¡ |d jd'ksHt t jdd(d d }td|i}t dd¡}td|i}| |¡}t jdd d t jd(d d t d¡g} td| i}t
|
||¡ |d jd)ksÖt d S )*Nz20100101 01:01ÚUTCÚ UTCdatetimeÚabcZ20140627r )Úperiods)r' Údatar z20121212 12:12ZxyzZ20140628z2010-01-01 01:01)Útzz2012-12-12 12:12z2010-01-01 01:01:00)rc rd r r$ )re Úfreq)r' r zdatetime64[ns, UTC]z
|
2015-01-01z
|
2015-01-05ZDATEz
|
2015-01-03rY r_ z
|
2011-01-03z
|
2011-01-04z
|
US/EasternrJ r r r ú
|
2012-01-01ú
|
2012-01-02ú
|
2012-01-03r9 zdatetime64[ns, US/Eastern]z
|
2015-01-02Úobject) r] Ú to_datetimeZ tz_localizer Z
|
date_ranger r^ r_ rP r( r r Z DatetimeIndex)
|
r rV rL rW r1 r` r Zdts1Údts2Úexp_dtsr r r! Útest_combine_first_timezoneÝ s ý ý þû÷
|
|
ÿ ÿ
|
ú÷
|
|
ý z5TestDataFrameCombineFirst.test_combine_first_timezonec C s® t ddddg¡}td|idddd gd
|
}t d d d g¡}td|idddgd
|
}| |¡}t dd dd ddg¡}td|idddddd gd
|
}t ||¡ |d jdksªtd S )Nz1 dayr_ z3 dayZ4dayZTDr rJ r r r z10 dayz11 dayz12 dayr r9 z4 dayztimedelta64[ns])r] ZTimedeltaIndexr r r r rP r( )r rV rL rW r1 r` ro r r r r! Ú test_combine_first_timedelta8 s
|
ÿ z6TestDataFrameCombineFirst.test_combine_first_timedeltac C sp t jddddgdd}td|idd d
|
d gd }t jd ddgdd}td|iddd
|
gd }| |¡}t jddddddgdd}td|iddd dd
|
d gd }t ||¡ |d j|jks¸tt jd ddgdd}td|iddd
|
gd }| |¡}t jdddt jd ddt j t jdddt jdddt jdddg}td|iddd dd
|
d gd }t ||¡ |d jdksltd S )Nz2011-01r_ z2011-03z2011-04ÚM)rh ÚPr rJ r r r ri z2012-02z2012-03r r9 z2012-01rj rk r$ rl )
|
r] Z PeriodIndexr r r r rP r( ÚPeriodr_ ) r rV rL rW r1 r` ro r rn r r r! Útest_combine_first_periodF s2
|
ÿ
|
ú z3TestDataFrameCombineFirst.test_combine_first_periodc C s t dddddgidd}t dddgidd}| |¡}t dddddgi}t ||¡ | |¡}t dddddgi}t ||¡ d S ) Nr
|
r r rJ r rZ ©rP r9 rU )r rL r1 rR Z expected_12rS Z expected_21r r r! Útest_combine_first_inte s
|
|
z0TestDataFrameCombineFirst.test_combine_first_intÚvalr? c C sH t d|gi}t ddgi}| |¡}t dg|gd}t ||¡ d S )NÚisNumÚisBoolT)rz ry rU )r rx rL r1 r` r r r r! Ú(test_combine_first_with_asymmetric_others s
|
|
zBTestDataFrameCombineFirst.test_combine_first_with_asymmetric_otherc C s t ddgtjgd d|d}t dgtjgd|d}|jddgdd |jddgdd | |¡}t ddgtjgd d|d ddg¡}t ||¡ d S )
|
NZ962Z85r r[ rv r
|
r T)Zinplace)r r] ZNAZ set_indexr r r )r Znullable_string_dtyper0 r1 r2 rM r r r! Ú'test_combine_first_string_dtype_only_na~ s ÿ
|
ÿþzATestDataFrameCombineFirst.test_combine_first_string_dtype_only_naN)Ú__name__Ú
|
__module__Ú __qualname__r" r3 r> rF rN rT ÚpytestÚmarkÚ parametrizer r] r_ rX ra rp rq ru rw r{ r| r r r r! r sF = ý ý ý ýðþ
|
|
[
|
|
r zscalar1, scalar2éä r r z
|
2020-01-01r$ z
|
2020-01-02z89 daysz60 min)ÚleftÚrightrJ r )r r
Úclosedc
|
C s¶ |}t ||ggddgd}t | |ggddgd}t|jd |jd g}t|dsf|jd |jd krl| }n|}| |¡}t |||ggdddgd} | d |¡| d< t || ¡ d S )Nr
|
r r& r4 rl )r r Zdtypesr r Zastyper r )
|
Zscalar1Zscalar2Z nulls_fixtureZna_valueÚframerE Z common_dtyperx r2 rM r r r! Ú test_combine_first_timestamp_bug s
|
r c C s t tjtjggddgd} t tdddtdddggddgd}| |¡}t tjtdddtdddggdddgd}t ||¡ d S )Nr
|
r r& r r r r4 )r r] r_ r r r r )r rE r2 rM r r r! Ú$test_combine_first_timestamp_bug_NaT s ÿ
|
ÿr c C sH t jdddddtjgdddddd ggddgd
|
} tdddddddgi| d }t jdddddd gddddddggddgd
|
}tdddddd g|d }| td |i¡}t jdddddddddd tjg ddddddddddd g gddgd
|
}ttjtjdddddtjdtjdg d dtjddtjtjdtjdtjg d|d }t ||¡ d S )Nr r4 r
|
r r rJ r9 r r ©Únamesr Údr? r7 r@ r6 rG rH )r4 r ) r Ú from_arraysrB rC r r r r r )Zmi1r0 Zmi2Úsr` Z mi_expectedrM r r r! Ú&test_combine_first_with_nan_multiindex¼ s0 ÿ ÿþû "þûr c C sÈ t ddgtdd} t tdtdd}t| |d}t ddgtddd}t tdd tddd}t||d
|
}tddtjtjgd d dd gtjtjddgd d d dd gd}| |¡}t ||¡ d S )Nr
|
r r r r r r éÿÿÿÿr )r r# r ©r r r# r )r r r rB rC r r r )Za_columnZb_columnrL Zc_columnr1 rM r r r r! Ú test_combine_preserve_dtypesÙ s
|
ý
|
ú
|
r c C sÐ t ddddgitjdddgtjdd ggd
|
d gd d } t ddddgitjdddgtjddggd
|
d gd d }t dddtjgddtjdgdtjddddgtjdd d ggd
|
d gd d }| |¡}t ||¡ d S ) NÚxé r% é r r rJ r r r
|
r r r Úyr8 é é r9 r g "@g $@g &@g (@g *@g ,@)r r r6 rH rI )r r r rB rC r r r )rL r1 rM r r r r! Ú7test_combine_first_duplicates_rows_for_nan_index_valuesï s& "þ "þ þ ÿû
|
r c C sx t dddgdddgd} t ddd gd
|
d d gd ddgd}| |¡}t dddgdddgd ddgd}t ||¡ d S )Nr r rJ r9 r r r é é é( é2 é< r8 é" éA r rU )Zdf_1Zdf_2r2 rM r r r! Ú,test_combine_first_int64_not_cast_to_float64 s
|
"
|
"r¡ c C s¶ t ddgtjtjgg¡} t ddgtjtjgg¡}tdd dgi| d}tdddgi|d}| |¡}t ddddgtjtjtjtjgg¡}tdtjdddgi|d}t ||¡ d S )Nr r r
|
r9 r rJ )r r rB rC r r r r )ZmidxZmidx2rL r1 r2 Z expected_midxrM r r r! Útest_midx_losing_dtype s
|
ÿr¢ ) r ÚnumpyrB r Zpandas.core.dtypes.castr Zpandas.core.dtypes.commonr Zpandasr] r r r r Zpandas._testingZ_testingr r r r rt Z TimedeltaZIntervalr r r r r r¡ r¢ r r r r! Ú<module> s2 | üþ
|
|
