U
|
®ý°d³ ã @ s d dl Zd dlZd dlZd dlmZmZmZ d dlm Z
|
dd Z dd Z dd Z ej d d
|
d d gd d gd d gd d ggejd d ddgddgdejddgd
|
gdfdd d d d gd d d d ggejddgddejdddd gd
|
dgdfg¡d d Zd d Zd!d" ZdS )#é N)Ú DataFrameÚ
|
MultiIndexÚSeriesc C sf t ddgddgg¡} tt d¡| td}| ¡ }|jtk ¡ sDt |jdd}|jtk ¡ sbt d S ) NÚbarÚfooÚoneÚtwoç @)ÚindexÚdtyper ©Úlevel)
|
r Ú from_productr ÚnpÚarangeÚobjectÚunstackZdtypesÚallÚAssertionError)ÚmiÚserZres1Zres2© r ú_d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/series/methods/test_unstack.pyÚ test_unstack_preserves_object s r c
|
|
C sn t ddgdddggddddgddddggd } tt d
|
¡| d }| ¡ }td tjd gddtjggddgdddgd}t ||¡ |jdd}t ||j ¡ t dgdddgddggddddddgddddddgddddddggd } ttj
|
d¡| d }t dddgddggddddddgddddddggd }td|j i|d j dd}| d¡ ¡ }t ||¡ t ddgdtjgg¡}tddg|d }| ¡ }ttjdgdtjggddgtjdgd}t ||¡ t dddddgdddddgddddtjgg¡}tdddd d g|d }tdd gdtjgtjd gdtjggddgd }d d dtjfd!g} t | ¡|_t |jdd|¡ d S )"Nr r r Zthreer é r é )ÚlevelsÚcodesr ©r
|
g @g @g g ð?)r
|
Úcolumnsr é ée éf g @ÚcatZdogÚaÚbgñ?g333333ó?gÍÌÌÌÌÌô?gffffffö?)r ©r$ r ©r$ r ©r% r )r r r r r r ÚnanÚtmÚassert_frame_equalÚTÚrandomZrandnÚvaluesZ
|
sort_indexÚ from_arraysÚ from_tuplesr
|
)
|
r
|
ÚsZ unstackedÚexpectedZ exp_indexÚidxÚtsÚleftÚrightZtplsr r r Ú test_unstack sb þý ,þ þ ÿ ýÿ þ r7 c C s t jdddgdddggddgd } td| d
|
}| d¡}tdddgdddgdddggt jd d d gdgd tjdddgddd}t ||¡ d S )Nr$ r% Úcr r é ©ÚAr$ )ÚBr% ©Únamesr ©r$ ©r% )r8 ©Úname©r r
|
)
|
r r r r r r0 ÚpdÚIndexr* r+ )r3 r Úresultr2 r r r Ú$test_unstack_tuplename_in_multiindexT s ÿ
|
ýrG z>unstack_idx, expected_values, expected_index, expected_columnsr: r )r r9 )r é )r r9 )r rH r< ÚCr= r? r@ )r: r< r9 rH rA r& r' r( )r% r c C sZ t jddgddgddggddd gd
|
}td|d }| | ¡}t|||d }t ||¡ d S ) Nr$ r% r r r9 rH r: r< rI r= r rC )r r r r r r* r+ )Z unstack_idxZexpected_valuesZexpected_indexZexpected_columnsr3 r rF r2 r r r Ú*test_unstack_mixed_type_name_in_multiindexd s ÿ
|
ÿrJ c C s¦ t ¡ ¡ j ddg¡} tdgt| | ddd}| ¡ }|jjd }t
|
dgt| ¡}t | ¡ | ¡ | ¡ | ¡ dt j tddd | d¡d
|
}t ||¡ d S ) NÚmajorÚminorr Úcategory)r
|
rB r r )r; r< rI ÚDÚABCDrA rC )r* ZmakeTimeDataFrameÚstackr
|
Úrenamer Úlenr r rD Z Categoricalr ÚcopyrE Úlistr+ )r r rF Zdtir8 r2 r r r Ú+test_unstack_multi_index_categorical_values s ýrU c C s ddgddgddgg} t j| dd}tddg|d}| d ¡}tdgdggtjdgd d
|
t jd d gd dgdd}t ||¡ d S )Nr$ r r ÚredÚblue)Úxr Úyr= r rX rA )r rV )r rW r rY rC )
|
r r/ r r r rD rE r0 r* r+ )Zarraysr3 r rF r2 r r r Ú test_unstack_mixed_level_names s
|
|
ýrZ )Únumpyr ZpytestZpandasrD r r r Zpandas._testingZ_testingr* r r7 rG ÚmarkZ parametrizer0 rE rJ rU rZ r r r r Ú<module> s6 ;ü
|
ÿüùþ
|
|
