U
|
ý°d & ã @ s¤ d dl Z d dlZd dlZd dlZd dlmZ ej
|
dd d ddggddggdg gdddd ggd
|
dd ggd g gg¡d d Z dd Z ej j
|
dd dgd dddggd ddgd ddddd gge d ddd dddggd dhd dddgggddddgddd Zdd Zdd Zd d Zej
|
dd d ddggd ddgd ddddd ggedded
|
dfd ddddd d ggg¡d d Zd d! Ze ¡ d"d# Zd$d% Zd&d' Zej
|
d(ddd)d*dd+d,d-g¡ej
|
d.d/d0g¡ej
|
d1d2d3g¡d4d5 Zej
|
d6dd dd7dd8g¡ej
|
d9dd dd7dd8g¡ej
|
d:dddg¡d;d< Zd=d> Zej
|
d:dddddg¡d?d@ Z e ¡ dAdB Z dCdD Z dEdF Z ej
|
dGe ej!ej"ej#g¡dHdI Z$dJdK Z%dS )Lé Nzarg, expected_rowsé é é é éÿÿÿÿé é éþÿÿÿé iúÿÿÿc C s$ |j | }| j| }t ||¡ d S ©N©Ú_positional_selectorÚilocÚtmÚassert_frame_equal©Ú slice_test_dfÚslice_test_groupedÚargÚ expected_rowsÚresultÚexpected© r úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/groupby/test_indexing.pyÚtest_int s
|
|
r c C s2 |j ddd
}| jddddg }t ||¡ d S )Nr r r r r r r ©r r r r r r r Ú
|
test_slice s r ÚlistÚnegativeÚrangeÚset)Úidsc C s$ |j | }| j| }t ||¡ d S r r r r r r Ú test_list' s
|
|
r" c C s0 |j d }| jddddddg }t ||¡ d S )N)r r r r r r r r r r r r r r Ú test_ints> s
|
r# c C s> |j d d
dd
f }| jdddddddg }t ||¡ d S ) Nr r r r r r r
|
r r r r r r Ú test_slicesF s r$ c C s< |j dddd
f }| jdddddddg }t ||¡ d S ) Nr r r r r r r
|
r r r r r r Útest_mixN s r% c C s. | j dddj| }| j| }t ||¡ d S )NÚGroupF)Úsort)Úgroupbyr r r r )r r r r r r r r Ú test_as_indexV s
|
|
r) c C s t jddgddgddgddgddggdd gd
|
} | jdd d }|jdd
}t jddgddggdd gddgd }t ||¡ |jd }t jddgddgddggdd gdddgd }t ||¡ d S )NÚar r r Úbr r ÚAÚB©ÚcolumnsF©Zas_index©r/ Úindex)r r )ÚpdÚ DataFramer( r r r )ÚdfÚgroupedr r r r r Útest_doc_examplesf s ÿ$
|
ÿr7 c C sz d} d}t jd| dd}dd t|D }i }|D ]B}|t dd ¡ }d
|
d |d |
D }|jd d d |||< q2|S )Néd é Z20130101ÚD)ZperiodsÚfreqc S s g | ]}d | qS )zitem r ©Ú.0Úir r r Ú
|
<listcomp> s z#multiindex_data.<locals>.<listcomp>r é c S s. g | ]&}|t d d¡d t d d¡d fqS )r i' r8 )ÚrandomÚrandint)r= Úitemr r r r?
s ÿc S s | d S )Nr r ©Úxr r r Ú<lambda> ó z!multiindex_data.<locals>.<lambda>)Úkey)r3 Z
|
date_ranger rA rB r' )ZndatesZnitemsÚdatesÚitemsÚdataÚdateZnitems_for_dateÚlevelsr r r Úmultiindex_data{ s
|
þ
|
rN c C sV i }| D ]0}| | D ]"}|d |d d|||d f< qqt jj|dd}d|j_|S )Nr r )r, r- r r2 )Zorient)ÚDateZItem)r3 r4 Ú from_dictr2 Únames)rK ÚrowsrL Úlevelr5 r r r Ú_make_df_from_data s "rT c sL t }|jddd tdd¡} fdd D }t |}t ||¡ d S )NrO Fr0 r éýÿÿÿc s i | ]}| | d d
qS )r rU r )r= rL ©rN r r Ú
|
<dictcomp> s z#test_multiindex.<locals>.<dictcomp>)rT r( ZnthÚslicer r )rN r5 r Zslicedr r rV r Útest_multiindex s
|
rY r é iè éûÿÿÿiâÿÿÿiüÿÿÚmethodÚheadÚtailÚ simulatedTFc sl d d} fddt |D fddt |D d}t |¡}|jddd }| d
|
krZ| n|| }|d krÚ|jd |
}|rÎg } t |D ]8}
|
t D ]*} |
|
| | k r| |
|
| ¡ qq|j| } n
|
| | ¡} n|j| d
}|rRg } t |D ]H}
|
t D ]:} ||
|
| | d
|
kr| ||
|
| | ¡ qqü|j| } n
|
| | ¡} t
|
|| ¡ d S ) Nr8 rZ c s$ g | ] }t D ]}d | qqS )úgroup ©r ©r= ÚjÚg©Zn_groupsr r r? ® s
|
z.test_against_head_and_tail.<locals>.<listcomp>c s* g | ]"}t D ]}d | d| qqS )r` z row ra rb re r r r? ± s
|
þ©ÚgroupÚvaluerg Fr0 r r] ) r r3 r4 r( r Úappendr r] r^ r r ) r r\ r_ Zn_rows_per_grouprK r5 r6 Úsizer Úindicesrc r> r r re r Útest_against_head_and_tail¥ s@
|
ÿ
|
þü
|
|
|
rl Ústarté
|
iöÿÿÿÚstopÚstepc C sd d}dg| t t|d}t |¡}|jddd}|j| ||
}|j| ||
}t ||¡ d S )NrZ zgroup 0rf rg Fr0 ) r r r3 r4 r( r r r r ) rm ro rp Zn_rowsrK r5 r6 r r r r r Útest_against_df_ilocÜ s
|
þ
|
rq c C sb t jdddddgdddddgd} | jd d
|
}|jdd
}t jddgddgd}t ||¡ d S ) Nr r r r r r* r+ )r2 r )rS )r3 ÚSeriesr( r r Zassert_series_equal)Zserr6 r r r r r Ú test_seriesð s
|
" rs c C s& dd t dD }|dd t dD 7 }|dd t dD 7 }tj|dd gd
|
}|jdd d }|jd d |
}d d t dd| D }|dd t dd| D 7 }|dd t dd| D 7 }dd t dd| D }|dd t dd| D 7 }|dd t dd| D 7 }tj|dd g|d}t ||¡ d S )Nc S s g | ]}d d | gqS rD r r< r r r r? ý s z test_step.<locals>.<listcomp>r c S s g | ]}d d | gqS ©Úyr r< r r r r? þ s r c S s g | ]}d d | gqS ©Úzr r< r r r r? ÿ s r r, r- r. Fr0 c S s g | ]}d d | gqS rD r r< r r r r? s r c S s g | ]}d d | gqS rt r r< r r r r? s c S s g | ]}d d | gqS rv r r< r r r r? s c S s g | ] }d | qS )r r r< r r r r?
|
s c S s g | ] }d | qS )r r r< r r r r? s c S s g | ] }d | qS )é r r< r r r r? s r1 )r r3 r4 r( r r r )rp rK r5 r6 r r2 r r r r Ú test_stepú s ry c C s< t jdddddddgdddddddggdd d
|
d d d dgdS )Nr r r r r r r
|
r, r- ÚCr: ÚEÚFÚGr. )r3 r4 r r r r Úcolumn_group_df s "þr~ c C sH | j | jd dd}|jdd
}| jd d
ddgf }t ||¡ d S )Nr )Zaxisr r )r( r r r r )r~ rd r r r r r Útest_column_axis s r c C s` t dd dD ¡} ddg}| | jdk ¡| D ]*\}}t |jt |¡¡ d|jks0tq0d S )Nc S s i | ]}|t d qS )rn ra )r= Úkr r r rW % s z(test_columns_on_iter.<locals>.<dictcomp>ÚABCr, r- r rz ) r3 r4 r( r, r Zassert_index_equalr/ ÚIndexÚAssertionError)r5 ÚcolsÚ_Zdgr r r Útest_columns_on_iter# s
|
r Úfuncc C st t ddgddgddgd¡}| d¡| dddg ¡ }t jd d
|
d ggdddgt jdgdd d }t ||¡ d S ) Nr r r r} )r, r- rz rz r, r- g ø?g @)Únamer1 )r3 r4 r( Zmeanr r r )r r5 r r r r r Ú test_groupby_duplicated_columns. s ýÿ
|
ÿr c C sb t jddd gdddgdddgdd } | jd
|
d gd }d }tjt|d | d¡ W 5 Q R X d S )NZa1Úa2Úb1Zb2r r r )r, r- Úval)rK r, r- )Zbyz ('a2', 'b1'))Úmatch)r r )r3 r4 r( ÚpytestZraisesÚKeyErrorZ get_group)r5 ZgrpsÚmsgr r r Ú#test_groupby_get_nonexisting_groups? s ýÿr )&rA ÚnumpyÚnpr Zpandasr3 Zpandas._testingZ_testingr ÚmarkZ parametrizer r r r" r# r$ r% rX r) r7 ZfixturerN rT rY rl rq rs ry r~ r r r r rr Úarrayr r r r r r Ú<module> s
|
úþ
|
üüø
|
$ýþ
|
|
4
|
|
|
|
|
