U
|
ý°d ã @ s~ d dl Zd dlZd dlm mZ d dlmZm Z m
|
Z
|
m Z m Z d dl mZ e d¡G dd dZe d¡G dd dZdS )é N)Ú CategoricalÚ DataFrameÚ
|
MultiIndexÚSeriesÚ
|
date_rangeÚxarrayc @ s2 e Zd Zejdd Zdd Zdd Zdd Zd S )
|
ÚTestDataFrameToXArrayc C sb t tdttddt dd¡ d¡tjddd d
|
d d d gttdtd ddtd ddddS )NÚabcé é é é Úu1g @g @Úfloat64©ÚdtypeTFZ20130101)Úperiodsz
|
US/Eastern)r Útz)ÚaÚbÚcÚdÚeÚfÚgÚh)r ÚlistÚrangeÚnpZarangeÚastyper r )Úself© r! úZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/generic/test_to_xarray.pyÚdf s
|
|
øÿzTestDataFrameToXArray.dfc C sÞ |}t |dkrt d¡ ddlm} |d d
|_d|j_d|j_| ¡ }|j d dks^t
|
t |j dkspt
|
t |j dkst
|
t t|j ¡ dg¡ t||s¨t
|
| ¡ }|d t¡|d < d |j_t | ¡ |¡ d S )
|
Nr z'Test doesn't make sense for empty index©ÚDatasetr ÚfooÚbarr
|
é r )ÚlenÚpytestÚskipr r% ÚindexÚnameÚcolumnsÚ to_xarrayÚdimsÚAssertionErrorÚcoordsÚ data_varsÚtmÚassert_almost_equalr ÚkeysÚ
|
isinstanceÚcopyr ÚobjectÚassert_frame_equalÚ to_dataframe)r Ú
|
index_flatr# r, r% ÚresultÚexpectedr! r! r" Útest_to_xarray_index_types! s"
|
z0TestDataFrameToXArray.test_to_xarray_index_typesc C sH ddl m} d|j_|dd
¡ }|jd dks6tt||sDtd S )Nr r$ r& )r r% r, r- r/ r0 r1 r7 )r r# r% r= r! r! r" Útest_to_xarray_empty; s
|
z*TestDataFrameToXArray.test_to_xarray_emptyc C sÜ ddl m} tjdgtdgddgd|_| ¡ }|jd dksDt|jd dksVtt |j
|
d kshtt |j d
|
ksztt t|j
|
¡ ddg¡ t||s¢t| ¡ }| ¡ }|d t¡|d < d |j_t ||¡ d S ) Nr r$ r r ÚoneÚtwo©Únamesr
|
é r( r )r r% r Ú from_productr r, r/ r0 r1 r) r2 r3 r4 r5 r r6 r7 r; r8 r r9 r. r- r: )r r# r% r= r> r! r! r" Ú test_to_xarray_with_multiindexC s z4TestDataFrameToXArray.test_to_xarray_with_multiindexN) Ú__name__Ú
|
__module__Ú __qualname__r* Zfixturer# r? r@ rG r! r! r! r" r s
|
|
r c @ s$ e Zd Zdd Zdd Zdd ZdS )ÚTestSeriesToXArrayc C s |}ddl m} ttt||dd}d|j_| ¡ }t| t|t|ksRt t|j
|
dksdt t t |j
|
¡ dg¡ t||st t | ¡ |¡ d S )Nr ©Ú DataArrayÚint64)r, r r& r
|
)r rM r r r) r, r- r/ Úreprr1 r2 r4 r5 r r6 r7 Úassert_series_equalÚ to_series)r r< r, rM Úserr= r! r! r" r? Y s z-TestSeriesToXArray.test_to_xarray_index_typesc C st ddl m} tg td}d|j_| ¡ }t|dks8tt|j dksJtt
|
t |j ¡ dg¡ t||sptd S )Nr rL r r& r
|
)r rM r r9 r, r- r/ r) r1 r2 r4 r5 r r6 r7 )r rM rR r= r! r! r" r@ k s z'TestSeriesToXArray.test_to_xarray_emptyc C s ddl m} tjddgtdgddgd}ttd d
|
|d }| ¡ }t|d ksTtt
|
t |j ¡ ddg¡ t||s|t| ¡ }t ||¡ d S ) Nr rL r r r rA rB rC r rN )r r, rE )r rM r rF r r r/ r) r1 r4 r5 r r2 r6 r7 rQ rP )r rM ÚmirR r= Úresr! r! r" rG v s z1TestSeriesToXArray.test_to_xarray_with_multiindexN)rH rI rJ r? r@ rG r! r! r! r" rK W s rK )Únumpyr r* Z pandas.util._test_decoratorsÚutilZ_test_decoratorsÚtdZpandasr r r r r Zpandas._testingZ_testingr4 Z
|
skip_if_nor rK r! r! r! r" Ú<module> s F
|
