U
|
®ý°d§ ã @ s8 d dl Zd dlZd dlmZ d dlmZ ejdd Z ejdd Z
|
ejg dgddggd d
|
d Z ejg dgddggd d d Z dddZ ej ddgdfdgdfddgdfddgdfg¡dd Zej ddgdgdfdgdgdfddgddgdfddgddgdfg¡dd Zej ddddgg¡dd ZdS ) é N)Ú DataFramec C sF t dddddddddddg dddddddddddg t ddd¡dS )Né é é é r é )ÚouterÚinnerZv1©r ÚnpZlinspace© r r úld:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/reshape/merge/test_merge_index_as_string.pyÚdf1 s ýÿr c C sJ t ddddddddddddg ddddddddddddg t ddd¡dS ) Nr r r r é
|
r é )r r Zv2r
|
r r r r Údf2 s ýÿr r r )Úparamsc C s | j }|r| |¡}|S )z`Construct left test DataFrame with specified levels
|
(any of 'outer', 'inner', and 'v1')
|
©ÚparamÚ set_index)Úrequestr Úlevelsr r r Úleft_df s
|
r c C s | j }|r| |¡}|S )zaConstruct right test DataFrame with specified levels
|
(any of 'outer', 'inner', and 'v2')
|
r )r r r r r r Úright_df* s
|
r c sú |dk r|| dd | j jD dd |j jD fddD }fdd D }|rn| j|dd} fd dD }|r|j|dd}fd
|
d D } | r´| j| d } fd dD }
|
|
|
rÖ|j|
|
d }| j||d } |rö| |¡} | S )a¿
|
Compute the expected merge result for the test case.
|
|
This method computes the expected result of merging two DataFrames on
|
a combination of their columns and index levels. It does so by
|
explicitly dropping/resetting their named index levels, performing a
|
merge on their columns, and then finally restoring the appropriate
|
index in the result.
|
|
Parameters
|
----------
|
df_left : DataFrame
|
The left DataFrame (may have zero or more named index levels)
|
df_right : DataFrame
|
The right DataFrame (may have zero or more named index levels)
|
on : list of str
|
The on parameter to the merge operation
|
left_on : list of str
|
The left_on parameter to the merge operation
|
right_on : list of str
|
The right_on parameter to the merge operation
|
how : str
|
The how parameter to the merge operation
|
|
Returns
|
-------
|
DataFrame
|
The expected merge result
|
Nc S s g | ]}|d k r|qS ©Nr ©Ú.0Únr r r Ú
|
<listcomp>Z s z$compute_expected.<locals>.<listcomp>c S s g | ]}|d k r|qS r r r r r r r [ s c s g | ]}|kr| kr|qS r r )r Úi)Ú left_levelsÚ right_levelsr r r ^ s c s g | ]}| kr|qS r r r ©Úleft_onr r r a s T)Zdropc s g | ]}| kr|qS r r r ©Úright_onr r r e s c s g | ]}| kr|qS r r r r" r r r j s )Úlevelc s g | ]}| kr|qS r r r r$ r r r n s ©r# r% Úhow)ÚindexÚnamesÚ reset_indexÚmerger ) Zdf_leftZdf_rightÚonr# r% r( Z output_levelsZ drop_leftZ
|
drop_rightZ
|
reset_leftZ reset_rightÚexpectedr )r r# r! r% r Úcompute_expected7 s*
|
|
r/ zon,howÚleftÚrightc C s4 t | |||d}| j|||d}tj||dd d S )N)r- r( T©Z
|
check_like©r/ r, ÚtmÚassert_frame_equal)r r r- r( r. Úresultr r r Ú!test_merge_indexes_and_columns_on| s r7 zleft_on,right_on,howc C s8 t | ||||d}| j||||d}tj||dd d S )Nr' Tr2 r3 )r r r# r% r( r. r6 r r r Ú-test_merge_indexes_and_columns_lefton_righton s ÿr8 Ú
|
left_indexc C sf | |¡}| ddg¡}| ¡ j|ddg|ddd |¡}|j|ddg|ddd}tj||dd d S )Nr r Ú_xZ_y)r- r( ZlsuffixZrsuffixTr2 )r r+ Újoinr4 r5 )r r r9 Z join_typer r r. r6 r r r Ú test_join_indexes_and_columns_on¤ s(
|
þüÿ ÿr< )NNNN)Únumpyr ZpytestZpandasr Zpandas._testingZ_testingr4 Zfixturer r r r r/ ÚmarkZ parametrizer7 r8 r< r r r r Ú<module> s@
|
|
|
|
|
|
|
E
|
|
üþ
|
üþ
|
|
