U
|
®ý°dñ ã @ sb d dl Zd dlZd dlZd dlmZmZ d dlmZ ej
|
dd Z ej
|
dd Z G dd dZ dS ) é N)Ú DataFrameÚ merge_orderedc C s t dddgdddgdS )NÚaÚcÚeé g @é )ÚkeyÚlvalue©r © r r údd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/reshape/merge/test_merge_ordered.pyÚleft s r c C s t ddddgddddgd S )
|
NÚbr ÚdÚfr é ç @é ©r Úrvaluer r r r r Úright s r c @ sæ e Zd Zdd Zdd Zdd Zdd Zej d d
|
g d fi d fd gd fd d gd fg¡dd Z
|
ej de gd e ge d gg¡dd Z dd Z ej de ddgddgddgde dgdgd d gd d gd e dgd dgd dddgejd ejgd!fe ddgddgddgde dgdgd d d d gd e dgd dgd dddgejd ejgd!fe dgdgd e ddgddgddgdd gd d d ge dddgejd ejgdgd dgd d"fg¡d#d$ Zd%d& Zd'd( Zd S ))ÚTestMergeOrderedc C s\ t ||dd}tddddddgd tjd
|
tjd tjgtjd d
|
d tjd gd }t ||¡ d S )Nr ©Úonr r r r r r r r r r ©r r
|
r ©r r ÚnpÚnanÚtmÚassert_frame_equal©Úselfr r ÚresultÚexpectedr r r Ú
|
test_basic s ýÿzTestMergeOrdered.test_basicc C sV t ||ddd}tdddddd gd
|
d d d d dgtjd d d d dgd}t ||¡ d S )Nr Úffill)r Ú fill_methodr r r r r r ç ð?r r r r r r r r! r r r Ú
|
test_ffill# s ýÿzTestMergeOrdered.test_ffillc C s
|
t j||gdd}dgd dgd |d< t||dddd }tddd
|
d d d gd ddddddgd tjdddddgd d}dgd dgd |d< t ||jd d
|j f ¡ t||dddd}t ||jd d
|j f ¡ t||ddd}|d
|
¡ ¡ st d S )NT)Z ignore_indexr r r Úgroupr r& )r Úleft_byr' r r r r r r( r r r r é )r Úright_byr' ©r r+ ) ÚpdÚconcatr r r r r r ÚlocÚcolumnsZnotnaÚallÚAssertionError)r" r r r# r$ Zresult2r r r Útest_multigroup. s6 ÿýÿ ÿ z TestMergeOrdered.test_multigroupc s< G fdddt |}|j|dd}t| s8td S )Nc s e Zd Ze fddZdS )z7TestMergeOrdered.test_merge_type.<locals>.NotADataFramec s S ©Nr )r" ©Ú NotADataFramer r Ú _constructorK s zDTestMergeOrdered.test_merge_type.<locals>.NotADataFrame._constructorN)Ú__name__Ú
|
__module__Ú __qualname__Úpropertyr9 r r7 r r r8 J s r8 r r )r ÚmergeÚ
|
isinstancer4 )r" r r Znadr# r r7 r Útest_merge_typeI s z TestMergeOrdered.test_merge_typezdf_seq, pattern)r ú [Nn]o objectsrA Nz objects.*Nonec C s( t jt|d t |¡ W 5 Q R X d S )N©Úmatch)ÚpytestÚraisesÚ
|
ValueErrorr/ r0 )r" Zdf_seqÚpatternr r r Útest_empty_sequence_concatT s z+TestMergeOrdered.test_empty_sequence_concatÚargc C s t |¡ d S r6 )r/ r0 )r" rI r r r Ú test_empty_sequence_concat_okc s z.TestMergeOrdered.test_empty_sequence_concat_okc C s¦ t tdddddddgdddgd d}t d dd
|
gdddgd }t||d d d}t tddd dd
|
dgd dddddgd tjddddgd d}t ||¡ d S )NZaaabbbr r r r r r )r* r r
|
r r r r& r* )r' r+ Z
|
aaaaabbbbb)r* r r
|
r ©r Úlistr r r r r r! r r r Útest_doc_examplei s ýÿüÿ z!TestMergeOrdered.test_doc_examplez,left, right, on, left_by, right_by, expectedÚgÚhr r )ÚGÚHÚTr )rR ÚErR rP rQ r( )rP rQ rR rS )rR rS rP rQ c C s" t |||||d}t ||¡ d S )N)r r r r+ r- )r r r )r" r r r r+ r- r$ r# r r r Útest_list_type_by s 4ûz"TestMergeOrdered.test_list_type_byc C s t dddgdddggtdd}t ddggtdd}t||d d
|
d gd }t dgd dgd dddgtjd tjgd}t ||¡ d S )NrN rO r r ÚGHE©r2 r ÚETrS rP rQ r. r( )rP rQ rS rR rK r! r r r Ú*test_left_by_length_equals_to_right_shape0¿ s (ÿz;TestMergeOrdered.test_left_by_length_equals_to_right_shape0c C sl t dddgdddggtdd}t ddggtdd}d }tjt|d
|
t||d d dgd W 5 Q R X d S )NrN rO r r rU rV r rW z!\{'h'\} not found in left columnsrB rS rP r. )r rL rD rE ÚKeyErrorr )r" r r Úmsgr r r Ú!test_elements_not_in_by_but_in_dfÊ s
|
z2TestMergeOrdered.test_elements_not_in_by_but_in_df)r: r; r< r% r) r5 r@ rD ÚmarkZ parametrizerH r rJ rM r r rT rX r[ r r r r r s
|
ûþ
|
|
ÿ
|
üÿú üÿú üÿúáþ2
|
r )Únumpyr rD Zpandasr/ r r Zpandas._testingZ_testingr Zfixturer r r r r r r Ú<module> s
|
|
|
