U
|
®ý°dÁ% ã @ sh d dl Zd dlZd dlmZmZmZmZmZm Z m
|
Z
|
d dl m Z dd Zdd Zdd Zej d d
|
d g¡d d Zdd Zdd Zdd Zdd Zdd Zej ddd
|
g¡dd Zd d Zej d d d ejejfe e feje fg¡d!d" Z d#d$ Z ej d d
|
dg¡d%d& Z d'd( Z d)d* Z d+d, Z!d-d. Z"d/d0 Z#ej d1dd
|
g¡ej d2dd
|
g¡d3d4 Z$dS )5é N)ÚNAÚ CategoricalÚCategoricalIndexÚIndexÚ
|
MultiIndexÚNaTÚ
|
RangeIndexc C sR d} t dddg}t ddddg¡}tjt| d tj||d
|
d W 5 Q R X d S ) NzáIndex are different
|
|
Index levels are different
|
\[left\]: 1, Index\(\[1, 2, 3\], dtype='int64'\)
|
\[right\]: 2, MultiIndex\(\[\('A', 1\),
|
\('A', 2\),
|
\('B', 3\),
|
\('B', 4\)\],
|
\)é é é ©ÚAr ©r r
|
©ÚBr ©r é ©ÚmatchF©Úexact)r r Ú from_tuplesÚpytestÚraisesÚAssertionErrorÚtmÚassert_index_equal©ÚmsgÚidx1Úidx2© r! ú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/util/test_assert_index_equal.pyÚ test_index_equal_levels_mismatch s
|
|
r# c C sV d}t ddddg¡}t ddddg¡}tjt|d tj||| d W 5 Q R X d S )NúÁMultiIndex level \[1\] are different
|
|
MultiIndex level \[1\] values are different \(25\.0 %\)
|
\[left\]: Index\(\[2, 2, 3, 4\], dtype='int64'\)
|
\[right\]: Index\(\[1, 2, 3, 4\], dtype='int64'\)r r r r r ©Ú check_exact©r r r r r r r ©r& r r r r! r! r" Ú test_index_equal_values_mismatch" s
|
r) c C sP d}t dddg}t ddddg}tjt|d tj||| d W 5 Q R X d S )NzIndex are different
|
|
Index length are different
|
\[left\]: 3, Index\(\[1, 2, 3\], dtype='int64'\)
|
\[right\]: 4, Index\(\[1, 2, 3, 4\], dtype='int64'\)r r
|
r r r r% ©r r r r r r r( r! r! r" Ú test_index_equal_length_mismatch0 s
|
r+ r FÚequivc C s* t dddg}td}tj||| d d S )Nr r r
|
r r )r r r r )r r r r! r! r" Útest_index_equal_class> s r- c C sV d}t dddg}t dddgtjd}tjt|d tj||d| d W 5 Q R X d S ) NzcIndex are different
|
|
Attribute "inferred_type" are different
|
\[left\]: integer
|
\[right\]: floatingr r
|
r ©Zdtyper T©r r& )r ÚnpÚfloat64r r r r r r( r! r! r" Ú)test_int_float_index_equal_class_mismatchF s
|
r2 c C sN d}t dddg}ttd}tjt|d tj||d| d W 5 Q R X d S )NzkIndex are different
|
|
Index classes are different
|
\[left\]: Index\(\[1, 2, 3\], dtype='int64'\)
|
\[right\]: r r
|
r r Tr/ )r r Úranger r r r r r( r! r! r" Ú%test_range_index_equal_class_mismatchT s
|
r4 c C sd t dddg}t dddg}| rPd}tjt|d tj||| d W 5 Q R X ntj||| d d S )Nr r
|
ç @go @zµIndex are different
|
|
Index values are different \(33\.33333 %\)
|
\[left\]: Index\(\[1.0, 2.0, 3.0], dtype='float64'\)
|
\[right\]: Index\(\[1.0, 2.0, 3.0000000001\], dtype='float64'\)r r% r* )r& r r r r! r! r" Ú test_index_equal_values_closeb s r6 c C sv t dddg}t dddg}| |d}| s2|dk rbd}tjt|d tj||f| W 5 Q R X ntj||f| d S ) Nr r
|
r5 g9ÖÅm4 @©r& Úrtolgü©ñÒMb@?z¯Index are different
|
|
Index values are different \(33\.33333 %\)
|
\[left\]: Index\(\[1.0, 2.0, 3.0], dtype='float64'\)
|
\[right\]: Index\(\[1.0, 2.0, 3.0001\], dtype='float64'\)r r* ©r& r8 r r Úkwargsr r! r! r" Ú"test_index_equal_values_less_closes s
|
r; c C sX t dddg}t dddg}| |d}d}tjt|d tj||f| W 5 Q R X d S )Nr r
|
r r r7 zIndex are different
|
|
Index values are different \(33\.33333 %\)
|
\[left\]: Index\(\[1, 2, 3\], dtype='int64'\)
|
\[right\]: Index\(\[1, 2, 4\], dtype='int64'\)r r* r9 r! r! r" Ú test_index_equal_values_too_far
s
|
r< Ú check_orderTc C sl t dddg}t dddg}d}|rTtjt|d tj||| |dd W 5 Q R X ntj||| |dd d S ) Nr r
|
r zIndex are different
|
|
Index values are different \(66\.66667 %\)
|
\[left\]: Index\(\[1, 2, 3\], dtype='int64'\)
|
\[right\]: Index\(\[3, 2, 1\], dtype='int64'\)r T)r& r8 r= Fr* )r& r8 r= r r r r! r! r" Ú$test_index_equal_value_oder_mismatch s& ÿ ÿr> c C s` t ddddg¡}t ddddg¡}| |d}d}tjt|d tj||f| W 5 Q R X d S )Nr r r r r7 r$ r r' r9 r! r! r" Ú&test_index_equal_level_values_mismatchª s
|
r? z name1,name2)NÚx)r@ r@ c C s t dddg| d}t dddg|d}| |ks4| |krBt ||¡ nX| dkrNdn| } |dkr^dn|}d| d| d }tjt|d
|
t ||¡ W 5 Q R X d S ) Nr r
|
r ©Únamer@ z'x'zBIndex are different
|
|
Attribute "names" are different
|
\[left\]: \[z\]
|
\[right\]: \[z\]r )r r r r r r )Zname1Zname2r r r r! r! r" Útest_index_equal_names¹ s ýürC c C sr d}t tddg}t tddgdddgd}| r^tjt|d tj||| d W 5 Q R X ntj||| d d S )NzÅIndex are different
|
|
Attribute "dtype" are different
|
\[left\]: CategoricalDtype\(categories=\['a', 'b'\], ordered=False\)
|
\[right\]: CategoricalDtype\(categories=\['a', 'b', 'c'\], ordered=False\)ÚaÚbÚc)Ú
|
categoriesr )Úcheck_categorical)r r r r r r r )rH r r r r! r! r" Ú"test_index_equal_category_mismatchÐ s rI c C sl d}t td}t ttd}| rV|rVtjt|d tj||ddd W 5 Q R X ntj||| |d d S )Nz¦Index are different
|
|
Index classes are different
|
\[left\]: RangeIndex\(start=0, stop=10, step=1\)
|
\[right\]: Index\(\[0, 1, 2, 3, 4, 5, 6, 7, 8, 9\], dtype='int64'\)é
|
r T)rH r ) r r Úlistr3 r r r r r )rH r r ZrcatZicatr! r! r" Ú!test_index_equal_range_categoriesâ s ÿrL c C sL d} t tt d¡g}t ttg}tjt| d t ||¡ W 5 Q R X d S )NzaIndex are different
|
|
Attribute "inferred_type" are different
|
\[left\]: mixed
|
\[right\]: datetimeZnatr )
|
r r r0 Z
|
datetime64r r r r r r r r! r! r" Ú0test_assert_index_equal_different_inferred_typesø s
|
rM c C sP t ddgdd} t ddgdd}tjtdd tj| |dd d
|
W 5 Q R X d S ) Nr r rD rA rE z"names" are differentr FT)r= Ú check_namesr* ©r r r! r! r" Ú9test_assert_index_equal_different_names_check_order_false s rP c C s" t dddg} tj| | dd d S )NZfooÚbaré* F©r= ©r r r )Úidxr! r! r" Ú#test_assert_index_equal_mixed_dtype s rV c C s4 t ddg| d}t ddg| d}tj||dd d S )Nr r r. FrS rT )Zany_numeric_ea_dtyper r r! r! r" Ú,test_assert_index_equal_ea_dtype_order_false s rW c C s4 t ddgdd} t ddgdd}tj| |dd d S )Nr r Úobjectr. FrS rT rO r! r! r" Ú/test_assert_index_equal_object_ints_order_false s rY rH rN c C sP t ddgdd}t dtgdd}tjtdd tj||| |d W 5 Q R X d S )Nr r
|
ZInt64r. z50.0 %r )rN rH )r r r r r r r )rN rH r r r! r! r" Ú*test_assert_ea_index_equal_non_matching_na# s ÿrZ )%Únumpyr0 r Zpandasr r r r r r r Zpandas._testingZ_testingr r# r) r+ ÚmarkZ parametrizer- r2 r4 r6 r; r< r> r? ÚnanrC rI rL rM rP rV rW rY rZ r! r! r! r" Ú<module> s@ $
|
|
þ
|
|
|
