U
|
ý°d ã @ sh d dl Zd dlZd dlZd dlmZmZ d dlmZ dd Z
|
dd Z dd Z d d
|
Z d d Zd d ZdS )é N)Ú DataFrameÚ
|
MultiIndexc C s | ¡ }| j}t ||¡ d S )N)Zto_numpyÚvaluesÚtmZassert_numpy_array_equal)ÚidxÚresultÚexp© r úad:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/multi/test_conversion.pyÚ test_to_numpy s r c C sd ddddg} t | ¡}|jdd}t| }t ||¡ | ¡ }||_t ||¡ ddddg} t j| ddgd }|jdd}t| }ddg|_t ||¡ | ¡ }||_t ||¡ t | ¡}|jdddgd
|
}t| }ddg|_t ||¡ |jddgd }||_ddg|_t ||¡ d }tj t
|
|d |jdd W 5 Q R X d}tj t |d |jdgd W 5 Q R X t t dtjdddg¡}|jdd}tt tjdddd¡t tjdddd¡d}t ||¡ | ¡ }||_t ||¡ |jdddgd
|
}tt tjdddd¡t tjdddd¡d}t ||¡ |jddgd }||_t ||¡ d S )N)é Úone)r Útwo)é r )r r F©ÚindexÚfirstÚsecond©Únames)r Úname)r z1'name' must be a list / sequence of column names.©Úmatchz<'name' should have same length as number of levels on index.é Z20130101é )ÚperiodsÚint64)Údtype)r r )r r )r Ú from_tuplesÚto_framer r Úassert_frame_equalr ÚcolumnsÚpytestÚraisesÚ TypeErrorÚ
|
ValueErrorZ from_productÚrangeÚpdÚ
|
date_rangeÚnpÚrepeatZarangeZtile)Ztuplesr r ÚexpectedÚmsgr r r
|
Ú test_to_frame sj
|
|
|
|
|
þÿ þÿ r- c
|
sî t jtjddddddddddgtjddddd d gd
|
d d d d dd d ggdddd gd fddt jD } ttjddddddddddgtjddddd d gd
|
d d d d dd d gd} jdd}|j
|
¡ }t ||¡ | |ksêt d S )NZ19910905é z
|
US/Eastern)r Útzr r ÚaÚbÚcT)ZorderedÚxÚyÚzÚdatesr c s i | ]\}}| j | jqS r )Úlevelsr )Ú.0Úir ©Úmir r
|
Ú
|
<dictcomp>f s z0test_to_frame_dtype_fidelity.<locals>.<dictcomp>)r6 r0 r1 r2 Fr )r Ú from_arraysr' r( Z CategoricalÚ enumerater r r ZdtypesÚto_dictr r ÚAssertionError)Zoriginal_dtypesZ expected_dfZdfZ df_dtypesr r: r
|
Ú test_to_frame_dtype_fidelity[ s( ü
|
ù üÿ
|
rA c C sN dddg} t jdddgdddgdd d
|
gg| d }| ¡ j ¡ }|| ksJtd S ) Nr5 r r0 r1 r2 r3 r4 ÚqÚwÚer )r r= r r! Útolistr@ )r+ r; r r r r
|
Ú$test_to_frame_resulting_column_orderw s
|
ÿrF c C sÌ ddg} ddg}t j| |d}tjtdd | ¡ W 5 Q R X |jdd}t| ||d }t ||¡ d d
|
g}t j| |d}tjtdd | ¡ W 5 Q R X |jdd}t| |d
|
d
|
gd }t ||¡ d S ) N)r r )r é r0 r z%Cannot create duplicate column labelsr T)Zallow_duplicates)r r! r ) r r r" r# r% r r r r )Údatar r r r+ r r r
|
Ú test_to_frame_duplicate_labels s rI c C s& t jddd}| ¡ }t ||¡ d S )N))Úfoor )rJ r )Úbarr )Zbazr )Úquxr )rL r F)Z tupleize_cols)r' ZIndexZ to_flat_indexr Zassert_index_equal)r r+ r r r r
|
Útest_to_flat_index s ÷ rM )Únumpyr) r" Zpandasr' r r Zpandas._testingZ_testingr r r- rA rF rI rM r r r r
|
Ú<module> s I
|
|
