a ▒<b╩8у@s`ddlZddlZddlZddlmZmZmZddlm Z gdвZGddДdГZGddДdГZ dS)щN)┌ DataFrame┌ MultiIndex┌Series)┌sum┌prod┌min┌maxZmedian┌meanZskewZmad┌std┌varZsemc @s|eZdZddДZejаdgdвбddДГZddДZd d ДZ ddДZ d dДZddДZddДZ ddДZejаdeбejаdddgбejаdddgбejаdddgбddДГГГГZejаdeбejаdddgбejаdddgбejаdddgбejаdddgбd d!ДГГГГГZejаd"d#d$gбd%d&ДГZd'd(ДZd)d*ДZd+d,ДZd-d.ДZd/d0ДZejаd1d2d1gбd3d4ДГZd5d6ДZd7d8ДZd9S):┌TestMultiLevelcCs|}tаtбП|jddН}WdГn1s00Y|j|jddН}|jddНаtjб}tа ||б|dj|jddН}|djddНаtjб}tj ||ddНtаtбП |jjdddН}WdГn1s╓0Y|j|jddН}|jddНаtjбj}tа ||бdS) N┌monthй┌levelщ┌AFйZcheck_namesй┌axisr)┌columnsr)┌tm┌assert_produces_warning┌ FutureWarningr┌reindex┌index┌groupby┌ transform┌np┌assert_frame_equal┌assert_series_equal┌T)┌self┌/multiindex_year_month_day_dataframe_random_data┌ymd┌ month_sums┌result┌expectedйr'·l/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/pandas/tests/test_multilevel.py┌test_reindex_levels*.z!TestMultiLevel.test_reindex_level┌opname)┌sub┌add┌mul┌divc Cs╪|}tt|Г}tаtбП|jddН}WdГn1s:0Y|||ddН}|jddНаtjб}|||Г}tа ||бtt |Г}||d|dddН}|djddНаtjб}||d|Г}d|_tа||бdS)Nr rr) ┌getattrrrrrrrrrrr┌namer) r!r*r"r#┌opr$r%Zbroadcastedr&r'r'r(┌test_binops_level3s * z TestMultiLevel.test_binops_levelcCs0|}|jddg}|jddg}tа||бdS)Nrщй┌foo┌oneй┌barr6)┌iloc┌locrr)r!┌ multiindex_dataframe_random_data┌framer&Z reindexedr'r'r(┌test_reindexJszTestMultiLevel.test_reindexcCsМ|}|jdddЕ}|а|б}|j|us,JВ|j|}|jа|бsFJВ|j}|j|dН}|j|usfJВ|jddЕ|f}|jа|бsИJВdS)Nщ йr)rrr:┌equalsr r)r!r"r#Z new_index┌chunkZymdTr'r'r(┌test_reindex_preserve_levelsQs z+TestMultiLevel.test_reindex_preserve_levelsc Cs^|}|d}|jаdб}|а|б}|аddДб}|аddДб}|а|jб}tj||ddНdS)NrrcSs|dSйNщr'й┌xr'r'r(┌lєz7TestMultiLevel.test_groupby_transform..cSs|dSrCr'rEr'r'r(rGmrHFr)r┌get_level_valuesr┌applyrrrr) r!r;r<┌sZgrouper┌groupedZappliedr&r%r'r'r(┌test_groupby_transformds z%TestMultiLevel.test_groupby_transformcCsVtdgdgdggdgdgdgggdвdН}ttjаdбggdв|d Н}|jd dНdS)Nr5r8┌bazr)r6┌two┌threeй┌levels┌codes┌namesщй┌a┌b┌c┌d)rrrPr)rrr┌randomZrandr)r!┌midx┌dfr'r'r(┌test_groupby_cornerqs¤z"TestMultiLevel.test_groupby_cornercCsrtаgdвб}tgdвgdвg|dН}|jddН|jаddДб}|jdd d Н}|аб}|jddgkабsnJВdS) N))┌f1┌s1)r_┌s2)┌f2r`)rbra)┌f3r`)rcra)rrDr3rUщщ)щщщ r>щщr?r)rcSs|ddvS)Nr)rbrcr')┌ur'r'r(rGИrHz:TestMultiLevel.test_groupby_level_no_obs..rrrbrc) r┌from_tuplesrr:r┌maprr┌all)r!r\r]Zdf1rLr%r'r'r(┌test_groupby_level_no_obs{s z(TestMultiLevel.test_groupby_level_no_obscCsJ|}|ddЕj}|d|d<t|jtГs.JВ|d|dkабsFJВdS)Nrd)щ╨rrf)rprr>)r ┌ isinstancerrrn)r!r"r#r]r'r'r(┌.test_setitem_with_expansion_multiindex_columnsОs z=TestMultiLevel.test_setitem_with_expansion_multiindex_columnscCs╕tgdвtаgdвбdН}tgdвtаgdвбdН}||}|jа|jб}|а|б|а|б}tа||б|dddЕ|dddЕ}|jа|jб}|а|б|а|б}tа||бdS)N)rrDr3))rr)rrDй┌Br3)┌datar)rUrdre))┌Zr)rvrDrsщ )rrrlr┌unionrrr)r!rF┌y┌resZ exp_index┌expr'r'r(┌test_alignmentШs zTestMultiLevel.test_alignmentr1rrr┌skipnaTF┌sortc sВ|}|j||dН}|аЗЗfddДб}tаtбП"t|ИГ|ИdН} WdГn1sX0Y|rr| j|dН} tа|| бdS)N)rr~cst|ИГИdНS)N)r}йr/rEйr1r}r'r(rG╕rHz:TestMultiLevel.test_series_group_min_max..)rr}r)r┌aggrrrr/┌ sort_indexr) r!r1rr}r~Zseries_with_multilevel_indexZserrL┌leftside┌ rightsider'rАr(┌test_series_group_min_maxмs0z(TestMultiLevel.test_series_group_min_maxrcs$|}tj|jdddgf<tj|jdddgf<|jj|}ИdkrF|}n|j}|j|И|dН} gЙЗЗЗЗfddД} | а| б}tа t бП$t|ИГ|ИИdН}WdГn1s░0Y|r┌|j|Иd Н}|j|Иd Н}|а Ибj|а|б} tа|а Иб| бtа|а Иб| бtа||бdS) NrrDrfr)rrr~csИа|бt|ИГИИdНS)N)r}r)┌appendr/rEйrr1┌piecesr}r'r(┌aggf╪s z1TestMultiLevel.test_frame_group_ops..aggf)rrr})rr)r┌nanr9rrTr rrБrrrr/rВZ _get_axisrR┌rename┌assert_index_equalr)r!r1rrr}r~r;r<Z level_namerLrЙrГrДZlevel_indexr'rЗr(┌test_frame_group_ops┐s( 2z#TestMultiLevel.test_frame_group_ops┌methrr cstаtаdбаdбtаtаdбdбgб}ttjаt |Гdб|dН}dЙЗЗfddД}t аtбП&t |dИГdИdН}WdГn1sК0Y|djdd На|б}t а||бt аtбП"t |ИГdИdН}WdГn1sш0Y|jdd На|б}t а||бdS) Nrdr>йrrUcst|ИГИdНS)N)┌ddofrrEйrРrОr'r(rGєrHz7TestMultiLevel.test_std_var_pass_ddof..r)rrРr)rZfrom_arraysrZarange┌repeatZtilerr[┌randn┌lenrrrr/rrБrr)r!rОrr]Zaltr%r&r'rСr(┌test_std_var_pass_ddofыs 40z%TestMultiLevel.test_std_var_pass_ddofcCsn|}tа||б}tаtбП |jddgdН}WdГn1s@0Y|jddgdНаб}tа||бdS)N┌yearr r)r┌get_objrrrrZassert_equal)r!r"Zframe_or_seriesr#r%r&r'r'r(┌test_agg_multiple_levels s.z'TestMultiLevel.test_agg_multiple_levelscCsЪ|}|jddgdНаб}|jаdб}|jаdб}|а||gбаб}tj||ddН|jj|jjddЕksnJВ|j|jjddЕdНаб}tа||бdS)NrrrFrrD)rr rrIrrrT)r!r"r#r%Zk1Zk2r&┌result2r'r'r(┌test_groupby_multilevel sz&TestMultiLevel.test_groupby_multilevelcCs@tаgdвб}ttjаddб||dН}|аdб|d<|аб}dS)N)r4)r5rOr7)r8rOrU)rrr)ZTotals┌)rrlrrr[rУrZ_consolidate)r!rr]r'r'r(┌test_multilevel_consolidates z*TestMultiLevel.test_multilevel_consolidatecCsъtgdвddgggdвgdвgdН}ttjаdб|dН}ttjаdd б|dН}|d }|jd }|ddЕ}|jаdб|_t а ||бt а ||бtjt dd НП|dWdГn1s╛0Y|jd }|аd б}|ddЕ}|jаdб|_t а||бt а||бtgdвddgggdвgdвgdН}ttjаdб|dН}ttjаdd б|dН}|d}|jd}|ddЕ}|jаdб|_t а ||бt а ||б|jd}|аdб}|ddЕ}|jаdб|_t а||бt а||бdS)N)йr5r8r)r5rNr)r5┌quxrrr)rrrrrDrD)rrrrrr)rRrSrerПrUrЭrDz^$\('foo', 'bar', 0$, 2\)$)┌match)rЭrD)йr5r8)r5rN)r5rЮrа)rrrr[rУrr:rZ droplevelrr┌pytestZraises┌KeyError┌xsr)r!rZseriesr<r%rЩr&r'r'r(┌test_level_with_tuples#sL■ & ■ z%TestMultiLevel.test_level_with_tuplescCs║|}|jddgddН}|jgdв}tа||б|jjddgdddН}tа||jб|jddg}tа||б|djddg}tа||dб|jjddЕddgf}tа||jбdS) Nr5rЮrr)rrrDrfrgrhrrr)rr9rrr r:r)r!r;r<r%r&r'r'r(┌$test_reindex_level_partial_selectionSsz3TestMultiLevel.test_reindex_level_partial_selectionrZrUcCstgdв}tdgdgdg|gdЬГ}||jdkа|б}|абj}tdgdgdgggggggdвdН}tа||бdS)N)rWrXrYrrDr3rVrQ)rrWr┌firstrrrrМ)r!rZZ group_keysr]┌gr%r&r'r'r(┌+test_empty_frame_groupby_dtypes_consistencyfs z:TestMultiLevel.test_empty_frame_groupby_dtypes_consistencycCsRgdв}gdв}tj|ddgdН}t||dН}|а|jбаб}t|ГdksNJВdS)N))┌600809Z20061231йrйZ20070331)rйZ20070630rк)┌demorлrлrлZSTK_IDZRPT_Date)rTrПr3)rrlrrrrжrФ)r!Zidx_tp┌dt┌idxrKr%r'r'r(┌test_duplicate_groupby_issuestsz,TestMultiLevel.test_duplicate_groupby_issuescCsPddgg}ddtjfdg}t|tа|бdН}|jjj}|jjj}tа ||бdS)NrFrrWrX)rWrYgr?) rrКrrrlZdtypesrWrXrr)r!rurr]r&r%r'r'r(┌test_subsets_multiindex_dtypeГs z,TestMultiLevel.test_subsets_multiindex_dtypeN)┌__name__┌ __module__┌__qualname__r)rб┌markZparametrizer2r=rBrMr^rorrr|┌ AGG_FUNCTIONSrЕrНrХrШrЪrЬrдrеrиrоrпr'r'r'r(rs@ ' 0 rc@seZdZdZddДZdS)┌ TestSortedz+everything you wanted to test about sortingcCs╕tgdвgdвggdвgdвgГ}tdtt|ГГi|ddН}|jjdusLJВ|аб}|jjd usdJВtdgd вitаgdвбddН}|jt j dd ЕddЕfddЕf}tа||бdS)N)rrt┌C)rYrXrW)rrrDrrrD)rrDrrrrD┌colZint64)rZdtypeFT)rrUrdrD))rtrW)rtrY)r╢rW)r╢rXrtr╢rWrY) rr┌rangerФrZis_monotonicrВrlr:┌pdZ IndexSlicerr)r!rнr]┌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╡Нsr╡)ZnumpyrrбZpandasr╣rrrZpandas._testingZ_testingrr┤rr╡r'r'r'r(┌st