U
|
ý°d«7 ã @ s d dl m Z d dlmZ d dlZd dlZd dlZd dlmZ d dl Z
|
d dl m Z m Z m Z mZmZmZmZ d dlmZ d dlmZ G dd dZd d
|
Zd d Zd d Zej dddg¡dd Z ej dddg¡ej ddd g¡dd Z dd Z dd Z!dd Z"d d Z#G d d d Z$dS )!é )Údatetime)ÚStringION)ÚPerformanceWarning)Ú DataFrameÚIndexÚ
|
MultiIndexÚPeriodÚSeriesÚ TimedeltaÚ
|
date_range)Úreshapec @ s e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z e j ddddddgddddgddddgddddggejdddd gd d gd ejd d!d"d#gd$dgd fd%ddddddddgddddddddgged&d'gd d(ejd)d*d+d,d-d.d/d0gd$dd gd fg¡d1d2 Zd3d4 Zd5d6 Zd7d8 Zd9d: Zd;d< Zd=d> Zd?d@ ZdAdB Ze j dCe dD¡e dD¡fe j dDdEdFe j dDdGdEdFff¡dHdI Z dJdK Z dLdM Z!e j dNdOdPdQdRdSddDdTdUge j"dVdge j"dDdgfddTdUdd'dWdXdPdQge j"dDdge j"dVdgff¡dYdZ Z#e j d[d\d ge$d$g¡d]d^ Z%d_d` Z&dadb Z'dcdd Z(e j dedfdgg¡e j dhe)e*d¡didj Z+e j dee, -dkdldmg¡¡e j dhe)e*d&¡e j dndodpg¡dqdr Z.dsdt Z/dudv Z0dwdx Z1dydz Z2d{d| Z3e j d}dOddd&d'gdOddd&gdOddd'gdOddgddd&gdd&d'gdOdgdOdgdOd&gdOgdgd'gd'd&dddOgd&dddOgd'dddOgdddOgd&ddgd'd&dgddOgddOgd&dOgg¡e j d~ddOddOdgddOgf¡dd Z4dd Z5e j dd
dg¡e j de)de)dg¡dd Z6e j dd
dg¡e j de)ddGdUdWdPddXgfe)dddXdWdPdGdUgfg¡dd Z7dd Z8e j ddOdOddge 9ddgdVdgg¡fdOdOdd&ge 9ddgdVdgg¡fdOddd&ge 9ddgdVdgg¡fg¡dd Z:e j d~dOdg¡dd Z;e j d~dOdg¡dd Z<d$S )ÚTestDataFrameReshapec C s¤ | ¡ }t t |j¡¡ |j¡|d d
< | ¡ }t||d}| ¡ }| ¡ }t
|
||¡ t
|
|d |¡ | d¡}| d¡}t
|
|j |¡ t
|
|d j |¡ d S )N)ÚfooÚbarr r ) ÚcopyÚnpÚarangeÚprodÚshaper Ústackr ÚunstackÚtmÚassert_frame_equalÚT) ÚselfÚ float_frameÚdfÚstackedZ
|
stacked_dfÚ unstackedZ unstacked_dfZunstacked_colsZunstacked_cols_df© r ú\d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/test_stack_unstack.pyÚtest_stack_unstack s "
|
|
z'TestDataFrameReshape.test_stack_unstackc C sä t ddddgddgg}td|d |d d}| ¡ }tdt |d d
¡d}t ||¡ td|d t |dd
¡d}| d¡}tdt |d |d g¡|d d}t ||¡ |ddg d¡}|ddg }t ||¡ d S ) Né ÚaÚbé é r ©ÚindexÚcolumns©r( ) Úranger r r r Ú from_productr Úassert_series_equalr )r Úlevelsr ÚresultÚexpectedr r r Útest_stack_mixed_level+ s
|
ÿ z+TestDataFrameReshape.test_stack_mixed_levelc C sl t ddtjgddtjgd}|dg }|d |d< |sLt|jjdksLt| ¡ }| ¡ }t ||¡ d S )Nr% r& ç @é ©ÚxÚyr5 r6 )
|
r r ÚNaNÚlenÚ_mgrÚblocksÚAssertionErrorr r r- )r Úusing_array_managerr Údf2Úresr0 r r r Ú test_unstack_not_consolidatedB s
|
z2TestDataFrameReshape.test_unstack_not_consolidatedc C s t ddddgtjd}t dddd g¡|_|jd
|
d }tdd
|
dgddd
|
gd d ddgtjd}t ||¡ |jdd }tdddgdddgd d ddgt
|
d}t ||¡ tdddgdddgdddgdddgd d ddg¡}|jd dgdd }d}|| }t dddg|j|d}t ||¡ | d dg¡}|j |jj¡|_| tj¡}|j|j }t ||¡ |d }|jd dgdd }|d }t ||¡ d S )Nr% r& r3 é ©Údtype©r5 r# ©r5 r$ ©r6 r$ ©Úzr# éÿÿÿÿ©Ú
|
fill_value©r# r$ r5 r6 rG ©r( rB ç à?r# r$ ÚjÚkr )r5 r6 rG Úw)rP r$ rN )r( ÚnamerP )r r Úint16r Ú from_tuplesr( r r r r ÚfloatÚ set_indexr- r Zreorder_levelsÚnamesÚastypeÚint64Úloc) r Údatar/ r0 r r Úkeyr Úsr r r Útest_unstack_fillN sJ
|
ÿ ÿ ÿ $ÿþ z&TestDataFrameReshape.test_unstack_fillc C sr ddgddgddgddgg}t |td tjd
|
}t d d d dg¡|_|jdd}ddddgddddgddddgg}t |tdtjd}t ddddg¡|_t
|
||¡ |d tj ¡|d< |d tj ¡|d< |jdd}|d tj ¡|d< |d tj ¡|d< t
|
||¡ |jdd}ddddgddddgddddgg}t |tdtd}t ddddg¡|_t
|
||¡ d S )Nr% r& r" r3 r@ é é é ZAB)r) rB rC rD rE rF rH rI ÚxyzrL ©ÚAr# )rc r$ )ÚBr# ©rd r$ rc rd rM )r Úlistr Úint32r rS r( r r) r r rW rR Úfloat64rT )r Úrowsr r/ r0 r r r Útest_unstack_fill_frame| s2
|
ÿ "
|
ÿ "
|
ÿz,TestDataFrameReshape.test_unstack_fill_framec C sÒ t dddj}t|}t ddddg¡|_| ¡ }t|d tj |d g|d
|
|d tj gd d ddgd}t
|
||¡ |j|d d}t|d |d |d g|d
|
|d |d gd d ddgd}t
|
||¡ d S )Nú
|
2012-01-01r3 ©ÚperiodsrC rD rE rF r r" r% r& rK r5 r6 rG r* rI ) r Úvaluesr r rS r( r r ÚpdÚNaTr r )r ZdvrZ r/ r0 r r r Ú test_unstack_fill_frame_datetime s"
|
ÿ(þ ,þz5TestDataFrameReshape.test_unstack_fill_frame_datetimec C sÖ dd t dD }t|}t ddddg¡|_| ¡ }t|d tj|d g|d
|
|d tjgd d ddgd}t
|
||¡ |j|d
|
d}t|d |d
|
|d g|d
|
|d |d
|
gd d ddgd}t
|
||¡ d S )Nc S s g | ]}t |d qS ))Údays)r
|
)Ú.0Úir r r Ú
|
<listcomp>¸ s zJTestDataFrameReshape.test_unstack_fill_frame_timedelta.<locals>.<listcomp>r3 rC rD rE rF r r" r% r& rK r5 r6 rG r* rI ) r+ r r rS r( r r ro rp r r )r ÚtdrZ r/ r0 r r r Ú!test_unstack_fill_frame_timedelta¶ s"
|
ÿ(þ ,þz6TestDataFrameReshape.test_unstack_fill_frame_timedeltac C sÜ t dt dt dt dg}t|}t ddddg¡|_| ¡ }t|d d |d
|
g|d |d d gd dddgd}t ||¡ |j|d d}t|d |d |d
|
g|d |d |d gd dddgd}t ||¡ d S )Nz2012-01z2012-02z2012-03z2012-04rC rD rE rF r r" r% r& rK r5 r6 rG r* rI ) r r r rS r( r r r r )r rm rZ r/ r0 r r r Ú test_unstack_fill_frame_periodÌ s. ü
|
ÿ$þ þûz3TestDataFrameReshape.test_unstack_fill_frame_periodc C sö t ddddgdd}t dddd g¡|_| ¡ }ttjtd
|
td d tjtd td d dtdd}t
|
||¡ d}t j t |d |jdd W 5 Q R X |jdd}ttjtdtd d tjtdtd d dtdd}t
|
||¡ d S )Nr# r$ ÚcÚcategoryrA rC rD rE rF ZaxaÚabc©Ú
|
categoriesZbcxrK ra r* z9Cannot setitem on a Categorical with a new category \(d\)©ÚmatchÚdrI ZacaZbcc)r r rS r( r r ro Ú Categoricalrf r r ÚpytestÚraisesÚ TypeError)r rZ r/ r0 Úmsgr r r Ú#test_unstack_fill_frame_categoricalê s.
|
ÿþû þûz8TestDataFrameReshape.test_unstack_fill_frame_categoricalc C sº t jdddgdddggddgd }tdgd
|
dgd
|
d |d }| d¡}tddddddgddddddgddddddggt jd dddddgd dgd tdddgddd}t ||¡ d S )Nr# r$ ry r% r& r" rb re ©rV é ©r Úer* ©r r# ©r r$ )r ry ©r r# ©r r$ )r ry ©rQ ©r) r( )r r, r r rS r r r ©r Úidxr r/ r0 r r r Ú$test_unstack_tuplename_in_multiindex s* ÿ
|
,ú÷ óz9TestDataFrameReshape.test_unstack_tuplename_in_multiindexz>unstack_idx, expected_values, expected_index, expected_columnsrb r% r& )r% r" )r% r3 ©r& r" )r& r3 rd ÚCr r r r r N)rb rd r" r3 r )r r# r% )r r# r& )r r$ r% )r r$ r& )r r# r% )r r# r& )r r$ r% )r r$ r& c C sl t jddgddgddggddd gd
|
}tdgd dgd d |d }| |¡}t|||d}t ||¡ d S )Nr# r$ r% r& r" r3 rb rd r r r` r r* r )r r, r r r r ) r Z unstack_idxZexpected_valuesZexpected_indexZexpected_columnsr r r/ r0 r r r Ú*test_unstack_mixed_type_name_in_multiindex% s & ÿ
|
ÿz?TestDataFrameReshape.test_unstack_mixed_type_name_in_multiindexc C s t dddgdddgtdddg d¡tj d¡d d
|
t d ¡td d dg d¡tdddg d¡dtd ddg d¡d }dd }| dd g¡}||d | dd g¡}||d | d d g¡}||d | d d!dg¡}||d! | d"d#g¡}||d" | dd g¡}|d# }||d d S )$NZILÚMIZNCr# r$ ry rz r" r% r Z20010102ç ð?g I@g Y@Úfloat32r2 g @g @rh FéÈ i2 Zint8) Ústater( Úsome_categoriesrc rd r ÚDÚEÚFÚGÚHc S s& | |g¡}| |¡}t ||¡ d S ©N©r r r )r Z column_nameZ
|
unstacked1Z
|
unstacked2r r r Úunstack_and_compareh s
|
zNTestDataFrameReshape.test_unstack_preserve_dtypes.<locals>.unstack_and_comparer r( r r r r rd r rc ) r r rW r ÚrandomZrandro Ú TimestamprU )r r r¤ Údf1r\ r r r Ú test_unstack_preserve_dtypesV s8
|
õÿ
|
|
|
|
|
z1TestDataFrameReshape.test_unstack_preserve_dtypesc C sÖ t ttjtddd¡}ttj dd¡|d}t
|
|j ddgd|j ddj dd¡ t
|
|j d d
|
gd|j ddj dd¡ | ¡ }|jjtdd d }|d ksªtt
|
|j ddgd|j ddj dd¡ d S ) Nr" ©Úrepeaté é ©r) r% r& ©ÚleveléþÿÿÿrH T)Zinplace)r rS rf Ú itertoolsÚproductr+ r r r¥ Úrandnr r r r r) Z set_namesr; )r r) r Zdf_namedZ return_valuer r r Útest_stack_ints s ( ÿ ÿz$TestDataFrameReshape.test_stack_intsc C s t jddddgdddgd}ttj d d ¡|d
|
}|jddgd }|jddgd }| ¡ }ddd g|j_ t
|
j |jdd gd |d d t
|
j |jdd gd |d d d}t j t|d |jddgd W 5 Q R X | ¡ }dddg|j_ t
|
j |jddgd |d d d S )N©rc ÚcatÚlong©rd r¶ r· ©rc ÚdogÚshort©rd rº r» ÚexpÚanimalÚ hair_lengthr r3 r r® r% F©Z check_nameszSlevel should contain all level names or all level numbers, not a mixture of the twor~ r )r rS r r r¥ r³ r r r) rV r r r r Ú
|
ValueError)r r) r Úanimal_hair_stackedÚexp_hair_stackedr= r
Údf3r r r Útest_stack_mixed_levels sD üù ÿ ÿÿ ÿz,TestDataFrameReshape.test_stack_mixed_levelsc C s: t jddddgdddgd}ttj d d ¡|d
|
}|jddgd }|jddgd }|jddgd }| ¡ }d d dg|j_ t
|
j |jd dgd |dd t
|
j |jd d gd |dd t
|
j |jd dgd |dd | ¡ }dd d g|j_ t
|
j |jd d gd |dd t
|
j |jdd gd |dd t
|
j |jdd gd |dd d S )Nrµ r¸ r¹ r¼ r½ r¾ r¿ r r3 r r® r r% r& FrÀ ) r rS r r r¥ r³ r r r) rV r r )r r) r Zexp_animal_stackedr rà r= rÄ r r r Útest_stack_int_level_names½ s\ üù ÿ ÿ ÿ ÿ ÿ ÿz/TestDataFrameReshape.test_stack_int_level_namesc C s| t ddgt ddgddgg¡dgd}| ¡ }t tjdtjgtjdggtdddgt ddgddgg¡d}t ||¡ d S ) NFr# r$ ry ÚlÚcolr' rA )
|
r r Ú from_arraysr r ÚarrayÚnanÚobjectr r )r r ÚrsÚxpr r r Útest_unstack_boolæ s ý ýz&TestDataFrameReshape.test_unstack_boolc C sì t ddgddgddggddddgddddgddddggd d
|
d gd }td|d }| ddg¡ d¡}t ddgddggddddgddddggd d
|
gd }ttjtjdgdtjgtjdgdtjggtjd|t ddgd dd}t
|
||¡ d S )Nr r ZoneZtwor# r$ r r% ÚfirstÚsecondÚthird©r. ÚcodesrV r* r& rA r r' ) r r r r r r rÊ rË rh r r r )r Úmir\ r/ Z expected_mir0 r r r Útest_unstack_level_bindingô s* ý ý" ÿûz/TestDataFrameReshape.test_unstack_level_bindingc C sî | ¡ }t|tst| ¡ j}t ||¡ tddtj gddtj gd}t
|
dddg|_ | ¡ }t d d
|
gdddggd d d dddgd ddd ddggd }tddtj d dtj g|d}t ||¡ | ¡ }tdD ] }| ¡ }qÐt ||¡ d S )Nr% r& r2 r3 r4 r# r$ ry r5 r6 r ©r. rÔ r" r* )r Ú
|
isinstancer r; r r r r r r7 r r( r r- r r+ ) r r rZ Zundor/ Úmidxr0 Zold_dataÚ_r r r Útest_unstack_to_series s"
|
þ
|
z+TestDataFrameReshape.test_unstack_to_seriesc C s° ddddgddddgddddgddddgg}t |tdd}|j}tt d¡gd tdd}t ||¡ | d d
|
g¡}| d
|
¡}|j}tt d¡gd t
|
j d d d d gddddggd dd}t ||¡ | d d
|
g¡}d|d < | d
|
¡}|j}tt d¡gd t d¡gd t
|
j d d d d gddddggd dd}t ||¡ d|d < | d
|
¡}|j}tt d¡gd t d¡gd t
|
j d d d d gddddggd dd}t ||¡ d S )Nr% r" r3 r& ÚABCDr rX r* rc rd r r )Nrd r r2 rh r rÌ ) r rf Údtypesr r rB r r- rU r r rÉ )r ri r r/ r0 r= rÄ r r r Útest_unstack_dtypes) sN ,
|
ÿþ
|
ÿþ
|
ÿþz(TestDataFrameReshape.test_unstack_dtypeszc, dr@ Zf8rA é
|
c C s® t dgd ||tdddd}|jd d
jdd}| d d
|
g¡}|d d ¡|d < |jd d
d ¡}| d d
|
g¡ d ¡}|d d ¡|d< |jdkstt
|
||¡ d S )Nr# r@ rk rl )rc r r rd r" T)Údeeprc rd r rX r )r r# )r" r& ) r r Úilocr rU rW r r r; r r )r ry r r ÚrightÚleftr r r Ú test_unstack_dtypes_mixed_dateU s
|
üÿ z3TestDataFrameReshape.test_unstack_dtypes_mixed_datec C sx t jddgddgd}tddg|d}d}tjt|d | d¡ W 5 Q R X tjt|d |j d¡ W 5 Q R X d S )
|
NrK )ry r Úc1r r% r& r* z5The name c1 occurs multiple times, use a level numberr~ ) r rS r r r rÁ r r r )r r r r
r r r Ú#test_unstack_non_unique_index_namess s z8TestDataFrameReshape.test_unstack_non_unique_index_namesc C s` t dgddddgg¡d d
}tddggd |d
|
}| ¡ }t ddgdddgg¡}tddddddggdg|d }t ||¡ |jjd |jd k ¡ st ddd gddd d gg}ddddgdd dd gg}t ||}t
|
d¡ d d ¡}tt
|
||d g¡|d
|
}| ¡ }tt
|
j |d |d d gdd|d}t ||¡ |jjd |jd k ¡ s\t d S )Nr# rc rd r r rH r% r r" r* r' r_ r& r3 ©Úaxisr )r r, r r r r r) r. Úallr; r r r Z concatenate) r r r r/ Zexp_colr0 r. rÔ Úblockr r r Útest_unstack_unused_levels} s&
|
ÿ z/TestDataFrameReshape.test_unstack_unused_levelsz"level, idces, col_level, idx_levelr é é r^ r r` é r# é é c C s² dddgddddgg}dd ddgddd dgg}t ||}t d
|
¡}t| d d¡|d } | j|d }
|
t d¡tj } || |< t ddg|g¡} t| dd¡|| d} t
|
|
|
| ¡ d S )Nr# r& ry r% r" r@ r_ r rH r` r3 r* r® é r^ r' ) r r r r r r ÚzerosrË r, r r )r r¯ ZidcesZ col_levelZ idx_levelr. rÔ r rZ r r/ Zexp_dataÚcolsr0 r r r Ú)test_unstack_unused_levels_mixed_with_nan s
|
|
z>TestDataFrameReshape.test_unstack_unused_levels_mixed_with_nanró rc c C s¤ t dddgdddggddd gd
|
}|jddd gd d }|jtd td df|f }| ¡ }|jd g | }tj|jdggd d gd|_|j
|
d ¡|_ t ||¡ d S )NiÚ r# ÚIiÛ r$ ZIIrc rd r r F)Zdropr r ) r rU rY Úslicer rá r r, r) r( Ú droplevelr r )r ró r ÚindZ selectionr/ r0 r r r Útest_unstack_unused_level s "
|
ÿz.TestDataFrameReshape.test_unstack_unused_levelc C s´ t dggtjdggdgdtjdddddddggddddd d
|
d gdd }| dddd d
|
d g¡}t dggtjdddddddggddddd d
|
d gdtdgdd d }t ||¡ d S )Nr% r rå r Úi1Úi2Úi3Úi4Zi5Zi6Zi7r r )r r rS r r r r ©r r r/ r0 r r r Útest_unstack_long_index½ s$ þýþ úz,TestDataFrameReshape.test_unstack_long_indexc C s~ t ddgddggtjddgddggddgdtjdd d
|
gdd d ggd d dgdd}| d d g¡jjdd
d d gksztd S )Nç rd r r rå Úc2r rß é r« é( rú rû rü r r° )r r rS r r) rV r; ©r r r r r Ú test_unstack_multi_level_colsÓ s ÿ ÿû z2TestDataFrameReshape.test_unstack_multi_level_colsc
|
C s t ddgddgddgddggt d d
|
d gd d dgg¡tjdddgdddgdddgdddggdddgdd}| ddg¡}| dg¡ dg¡}t ||¡ d S )Nr% r& r" r3 rH r° éýÿÿÿéüÿÿÿr# r$ ry r r ÚfÚm1ZP3éÞ ZA5éo Úm2rú rû rü r r )r r rS r r r rþ r r r Ú&test_unstack_multi_level_rows_and_colsà s üùý z;TestDataFrameReshape.test_unstack_multi_level_rows_and_colsr ©ÚjimÚjoe)r r Úlevc C s0 dd }t ddtjdgdddd gd
|
d d d gd}| ddg¡ ¡ d }| ddg¡ ¡ d j}t ||¡ | t|¡}|j|d}| ¡ j
|
¡ t |ks t dd } |d ¡ j
|
¡ \}
|
} t|
|
| D ]b\} } t|d j| | f d¡}| |d j| | |d j| }tt||}||ksÈt qÈd S )Nc S s | | kr dn| }|dS ©NÚ Ú1r ©ÚvalZval_strr r r Úcast÷ s z:TestDataFrameReshape.test_unstack_nan_index1.<locals>.castr# r$ r rP r5 r6 rG za.wzb.xz .yzd.z)r r Újolier r r r® c S s t | trt| S | gS r¢ ©rØ Útuplerf ©r# r r r Ú<lambda>
|
ó z>TestDataFrameReshape.test_unstack_nan_index1.<locals>.<lambda>Ú.)r r rË rU r r r r rf Únotnarn Úsumr8 r; ÚnonzeroÚzipÚsortedrá Úsplitr( r) Úmap)r r r r r rã râ rÕ ÚudfÚmk_listri ró rt rN r r r Útest_unstack_nan_index1ó s(
|
|
ýÿ $z,TestDataFrameReshape.test_unstack_nan_index1Ú1stÚ2ndÚ3rdrÈ Ú4thÚ5thc sÆ dd t dgd tjgd dgd dgd d gd d
|
gd d gd d gd tjgd d gd tjgd dgd tjgd ddddddddddddd dd d d dd d!gd"}|j fd#d$d%d&|j fd'd$d%d& |d(< |d)< | t|¡}|j|d*}| ¡ j ¡ dt
|
| ks2t d+d$ }|| ¡ j ¡ \}} t || D ]f\}
|
} t|| j|
|
| f d,¡} ||| j|
|
||| j| } tt | } | | ksZt qZd S )-Nc S s | | kr dn| }|dS r r r r r r r s zATestDataFrameReshape.test_unstack_nan_index_repeats.<locals>.castr r" r@ r# r& ry r r$ r6 rP rG r3 r5 éC é' é5 éH é9 éP é rñ rî r« é; é2 é> éL é4 é é< é3 )r) r* r+ c s d t | ¡S )Nr )Újoinr% ©Úr©r r r r C r zETestDataFrameReshape.test_unstack_nan_index_repeats.<locals>.<lambda>r% rç c s d t | jd d d
¡S )Nr rH )r= r% rá r> r@ r r r D r r, r- r® c S s t | trt| S | gS r¢ r r r r r r J r r )r r rË ÚapplyrU rf r r rn r r8 r; r! r" r# rá r$ r( r) r% )r r r rÈ r rÕ r& r' ri ró rt rN rã râ r r@ r Ú test_unstack_nan_index_repeats s|
|
ÿþýüûÿ
|
þý
|
üû
|
úìòÿ)þ $z3TestDataFrameReshape.test_unstack_nan_index_repeatsc C s t tdtdtdd}| ddi¡}tj|jd< | ddg¡ d¡}d dd
|
d tj tj tj tj gtj tj tj tj d d ddgg}tt
|
tt | }t tj dd
|
d d d ddgdd}t dgddggddgdd
|
ggd dgd}t |||d}t ||¡ t tdttd d tdd}| ddi¡}tj|jd< | ddg¡ d¡}d tj gdd gd
|
d gtj dgd dgg}t dgddggddgdd
|
ggd dgd}t tj dd
|
d d gdd}t |||d}t ||¡ t tdttd d tdd}| ddi¡}tj|jd< | ddg¡ d¡}d tj gdd gd
|
d gd dgtj dgg}t dgddggddgdd
|
ggd dgd}t tj dd
|
d d gdd}t |||d}t ||¡ d S )NZaaaabbbbr` ©rc rd r rd rT ©r" r% rc r r" r% r& r3 r@ r^ r_ r r r# r$ rÓ r )r& r% )r rf r+ rW r r7 rá rU r rË r% r" r r r r )r r rã Úvalsr ró râ r r r Útest_unstack_nan_index2R sT þ ÿ $ & ÿ $ & ÿz,TestDataFrameReshape.test_unstack_nan_index2c C s
|
t tdtddd ¡ d t d¡d}tj|jd< | d d
|
g¡ ¡ }t
|
d d d dtj dgtj dddddgg¡}t ddgd d}t dgtdddgd d d d d d gdd d dd dggd d
|
gd}t |||d}|rú|jd dd dg }|| |d j¡||< t ||¡ d S )NZ
|
aaaaabbbbbrk r@ rl r& rß rC rD rc rd r" r r% r3 r^ r_ r` r r# r$ r r rH rÓ r )r rf r Útolistr r r7 rá rU r rÊ rË r r r) rW rB r r )r r< r rã rE r ró râ r r r Útest_unstack_nan_index3 s( ýÿ * ýz,TestDataFrameReshape.test_unstack_nan_index3c C sV dt jt jdgddt jdgddt jdgddd d
|
gd dd d
|
gdd d d
|
gg}t|d dddgddddddgd}| ¡ ddd g¡ ¡ }t jt jdt jdgdd t jdt jgg}tdd
|
gd ggddgddggddgd}tdgd ddddggdddddgddd d d ggd d gd}t|||d}t ||¡ |j dd
¡ ddd g¡}t | ¡ |¡ d S ) NZHgi¼ç(ÚUr ZPbg¢ VÍZ§Ý>ZSngº@¹ÕÂø>gÌ]KÈ=?i)=(ZAgga2U0*©#¿ZagentZchangeZdosageZs_idioC ipC iqC irC isC itC r r r% rH rÓ r& r" r3 )
|
r rË r r rU r r r r rY )r rE r rã r ró râ r r r Útest_unstack_nan_index4 s<
|
|
|
ú
|
ýþ ýý z,TestDataFrameReshape.test_unstack_nan_index4c C s t ddddddgtdddddt d¡ tj d¡d d¡d }|d
|
t d ¡ |d < tj |j
|
d < |j
|
d< tj |j
|
d< |j
|
d< | dd
|
d g¡ d
|
d g¡}| ¡ j ¡ dt| ks¾tdD ]F}| ¡ D ]8\}}|d ||d
|
|d ff}|| |j
|
| ksÎtqÎqÂd S )Nr% r& z
|
2014-02-01r^ r )rm Úfreqéd rß )r) r* r r r* z
|
2014-02-02r+ )r% r* )r" r* )r% r+ )r3 r+ r) r )r r r r r¥ r³ Úroundro r¦ rË rY rU r r rn r r8 r; Ziterrows)r r rã rÈ rÚ r? r[ r r r Útest_unstack_nan_index5È s üÿ z,TestDataFrameReshape.test_unstack_nan_index5c C sz t ddd}tddddgt |ddfg¡d}| ¡ }t d d
|
g¡}t |dfg¡}tddddg||d }t ||¡ d S ) NiÞ r% r& r" r3 rc rd r ©r r% r& r" )rd r' )r r r rS r r, r r )r Útr r/ ZeidxZecolsr0 r r r Ú%test_stack_datetime_column_multiIndexß s "z:TestDataFrameReshape.test_stack_datetime_column_multiIndexÚmultiindex_columnsr¯ rH c C sä t jdddddgddgd}|| }tt d t| ¡ d t|¡|d
|
}|j|d d }t|t r|j|d d }t|t
|
rt ||¡ n t ||¡ t j|j ¡ |jjd|_|j|d d }t|t
|
rÔt ||¡ n t ||¡ d S )N©rd r5 ©rd rG ©rc r6 ©r r5 ©r ÚuÚUpperÚLowerr r" r F)r¯ ÚdropnaT)r rS r r r r8 r r rØ Úintr r r- r r) Úto_numpyrV )r rR r¯ Úfull_multiindexZ
|
multiindexr r/ r0 r r r Ú test_stack_partial_multiIndexê s. þ þ
|
|
ÿ
|
z2TestDataFrameReshape.test_stack_partial_multiIndexc C sÜ t jdddddgddgd}tt d ¡ d
|
d ¡|d d d g d}|jdd}td d
|
gd tjgd dgdtjggt d d gddddggd d d d gd d d d ggd dgdtddgddd}|d |j
|
d ¡|d< t ||¡ d S ) NrS rT rU rV rW rY rZ r r^ r& r" r r% r F©r[ r@ r3 rX r5 r6 rG rÓ rd r r r' ) r rS r r r r r rË r rW rÝ r r )r r^ r r/ r0 r r r Útest_stack_full_multiIndex% s" þ$ ýù z/TestDataFrameReshape.test_stack_full_multiIndexÚorderedFTÚlabelsZyxzZyxyc C sb t j|td|d}tdddgg|d}| ¡ }t |j|g¡}tdddg|d}t
|
||¡ d S )Nra ©r} rb rß rî rï r r* ) ro ÚCategoricalIndexrf r r r r, r( r r r- )r rb rc Úcidxr r/ rÙ r0 r r r Ú%test_stack_preserve_categorical_dtype9 s z:TestDataFrameReshape.test_stack_preserve_categorical_dtypez labels,datara r: Zzyxc C s t j|t||d}t jddg|d}t ||g¡}tt|g|d}| ddg¡}t jt||d} t|t dg| |g¡d}
|
t ||
|
¡ d S ) Nrd rX Úv)rb r r r% r* )
|
ro re r# r r, r r r r r- ) r rb rc rZ rf Zcidx2rÙ r r/ Zs_cidxr0 r r r Ú+test_stack_multi_preserve_categorical_dtypeH s
|
z@TestDataFrameReshape.test_stack_multi_preserve_categorical_dtypec C st t ddddg¡}t||d}| ¡ }t ddddgd d
|
gg¡}tt ddddddddg¡|d }t ||¡ d S ) Nr# r$ ry ©rc rd r r% r& r" rc rd r* ) ro r r r r r, r r r- )r r¶ r r/ r( r0 r r r Ú,test_stack_preserve_categorical_dtype_values] s ÿzATestDataFrameReshape.test_stack_preserve_categorical_dtype_valueszindex, columnsr$ c C st t ||d d¡}| ¡ }t |j ¡ ¡}t | ¡ ||jd}t ||¡ t
|
|jj ¡}t
|
|j ¡}t ||¡ d S )Nr' r% )r Úfillnar r rS r( r] r) r r r ZasarrayrÔ Zassert_numpy_array_equal) r r( r) r r Ú new_indexr0 Z stacked_codesZexpected_codesr r r Ú)test_stack_multi_columns_non_unique_indexh s
|
ÿ z>TestDataFrameReshape.test_stack_multi_columns_non_unique_indexc C s® t jdddgddgd}ttjddd gd d
|
t dddg¡d |d }|j|d }| t¡j|d }t |j
|
j gd |j j gd |j d }t |j|¡ t | t¡|¡ d S )N)rc r ©rc r% )rd r% r# r$ r r r% ZInt64rA rj r* r® r& )r rS r ro rÊ r r rW rÌ r rc rB rd r) r r- rÝ r )r r¯ r( r r/ r0 Zexpected_dtypesr r r Ú"test_unstack_mixed_extension_types} s þû ÿz7TestDataFrameReshape.test_unstack_mixed_extension_typesÚbazc C s t jdgddggddgd}tddgdd gg|d
|
d gd }d |j_td dddggt jddddgdd gdd}d|j_| ¡ jddj d|d}t
|
||¡ d S )Nr r ry r rq r r& r% r" rd rc r' r )ry rc )ry rd )r rc )r rd r rç )rè r¯ ) r r, r r) rQ rS r( r Ú swaplevelÚ
|
sort_indexr r )r r¯ rÕ r r0 r/ r r r Ú test_unstack_swaplevel_sortlevel s
|
ÿþz5TestDataFrameReshape.test_unstack_swaplevel_sortlevel)=Ú__name__Ú
|
__module__Ú __qualname__r! r1 r? r] rj rq rw rx r r r ÚmarkÚ parametrizer rS r r r¨ r´ rÅ rÆ rÏ rÖ rÛ rÞ r rò r rä ræ rë rË rô rö rù rÿ r r rf r+ r( r± Ú permutationsrB rF rH rJ rN rQ r_ ra rg ri rk r, rn rp rt r r r r r s .$ "*
|
ÿ
|
þú &ø
|
õüôþ"
|
*,), þþ
|
|
**þþ
|
|
=1 )
|
|
|
|
ëþ þþ ýþ
|
|
r c C s t ddddgdd} t dddd g¡| _| ¡ }tdtjdgddtjgd
|
td d }t
|
||¡ | jd d}tdd dgddd gd
|
td d }t
|
||¡ d S )Nr# r$ ry rÌ rA rC rD rE rF rK ra r* r rI ) r r rS r( r r r rË rf r r )rZ r/ r0 r r r Ú test_unstack_fill_frame_object¤ s
|
ÿ ÿ ÿr{ c C s t tjdddgdgdgdgdddddgd } | ddg¡ ¡ }t tjddddggtdgdd
|
tddgdggd d gd d ggd dgd d}t ||¡ d S )Nz 2017-08-27 01:00:00.709949+0000ÚUTC)Útzr# r$ ry )Ú timestampr# r$ ry r~ r r r r% rÓ r' ) r ro r¦ rU r r r r r )r r/ r0 r r r Ú"test_unstack_timezone_aware_valuesº s& ü
|
ù ýý r c C sn t ddddd} td| iddd gd
|
}| ¡ }t| tddd gdggd d d gd d d ggdd
|
}t ||¡ d S )Nr Z20180101Z20180103zAmerica/New_York)rK ÚstartÚendr} rc r# r$ ry r* r r% r& r× )r r r r r r r- )Útsr r/ r0 r r r Ú test_stack_timezone_aware_valuesÒ s &þr r[ TFc C s^ t jg t jdt jg t jdg}tt jt|g g gdd}tt jdj| d}t ||¡ d S )NrA r× )rB r( r` )
|
r rÊ rX r rh r r r r r- )r[ r. r0 r/ r r r Útest_stack_empty_frameÞ s r rJ c C s8 t tjdj| dj|d}t tjd}t ||¡ d S )NrA r` rI )r r rX r r r r )r[ rJ r/ r0 r r r Ú test_stack_unstack_empty_frameç s ÿ r
c C s2 d} t jt| d ttjd ¡ W 5 Q R X d S )Nz'index must be a MultiIndex to unstack.*r~ rA )r r rÁ r r rX r )r
r r r Ú test_unstack_single_index_seriesò s r c C sÌ t ddgddgddgddgdd gd
|
} | d d d dg¡} | jdgdd} | jd ddjd dd}| jd d gdd}t ddddgddddggtddgd dtjddddgd dd d gdd}t ||¡ d S )NZAliceZBobg #@r` FTr ÚfemaleÚmale)rQ ÚscoreÚemployedÚkidsÚgenderrQ r r r rI r g @r )r r Fr )r r Tr )r r Fr )r r Tr r r' )r rU r r r rS r r )r r0 r/ r r r Ú test_unstacking_multi_index_dfù s2 ûÿ ü
|
ùý r c C s t jddgddgd} tddddgg| d}| d¡}tdd gdd
|
}t jd d gd dgd}tddgddgg||d }t ||¡ d S )Nr4 )r6 rG r# r r% r r r6 rG r )r r5 )r r6 r' )r r, r r r rS r r )r) r r/ Z new_columnsrm r0 r r r Ú2test_stack_positional_level_duplicate_column_names s
|
r c s² t tddddgg¡}ttj d¡tj d¡ tj¡tj d¡tj d¡d|d | stt dd j
|
j D stt ¡ }tj fd
|
d td D td d d}t ||¡ d S )Nr@ rc rd r rð rO r* c s s | ]}|j j V qd S r¢ )Zmgr_locsZ is_slice_like©rs r5 r r r Ú <genexpr>3 s z5test_unstack_non_slice_like_blocks.<locals>.<genexpr>c s g | ]} | ¡ qS r )r )rs Ún©r r r ru 7 s z6test_unstack_non_slice_like_blocks.<locals>.<listcomp>r3 r% )Úkeysrè )r r, r+ r r r¥ r³ rW rX Úanyr9 r: r; r ro Úconcatr r )r< rÕ r> r0 r r r Ú"test_unstack_non_slice_like_blocks% s
|
|
|
üù &r c @ s> e Zd Zdd Zej dddddddgddddddggdd d
|
d d d gdddddgfdddddgdddddggdd d
|
d d gddddgfdddddgdddddggdd d
|
d d gddddgfg¡dd Zdd Zdd Z ej de
|
ddddge ddgdddgge e d¡ d¡d¡e e d¡d ¡gd ge
|
ddddge ddgddgge e d¡ d¡d¡e d dd gd ¡gd ge d ¡dddge ddgddgddgge e d¡ d¡d¡e dd dgdd dg¡e d dd gd ¡gd gg¡d!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zej d+d,d-g¡d.d/ Zd0d1 Zd2d3 Zd4d5 Zd6d7 Zd8d9 Zd:d; Z d<d= Z d>d? Z d@dA Z dBdC Z dDdE Z!dFdG Z"ejj#dHdI Z$ej dJe%j& 'dKdL dMD ¡¡ej dNe(d¡dOdP Z)dQdR Z*dSdT Z+dUdV Z,dWdX Z-dYdZ Z.d[d\ Z/d]d^ Z0d_d` Z1dadb Z2dcdd Z3dedf Z4dgdh Z5didj Z6dS )kÚTestStackUnstackMultiLevelc C s6 |}| ¡ }| ¡ | t¡ ¡ | tj¡ ¡ d S r¢ )r rW r\ r rg ©r Ú/multiindex_year_month_day_dataframe_random_dataÚymdr r r r Ú test_unstack< s
|
z'TestStackUnstackMultiLevel.test_unstackz5result_rows,result_columns,index_product,expected_rowr% Ng >@r& Úix1Úix2Zcol1Zcol2Zcol3Zcol4c C sp t ||d ddg¡}|jdd
d¡}t |gtj|dd
|ggd dgdtdgddd}t ||¡ d S ) Nr r r r% r& r r r ) r rU rá r r r, r r r )r Z result_rowsZresult_columnsZ index_productZ expected_rowr/ r0 r r r Útest_unstack_partialI s ÿ ÿ ûz/TestStackUnstackMultiLevel.test_unstack_partialc C sR t ddddg¡}ttj d¡|d}| ddg¡}|jdd d
|
}t ||¡ d S ) N)r r r )r r r )r% rq r% )r% Zquxr% r3 r* r% r& ré ©rè Úhow)
|
r rS r r r¥ r³ r r[ r r )r r( r\ r r0 r r r Ú&test_unstack_multiple_no_empty_columnst s
|
ÿzATestStackUnstackMultiLevel.test_unstack_multiple_no_empty_columnsc
|
C sî |}| ¡ }| ¡ }t ||¡ |jdd}| d¡}| ¡ }t |jdd|¡ |d d d
}| d¡}| ¡ dd¡}t |jdd|¡ | dd¡}| d¡jdddd}| d¡ dd¡}t |jdd|¡ | ¡ }|jddd}| ¡ }t ||¡ | d¡ d¡}| d¡}| ¡ }t ||¡ | d¡}| d¡}t ||¡ | d¡}| ¡ d¡ d¡}t ||¡ | d¡jd d
d d d
f }| ¡ ¡ }| ¡ } t || |j ¡¡ | d¡ d
|
¡}| d¡ d¡}t
|
||¡ d S ) Nr& r® r rH r% rç F)rè Z ascendingr" r° ) r r r r rs rr rY r- Úreindexr( Z assert_equal)
|
r r r r Ú restackedZ unlexsortedr/ r0 r Z ymd_stackedr r r Ú
|
test_stack sJ
|
|
|
|
|
|
z%TestStackUnstackMultiLevel.test_stackzidx, columns, exp_idxÚababr) r* r+ r# r$ r" r3 r× r ))r# r& )r$ r% )r# r% )r$ r& r^ c C sp t t d¡ dd¡||d}| ¡ }tt d¡|d}t ||¡ |jj dksRt
|
|j|j }}t ||¡ d S )Nrï r3 r" r' r* F) r r r r r r r r- r( Z is_uniquer; Úassert_index_equal) r r r) Zexp_idxr r/ r0 ÚliÚrir r r Útest_stack_duplicate_index· s &ý z5TestStackUnstackMultiLevel.test_stack_duplicate_indexc C s@ d}t t|¡ dddg¡}| d¡}| ¡ }t ||¡ d S )Na day,time,smoker,sum,len
|
Fri,Dinner,No,8.25,3.
|
Fri,Dinner,Yes,27.03,9
|
Fri,Lunch,No,3.0,1
|
Fri,Lunch,Yes,13.68,6
|
Sat,Dinner,No,139.63,45
|
Sat,Dinner,Yes,120.77,42
|
Sun,Dinner,No,180.57,57
|
Sun,Dinner,Yes,66.82,19
|
Thu,Dinner,No,3.0,1
|
Thu,Lunch,No,117.32,44
|
Thu,Lunch,Yes,51.51,17ÚdayÚtimeZsmokerr& )ro Zread_csvr rU r r r r )r rZ r r/ Úreconsr r r Útest_unstack_odd_failureé s
|
|
z3TestStackUnstackMultiLevel.test_unstack_odd_failurec C sr |}|j }d|d< |jddd}| ¡ }|d ¡ ¡ }tj|d |dd |jd ksZt|d jtj ksntd S )Nr )r Zfourr% )r¯ rè FrÀ r )
|
r rs r r r- rQ r; rB r Zfloat_)r Ú multiindex_dataframe_random_dataÚframer r r/ r r r Útest_stack_mixed_dtypeÿ s z1TestStackUnstackMultiLevel.test_stack_mixed_dtypec
|
C s t ddddddgddddddgddddddgd d d
|
d
|
d
|
d gt d ¡d }| ddddg¡ t¡}| ¡ }| ¡ }t ||
|
|j ¡ t ¡¡ d S )NZnaiveÚactiver# r$ r% r& r" r3 ÚhiZbyeZpeaceg @)r r½ Úbarcoderh Úextrar r½ r³ rh )r r r ÚgroupbyrA r8 r r r r- r¢ r( rW rT )r r r/ r r£ r r r Útest_unstack_bug s ûÿ
|
z+TestStackUnstackMultiLevel.test_unstack_bugc C sP |}| ¡ }|jjdks t|jjddgks0t| ¡ }|jj|jjksLtd S )NrÐ r½ rÑ )r r( rQ r; r) rV r )r r® r¯ r r£ r r r Ú!test_stack_unstack_preserve_names s z<TestStackUnstackMultiLevel.test_stack_unstack_preserve_namesÚmethodr r c C s| |}|j d }tjtdd t||d W 5 Q R X |dkrx|jd d
df }tjtdd t||d W 5 Q R X d S )Nr zdoes not match index namer~ Zmistaker r )rY r r ÚKeyErrorÚgetattrrá )r r¸ r® r¯ r r\ r r r Ú#test_stack_unstack_wrong_level_name' s
|
z>TestStackUnstackMultiLevel.test_stack_unstack_wrong_level_namec C s* |}| d¡}|j dd}t ||¡ d S )NrÑ r% r® r£ )r r® r¯ r/ r0 r r r Útest_unstack_level_name: s
|
z2TestStackUnstackMultiLevel.test_unstack_level_namec C sT |}| d¡}| d¡}| ¡ d¡}t ||¡ | d¡}| ¡ }t ||¡ d S )NrÑ r½ r )r r r r r- )r r® r¯ r r/ r0 r r r Útest_stack_level_nameA s
|
|
|
z0TestStackUnstackMultiLevel.test_stack_level_namec C s2 |}| ddg¡}| d¡ d¡}t ||¡ |jj|jjksBt|d }| ddg¡}t ||d ¡ | ddg¡}| dd¡ dd¡}|jdd}t ||¡ |j j|j jks¶t| ddg¡}| d¡ d¡j
|
ddd }t ||¡ | ddg¡}| d¡ d¡j
|
ddd }t ||j d d
|jf ¡ d S )
|
NÚyearÚmonthrc r r% r& r® ré r ) r r r r) rV r; r rr rs r( r[ rY )r r r r r0 r\ Z s_unstackedr£ r r r Útest_stack_unstack_multipleM s& z6TestStackUnstackMultiLevel.test_stack_unstack_multiplec C s> |}| ddg¡}tjtdd | ddg¡ W 5 Q R X d S )Nr¾ r¿ zlevel should containr~ r )r r r rÁ r r r r r Ú test_stack_names_and_numbersl s z7TestStackUnstackMultiLevel.test_stack_names_and_numbersc C sf |}| ddg¡}tjtdd | ddg¡ W 5 Q R X tjtdd | dd g¡ W 5 Q R X d S )
|
Nr¾ r¿ zToo many levelsr~ r& r" znot a valid level numberr r )r r r Ú
|
IndexErrorr r r r r Ú!test_stack_multiple_out_of_boundsw s z<TestStackUnstackMultiLevel.test_stack_multiple_out_of_boundsc C s t jddddddgddd}tddgd d
|
d }d d d dddg}t ||g¡}t||d}| ¡ }|jd d}|jdd}t jdddgddd} td d dgd ddgd| ddgd}
|
d
|
|
|
j_ t
|
||
|
¡ t
|
||
|
¡ t
|
||
|
j ¡ t jddddddgddd}t jdddddd gdd d}t ||g¡}t||d}| ¡ }|jd d}|jdd}t jdddgddd} t jd dddddgdd d} tt jt jt jt jd d gt jt jdd t jt jgddt jt jt jt jgg| | d}
|
t
|
||
|
¡ t
|
||
|
¡ t
|
||
|
j ¡ d S ) Nz2013-01z2013-02z2013-03ÚMZperiod©rK rQ rc rd r" Ústrr r% r& r3 r@ r^ r* r® r rj r' Úperiod1ú2013-12z2013-11ú2013-10z2013-09z2013-08z2013-07Úperiod2)ro Ú PeriodIndexr r rÉ r r r r) rQ r r r r rË ) r Úidx1Úidx2Úvaluer r\ Úresult1Úresult2Úresult3Ze_idxr0 Úe_colsr r r Útest_unstack_period_series s| ý ÿ ÿ ýý ÿýýù
|
z5TestStackUnstackMultiLevel.test_unstack_period_seriesc C s t jddddddgddd}t jddddddgddd}d d
|
d d d dgdd d d d
|
d gd}t ||g¡}t||d}| ¡ }|jd d}|jdd}t jddgddd} t jddddddgddd}
|
t d ¡ |
|
g¡} td d dd
|
dd gd d
|
d d d d gg| | d} t || ¡ t || ¡ t jddddgddd} t jdddgddd}
|
t d ¡ | g¡} td d d
|
d gd d
|
dd gdd d d gg|
|
| d} t || ¡ d S )Nz2014-01z2014-02rÄ rÇ rÅ rÈ rÉ rÊ r% r& r" r3 r@ r^ rj r* r® r z A A A B B Br' zA A B B) ro rË r rÉ r r r$ r r ) r rÌ rÍ rÎ r r rÏ rÐ rÑ Ze_1Ze_2rÒ r0 r r r Útest_unstack_period_frameÉ s^ ýý" ý ÿ
|
ÿ ÿ ÿz4TestStackUnstackMultiLevel.test_unstack_period_framec C sþ dgd dgd }dgd dgd }t dddgd ¡}tj d d
|
d ¡}t||||d }| d dg¡}d|j_| d¡}t
|
j t dd | d¡ ¡ W 5 Q R X | d¡jdd}| d¡} |jd d
dgf d¡ ¡ d¡}
|
d|
|
j_t | |
|
¡ d S )Nr% r" r& r# r$ z
|
2013-01-03z
|
2013-01-04z
|
2013-01-05r rL r^ )ÚIDÚNAMEÚDATEÚVAR1r× rÕ ZParamszCould not convertr~ zW-THUT)Z numeric_onlyrØ )ro Ú to_datetimer r¥ Úrandintr rU r) rQ r r r r ZresampleZmeanr rY r r ) r Zid_colrQ ÚdateZvar1r ZmultiZunstZdownrÍ rÎ r r r Útest_stack_multiple_bugø s
|
|
$z2TestStackUnstackMultiLevel.test_stack_multiple_bugc C sv t ddgddgddgd}| ddg¡}| ¡ jd d
|
}t|t| ¡ ksRt| ¡ jd d
|
}t || ¡ ¡ d S ) NÚa1Úa2Úb1Úb2r% rC rc rd Fr` T) r rU r r r8 r[ r; r r )r r r r r r Útest_stack_dropna s z,TestStackUnstackMultiLevel.test_stack_dropnac C s t ddddddddgddddddddgddddddddggddddgddddggd}dddg|j_ddg|j_| ddg¡ d S ) Nr r% r' r# r$ ry r r )r r( rV r) r r r r r Ú"test_unstack_multiple_hierarchical s ýú z=TestStackUnstackMultiLevel.test_unstack_multiple_hierarchicalc C s d}t tjjd|dtjjd|dtjjdd|dtjjdd|dtjjd |dtj |¡d
|
}| d d d ddg¡}| d¡ d S )Néè rL )Úsizei, iùÿÿÿr_ iíÿÿÿé i¸ )rc rd r r r r rc rd r r r )r r r¥ rÚ r³ rU r )r ZNUM_ROWSr Zidfr r r Ú test_unstack_sparse_keyspace& s
|
úÿ z7TestStackUnstackMultiLevel.test_unstack_sparse_keyspacec C s~ ddgddddgg}ddddgddddgg}t ||}ttj dd¡|d}| ¡ }t|jdksft| ¡ }t
|
||¡ d S )Nr r% r& r" r3 r* ) r r r r¥ r³ r r8 r) r; r r r )r r. rÔ r( r r/ r¬ r r r Ú test_unstack_unobserved_keys; s
|
z7TestStackUnstackMultiLevel.test_unstack_unobserved_keysc C s G dd dt j}| ¡ z}| t d|¡ ttj dd¡t d¡t d¡gd}d}t j
|
t |d( t j td d | ¡ W 5 Q R X W 5 Q R X W 5 Q R X d S )
|
Nc s" e Zd Zdd fdd Z ZS )zaTestStackUnstackMultiLevel.test_unstack_number_of_levels_larger_than_int32.<locals>.MockUnstackerN)Úreturnc s t j|| tdd S )NúDon't compute final result.)ÚsuperÚ__init__Ú Exception)r ÚargsÚkwargs©Ú __class__r r rë P s zjTestStackUnstackMultiLevel.test_unstack_number_of_levels_larger_than_int32.<locals>.MockUnstacker.__init__)ru rv rw rë Ú __classcell__r r rï r Ú MockUnstackerO s rò Ú
|
_Unstackeri r& r* z$The following operation may generater~ ré )Ú reshape_libró ÚcontextÚsetattrr r r¥ r³ r r Zassert_produces_warningr r r rì r )r Z monkeypatchrò Úmr r
r r r Ú/test_unstack_number_of_levels_larger_than_int32J s
|
þzJTestStackUnstackMultiLevel.test_unstack_number_of_levels_larger_than_int32r. c c s* | ]"}t jt d ddg|¡ddV qdS )r r% r& r© N)r± r² rz )rs Úwidthr r r r b s ÿz$TestStackUnstackMultiLevel.<genexpr>r Ú stack_levc s` t |ddddgddddggd}t|tdgd ¡t fdd jD s\td S )Nr r% r× r3 )r) rZ c 3 sF | ]>} j D ]2} j||f j|| f|d f kV q qdS )r% N)r) rY ©rs ÚrowrÈ ©r Z
|
df_stackedrú r r r n s ý ÿzSTestStackUnstackMultiLevel.test_stack_order_with_unsorted_levels.<locals>.<genexpr>)r r r+ r ré r( r; )r r. rú r) r rý r Ú%test_stack_order_with_unsorted_levels` s
|
ýz@TestStackUnstackMultiLevel.test_stack_order_with_unsorted_levelsc s t dddgdddggt tdd¡t tdd¡gd}t|tdt dt| ¡ dd¡d t fd d
|
j
|
D s|t d S ) Nrc r rd r" r× r@ rH )r) r( rZ c 3 sH | ]@} j D ]4} j||f d ¡j||d f|d f kV q qdS )r r% N)r) rY r rû r r r r s þz]TestStackUnstackMultiLevel.test_stack_order_with_unsorted_levels_multi_row.<locals>.<genexpr>) r r rª r+ Útiler r r8 r ré r( r; )r rÕ r r r Ú/test_stack_order_with_unsorted_levels_multi_rowu s þ ÿ þzJTestStackUnstackMultiLevel.test_stack_order_with_unsorted_levels_multi_rowc C sì t d¡}t dd |D dd |D g¡}t|jddgd}d|j_d |i}tj|d
|
d }d dg|j_ |j
|
t |jd
|
d }| dd g¡ dd g¡}td dgddgddgddgddggdd
|
dddgtjdd gdd gd d }t ||¡ d S ) Nr@ c S s g | ]}d | qS )r$ r r r r r ru s zVTestStackUnstackMultiLevel.test_stack_unstack_unordered_multiindex.<locals>.<listcomp>c S s g | ]}d | qS r r r r r r ru s r$ r# r rÐ r5 r% rç rÑ Za0Zb0rÝ rß rÞ rà Úa3Zb3Za4Zb4r r& r" r3 )r# r5 )r$ r5 r r' )r r Zvstackr r r) rQ ro r rV r¢ r# r r r rS r r )r rn rZ r Zsecond_level_dictZmulti_level_dfr/ r0 r r r Ú'test_stack_unstack_unordered_multiindex s,
|
þÿ ÿýzBTestStackUnstackMultiLevel.test_stack_unstack_unordered_multiindexc C s^ |}d|d< d|d< | d¡}|d jtjks2t|d jtjksFt|d jtjksZtd S ) Nr r r& r r¿ ro )r r% )r r% )r rB r rh r; Zobject_r r r r Útest_unstack_preserve_types s
|
z6TestStackUnstackMultiLevel.test_unstack_preserve_typesc C sj t t d¡d¡}t d¡}t|gd ddgg |gd t d¡ d¡g d}tt d¡|d}| ¡ }|jd ksxt| ¡ }t
|
|| |j ¡¡ tddgg|gd t d¡ d¡g|gd d}tt d¡|d}| d¡}|jd ksòtt|gd
|
ddgg |gd
|
|gd
|
t d¡ d¡g |gd
|
d}tt d¡|d}| d
|
¡}|jd ksftd S ) Néô r& r` r r% r× rã r* )r r& r3 )r rÿ r r rª r r r r; r r r- r¢ r( )r rÔ r¯ r( r\ r/ r r r r Ú!test_unstack_group_index_overflow¬ s2
|
þþ
|
$þ
|
z<TestStackUnstackMultiLevel.test_unstack_group_index_overflowc C sè t dddgdddgdgd d dd g¡}d
|
|d < |sLt|jjd ksLt| d ¡}|d d¡|d < t ddddgtj dddggt
|
ddgdddt j ddd dgd d gdd}|rØ|d d¡|d< |d d¡|d< t ||¡ d S )Nrc rd ÚcaÚcbrß r" )r# r$ rh r# r$ r% Úis_r& )r r r g $@r r rÌ )rB rQ )rh r )rh r )r r r r' rX )r rU r8 r9 r: r; r rl r rË r r rS rW r r )r r< r r/ r0 r r r Ú+test_unstack_with_missing_int_cast_to_floatÑ s2 ýÿú
|
|
þýzFTestStackUnstackMultiLevel.test_unstack_with_missing_int_cast_to_floatc C sÐ t ddddgddddgddddgddddgd}| dddg¡}d d
|
d d g}|jj|dd |_|jj|dd |_| d¡d ¡ j}td d
|
d d gd d
|
d d ggddddgddddggddgd}t ||¡ d S )Nr% r& r" r3 )ÚL1ÚL2ÚL3r5 r
|
r r Zn1Zn2Zn3)r. r¯ )r5 r% r rÓ ) r rU r( Z
|
set_levelsr rs r r r¦ )r r§ Z
|
new_levelsr/ r0 r r r Ú test_unstack_with_level_has_nanñ s$
|
|
|
|
üÿ ýz:TestStackUnstackMultiLevel.test_unstack_with_level_has_nanc C s t t ddg¡t dddddg¡d}| d ¡}t d
|
tjtjgtjd
|
d
|
gtjd
|
d
|
ggtd d d gtd ddgd}t ||¡ d S )Nr% r@ )r NN)r r& r )r r& r% )r r" r )r r" r% r r& r )r N)r r ©r r% )r r& )r r" r' )
|
r r rò r rS r rË r r r rþ r r r Ú$test_stack_nan_in_multiindex_columns s$ ûÿþ
|
" ýz?TestStackUnstackMultiLevel.test_stack_nan_in_multiindex_columnsc C sî t dgd dgd t td¡t td¡g¡}tt d¡ dd¡|d}| d dg¡}td
|
tj
|
gtj
|
dgd tj
|
gtj
|
d gdtj
|
gtj
|
d gd tj
|
gtj
|
dggddgt d
|
gd d gd t td¡t td¡g¡d}t ||¡ d S )Nrc r& rd r¥ Zccddr` r3 r r% r r" r^ r@ r_ ZaabbaabbZcdcdcdcdr ) r rÉ ro r rf r r r r r rË r r )r rÙ r r/ r0 r r r Ú"test_multi_level_stack_categorical s6 ýÿø
|
ýÿôz=TestStackUnstackMultiLevel.test_multi_level_stack_categoricalc C s¸ t t d¡ dd¡tjdtjfdgddgdtdd gd
|
d tjd }| ¡ }t d tjgtjd gdtjgtjdggtddgdd tjdtjfdd tjfdgd
|
dgdd}t
|
||¡ d S )Nr3 r& rc re rY rZ r r r% ZNumr ©r) r( rB r g @r2 rd )r r$ )r% r$ r ) r r r r r rS rË r rh r r r )r Zdf_nanr/ r0 r r r Útest_stack_nan_level@ s& ÿú" ÿýz/TestStackUnstackMultiLevel.test_stack_nan_levelc C s t dgddgg¡}tdt ddg¡i|d}| ¡ }ttjdgddgdtjdgddgdd dgd}t d
|
d g¡|_t ||¡ d S ) Nrc r r% r¶ r# r$ r* r| r )r¶ r )r¶ r% )
|
r r, r ro r r rS r) r r r r r r Ú test_unstack_categorical_columnsT s þûz;TestStackUnstackMultiLevel.test_unstack_categorical_columnsc C s¾ ddg}ddg}dddg}t j|||gdd d
|
gd }ttt|}t||d gd }| d d
|
g¡}|j d¡|_d|j d d
df< |
|
d d
|
g¡ ¡ }|j dd
|
d d
|
g¡ ¡ }t ||¡ d S )NZITAZFRAÚA1ÚA2ZCRTZDBTÚNETÚPAEÚVARÚTYPr ZVALUE)rZ r( r) r i' )ZA0r r% rç )r r, rf r+ r8 r r r) r÷ rY r rs r r- ) r r r r r ÚVZDFr/ r0 r r r Útest_stack_unsortedc s
|
z.TestStackUnstackMultiLevel.test_stack_unsortedc C sê t jddgddggddgd}tdd d
|
gd d }t d dddgddddgddddgg¡}t|||t ¡ d}| d¡}| tj
|
¡ d¡ t ¡ ¡}t ||¡ ||j d t ¡ ¡||j d < | d¡}| t¡ d¡}t ||¡ d S )NZ54511Z54515r? Zt_meanZstationÚelementr r% r& r" r« r r6 éâ rß é× r éÜ i1 éè r r r )r r, r r rÊ r ro Z
|
Int64Dtyper rW rX r r r) Z Float64DtyperÌ )r r) r( Zarrr r/ r0 r r r Útest_stack_nullable_dtypeu s ÿ(
|
"
|
z4TestStackUnstackMultiLevel.test_stack_nullable_dtypec C s ddgddgddgg}t j|dd}tdddgi|d }| d
|
¡}tdgdggt jd gd d
|
gdt jd d gddgdd}t ||¡ d S )Nr# r% r& ÚredÚblue)r5 r r6 r r÷ r* r5 )r÷ r# )r% r" )r& r# r r6 r )r rÉ r r rS r r )r Zarraysr r r/ r0 r r r Ú test_unstack_mixed_level_names s
|
|
ýz9TestStackUnstackMultiLevel.test_unstack_mixed_level_names)7ru rv rw r r rx ry r r¡ r¤ rf r r rÿ r rª rS r© r r° r¶ r· r» r¼ r½ rÀ rÁ rà rÓ rÔ rÜ rá râ ræ rç Zslowrø r± ÚchainÚ from_iterabler+ rþ r r r r r r r r r r r r! r$ r r r r r ; s¾
|
ü ü üóþ
|
8þþý (þý ýþýìþ$
|
|
|
E/
|
þþ % !r )%r Úior r± Únumpyr r Z pandas.errorsr Zpandasro r r r r r r
|
r Zpandas._testingZ_testingr Zpandas.core.reshaper rô r r{ r r rx ry r r
r r r r r r r r r Ú<module> s@ $
|
|
