U
|
¬ý°dì' ã @ sX d dl Zd dlZd dlZd dlmZmZmZ d dlm Z
|
G dd dZ G dd dZ dS )é N)Ú DataFrameÚ
|
MultiIndexÚSeriesc @ s e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zej dddg¡dd Zd d Zd d Zd!S )"ÚTestMultiLevelc C sÈ |}| d¡ ¡ }|j|jdd}|j dd tj¡}t ||¡ |d j|jdd}|d j dd tj¡}tj||dd |j j ddd ¡ }|j|jdd}|j dd tj¡j }t ||¡ d S ) NÚmonthé ©ÚlevelÚAF©Z check_names©Úaxis)Úcolumnsr )
|
ÚgroupbyÚsumÚreindexÚindexÚ transformÚnpÚtmÚassert_frame_equalÚassert_series_equalÚT)ÚselfÚ/multiindex_year_month_day_dataframe_random_dataÚymdZ
|
month_sumsÚresultÚexpected© r úSd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/test_multilevel.pyÚtest_reindex_level s z!TestMultiLevel.test_reindex_levelc C s0 |}|j ddg }|jddg }t ||¡ d S )Nr é ©ÚfooÚone©Úbarr$ )ÚilocÚlocr r )r Ú multiindex_dataframe_random_dataÚframer Z reindexedr r r Ú test_reindex# s zTestMultiLevel.test_reindexc C s |}|j d d d
}| |¡}|j |ks,t|j| }|j |¡sFt|j}|j|d}|j|ksft|jd d
|f }|j |¡std S )Né
|
©r )r r ÚAssertionErrorr( Úequalsr r )r r r Z new_indexÚchunkZymdTr r r Ú test_reindex_preserve_levels* s
|
|
z+TestMultiLevel.test_reindex_preserve_levelsc C sb |}|d }|j d¡}|j|dd}| dd ¡}| dd ¡}| |j ¡}tj||dd d S ) Nr
|
r F)Ú
|
group_keysc S s | d S ©Né r ©Úxr r r Ú<lambda>E ó z7TestMultiLevel.test_groupby_transform.<locals>.<lambda>c S s | d S r3 r r5 r r r r7 F r8 r )r Úget_level_valuesr Úapplyr r r r ) r r) r* ÚsZgrouperÚgroupedZappliedr r r r r Útest_groupby_transform= s z%TestMultiLevel.test_groupby_transformc C s\ t dgdgdggdgdgdggdddgd}ttj d ¡gd
|
d d d g|d}|jdd d S )Nr# r& Úbazr r$ ÚtwoZthree©ÚlevelsÚcodesÚnamesé ÚaÚbÚcÚd)r r r )r r r ÚrandomZrandr )r ÚmidxÚdfr r r Útest_groupby_cornerJ s ý z"TestMultiLevel.test_groupby_cornerc C s t ddddddg¡}tddd d
|
d d gd dddddgg|d}|jdd|j dd ¡ }|jddd}| ¡ }|jddgk ¡ st d S )N)Úf1Ús1)rM Ús2)Úf2rN )rP rO )Úf3rN )rQ rO r r4 r! rD é é é é é r, é é r- r c S s | d dkS )Nr )rP rQ r )Úur r r r7 a r8 z:TestMultiLevel.test_groupby_level_no_obs.<locals>.<lambda>r ©r r rP rQ )
|
r Ú from_tuplesr r( r Úmapr r Úallr. )r rJ rK Zdf1r< r r r r Útest_groupby_level_no_obsT s úÿ
|
( z(TestMultiLevel.test_groupby_level_no_obsc C sJ |}|d d
j }|d |d< t|jts.t|d |d k ¡ sFtd S )NrR )éÐ r rT )r_ r r, )r Ú
|
isinstancer r r. r] )r r r rK r r r Ú.test_setitem_with_expansion_multiindex_columnsg s
|
z=TestMultiLevel.test_setitem_with_expansion_multiindex_columnsc C sÀ t dddgt dddg¡d}t dd d
|
gt d d dg¡d}|| }|j |j¡}| |¡| |¡ }t ||¡ |d d d
|d d d
}|j |j¡}| |¡| |¡ }t ||¡ d S )Nr r4 r! )r
|
r )r
|
r4 )ÚBr! )Údatar rD rR rS )ÚZr )rd r4 éÿÿÿÿ)r r r[ r Úunionr r r )r r6 ÚyÚresZ exp_indexÚexpr r r Útest_alignmentq s ÿ ÿ z TestMultiLevel.test_alignmentc C s |}|j ddgd ¡ }|j d¡}|j d¡}| ||g¡ ¡ }tj||dd |jj|jjd d
ksnt|j |jjd d
d ¡ }t ||¡ d S )Nr r r Fr r4 )r Zmeanr r9 r r rC r. )r r r r Zk1Zk2r Úresult2r r r Útest_groupby_multilevel
s z&TestMultiLevel.test_groupby_multilevelc C sD t ddddg¡}ttj dd¡||d}| d¡|d< | ¡ }d S ) Nr" )r# r? r% )r& r? rD )r r r )ZTotalsÚ )r r[ r r rI Úrandnr Z _consolidate)r r rK r r r Útest_multilevel_consolidate s
|
ÿz*TestMultiLevel.test_multilevel_consolidatec C sú t dddgddggddddddgddddddggd}ttj d¡|d }ttj dd
|
¡|d }|d }|jd }|d d
}|j d¡|_t
|
||¡ t
|
||¡ t j t d d |d W 5 Q R X |jd }| d¡}|d d
}|j d¡|_t ||¡ t ||¡ t dddgddggddddddgddddddggd}ttj d¡|d }ttj dd
|
¡|d }|d }|jd }|d d
}|j d¡|_t
|
||¡ t
|
||¡ |jd }| d¡}|d d
}|j d¡|_t ||¡ t ||¡ d S )N©r# r& r )r# r> r )r# Úquxr r r r4 )rA rB rS ©r rD z ^\(\('foo', 'bar', 0\), 2\)$)Úmatch)rp r4 )r# r& )r# r> )r# rq )r r r rI rn r r( r Z droplevelr r ÚpytestZraisesÚKeyErrorÚxsr )r r Zseriesr* r rk r r r r Útest_level_with_tuples sL þ
|
|
|
þ
|
|
|
z%TestMultiLevel.test_level_with_tuplesc C s |}|j ddgdd}|jdddddd g }t ||¡ |jj ddgddd
|
}t ||j¡ |jddg }t ||¡ |d jddg }t ||d ¡ |jjd d
ddgf }t ||j¡ d S ) Nr# rq r r r r4 rT rU rV rZ r
|
)r r' r r r r( r )r r) r* r r r r r Ú$test_reindex_level_partial_selectionÎ s z3TestMultiLevel.test_reindex_level_partial_selectionrH rD c C sx dddg}t dgdgdg|gd}||jdk |¡}| ¡ j}tdgdgdggg g g gdddgd}t ||¡ d S ) NrE rF rG r r4 r! )rE rF rG rH r@ )r rE r Úfirstr r r Zassert_index_equal)r rH r2 rK Úgr r r r r Ú+test_empty_frame_groupby_dtypes_consistencyá s
|
|
ÿz:TestMultiLevel.test_empty_frame_groupby_dtypes_consistencyc C sZ ddddg}ddddg}t j|ddgd}t||d}| |j¡ ¡ }t|d ksVtd S )
|
N)Ú600809Z20061231)r| Z20070331)r| Z20070630ZdemoZSTK_IDZRPT_Date)rC rr r! )r r[ r r r ry Úlenr. )r Zidx_tpÚdtÚidxr; r r r r Ú test_duplicate_groupby_issuesï s ü z,TestMultiLevel.test_duplicate_groupby_issuesc C sP ddgg}ddt jfdg}t|t |¡d}|jjj}|jjj}t ||¡ d S )Nr6 r rE rF )rE rG g r- )
|
r Únanr r r[ ZdtypesrE rF r r )r rc r rK r r r r r Ú test_subsets_multiindex_dtypeþ s
|
|
|
z,TestMultiLevel.test_subsets_multiindex_dtypeN)Ú__name__Ú
|
__module__Ú __qualname__r r+ r1 r= rL r^ ra rj rl ro rw rx rt ÚmarkZ parametrizer{ r r r r r r r s
|
|
0
|
r c @ s e Zd ZdZdd ZdS )Ú
|
TestSortedz+everything you wanted to test about sortingc C sÔ t dddgdddggddd ddd gdd dddd gg}td
|
tt|i|d d }|jjd ks`t| ¡ }|jjdksxttd
|
dddd git ddddg¡d d }|j t
|
j dd
dd
f d d
f }t ||¡ d S )Nr
|
rb ÚCrG rF rE r r r4 ÚcolÚint64)r ZdtypeFTrD rR )rb rE )rb rG )r rE )r rF )r r Úranger} r Zis_monotonic_increasingr. Z
|
sort_indexr[ r( ÚpdZ
|
IndexSlicer r )r r rK Úsortedr r r r r Útest_sort_non_lexsorted s" ÿ
|
ÿû$z"TestSorted.test_sort_non_lexsortedN)r r r
Ú__doc__r r r r r r s r ) Únumpyr rt Zpandasr r r r Zpandas._testingZ_testingr r r r r r r Ú<module> s |
|
