U
|
ý°d& ã
|
@ sH d dl Zd dlZd dlmZ d dlZd dlmZ ej
|
dd dddg¡dd Z ej
|
d d
|
d d g¡d d Z dd Zdd Zej
|
dd dg¡dd Zdd Zdd Zej
|
dddgddd d d d g¡d d Zej
|
d!d"ejfejejfejd"fg¡d#d$ Zej
|
d%d&d"d"ejejfd"ejd"ejfejejejejfg¡d'd( ZdS ))é N)Ú is_numpy_devÚ
|
align_axisé ÚindexÚcolumnsc C s t jdddgddtjgdddgddd d
|
gd }| ¡ }d|jd < d |jd< |j|| d}| dkr¶t ddg¡}t j dd
|
gddgg¡}t jddtjtjgtjtjdd gg||d}nVt j ddgddgg¡}t dd
|
g¡}t jdtjgdtjgtjdgtjd gg||d}t
|
||¡ d S )NÚaÚbÚcç ð?ç @ç @©Úcol1Úcol2Úcol3r r r ©r ©r r ç @©é r ©r )r r r r ÚselfÚother©r r © ÚpdÚ DataFrameÚnpÚnanÚcopyÚlocÚcompareZIndexÚ
|
MultiIndexZ from_productÚtmÚassert_frame_equal)r ÚdfÚdf2ÚresultÚindicesr Úexpected© r* ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_compare.pyÚtest_compare_axis
|
s0 þ
|
|
ý"ýr, zkeep_shape, keep_equal)TF)FT)TTc C s t jdddgddtjgdddgddd d
|
gd }| ¡ }d|jd < d |jd< |j|| |d}| r(t dddg¡}t j dd d
|
gddgg¡}|rÐt jddddddgddddddgddtjtjdd gg||d}nVt jddtjtjtjtjgtjtjtjtjtjtjgtjtjtjtjdd gg||d}nJt ddg¡}t j dd
|
gddgg¡}t jddddgdddd gg||d}t
|
||¡ d S )Nr r r r
|
r r r r r r r r r r )Ú
|
keep_shapeÚ
|
keep_equalr r r r r r r )r- r. r% r& r' r( r r) r* r* r+ Ú test_compare_various_formats* sL
|
þ
|
|
ÿýù
|
ýù
|
ÿr/ c C s t jdddgddtjgdddgddd d
|
gd } | ¡ }d|jd < | |¡}t d g¡}t j dgddgg¡}t jddgg||d}t
|
||¡ d S )Nr r r r
|
r r r r r r r r r r r r r ©r% r& r' r( r r) r* r* r+ Ú test_compare_with_equal_nulls` s þ
|
|
r1 c C s¼ t jdddgddtjgdddgddd d
|
gd } | ¡ }d|jd < tj|jd < | |¡}t ddg¡}t j dd
|
gddgg¡}t jddtjtjgtjtjdtjgg||d}t
|
||¡ d S )Nr r r r
|
r r r r r r r r r r r r r r r r0 r* r* r+ Ú!test_compare_with_non_equal_nullsq s þ
|
|
ýr2 c C s~ t dddgddtjgdddgd¡}t j dddgdd d
|
gg¡|_t j d d d gd ddgg¡|_| ¡ }d|j d< d|j d< |j
|
|| d}| d krüt j d d d d gd d ddgddddgg¡}t j ddgdd
|
gg¡}dtjgdtjgtjdgtjdgg}nbt j d d gd dgg¡}t j ddddgddd
|
d
|
gddddgg¡}ddtjtjgtjtjddgg}t j|||d}t ||¡ d S )Nr r r r
|
r r r r r r ÚxÚyr r r )r r r )r r r r r )Údatar r ) r r r r r" Z from_arraysr r r Zilocr! r# r$ )r r% r& r' r( r r5 r) r* r* r+ Útest_compare_multi_index s2 ÿ
|
|
ÿ&
|
|
|
ýÿ r6 c C s¨ d} t jt| dD tjdddgdddgd }tjdddgddd
|
gd }| |¡ W 5 Q R X d} t jt| d0 t t d ¡¡}t t d ¡¡}| |¡ W 5 Q R X d S ) NzQCan only compare identically-labeled \(both index and columns\) DataFrame objects©Úmatchr r é r r r )r Úd)r9 r9 )r r ) ÚpytestÚraisesÚ
|
ValueErrorr r r! r ZonesÚzeros)ÚmsgÚdf1r& r* r* r+ Ú test_compare_unaligned_objects« s ÿÿrA c C s¢ t dddgddtjgdddgd¡} t dddgddtjgddtjgd¡}| j|dd }t dtjd
|
dtjd
|
tjdd
|
tjtjd
|
d ¡}t ||¡ d S ) Nr r r r
|
r r r )ÚleftÚright©Ú result_names)r r ))r rB )r rC )r rB )r rC )r r r r r! r# r$ )r@ r& r' r) r* r* r+ Útest_compare_result_namesÁ s$ ÿ
|
|
ýÿ
|
|
|
üÿrF rE r ZHKr9 )Ú2Ú3r c C s t dddgddtjgdddgd¡}t dddgddtjgddtjgd¡}tjtdt| d d
|
|j|| d W 5 Q R X d S ) Nr r r r
|
r r r z Passing 'result_names' as a z= is not supported. Provide 'result_names' as a tuple instead.r7 rD ) r r r r r; r< Ú TypeErrorÚtyper! )rE r@ r& r* r* r+ Ú test_invalid_input_result_namesÙ s ÿ
|
|
ýÿýrK z val1,val2é c C s° d| g}t jd|gdd}t |ddgd¡}t |ddgd¡}t ||tjtjd¡}| t jkrtrtjt d d
|
|j
|
|d d }W 5 Q R X n|j
|
|d d }t ||¡ d S ) Nr r ÚInt64©Zdtyper
|
r ©r r ©)r r )r r )r r )r r z boolean value of NA is ambiguousr7 T©r- ) r ÚSeriesr r r ÚNAr r; r< rI r! r# r$ )Zval1Zval2ZarrZserr@ r& r) r' r* r* r+ Ú test_compare_ea_and_np_dtypeù s üÿrT z$df1_val,df2_val,diff_self,diff_other)rL r9 rL r9 c C s t t j| t jgddddgd¡}| ¡ }||jd< t t j|t jgddt j|t jgddtjtjd¡}|j|dd }t
|
||¡ d S )
|
NrM rN r
|
r rO )r r rP TrQ ) r r rR rS r r r r r! r# r$ )Zdf1_valZdf2_valZ diff_selfZ
|
diff_otherr@ r& r) r' r* r* r+ Ú!test_compare_nullable_int64_dtype s $
|
üÿrU )Únumpyr r; Z pandas.compatr Zpandasr Zpandas._testingZ_testingr# ÚmarkZ parametrizer, r/ r1 r2 r6 rA rF rK rS rT rU r* r* r* r+ Ú<module> sX
|
ýþ
|
-
|
"ûþ
|
|
þ
|
üþ
|
