U
|
ý°d^ ã @ s® d Z ddlZddlZddlZddlmZmZmZm Z m
|
Z
|
m Z ddl m Z G dd dZdd Zdd Zd
|
d Zejjd d dd Zdd Zejjdd dd Zdd ZdS )z±
|
Tests for values coercion in setitem-like operations on DataFrame.
|
|
For the most part, these should be multi-column DataFrames, otherwise
|
we would share the tests with Series.
|
é N)Ú DataFrameÚ
|
MultiIndexÚNaTÚSeriesÚ TimestampÚ
|
date_rangec @ s& e Zd Zej dddg¡dd ZdS )Ú TestDataFrameSetitemCoercionÚ consolidateTFc C sð t tjdtjd}tj||gdddgd}|r8| ¡ }tjdtjd|jdd
dt ddff< |j
|
tjk ¡ stt tjdtjd|jd d
|
dt ddff< |j
|
tjk ¡ s°t tjdtjd|jd d
dt ddff< |j
|
tjk ¡ sìt d S ) N)é é ©Údtypeé é )ZaxisÚkeys)r r é )r
|
r r r ) r ÚnpÚzerosÚfloat32ÚpdÚconcatZ _consolidateZonesÚlocÚsliceÚdtypesÚallÚAssertionError)Úselfr ÚA© r ú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/indexing/test_coercion.pyÚ#test_loc_setitem_multiindex_columns s (((z@TestDataFrameSetitemCoercion.test_loc_setitem_multiindex_columnsN)Ú__name__Ú
|
__module__Ú __qualname__ÚpytestÚmarkZ parametrizer r r r r r s r c C s° t dddgdddgd} t dddgdddgd}| ¡ }d|jd< t ||¡ | ¡ }d|jd< t ||¡ | ¡ }d|jd < t ||¡ | ¡ }d|jd < t ||¡ d S )
|
Nr r r é r )r ÚBg333333ó?)r r' )r r )r ÚcopyÚatÚtmÚassert_frame_equalr ZiatÚiloc)ÚorigÚexpectedÚdfr r r Ú
|
test_374770 s
|
|
|
|
r0 c C sx t d}t d}t d}t|dd}t|ddgd}| ¡ }|| ||df< d |d< ||j|df< |jd
|
|ksttd S ) Nz
|
2014-04-01z2014-04-23 12:42:38.883082z2014-04-24 01:33:30.040039r )ZperiodsZtimenowZLive©ÚindexÚcolumnsT)r r )r r r r( r) r, r )Ú
|
indexer_alÚstartÚt1Út2Zdtir- r/ r r r Ú test_6942F s r8 c C s t dddgd}d|d< d| |d< t ddddgidddgtjd }t ||¡ d
|
| |d< t dddd
|
gidddgtjd }t ||¡ d | |d< t dddd gidddgtd }t ||¡ d S ) Nr r' ÚC©r2 r ÚDr )r9 r; )r2 r g @F@Zhello)r r Úint64r* r+ Úfloat64Úobject)r4 r/ r. r r r Ú
|
test_26395X s r? zunwanted upcast)Úreasonc C s t ddgddggddgd} tdd d
|
| jd< | jtjk ¡ sDttdd i| jd< ttjtjgt ddgd }t
|
| j|¡ d S ) Nr r r r& ÚaÚb)r3 r r
|
)rA rB é )r r2 ) r r r r r r< r r r= r> r* Úassert_series_equal)r/ Ú
|
exp_dtypesr r r Ú
|
test_15231j s rF c C st t tjddgtjdtjddgtjdddgd} | ¡ }| d j dd¡}|| jd d
d d
f< t | |¡ d S )
|
Nr r r r r& rA rB )r r r r )
|
r r Úarrayr r( ÚvaluesZreshaper, r* r+ )r/ r- rH r r r Ú.test_iloc_setitem_unnecesssary_float_upcastingw s ýÿrI zunwanted casting to dt64c C s t ddd} d| dg}t|}| ¡ }tjtg|jd< tdtjgt| tgddd}t ||¡ d| dg}| ¡ }tjtg|jdd d
f< t ||¡ d S ) Nz2016-03-01 03:13:22.98986ÚUTC)Útzr )ZoneZtwor zdatetime64[ns, UTC]r )
|
r r r( r Únanr r r r* r+ )ÚtsÚdatar- r/ r. r r r Ú
|
test_12499 s ÿ rO c C sâ t ddgdddgg¡} tdtd| d}td d
|
d ggd tddddgd}||d< td d d gd
|
d
|
d
|
gd d d gdddgdddgdddgd }| |_tt tj¡gd
|
t tj ¡g t tj¡gd | d }t
|
|j |¡ d S )Nr r' rA rB Úcéÿÿÿÿr r1 r r g @)r r r r r& r r: ) r Z from_productr Úranger3 r r r r< r= r* rD r )Úmir/ Zfillerr. rE r r r Ú
|
test_20476 s& $úÿ
|
0þrT )Ú__doc__Únumpyr r$ Zpandasr r r r r r r Zpandas._testingZ_testingr* r r0 r8 r? r% ZxfailrF rI rO rT r r r r Ú<module> s
|
|
|
