U
|
ý°d4a ã @ s d dl mZm Z mZ d dlmZ d dlZd dlZd dlm Z d dl
|
Z d dl
|
m Z m Z mZmZmZmZmZmZ d dlmZ d dlmZ d dlmZ d dl m Z ej d d
|
gd d d Z ej d gd dgd fe!d de!ddfgd dd Z"G dd dZ#G dd dZ$dS )é )ÚdateÚdatetimeÚ timedelta)ÚproductN)ÚPerformanceWarning)Ú CategoricalÚ DataFrameÚGrouperÚIndexÚ
|
MultiIndexÚSeriesÚconcatÚ
|
date_range)ÚCategoricalDtype)Úreshape©Ú pivot_tableTF)Úparamsc C s | j S ©N)Úparam)Úrequest© r úVd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/reshape/test_pivot.pyÚdropna s r é é é c C s | j \}}ttj |||¡S r )r r ÚpdÚ IntervalIndexÚ from_arrays)r ÚclosedÚleftÚrightr r r Úinterval_values# s
|
r# c @ s* e Zd Zejdd Zdd Zdd Zdd Zd d
|
Z d d Z
|
d d Z dd Z dd Z dd Zdd Zdd Zdd Zdd Zd d Zej d d d!d"gfd d!d"d gfd#d!d"d gfg¡d$d% Zd&d' Zd(d) Zd*d+ Zej d,d-d.g¡d/d0 Zej d,d-d.g¡d1d2 Zej d,d-d.g¡d3d4 Z d5d6 Z ej d,d-d.g¡d7d8 Z d9d: Z ej d;d<d=ge !d<d=g¡e"d<d=ge#d<d=gg¡ej d,d-d.g¡d>d? Z$ej d;d@d<ge !d@d<g¡e"d@d<ge#d@d<gg¡ej d,d-d.g¡dAdB Z%dCdD Z&ejj'dEdFej d,d-d.g¡dGdH Z(ej d,d-d.g¡dIdJ Z)dKdLgdMgdNfdOdPZ*dQdR Z+dSdT Z,ej dUdKe j-dVdVdWdWgdXdXdYdYgge#d@dNdZdNgdKd[fdKdLgd\d]d^d_d`dadbgdcdddedbdbd_gge.j/dfdgdhdidjdkgdKdLgdlfg¡dmdn Z0dodp Z1dqdr Z2ej dsdtdudvdwg¡dxdy Z3dzd{ Z4d|d} Z5d~d Z6dd Z7dd Z8dd
Z9dd Z:dd Z;dd Z<ej ddZdddddgg¡dd Z=dd Z>dd Z?dd Z@dd ZAej deBdd¡dd ZCej deBdd^¡d¡d¢ ZDd£d¤ ZEd¥d¦ ZFd§d¨ ZGd©dª ZHd«d¬ ZIdd® ZJd¯d° ZKd±d² ZLd³d´ ZMdµd¶ ZNd·d¸ ZOej d¹d\e jPfdºe j-fd»e jQfd\dºge jPe j-gfd\d»ge jPe jQgfd»dºge jQe j-gfg¡d¼d½ ZRejjSd¾d¿ ZTdÀdÁ ZUdÂdà ZVej dÄd-d.g¡dÅdÆ ZWdÇdÈ ZXdÉdÊ ZYdËdÌ ZZdÍdÎ Z[dÏdÐ Z\dÑdÒ Z]ej dÓd-d.g¡dÔdÕ Z^dÖd× Z_dØdÙ Z`dS )ÚÚTestPivotTablec C sp t dddddddddddg dddddddddddg dddddddddddg tj d¡tj d¡tj d¡dS ) NÚfooÚbarÚoneÚtwoÚdullÚshinyé )ÚAÚBÚCÚDÚEÚF)r ÚnpÚrandomÚrandn)Úselfr r r Údata* sT õõõ
|
|
|
ÖÿzTestPivotTable.datac C sä ddg}d}t |d|||d}|j d|||d}t ||¡ t |d||d t|dkrn|jjt|kstn|jj|d kstt|dkr |j j|ks´tn|j j|d ks´t|
|
||g ¡d t j ¡ ¡ }t ||¡ d S ) Nr, r- r. r/ )ÚvaluesÚindexÚcolumnsÚobserved)r7 r8 r: r r )r ÚtmÚassert_frame_equalÚlenr8 ÚnamesÚtupleÚAssertionErrorÚnamer9 ÚgroupbyÚaggr2 ÚmeanÚunstack)r5 r: r6 r8 r9 ÚtableÚtable2Úexpectedr r r Útest_pivot_table[ s2 ÿ ÿ z TestPivotTable.test_pivot_tablec C s t tdtddddddgd}|jd d
|
d tjd d }|j d¡|_|j d¡|_|j d¡|_|j d¡|_ |jd d
|
d tjd |d}t
|
||¡ d S )NZabcdeZfghijr é r r é ©Úcol1Úcol2Úcol3rM rO rN r )r8 r7 r9 ÚaggfuncÚ
|
fill_valueÚcategory)r8 r7 r9 rP rQ r: ) r Úlistr r2 Úsumr8 Úastyper9 rM rN r; r< )r5 r: ÚdfrH Úresultr r r Ú+test_pivot_table_categorical_observed_equalw s. ÿ ÿú z:TestPivotTable.test_pivot_table_categorical_observed_equalc C s t dddgdddgddd gd
|
}|jd tjd }|jd tjd j}t ||¡ |jd ddid }|jd ddid j}t ||¡ d S )NÚaÚbÚcÚxÚyÚzr rJ r )ÚrowsÚcolsr7 r` )r9 rP ©r8 rP r7 rD )r r r2 rT ÚTr; r< )r5 rV ÚrsZxpr r r Útest_pivot_table_nocols s ÿ z&TestPivotTable.test_pivot_table_nocolsc C s¼ t ddddddddddd d
|
d d dd dddddddddd}|jddddgdd}|jdddgddd}tjdd d d d d d!d"d#d$d%d&g ddgd'}t |j|¡ t |j|¡ d S )(Ni`ê i iPÃ i0u )r r rJ r r, r- r. i[ i] i\ i^ rY rZ r[ Údi i ¡ i@B )ÚamountÚcustomerÚmonthr Úquantityri rh rg r F)r )r, rY )r, rZ )r, r[ )r, re )r- rY )r- rZ )r- r[ )r- re )r. rY )r. rZ )r. r[ )r. re ©r> )r r r Ú from_tuplesr; Úassert_index_equalr9 r8 )r5 rV Zpv_colZpv_indÚmr r r Útest_pivot_table_dropna sL ûÿ ÿ ÿôñz&TestPivotTable.test_pivot_table_dropnac C s¢ t ddddgdddgdd}t ddddgdddgdd}t||d d
|
d d gd }t|dddgdd}tj||gddgd}tdd d
|
d d gi|d}t ||¡ d S )NrY rZ r^ T©Ú
|
categoriesÚorderedr[ re r] r rJ r r )r, r- r7 r7 r, r- )r7 r8 r rj ©r8 )r r r r r r; r< )r5 Zcat1Zcat2rV rW Z exp_indexrH r r r Ú test_pivot_table_categoricalà s
|
ÿ
|
ÿz+TestPivotTable.test_pivot_table_categoricalc C sø ddddg}t dddddddddg dddddddddg tdd d
|
}|d t|d d ¡|d < |jdd d|d}tdddgd d}| t|d d ¡}tdddgdd}t dddgdddgdddgg||d}|sè|jt|d d¡}t ||¡ d S )NrY rZ r[ re r rJ r r é ©r, r- r. r, F©rq r- r. )r8 r9 r7 r ©rA é r é rK é ©r8 r9 ©r9 Úfloat)
|
r ÚrangerU ÚCDTr r Zreindexr r; r< )r5 r rp rV rW Úexpected_columnsÚexpected_indexrH r r r Ú$test_pivot_table_dropna_categoricalsÑ s( ýÿýz3TestPivotTable.test_pivot_table_dropna_categoricalsc C s t ttjddddgddgddddddd gd
|
}|jd d |d }t d ddgittjddgddgddd dd}t ||¡ d S )NÚlowÚhighTro ç ç ð?ç @ç @ç @©r, r- r, r- ©r8 r7 r r r rw rr ) r r r2 Únanr r
|
Ú
|
from_codesr; r< ©r5 r rV rW rH r r r Ú%test_pivot_with_non_observable_dropnaí s, ý úÿ
|
ÿüþ
|
z4TestPivotTable.test_pivot_with_non_observable_dropnac C s t tdddddgdddgddtdd}|jdd |d
|
}t d d d d gittjd dd gdddgddddd}|s|d t¡|d < t ||¡ d S )Nr! r r Tro rK r r, r- r rJ r r r rw rr )
|
r r r~ r r
|
r rU r} r; r< r r r r Ú/test_pivot_with_non_observable_dropna_multi_cat s0 ýúÿ ÿüþ z>TestPivotTable.test_pivot_with_non_observable_dropna_multi_catc C sX t |dd}|jdd|d}t ddit| ¡ ddd}|sH| t¡}t ||¡ d S )Nr r r, r- r rw rr )r r r
|
ÚuniquerU r} r; r< )r5 r# r rV rW rH r r r Ú test_pivot_with_interval_index# s
|
z-TestPivotTable.test_pivot_with_interval_indexc C s¸ t j ddddgddddg¡}ttjdddtjdddddgt|d d
|
jd d d }t |ddddd d}|d }t
|
dddgt t dd¡t dd¡dgdddtjd}t ||¡ d S )Nr r rJ r éÿÿÿÿ©ÚdtyperY rZ Trv F)Z ascendingru r. r- r, rT ©r8 r9 r7 rP ÚmarginsÚAllr ry é
|
rw )r8 rA r )r r r r r2 ÚarangeZintpr Z sort_valuesr r r
|
ZIntervalr; Úassert_series_equal)r5 Z ordered_catrV Z pivot_tabrW rH r r r Ú&test_pivot_with_interval_index_margins, s2
|
ÿýÿ
|
ÿ üz5TestPivotTable.test_pivot_with_interval_index_marginsc C s4 |j d|j|jd}|j dddd}t ||¡ d S )Nr/ r{ r, r. )r r, r. r; r< ©r5 r6 rW rH r r r Útest_pass_arrayF s z TestPivotTable.test_pass_arrayc C s< |j ddd |jd}|j d|jd dd}t ||¡ d S )Nr/ c S s | d S )NrK r ©r\ r r r Ú<lambda>L ó z3TestPivotTable.test_pass_function.<locals>.<lambda>r{ rK r. )r r. r8 r; r< r r r r Útest_pass_functionK s z!TestPivotTable.test_pass_functionc C sF ddg}d}t |||d}| ||g ¡ tj¡ ¡ }t ||¡ d S )Nr, r- r. r{ )r rB rC r2 rD rE r; r< )r5 r6 r8 r9 rF rH r r r Útest_pivot_table_multipleP s
|
z(TestPivotTable.test_pivot_table_multiplec C s t ddddgddddgddddgd }|jd
|
d ks:tt|d
|
dgd gd tjd}|j}tt d ¡gd tt dd dd}t
|
||¡ t ddddgddddgddddgd }|jd
|
dksÄtt|d
|
dgd gd tj d}|j}tt d¡gd tt dd dd}t
|
||¡ d S )NÚcatZbatr rJ r r rY rZ )rY ÚvÚir¥ Úint64r¦ r )r7 r8 r9 rQ rP Úabrw rr ç ø?ç @ç @ç @Úfloat64) r Údtypesr@ r r2 rT r r r
|
rS r; r rD )r5 Úfr^ rW rH r r r Útest_pivot_dtypesW sH
|
|
|
ýÿ ÿ$
|
|
|
ýÿ ÿ$z TestPivotTable.test_pivot_dtypeszcolumns,valuesÚbool1Úfloat1Úfloat2Úbool2c C sj t jdt jd}t||d |dk|dkd}| ¡ jd||d}t|j}d d
|
|D }||ksftd S ) NrK r r rJ r )r² r³ r± r´ r8 ©r8 r9 r7 c S s i | ]}|t d ¡qS )r )r2 r )Ú.0Úcolr r r Ú
|
<dictcomp> s z=TestPivotTable.test_pivot_preserve_dtypes.<locals>.<dictcomp>) r2 r r r Ú reset_indexr Údictr® r@ )r5 r9 r7 r¥ rV Zdf_resrW rH r r r Útest_pivot_preserve_dtypesz s
|
ÿ ÿ
|
z)TestPivotTable.test_pivot_preserve_dtypesc C sx t dddddg¡}tdddddd gi|d
|
}|j|jj|jjd }t d d g¡}|j |j
|
d t j ¡dd}tddgdt jggtddgt j d|d }t ||¡ tddddd gtddd dd|d
|
}|j|jjtdddd }t dt d¡fg¡}d dg|_tddgtddgt j d|d }t ||¡ |jtddtdddd }tdgt jd gdd|d }t ||¡ d S ) Nz
|
2011-01-01z
|
2011-02-01z
|
2011-01-02r, r rJ r r rK rr r{ )r, r )r, rJ )Úlevelrª r r r r/ )ÚfreqÚperiods)r, Údtr¿ ÚM)Úkeyr½ z
|
2011-01-31ç
|
@©r½ z
|
2011-12-31)r Ú DatetimeIndexr r r8 rh Údayr rk Z
|
set_levelsÚlevelsrU r2 Úint32r r
|
r; r< r r Ú Timestampr> )r5 ÚidxrV ÚresZ exp_columnsÚexpr r r Útest_pivot_no_values sR ÿ ÿý þû
|
ÿ
|
ÿ ÿz#TestPivotTable.test_pivot_no_valuesc C sJ t |ddgdddgdd}t |jdgd d
|
dddgdd }t ||¡ d S ) Nr/ r0 r, r- r. r )r7 r8 r9 rQ r1 r ©Úaxis)r8 r9 rQ )r Údropr; r< r r r r Útest_pivot_multi_values» s ÿ ÿz&TestPivotTable.test_pivot_multi_valuesc s¨ fdd}|t jt jg}|t j}|t j}t||gddgdd}t ||¡ fdd}|t jt jg}|t j}|t j}t||gddgdd}t ||¡ d S )Nc s t ddgddgd| dS )Nr/ r0 r, r- r. ©r7 r8 r9 rP r ©Úfunc©r6 r r r Å s ÿz;TestPivotTable.test_pivot_multi_functions.<locals>.<lambda>rD Ústdr )ÚkeysrÎ c s t ddgddgd| ddS )Nr/ r0 r, r- r. T)r7 r8 r9 rP r r rÒ rÔ r r r Ï s ú)r2 rD rÕ r r; r< )r5 r6 r¯ rW ZmeansZstdsrH r rÔ r Útest_pivot_multi_functionsÄ s
|
|
|
|
z)TestPivotTable.test_pivot_multi_functionsÚmethodTFc C sÖ t j}tdd|dgddddgdd d
|
d gd }|rD|jd ddd}ntj|d ddd}t||d
|
|gd|||g|d ||g|||d ggt|dddgd dtddddgddd}t ||¡ t |jdd dd|j¡ d S )NZR1ZR2ZR4ÚC1ÚC2ZC3ZC4r é é é ©rY rZ r[ rY rZ r[ rµ rw r{ ) r2 r r Úpivotr r
|
r; r< rb )r5 rØ r rV rW rH r r r Útest_pivot_index_with_nanÝ s*
|
|
|
ýÿ
|
|
|
|
üø
|
z(TestPivotTable.test_pivot_index_with_nanc C s t tdddddt d¡ d}|d t d¡ |d < tj |jd
|
< |jd < tj |jd < |jd < |rz|jdd dd}ntj|dd dd}| ¡ j
|
¡ t |ks¦t | ¡ D ]*\}}|j|d |d f |d ks®t q®|rð|jd ddd}ntj|d ddd}t ||j¡ d S )Nz
|
2014-02-01rx r/ )r¾ r½ éd )rY r[ rY z
|
2014-02-02rZ )r rY )r rY ©r rZ )r rZ r[ rµ )r r r2 r r rÈ r Úlocrß Znotnar7 rT r= r@ Ziterrowsr; r< rb )r5 rØ rV ÚpvÚ_ÚrowrW r r r Ú test_pivot_index_with_nan_datesù s$ þÿ$z.TestPivotTable.test_pivot_index_with_nan_datesc C sÐ t tdddddtdddddtdddddtdddddgtdddddtdddddtdddddtdddddgtjddd tjddd d
|
}|d d d ¡|d < |d dd ¡|d< tddddg}tjddgd ddd}t ||g¡}t ddddgddddggtjddgd dd|d}|r@|j
|
d dd}ntj
|
|d dd}t ||¡ t ddgddggtjddgd ddtjddgdddd}|r®|j
|
d ddd}ntj
|
|d ddd}t ||¡ d S ) NéÝ r rt r rJ éÞ r r§ r )Údt1Údt2Údata1Údata2rê c S s t j| ddS ©Nú
|
US/Pacific©Útz©r rÈ ©re r r r r + r¡ z3TestPivotTable.test_pivot_with_tz.<locals>.<lambda>rë c S s t j| ddS ©Nú
|
Asia/Tokyorð rò ró r r r r , r¡ rì rí z2014/01/01 09:00z2014/01/02 09:00rõ )rA rñ r z2013/01/01 09:00z2013/01/02 09:00rï r{ rµ ) r r r2 r Úapplyr
|
r rÄ r r rß r; r< ©r5 rØ rV Úexp_col1Úexp_col2Úexp_colrH rä r r r Útest_pivot_with_tz sl üü òÿ
|
ÿ ÿû ÿ ÿû
|
z!TestPivotTable.test_pivot_with_tzc
|
C s t dtjdddddtjdddddtjdddddtjdddddtjd dddg}| d
|
¡ ¡ }|j d d ¡}t| d
|
¡ ¡ d
|
d g|gtj d}t tjdddtjdddggt
|
dgd dtj tjdddtjdddgd
|
dd}t ||¡ d S )NZaaz2016-08-12 13:00:00-0700rï rð )ÚuidÚtsz2016-08-12 08:00:00-0700z2016-08-12 14:00:00-0700z2016-08-25 11:00:00-0700z2016-08-25 13:00:00-0700rý c S s | j dddddS )Nr )ÚhourÚminuteÚsecondÚ microsecond)Úreplacer r r r r m r¡ z8TestPivotTable.test_pivot_tz_in_values.<locals>.<lambda>rü rÑ rw z2016-08-12 00:00:00z2016-08-25 00:00:00r{ )r r rÈ Ú set_indexr¹ rý Úmapr r2 Úminr
|
rÄ r; r< )r5 rV ZminsrW rH r r r Útest_pivot_tz_in_valuesQ sR þ þ þ þ þïÿ û þÿ þûøz&TestPivotTable.test_pivot_tz_in_valuesc C s t t dd¡t dd¡t dd¡t dd¡gt dd¡t dd¡t dd¡t dd¡gtjddd tjddd d
|
}td d d d g}tjddgd ddd}t ||g¡}t dd dd gddddggtjddgddd|d}|rò|j ddd}ntj |ddd}t
|
||¡ t dd gddggtjddgdddtjddgdddd}|r`|j ddd d}ntj |ddd d}t
|
||¡ d S )Nz
|
2013-01-01r/ z
|
2013-01-02z2013-01rÀ z2013-02r r§ r )Úp1Úp2rì rí rì rí rJ r )rA r½ r r r r r{ rµ ) r r ÚPeriodr2 r r
|
Z PeriodIndexr r rß r; r< r÷ r r r Útest_pivot_periods sH
|
|
|
|
ü
|
|
|
|
ü òÿý ýz!TestPivotTable.test_pivot_periodsc C s t ddddgt d¡t d¡t d¡t d¡gdd}t dtdddgdd tt d¡t d¡dgd
|
d d }|jdd
|
d d d}t ||¡ d S )Nr rJ Z2019Q1Z2019Q2r )rY rZ r\ r rY rw rZ )r6 r8 r9 r\ T)r8 r9 r7 r )r r r r
|
r r; r< )r5 rV rH rW r r r Ú test_pivot_periods_with_margins¶ s"
|
üøÿ ýz.TestPivotTable.test_pivot_periods_with_marginsr7 ÚbazÚzooc
|
C sö t ddddddgddddddgdddd d
|
d gd d ddddgd}|rX|jdd|d}ntj|dd|d}dddd d dgd d
|
d dddgg}tddgdd}tddgdddggddddddgddddddggd dgd}t |||dd }t ||¡ d S ) Nr' r( r, r- r. r rJ r r rK rx r\ r] r^ ÚqÚwÚt©r% r& r r r% r& rµ ©r6 rA r r r ©rÆ Úcodesr> Úobject©r6 r8 r9 r )r rß r r
|
r r; r< © r5 r7 rØ rV rW r6 r8 r9 rH r r r Ú test_pivot_with_list_like_valuesÎ s& üÿ ýz/TestPivotTable.test_pivot_with_list_like_valuesr& c
|
C s, t ddddddgddddddgdddd d
|
d gd d ddddgd}|rX|jdd|d}ntj|dd|d}tjdtjd gtjdtjd gtjdtjd
|
gdtjdtjgdtjdtjgdtjdtjgg}tdddd d dgdd}tddgddggddddgddddggd dgd}t |||dd }t ||¡ d S ) Nr' r( r, r- r. r rJ r r rK rx r\ r] r^ r r r r r r% rµ r r& r r r r r ) r rß r r2 r r
|
r r; r< r r r r Ú%test_pivot_with_list_like_values_nansò s2 üÿ úýz4TestPivotTable.test_pivot_with_list_like_values_nansc C sR t dddgdddgdddgd}d}tjt|d |jd
|
d d W 5 Q R X d S ) NrY rZ r[ r rJ r rL z=pivot\(\) missing 1 required keyword-only argument: 'columns'©ÚmatchrM rO )r8 r7 ©r ÚpytestÚraisesÚ TypeErrorrß )r5 rV Úmsgr r r Ú#test_pivot_columns_none_raise_error s "z2TestPivotTable.test_pivot_columns_none_raise_errorzBMultiIndexed unstack with tuple names fails with KeyError GH#19966©Úreasonc C s t ddddddgd}dd dd
|
gdd dd gdd ddgdd ddgdd ddgdd ddgg}tddgddggddddgddddggd}t|||dd}|r¨|jdd d d }ntj|dd d d }tddgddgd tddgddgd tddgddgd d }t|}t ||¡ d S )!Nr r rJ r r rK rÔ r' r, r\ r- r] r. r^ r( r r rx r r& r Úfirstr )rÆ r r r )r& r$ )r& r )r r$ rµ rr ru )r
|
r r rß r r r; r< )r5 rØ r8 r6 r9 rV rW rH r r r Útest_pivot_with_multiindex$ s>
|
|
|
|
|
|
úþýüýz)TestPivotTable.test_pivot_with_multiindexc
|
C s t ddddddgddddddgdddd d
|
d gd d ddddgd}tjtdd. |rh|jdddd ntj|dddd W 5 Q R X d S )Nr' r( r, r- r. r rJ r r rK rx r\ r] r^ r r r r z^\('bar', 'baz'\)$r r r% )r& r rµ )r r r ÚKeyErrorrß r )r5 rØ rV r r r Ú test_pivot_with_tuple_of_valuesN s üÿz.TestPivotTable.test_pivot_with_tuple_of_valuesr, r- r. r c C sÊ |j |jd d
|f }| |¡| ¡ }tj||dd |j|ksHt| ¡ }|j |df j d d
} | |¡| ¡ }
|
tj| |
|
dd | j|dfkst|j |df|f } || ¡ } | | ksÆtd S )Nr F©Z check_namesÚ )
|
rã r8 rB rD r; r rA r@ Z
|
sort_indexZiloc) r5 rW Z
|
values_colr6 r8 r9 Ú margins_colZ col_marginsZexpected_col_marginsZ index_marginsZexpected_ix_marginsZgrand_total_marginsZexpected_total_marginsr r r Ú _check_output` s z TestPivotTable._check_outputc C s |j dddgddtjd}| |d|¡ |j dddgddtjdd}| j|d|dd |j ddgddtjd
|
}|jjd D ]}| || ||¡ qzd S ) Nr/ r, r- r. T)r7 r8 r9 r rP ZTotals)r7 r8 r9 r rP Ú margins_name)r* ©r8 r9 r rP r )r r2 rD r+ r9 rÆ )r5 r6 rW rF Ú value_colr r r Ú test_marginsy s2 ÿú ÿzTestPivotTable.test_marginsc C s dd |j D |_ tjtdd |jddgdtjd W 5 Q R X |jd d
|
jddgdtjd}|j D ]&}|jd |f }||| ¡ ksft qftjtdd |jddgdd d W 5 Q R X |jd d
|
jddgdd d}d D ]&}|jd |f }||| ¡ ksÜt qÜd S )Nc S s g | ] }|d qS )rJ r )r¶ Úkr r r Ú
|
<listcomp> s z.TestPivotTable.test_no_col.<locals>.<listcomp>úCould not convertr ZAAÚBBT)r8 r rP ÚCCr| ©r r) rD )ZDDZEEZFF)
|
r9 r r r r r2 rD rÏ rã r@ )r5 r6 rF r. ZtotalsÚitemr r r Ú test_no_col s* ÿ
|
ÿzTestPivotTable.test_no_colz*columns, aggfunc, values, expected_columnsç @g@ç @g@r% rw rT rt é é rK rx r+ é é é )r& r' )r& r( )r& r )r% r' )r% r( )r% r rj c C sò t dddddddddg dddddddddg dddddddddg dddd d d
|
d d d g dd
|
d d d d dddg d}|dkr tjtdd |j|d|d W 5 Q R X d|kr´|jdd}|jddj|d|d}t |tddg|d}t ||¡ d S ) Nr% r& r' r( ÚsmallÚlarger rJ r r rK rx ry rz rt ©r, r- r. r/ r0 rT r2 r T)r9 r rP r- r| r. r/ r0 r{ ) r r r r r rÏ r
|
r; r< )r5 r9 rP r7 r rV rW rH r r r Ú%test_margin_with_only_columns_defined© s: "÷ ñÿ ÿz4TestPivotTable.test_margin_with_only_columns_definedc C sÐ | ¡ }t t|d ¡ t|d¡ d¡|dddg< ttddgdd gd
|
g }tj |d d }t
|
d ddddgdddddgd|dj ddd}|d |d |d < |j dd d gdd tj dd!}t ||¡ d S )"Nr Úi8r/ r0 r1 r& r% r' r( r5 r rj é é rt é- é! r é$ é3 éx ©r) r* rr r. r rÍ r) r* r r, r- T©r7 r8 r9 r rP rQ )Úcopyr2 r r= r rU rS r r rk r Ú rename_axisr rT r; r< )r5 r6 rV Úmi_valÚmirH rW r r r Útest_margins_dtypeê s* . ÿ þú z!TestPivotTable.test_margins_dtypec C s t tddgddgdg }tj|dd}tddd dd
|
gd d d d d gd |djddd}|d |d |d< |jdddgddtd d}t ||¡ d S )Nr& r% r' r( r5 r rj r rJ rK r rx rK rr r. rÍ r) r* r r/ r, r- TrL )
|
rS r r rk r rN r r= r; r< )r5 r6 rO rP rH rW r r r Útest_margins_dtype_len s& ÿ þú z%TestPivotTable.test_margins_dtype_lenr` )r rJ ©rY rZ râ )rY r c C sx t |d dddg|d dddgddddgi}|jd|d }t dddggtjd
|
d d g|d tdgd}t ||¡ d S )Nr r rJ r r¥ r rK rx )r7 r9 )r r )rJ rJ )r r rj ©r9 r8 )r r r rk r
|
r; r< )r5 r` Údf2rW rH r r r Ú test_pivot_table_multiindex_only s .
|
ýz/TestPivotTable.test_pivot_table_multiindex_onlyc C sZ t dddd}ttj d¡tj d¡|d}|jddgdd }|jjd
|
|¡sVt d S ) Nz
|
2016-01-01r zEurope/Amsterdam)r¾ rñ ru r- r. F)r8 r r )
|
r r r2 r3 r4 r r8 rÆ Úequalsr@ )r5 ZdtirV rW r r r Útest_pivot_table_retains_tz# s z*TestPivotTable.test_pivot_table_retains_tzc s t j ttddgdddgddg fdd td
|
D d g}t|}|jd d ddgdgd}|jtd}|jddddgdgd}t j
|
||dd d S )Nr% r& r, r- r. Úx1Zx2c s g | ]} t | qS r )r )r¶ r¦ ró r r r1 4 s z=TestPivotTable.test_pivot_integer_columns.<locals>.<listcomp>rÝ r r r r r rJ )r7 r8 r9 r| Ú4Ú0Ú1Ú3Ú2Fr( ) r r rS r r~ r r ÚrenameÚstrr; r< )r5 r6 rV rF rU rG r ró r Útest_pivot_integer_columns+ s ûÿ z)TestPivotTable.test_pivot_integer_columnsc
|
C s¬ t ddddddddgd ddddddddgd dgd dgd d tj d ¡d
|
}|jd ddd gd }| ddd g¡d ¡ }| d¡ d ¡jddd}t
|
||¡ d S )NrY rZ rJ r r r% r r& é )rY rZ r[ Úvaluerc r[ r{ Úall)rÎ Úhow) r r2 r3 r4 r rB rD rE r r; r< )r5 r6 rF ZgroupedrH r r r Útest_pivot_no_level_overlap@ s
|
üÿ z*TestPivotTable.test_pivot_no_level_overlapc
|
|
C s@ d}t dtfdtfdtfdtfdtfdtfdtfg¡}t jd d
|
d d d ddddg dtfdtfgd}t j||d}t j dt ||¡}|d | |d< |d | |d< t
|
t dddt ddd}|t j dt ||¡ }|j |d< |j |d< |j|d< t j dd|¡|d< t|}|jdddgdddgdd } | jjs<td S ) Ni' r
|
ZSymbolÚYearÚMonthZDayÚQuantityZPrice)ÚSP500ZADBE)rj ZNVDA)rj ZORCL)ÚNDQ100ZAAPL)rk ZMSFT)rk ZGOOG)ÚFTSEzDGE.L)rl zTSCO.L)rl zGSK.Lr r iÐ r iÚ rD é r r rD ©r8 r9 rP )r2 r r Úintr} ÚarrayÚemptyr3 Úrandintr= r r Úyearrh rÅ Z lognormalr r r9 Zis_monotonic_increasingr@ )
|
r5 Únr ZproductsÚitemsZiproductZdrÚdatesrV Úpivotedr r r Ú test_pivot_columns_lexsortedR sT ùÿ ÷ ô
|
|
|
üz+TestPivotTable.test_pivot_columns_lexsortedc C sH dgdgd}| ddg¡ |¡ d¡}|jdd|d}t ||¡ d S )NrÕ rT ©r/ r0 r, r- rn )rB rC rE r r; r< )r5 r6 r¯ rH rW r r r Útest_pivot_complex_aggfunc
s z)TestPivotTable.test_pivot_complex_aggfuncc C sD |ddg j ddgtdd}| ¡ }t|d d
|d ks@td S )Nr, r- T)r8 rP r r )r r= ÚtolistrT r@ )r5 r6 rW Z result_listr r r Ú test_margins_no_values_no_cols s ÿz-TestPivotTable.test_margins_no_values_no_colsc C s@ |dddg j ddgdtdd}|j ¡ ddddd gks<td S )
|
Nr, r- r. T©r8 r9 rP r r r r ç &@©r r= r r{ r@ ©r5 r6 rW r r r Ú test_margins_no_values_two_rows s ÿz.TestPivotTable.test_margins_no_values_two_rowsc C s6 |ddg j ddtdd}|j ¡ dddgks2td S )Nr, r- Tr} r ç @r~ r r r r r Ú&test_margins_no_values_one_row_one_col s ÿz5TestPivotTable.test_margins_no_values_one_row_one_colc C sd ddddddddd d
|
d g |d < |d ddd g j d dgdd gtdd}|j ¡ dddddgks`td S )NrY rZ r[ re Úer¯ ÚgÚhr¦ Újr0 r/ r, r- r. Tr} r r r r~ r r r r r Ú'test_margins_no_values_two_row_two_cols¤ s ÿz6TestPivotTable.test_margins_no_values_two_row_two_colsÚ margin_namer' i NrY rZ c C sª d| d}t jt|d t|dddgdgd|d W 5 Q R X t jt|d t|ddgddgd|d W 5 Q R X t jt|d t|ddgdgd|d W 5 Q R X d S )
|
NzConflicting name "z3" in margins|margins_name argument must be a stringr r/ r, r- r. T)r7 r8 r9 r r, )r r Ú
|
ValueErrorr )r5 r r6 r r r r Ú-test_pivot_table_with_margins_set_margin_name s:
|
ÿúúúz<TestPivotTable.test_pivot_table_with_margins_set_margin_namec C sz t d ¡ d ¡ ddddddddgtdddtdddtdd dtdd d
|
tdd dtdd d
|
tdd d
|
tdd d
|
gd d ¡}t tjd ddgdd dd¡tjtdd dgddd ¡ d}d |j _
|
d|j _
|
t |t ddddtjd}t ||¡ t |dt dddtjd}t ||j¡ t t dtjdddtjg¡ d
|
d¡tjtdddtdddgddd ¡ d}d |j _
|
d|j _
|
|r|d d¡|d< t |t ddddtjd}t ||¡ t |dt dddtjd}t ||j¡ | ¡ }t |t dd d ddtjd}t ||¡ t |dt dd d dtjd}t ||j¡ d }tjt|d " t |t dd d ddtjd W 5 Q R X tjt|d " t |dt dd d dtjd W 5 Q R X | d ¡}t |t dd d ddtjd}t ||¡ t |dt dd d dtjd}t ||j¡ d!}tjt|d " t |t dd d ddtjd W 5 Q R X tjt|d " t |dt dd d dtjd W 5 Q R X d S )"NúA A A A A A A Bú$Carl Mark Carl Carl Joe Joe Joe Carlr r rK rz rt rè r rJ rD )ÚBranchÚBuyerri ÚDater r= r§ r rm r, rà z Carl Joe Markr{ r ri ©r8 r9 r7 rP ry Z6MSZCarl©r½ rÁ z#'The grouper name foo is not found'r r% )r½ r¼ zThe level foo is not valid)r Úsplitr r r2 rp r r rÄ r8 rA r9 r r rT r; r< rb r rU r¹ r r r& r )r5 Úusing_array_managerrV rH rW r r r r Útest_pivot_timegrouperÓ s
|
|
|
|
|
|
|
|
øüÿðýû û ÿûû û
|
û
|
û
|
û
|
û
|
|
û
|
û
|
û
|
ûz%TestPivotTable.test_pivot_timegrouperc C s t d ¡ d ¡ ddddddddgtdd dd
|
d tdddd
|
dtdd dd d tdd dd d tdd dd d tdd dd d tdd ddd tddddd gtdd dd d tdd dd
|
dtdddd d tdd dd d tdd dd d tdddd d tddddd tdd d dd gd}t|tdddtddddtjd}t t tjdtjtjd tjddtjdtjtjtjtjdtjg¡ dd¡t
|
j tdddtdd d tdd dtddd gdd t
|
j tdddtdd d tdd dtddd gdd d }d|j _ d|j_ t ||¡ t|tdddtddddtjd}t ||j¡ tdddtdd d ftdd d tdddftdd d tdd dftdd d tddd ftdd dtdd d ftddd tdd dfg}tj|ddgd }t t dtjd tjdtjdtjdtjtjdg ¡ d d¡|d!d"gd }d#|j_ t|tdddtdddgd#gdtjd}t ||¡ t|d#gtdddtdddgdtjd}t ||j¡ d S )$Nr r r r rK rz rt rè r+ r: r r rÝ rJ rD r< r rÛ ry é )r r ri r ÚPayDayrÀ r r r ri r rx rm rà r{ rj r, r- r )r r r r r r2 rT rp r r r rÄ r8 rA r9 r; r< rb r rk )r5 rV rW rH ZtuplesrÉ r r r Ú test_pivot_timegrouper_doublek sê ø øòÿ
|
|
ûðÿ í
|
|
|
|
üù
|
|
|
|
üùâ(
|
|
ûú&ÿ þûû ûz,TestPivotTable.test_pivot_timegrouper_doublec C sÔ ddddddg}ddddddg}t ddddddg||tjdd d
|
d d gd d}|d dd ¡|d< |d dd ¡|d< tjdddgddd}tddg}tddgdd}t ||g¡}t dd gd dgd dgg||d }t |dgdgdgd } t
|
| |¡ td d d d d d d d g}tddd d gd }tjddgd d!dd}
|
t |||
|
g¡}t tj dd d d dd d d gd dd d d dd d gd dd d d dd d ggd d
|
||d }t |dgdgdd gtj tjgd"} t
|
| |¡ d S )#Nú2011-07-19 07:00:00ú2011-07-19 08:00:00ú2011-07-19 09:00:00ú2013-01-01 15:00:00ú2013-02-01 15:00:00rY rZ rx r§ r r rJ r ©Úlabelrê rë Úvalue1Úvalue2rê c S s t j| ddS rî rò ró r r r r r¡ z7TestPivotTable.test_pivot_datetime_tz.<locals>.<lambda>rë c S s t j| ddS rô rò ró r r r r r¡ rï )rñ rA r r rw r r rK r{ rµ rT rD r¡ rõ r )r r2 r rö r rÄ r
|
r r r r; r< rp rT rD ) r5 Údates1Údates2rV Úexp_idxrø rù rú rH rW Zexp_col3r r r Útest_pivot_datetime_tzê s| ú ú
|
ûÿ ý
|
ýýúö
|
ûz%TestPivotTable.test_pivot_datetime_tzc
|
C s ddddddg}ddddddg}t ddddddg||tjdd d
|
d d gd d}|d dd ¡|d< |d dd ¡|d< t|d|d jjdd}tddgdd}t dd gd dgd dgd|td d d gtjdd d }t
|
||¡ t||d jj |d jjdd}t dd gd dgd dgdtd d gtjdd td d d gtjdd d }t
|
||¡ t||d jj j |d jj|d jj gdd}tjtjd d d d d d gtjd
|
tjd d gd tjd
|
gddgd!}t tjdd d dd dggd d
|
td"gtjd
|
|d }t
|
||¡ t|t d#d#d#d#d$d$g¡|d jj|d jj gdd}t t dd d tjd tjgtjtjtjdtjdgg¡d#d$g|d }t
|
||¡ d S )%Nr r r r r rY rZ rx r§ r r rJ r r rê c S s
|
t | ¡S r rò ró r r r r L r¡ z6TestPivotTable.test_pivot_dtaccessor.<locals>.<lambda>rë c S s
|
t | ¡S r rò ró r r r r M r¡ r r rµ rw r r rK )ry rz rt ry rz rt ©r rA r{ rj rè ÚXÚY)r r2 r rö r r¿ rþ r
|
rÇ r; r< rh rs r7 r r rp r )r5 r¢ r£ rV rW r¤ rH rú r r r Útest_pivot_dtaccessor1 s¢ ú ú
|
ûÿ
|
ÿý
|
|
ÿý üþûý ü*ÿûz$TestPivotTable.test_pivot_dtaccessorr¦ r io c C s t dddd}ttj t|¡|d}tt||jj |jj
|
d}|j d¡|_ t |jj
|
¡}|||k }|jj |_|| ¡ }tj||dd |j|kstd S )
|
Nú1/1/2000ú
|
12/31/2004r/ rà rr r{ r Fr( )r r r2 r3 r4 r= r r r8 rs Z dayofyearr9 Ú droplevelZasarrayr r; r rA r@ )r5 r¦ Úrngrý ÚannualZdoyÚsubsetrW r r r Ú
|
test_daily s ÿ
|
zTestPivotTable.test_dailyc C s t dddd}ttj t|¡|d}tt||jj |jj
|
d}|j d¡|_ |jj
|
}|||k }|jj |_|| ¡ }tj||dd |j|kstd S )
|
Nrª r« rÀ rà rr r{ r Fr( )r r r2 r3 r4 r= r r r8 rs rh r9 r¬ r r; r rA r@ )r5 r¦ r rý r® rh r¯ rW r r r Ú test_monthly s
|
zTestPivotTable.test_monthlyc C sv ddd}t |dgt| ¡ |d}t |dg| ¡ |d}t ||¡ dd |D }t |dg||d}t ||¡ d S )NrT rD ry r, )r8 r7 rP c s s | ]
|
}|V qd S r r )r¶ rc r r r Ú <genexpr>¹ s zGTestPivotTable.test_pivot_table_with_iterator_values.<locals>.<genexpr>)r rS rÖ r; r< )r5 r6 ZaggsZpivot_values_listZpivot_values_keysZagg_values_genZpivot_values_genr r r Ú%test_pivot_table_with_iterator_values¬ s,
|
|
ÿ ÿ ÿz4TestPivotTable.test_pivot_table_with_iterator_valuesc C sª d}t ddddgddddgddd d gd
|
}|jd d d |tjtgd}tdd|gdd d}dddd|fdddd|fg}tj|d d d gd}t |j||d}t
|
||¡ d S )NZWeeklyZbaconZcheeserª r¬ g @gffffff
|
@rÀ rb )r6 ÚcostrÅ r6 rÅ T)r8 r9 r r, rP r r¦ )rD r´ rÀ )rD r´ rb rD r´ )Úmaxr´ rÀ )rµ r´ rb rµ rj r{ ) r r r2 rD rµ r
|
r rk r7 r; r< )r5 r, ZcostsrF ÚixZtupsr` rH r r r Ú/test_pivot_table_margins_name_with_aggfunc_list¿ s2
|
|
|
ýÿûúz>TestPivotTable.test_pivot_table_margins_name_with_aggfunc_listc C s° |r|j tjjdd¡ tt d¡t d¡d t d¡d d}tddd gd
|
d d gd ddgg}tdddgdd|_ tdddgdd|_
|
|j ddd|dd}t ||¡ d S )Nú1GH#17035 (np.mean of ints is casted back to ints)r" rz r rJ ©r\ r] r^ r r r© rK rx r8 r r« r r r r] rw r^ r\ T©r r )ÚnodeÚ
|
add_markerr ÚmarkÚxfailr r2 r r
|
r8 r9 r r; r< ©r5 r: r rV rH rF r r r Útest_categorical_marginsÝ s ÿÿ$ÿ z'TestPivotTable.test_categorical_marginsc C sÌ |r|j tjjdd¡ tt d¡t d¡d t d¡d d}tddd gd
|
d d gd ddgg}tdddgdd|_ tdddgdd|_
|
|j d¡|_ |j d¡|_ |jddd|dd}t ||¡ d S )Nr¸ r" rz r rJ r¹ r r r© rK rx r8 r r« r r r r] rw r^ rR r\ Trº )r» r¼ r r½ r¾ r r2 r r
|
r8 r9 r] rU r^ r r; r< r¿ r r r Ú!test_categorical_margins_categoryð s ÿÿ$ÿ z0TestPivotTable.test_categorical_margins_categoryc C s t ddddgddddgddddgddd d gd
|
}t|d d d }t dddgdddgdddgdtdd dgd dd}t ||¡ d S )NrJ r rx rz r rK r r§ r¨ )r, r- r. r/ r/ T)r8 r ry rª g @r¬ r« ru r rw rr )r r r
|
r; r< ©r5 rV rW rH r r r Ú test_margins_casted_to_float s
|
|
|
|
üÿ þz+TestPivotTable.test_margins_casted_to_floatc C sþ t jdddt jg}t jddt jdg}tt|ddg|dt|ddg|dtddd}|jd d
|
d |d }tjddg|d
|
d }tdt jgt jdgg|d}ttddgddg|dd d|_ t
|
||¡ |jd
|
d |d}tddgg|td gd}t
|
||¡ d S )Nr r r, r- ro r rx )ÚInÚColÚValrÄ rÅ rÆ )r8 r9 r7 r: )rq rA r r )r6 r9 rw )r9 r7 r: r« )r6 r9 r8 ) r2 r r r r~ r r ÚCategoricalIndexr
|
r8 r; r< ) r5 r: rq rÉ r· rV rW Z expected_colsrH r r r Útest_pivot_with_categorical s8 ýÿ ÿ þ ÿz*TestPivotTable.test_pivot_with_categoricalc C s¼ t ddddgddddgdddd gd
|
}|d d ¡|d < |jd d d|dd}tjdddgdddgdd d}tddgdd}tjddgddgddggtjd}t |||d}t
|
||¡ d S )Nr, r- r. rY rZ r rJ r r )rÙ rÚ ÚVrÙ rR rÉ rÚ Úcount)r8 r9 r rP F©rp rq rA rw r r r{ ) r rU r r rÇ r
|
r2 rp r§ r; r< )r5 r: rV rW r r Ú expected_datarH r r r Útest_categorical_aggfunc9 s2 "ÿ ÿ ÿ" ÿz'TestPivotTable.test_categorical_aggfuncc C sÖ t dddgdddgdddgd}ddd d
|
d d d dddddg }|d d¡j |¡|d< |jddd|dd}tddgddd}tj||ddd }d dggd d ggd }t |||d }|rÆ|jdg }t
|
||¡ d S )!Nrá rJ éÜ ÚJanuaryrè ré )ÚSalesrh rg ÚFebruaryZMarchZAprilÚMayZJuneZJulyZAugustZ SeptemberZOctoberZNovemberZDecemberrh rR rÐ rg rT )r7 r8 r9 r: rP r§ )rA r FrË i@ r r+ r{ ) r rU r¤ Zset_categoriesr r
|
r rÇ rã r; r< ) r5 r: rV ÚmonthsrW r r rÌ rH r r r Ú%test_categorical_pivot_index_orderingM sV ýÿôû ÿ ÿ z4TestPivotTable.test_categorical_pivot_index_orderingc C sB t dddgdddgdddgd }|jd
|
d d gtjd }tjdddgdddggd d gd}t dddg|d
|
gd}t ||¡ |jd
|
d d tjd}t dtjtjgtjdtjgtjtjdggt dddgd dt dddgd dd}t ||¡ |jd
|
d tjgd }t dgd
|
gg¡}t dddgt dddgd d|d}t ||¡ d S )Nr r rK r. r/ r0 r rt rL rM rO rN ra rj r{ rn rw rT )
|
r r r2 rT r r r; r< ÚNaNr
|
)r5 rV rW rm rH r r r Útest_pivot_table_not_seriesy s "" &ý "z*TestPivotTable.test_pivot_table_not_seriesc C s\ d}t ddddgi}t|dgtd|d}tddd|gddd }t |g d
|
}t ||¡ d S ) Nu Îοκιμήr% r rJ r T)r8 rP r r, r r¦ r{ )r r r= r
|
r; r< )r5 ÚgreekÚframerF r8 rH r r r Ú test_pivot_margins_name_unicode s ÿ z.TestPivotTable.test_pivot_margins_name_unicodec C sB t dddddddddddg dddddddddddg tdd}t|ddd d
|
}td gddggd d gd d ggd dgd}t ddddddd|d d¡}t ||¡ t|ddd dgd
|
}td dgd gddggd d d d gd d d d gd d d d ggd d dgd}t ddddddddddddd|d d¡}t ||¡ d S ) Nr% r& r' r( r+ ru r, r- rT rn r. r r r rÛ r: )r& r% ry rÝ ))r. r' )r. r( r| rD ç @r r g«ªªªªª@))rD r. r' )rD r. r( )rT r. r' )rT r. r( )r r~ r r rN r; r< )r5 r6 rW rP rH r r r Útest_pivot_string_as_func st õõ åÿ ÿþý ýüùø z(TestPivotTable.test_pivot_string_as_funcz
|
f, f_numpyrD rÕ c C s< |j dd}t|dd|d}t|dd|d}t ||¡ d S )Nr. r| r, r- rn )rÏ r r; r< )r5 r¯ Zf_numpyr6 rW rH r r r Ú test_pivot_string_func_vs_funcÞ s z-TestPivotTable.test_pivot_string_func_vs_funcc C s G dd dt j}| ¡ z}| t d|¡ tt d¡t d¡dd}d}tjt |d2 t
|
j t d d |j d
|
d d d 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 )zSTestPivotTable.test_pivot_number_of_levels_larger_than_int32.<locals>.MockUnstackerN)Úreturnc s t j|| tdd S )NúDon't compute final result.)ÚsuperÚ__init__Ú Exception)r5 ÚargsÚkwargs©Ú __class__r r rà ö s z\TestPivotTable.test_pivot_number_of_levels_larger_than_int32.<locals>.MockUnstacker.__init__)Ú__name__Ú
|
__module__Ú __qualname__rà Ú __classcell__r r rä r Ú MockUnstackerõ s rê Ú
|
_Unstackeri r )Úind1Úind2rÊ z$The following operation may generater rÞ rì rí rÊ r )Ú reshape_librë ÚcontextÚsetattrr r2 r r; Úassert_produces_warningr r r rá r )r5 Z monkeypatchrê rm rV r r r r Ú-test_pivot_number_of_levels_larger_than_int32ñ s
|
ÿ ÿz<TestPivotTable.test_pivot_number_of_levels_larger_than_int32c
|
C sÊ t dddgdddgdddgd}dd }d
|
d }d d }t|d|||g|d}ddtjtjddgddtjtjddgg}tjdddgddggd dgd}t |ddg|d} |rº| jdd} t || ¡ d S )NZappleZpeachr rJ ry rx )ÚfruitÚsizeÚtastec S s dS )Nr r r r r r Úret_one s z?TestPivotTable.test_pivot_table_aggfunc_dropna.<locals>.ret_onec S s t | S r )rT r r r r Úret_sum s z?TestPivotTable.test_pivot_table_aggfunc_dropna.<locals>.ret_sumc S s t jS r )r2 r r r r r Úret_none s z@TestPivotTable.test_pivot_table_aggfunc_dropna.<locals>.ret_noneró ©r9 rP r r r: r÷ rø rö rj rô rõ r{ r9 rÍ ) r r r2 r r Z from_productr r; r< )
|
r5 r rV rö r÷ rø rW r6 r· rH r r r Ú test_pivot_table_aggfunc_dropna s0 ýÿ ÿ(þ z.TestPivotTable.test_pivot_table_aggfunc_dropnac C s t dddgdtjdgdtjtjgd}t|dtj|d}d tjgdtjgg}tddgdd
|
}t |d d g|d }|r|jdd}t ||¡ d S )Nr' r( r rJ r )r, r\ r] r, rù rª rw r\ r] r{ r9 rÍ ) r r2 r r rD r
|
r r; r< )r5 r rV rW r6 r· rH r r r Ú&test_pivot_table_aggfunc_scalar_dropna+ s "ÿ z5TestPivotTable.test_pivot_table_aggfunc_scalar_dropnar c C s t dddddgdddddgdd d dd gd d d d d gd
|
}|jd d d tj|d}tg d d}t tg dd d|d}t ||¡ d S )NrJ r rK rx ry rz rt Úpr )r, Úidr. r/ r, r/ rý r rw r§ r¦ r{ )r r r2 rô r
|
r; r< )r5 r rV rW Zexp_colsrH r r r Ú test_pivot_table_empty_aggfunc< s" üÿ ÿ z-TestPivotTable.test_pivot_table_empty_aggfuncc C s` dd }t ddddgddddgddddgd }tjtd
|
d |jd
|
d d |d W 5 Q R X d S )Nc S s
|
t | ¡S r )r2 rD )Zarrr r r rC P s z=TestPivotTable.test_pivot_table_no_column_raises.<locals>.aggr r r rÝ r é( )r§ r¨ ÚZZ
|
notpresentr r§ r¨ )rP )r r r r& r )r5 rC rV r r r Ú!test_pivot_table_no_column_raisesN s (z0TestPivotTable.test_pivot_table_no_column_raisesc C s: t dddddddddg dddddddddg dddddddddg dddd d d
|
d d d g dd
|
d d d d dddg d}t|ddgddgtjtttjgdd}t ddddg¡}tjdd d d gddgd }t d d!d"d#gd d!d$d%gd&d'd(d)gd*d#d+d&gg¡}t |||d,}|d tj
|
¡|d< |d tj
|
¡|d< t ||¡ d S )-Nr% r& r' r( r? r@ r rJ r r rK rx ry rz rt rA r/ r0 r, r. ry )r7 r8 rP )r/ rD )r0 rµ )r0 rD )r0 r )r& r@ )r& r? )r% r@ )r% r? rj r8 g "@g @ç @g !@r9 r rÚ r¬ r g["8ªªª@g. UUU@rT ) r r r2 rD r rµ r rk rp rU r§ r; r< )r5 rV rF r` r8 ÚvalsrH r r r Ú0test_pivot_table_multiindex_columns_doctest_caseW sR ÷ ñÿü
|
ÿ
|
þ
|
|
|
|
üÿz?TestPivotTable.test_pivot_table_multiindex_columns_doctest_casec C s¢ t dddgdddgddd gd
|
d
|
d gd }|jdd gddddd}t dtjgdtjgtjd ggtd
|
d gddtjdddgdddggdd gdd}t ||¡ d S )NÚd1Zd4Zd3rY rZ r[ é rE é" Z2018Z2019)rY r· Únumrs r· rs r rT F)r8 r9 r7 rP Úsortrw rj rT ) r r r2 r r
|
r r r; r< r r r r Útest_pivot_table_sort_false s. üÿ ÿ ÿýz*TestPivotTable.test_pivot_table_sort_falsec C s t dddgtddgddd}|jd d
|
d d d }t ddgddgddggtdddgd dtjddgd d
|
gddd}t ||¡ d S )Nr, r rJ r r+ ÚInt64r )rY rZ ÚsalesrZ rY TrT r- rE r rw ©r r, ©r r rj ©r8 r9 r )r r r r
|
r rk r; r< r r r r Ú!test_pivot_table_nullable_margins¡ s ÿ ÿúz0TestPivotTable.test_pivot_table_nullable_marginsc C sz t ddgddgddgddgd }|jd
|
d gd d gdd}t ddgddggd d gtjddgd
|
d gdd}t ||¡ d S )NÚJohnÚMichaelÚFooÚBaré é¶ é/ rG )Ú firstnameÚlastnameÚheightÚager r r r F)r8 r7 r )r r )r r rj rT )r r r rk r; r< r r r r Ú0test_pivot_table_sort_false_with_multiple_values² s* üÿ ÿþýz?TestPivotTable.test_pivot_table_sort_false_with_multiple_valuesc C sª t dddgdddgdddgddd gg}d
|
d d g|_|jd
|
d d d ddd}t ddddgddd dgddd d
|
gg}ddddg|_dddg|_d |j_d
|
|j_t ||¡ d S )NrY r\ r r] rJ rZ r r^ r r rÝ r rT r T)r8 r9 r7 rP rQ r ry rK r )r r9 r r8 rA r; r< r r r r Ú1test_pivot_table_with_margins_and_numeric_columnsÈ s ( ÿ& z@TestPivotTable.test_pivot_table_with_margins_and_numeric_columnsr c C sh t ddtddgddd}|jdd d
|
d |d }t d ggtdgddtdgd ddd}t ||¡ d S )NrY rZ rÝ rÿ r r )r\ r] r r\ r] r rD )r8 r9 r7 rP r r rw ZFloat64r )r r r r
|
r; r< r r r r Útest_pivot_ea_dtype_dropnaÙ s ÿ üz)TestPivotTable.test_pivot_ea_dtype_dropnac
|
C s t dddgt d¡ddgd}t d ¡ |jdd gd
|
d d d }W 5 Q R X t ddgddgddggt tdddgddtt d¡t d¡dgt d dg¡tj
|
ddgd d
|
gdd}t ||¡ d S )Nr, r rJ z
|
2019-12-31g $@r+ )rY rZ r r rZ r rY TrT r- r~ g 5@r rw r) r¦ r r rj r{ ) r r rÈ r; rñ r r r r
|
r rk r< r r r r Ú!test_pivot_table_datetime_warningè s> üÿ ÿýþÿ
|
ÿôz0TestPivotTable.test_pivot_table_datetime_warningc C sB t dddddddddg dddddddddg dddddddddg dddd d d
|
d d d g dd
|
d d d d dddg dddddddddg dddddddddg dddddddddg d}t|dddgdgtjd}t ddgddgddgtjdggtddgddtjddddgddgd gddgd d }|r2|d d ¡|d< t
|
||¡ d S ) Nr% r& r' r( r? r@ r rJ r r rK rx ry rz rt )r, r- r. r/ r0 )Zcol5)Zcol6rx ©ry Zsevenr/ r, r- r rÑ r rÚ r r r rw rj rT r§ ) r r r2 rT r r
|
r r rU r; r< )r5 r rV rW rH r r r Ú)test_pivot_table_with_mixed_nested_tuples s ÷ ÷ ÷ ÷Úÿ4 ÿ ÿýz8TestPivotTable.test_pivot_table_with_mixed_nested_tuples)aræ rç rè r Úfixturer6 rI rX rd rn rs r r r r r r r¢ r£ r° r½ Z parametrizer» rÌ rÐ r× rà rç rû r r
|
r r2 rp r r
|
r r r! r¾ r% r' r+ r/ r7 rD r rk rB rQ rR rV rX ra rf rx rz r| r r r r r r r¥ r© r~ r° r± r³ r· rÀ rÁ rà rÈ rÍ rÔ rÖ rÙ rÛ rT rÕ rÜ Zslowrò rú rû rþ r r r
|
r r r r r r! r r r r r$ ) s>
|
0 % #
|
ýþ
|
*
|
|
|
;7
|
-
|
|
üþ
|
|
üþ !ÿ &
|
ù
|
üþú÷ùùþ
|
#
|
3
|
% GY
|
|
#, >úþ
|
|
#
|
4
|
r$ c @ sd e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dS )Ú TestPivotc C sÔ ddddddgddddddgddddddgd }t |}|jd
|
d d d }t dddd dddd d}d\|j_|j_t ||¡ |jjd
|
kst|jjd ks¢t|jd
|
d d}|jjd
|
ksÀt|jjdksÐtd S )Nr, r- r. ÚOneÚTwor r r rµ r8 r9 r7 ru ©r$ r% r{ ©Nr9 ) r rß r8 rA r9 r; r< r@ r> )r5 r6 rØ rw rH r r r Ú
|
test_pivotO s$ ý
|
|
þÿ zTestPivot.test_pivotc C s\ t dddddgdddddgdddddgd }tjtd
|
d |jd d dd W 5 Q R X d S )Nr& r% r' r( r r r r rÞ zduplicate entriesr rY rZ r[ rµ )r r r r rß )r5 r6 r r r Útest_pivot_duplicatesl s ýÿz TestPivot.test_pivot_duplicatesc C s@ t dddgd}|jdddd}t g g d}tj||dd d S ) NrY rZ r[ r| rµ r{ Fr( )r rß r; r< r r r r Útest_pivot_emptyw s zTestPivot.test_pivot_emptyc C sD t ddgd}|jdddd}t| t |jtdd gdd
|
¡ d S ) N)r, r\ ÚA1)r- r^ ZB2rÔ r r rJ rµ r, r- rw )r rß Úreprr; rl r9 r
|
)r5 rV rW r r r Útest_pivot_integer_bug} s z TestPivot.test_pivot_integer_bugc C s ddddddgddddddgddddddgd }t | d
|
¡}|jd d d }t ddddddddd}d\|j_|j_t ||¡ |jd d}tj ddgd d gd|_d
|
|j_tj||dd |jjd
|
ksÒt
|
|jj dksât
|
|j d¡|_|jd d d }d |j_t ||¡ d S )Nr, r- r. r$ r% r r r rµ r8 r9 r7 ©r9 r7 ru r& r{ r| )r7 r$ )r7 r% rj Fr( r' r ) r r rß r8 rA r9 r; r< r rk r@ r> r¬ )r5 r6 rØ rW rH r r r Útest_pivot_index_none s4 ý
|
|
þÿ ÿz TestPivot.test_pivot_index_nonec C s t ddddddgddddddgddddddgddddddgddddddgd}d d
|
g}d g}|j||d }t t d ddd gdtjdtjgddddgtjdtjdgg¡tjddgd d
|
gdtjddgd d gdd }t ||¡ |d d
|
gksðt |d gksþt d S )Nr rJ r r rK rx r )Úlev1Úlev2Úlev3Úlev4r7 r0 r1 r2 r{ r r r
r rÚ r r )r r rJ rJ )r rJ r rJ rj )r3 r3 r7 r7 )
|
r rß r2 rp r r r r; r< r@ )r5 rV r8 r9 rW rH r r r Ú0test_pivot_index_list_values_none_immutable_args¨ s> ûÿ
|
|
üÿ ÿþô z:TestPivot.test_pivot_index_list_values_none_immutable_argsc C s6 t dgdd}tjtdd | ¡ W 5 Q R X d S )Nr rS z(missing 1 required keyword-only argumentr r )r5 rV r r r Ú test_pivot_columns_not_givenÎ s z&TestPivot.test_pivot_columns_not_givenc C s¨ t dgddd}|jd d}t dgdd}t ||¡ |jd dd}t d ditdgdd
|
d }t ||¡ |jd dd d }t dditdgdd
|
d }t ||¡ d S )Nr rJ r ©NrZ r[ r| ))rZ r ©r[ r rZ rT r7 rw rr r[ ©r9 r8 r7 )r rß r; r< r
|
r r r r Útest_pivot_columns_is_noneÔ s z$TestPivot.test_pivot_columns_is_nonec C s t dgddd}|jdd d}t ddidgd}d dg|j_t ||¡ |jdd d d
|
}t ddgtdgdd d }t ||¡ d S ) Nr rJ r r6 rZ rT )r[ rJ rr r[ r8 rw r{ )r rß r9 r> r; r< r
|
r r r r Útest_pivot_index_is_noneã s z"TestPivot.test_pivot_index_is_nonec C s t dgddd}|jddd d}t dtdgddtdgddd }t ||¡ |jdd d
|
}t dd gtdgddd }t ||¡ d S ) Nr rJ r r6 rZ r[ r8 rw r{ r. r )r rß r
|
r; r< r r r r Útest_pivot_values_is_noneð s ÿ z#TestPivot.test_pivot_values_is_nonec C s> t dgddd}|jdd}|jddd d
|
t ||¡ d S ) NrY r r )r' r( ÚthreeT)Údeepr' r( r< rµ )r rM rß r; r< )r5 rV rH r r r Ú"test_pivot_not_changing_index_nameþ s z,TestPivot.test_pivot_not_changing_index_nameN)ræ rç rè r( r) r* r- r/ r4 r5 r9 r: r; r> r r r r r# N s $& r# )%r r r Ú itertoolsr Únumpyr2 r Z pandas.errorsr Zpandasr r r r r
|
r r r r Zpandas._testingZ_testingr; Zpandas.api.typesr r Zpandas.core.reshaper rî Zpandas.core.reshape.pivotr r" r r~ r# r$ r# r r r r Ú<module> sD (
|
|
.
|
7
|
