U
|
®ý°dN2 ã
|
@ sê d dl Z d dlZd dlmZ d dlmZ e jddgddd Ze jdd gdd
|
d Z d d Z
|
e j dddg¡dd Z e j deddddgiedddddgifedddgdddgdeddddgifg¡dd Ze j d ejddgd d gd d gd!ejd"d#gd d gd d gd!d$fejddgd%d&gd d gd'd d(gd!ejd"d#gd%d&gd d gd'd d(gd!d)fg¡d*d+ Zd,d- Ze j dddg¡d.d/ Ze j dddg¡d0d1 Zd2d3 Ze j d ed4d5d6gd7d8d9gd:ed4d5d6gd7d8d;gd:d<fed4d5d6gd7d8d9gd:ed d d gd=d=d=gd:d>fg¡d?d@ ZdAdB ZdCdD Ze j dEdFdGg¡dHdI Ze j dJdKdLdMg¡dNdO ZdPdQ ZdRdS ZdTdU Z e j dVdWdXg¡dYdZ Z d[d\ Z d]d^ Z d_d` Z dadb Z!dcdd Z"dedf Z#dS )gé N)Ú DataFrameTF)Úparamsc C s | j S ©N©Úparam©Úrequest© r ú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/util/test_assert_frame_equal.pyÚby_blocks_fixture s r r ÚSeriesc C s | j S r r r r r r
|
Ú obj_fixture s r c K s$ t j| |f| t j|| f| dS )a4
|
Check that two DataFrame equal.
|
|
This check is performed commutatively.
|
|
Parameters
|
----------
|
a : DataFrame
|
The first DataFrame to compare.
|
b : DataFrame
|
The second DataFrame to compare.
|
kwargs : dict
|
The arguments passed to `tm.assert_frame_equal`.
|
N)ÚtmÚassert_frame_equal)ÚaÚbÚkwargsr r r
|
Ú_assert_frame_equal_both s r Ú
|
check_likec C s t dddgdddgddd d
|
gd }t dddgdddgdd
|
d dgd }| s| d }tjt|d tj||| |d W 5 Q R X nt||| |d d S )Né é é é é é ©ÚAÚBr r Úc©Úindexz.index are different©Úmatch©r Úobj)r ÚpytestÚraisesÚAssertionErrorr r r )r r Údf1Údf2Úmsgr r r
|
Ú#test_frame_equal_row_order_mismatch% s $$
|
r+ zdf1,df2r r r r r r r r c C s8 | d}t jt|d tj| ||d W 5 Q R X d S )Nz are differentr! ©r$ ©r% r& r' r r )r( r) r r* r r r
|
Ú test_frame_equal_shape_mismatch2 s
|
r. z df1,df2,msgÚl1Úl2)r r r r g ð?g @z DataFrame\.index are differentgÍÌÌÌÌÌ @g ø?)r r r r z$MultiIndex level \[0\] are differentc C sL d|i}|r8t jt|d tj| |f| W 5 Q R X ntj| |f| d S )NÚcheck_index_typer! r- )r( r) r* r1 r r r r
|
Ú%test_frame_equal_index_dtype_mismatch@ s
|
r2 c C s~ ddg}t |d}t |d}d| i}|d d¡|d< | rjd}tjt|d tj||f| W 5 Q R X ntj||f| d S )NZcol1Zcol2)ÚcolumnsÚ check_dtypeÚint64z)Attributes of DataFrame\..* are differentr! )r Úastyper% r& r' r r )r4 r3 r( r) r r* r r r
|
Útest_empty_dtypes_ s
|
|
r7 c C s | d| d}t dddgdddgd d
|
d d gd }t dddgdddgd d
|
d dgd }tjt|d tj||| |d W 5 Q R X d S )Nz\.index are different
|
|
zÂ\.index values are different \(33\.33333 %\)
|
\[left\]: Index\(\['a', 'b', 'c'\], dtype='object'\)
|
\[right\]: Index\(\['a', 'b', 'd'\], dtype='object'\)
|
At positional index 2, first diff: c != dr r r r r r r r r r r Údr! r# ©r r% r& r' r r ©r r r* r( r) r r r
|
Ú test_frame_equal_index_mismatcho s
|
$$r; c C s | d| d}t dddgdddgd d
|
d d gd }t dddgdddgdd
|
d d gd }tjt|d tj||| |d W 5 Q R X d S )Nz\.columns are different
|
|
z\.columns values are different \(50\.0 %\)
|
\[left\]: Index\(\['A', 'B'\], dtype='object'\)
|
\[right\]: Index\(\['A', 'b'\], dtype='object'\)r r r r r r r r r r r )r r r! r# r9 r: r r r
|
Ú!test_frame_equal_columns_mismatch s
|
$$r< c C sx |}| d| d}t dddgdddgd }t dddgddd
|
gd }tjt|d tj||| |d W 5 Q R X d S ) Nz2\.iloc\[:, 1\] \(column name="B"\) are different
|
|
z\.iloc\[:, 1\] \(column name="B"\) values are different \(33\.33333 %\)
|
\[index\]: \[0, 1, 2\]
|
\[left\]: \[4, 5, 6\]
|
\[right\]: \[4, 5, 7\]r r r r r r r é r! ©Z by_blocksr$ r9 )r r r$ r* r( r) r r r
|
Ú test_frame_equal_block_mismatch s r? õ áõ à õ äõ éõ èõ ë)r ÚEu eÌuÏ {obj}\.iloc\[:, 1\] \(column name="E"\) are different
|
|
{obj}\.iloc\[:, 1\] \(column name="E"\) values are different \(33\.33333 %\)
|
\[index\]: \[0, 1, 2\]
|
\[left\]: \[é, è, ë\]
|
\[right\]: \[é, è, eÌ\]ÚeuÈ {obj}\.iloc\[:, 0\] \(column name="A"\) are different
|
|
{obj}\.iloc\[:, 0\] \(column name="A"\) values are different \(100\.0 %\)
|
\[index\]: \[0, 1, 2\]
|
\[left\]: \[á, à , ä\]
|
\[right\]: \[a, a, a\]c C s< |j |d}tjt|d tj| |||d W 5 Q R X d S )Nr, r! r> )Úformatr% r& r' r r )r( r) r* r r r r r
|
Útest_frame_equal_unicode s rI c C sb t ddddgidd} | t¡}d}tj| |dd tjt|d
|
tj| |d d W 5 Q R X d S ) Nr r r r ÚInt64©ÚdtypezAttributes of DataFrame\.iloc\[:, 0\] \(column name="a"\) are different
|
|
Attribute "dtype" are different
|
\[left\]: Int64
|
\[right\]: int[32|64]F©r4 r! T)r r6 Úintr r r% r& r' ©ÚleftÚrightr* r r r
|
Ú0test_assert_frame_equal_extension_dtype_mismatchÁ s
|
ÿrR c C sf t dt dd¡gidd} | t¡}d}tj| |dd tjt |d tj| |d
|
d W 5 Q R X d S ) Nr r r ÚintervalrK zAttributes of DataFrame\.iloc\[:, 0\] \(column name="a"\) are different
|
|
Attribute "dtype" are different
|
\[left\]: interval\[int64, right\]
|
\[right\]: objectFrM r! T)
|
r ÚpdZIntervalr6 Úobjectr r r% r& r' rO r r r
|
Ú/test_assert_frame_equal_interval_dtype_mismatchÔ s
|
ÿrV Ú right_dtypeZInt32r5 c C s@ t ddddgidd}t ddddgi| d}tj||dd d S ) Nr r r r rJ rK FrM ©r r r )rW rP rQ r r r
|
Ú7test_assert_frame_equal_ignore_extension_dtype_mismatchç s rY rL ztimedelta64[ns]zdatetime64[ns, UTC]z Period[D]c C s0 t dg i| d}t dg i}tj||dd d S )Nr rK FrM rX )rL r( r) r r r
|
Ú4test_assert_frame_equal_datetime_like_dtype_mismatchï s rZ c C s t } t jdd}t | | ¡ t ||¡ tj| |dd tj|| dd tjtdd t | |¡ W 5 Q R X tjtdd t | |¡ W 5 Q R X d S )NF)Úallows_duplicate_labels©Z check_flagsz<Flagsr! )r Z set_flagsr r r% r& r' ©rP rQ r r r
|
Ú test_allows_duplicate_labelsý s r^ c C s6 t dddggdddgdddgd} tj| | d d
|
d S ) Nr r r ZfooÚbaré* Útest)r3 r T©r rX )Zdfr r r
|
Ú+test_assert_frame_equal_columns_mixed_dtype s "rc c C s$ | ddg|d}t j||dd d S )Nr r rK T©Ú check_exact)r Ú assert_equal)Úframe_or_seriesÚany_numeric_ea_dtyper$ r r r
|
Ú test_frame_equal_extension_dtype s ri Úindexer)r r )r r c C sl |df}| ddg||d d}| ddg||d d}d}t jt|d tj||dd d
|
W 5 Q R X d S ) Nr5 r r r rK zI(Series|DataFrame.iloc\[:, 0\] \(column name="0"\) classes) are differentr! TF)re r4 )r% r& r' r rf )rg rh rj ZdtypesZobj1Úobj2r* r r r
|
Ú test_frame_equal_mixed_dtypes s rl c C sX t tjg ddd} t tjddddd}tjtdd tj| |d d
|
W 5 Q R X d S ) NrU rK r r r )ÚstartÚstopÚstepz DataFrame.index are differentr! Trb ) r rT ZIndexZ
|
RangeIndexr% r& r' r r )r( r) r r r
|
Ú4test_assert_frame_equal_check_like_different_indexes" s rp c C sv t ddgddgg} d| j_t ddgddgg}d|j_tj| |dd tjtdd tj| |dd W 5 Q R X d S )
|
Nr r r r FTr\ r[ r! )r Úflagsr[ r r r% r& r' r] r r r
|
Ú0test_assert_frame_equal_checking_allow_dups_flag* s rr c
|
C s t dgdgdggtj t dddg¡t dddg¡g¡d} t dgdgdggtj t dddg¡t dddg¡g¡d}tj| |dd d S )
|
Nr r r r r r r Trb )r rT Z
|
MultiIndexZ from_arraysZ Categoricalr r r] r r r
|
Ú3test_assert_frame_equal_check_like_categorical_midx7 s" þÿþ þÿþ rs c C sd t dtjtjdgddi} t dtjddgddi}d}tjt|d t | |¡ W 5 Q R X d S )Nr r rJ rK zGDataFrame.iloc\[:, 0\] \(column name="a"\) NA mask values are differentr! © r rT r ZNAr% r& r' r r ©r( r) r* r r r
|
Ú>test_assert_frame_equal_ea_column_definition_in_exception_maskN s
|
rv c C s t dtjtjdgddi} t dtjtjdgddi}d}tjt|d t | |¡ W 5 Q R X tjt|d tj| |dd W 5 Q R X d S )
|
Nr r rJ rK r ú?DataFrame.iloc\[:, 0\] \(column name="a"\) values are differentr! Trd rt ru r r r
|
Ú9test_assert_frame_equal_ea_column_definition_in_exceptionX s rx c C sf t dt d¡t d¡gi} t dt d¡t d¡gi}d}tjt|d t | |¡ W 5 Q R X d S )Nr z
|
2019-12-31z
|
2020-12-31rw r! )r rT Ú Timestampr% r& r' r r ru r r r
|
Ú!test_assert_frame_equal_ts_columne s
|
rz )$r% ZpandasrT r Zpandas._testingZ_testingr Zfixturer r r ÚmarkZ parametrizer+ r. Z from_recordsr2 r7 r; r< r? rI rR rV rY rZ r^ rc ri rl rp rr rs rv rx rz r r r r
|
Ú<module> s
|
|
|
$*þþ
|
ý ÿ ÿùøþ
|
|
|
|
ý ýõþ
|
|
|
ýþ
|
|
|
|
