U
|
ý°dn" ã @ s¸ d dl Zd dlZd dlmZ d dlZd dlmZmZm Z d dl
|
m Z dd Z dd Zdd Zd
|
d Zd d Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zd d Zd d ZdS ) é N)Úis_any_real_numeric_dtype)ÚIndexÚ
|
MultiIndexÚSeriesc C sÌ | | ¡st| | ¡ ¡s t| | t¡¡s4t| | ¡ ¡sFt| | ¡ d¡¡s^t| t| ¡rpt| t | ¡¡rtt | td}| |¡st| | ¡s¬t| j
|
dkrÈ| t | ¡rÈtd S )NÚcategory©Zdtypeé ) ÚequalsÚAssertionErrorÚcopyZastypeÚobjectZ to_flat_indexÚlistÚnpÚarrayr Znlevelsr )ÚidxZ same_values© r úbd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/multi/test_equivalence.pyÚ test_equals s
|
r c C s¦ | }t |}|dd
}|dd
|dd
¡}|dd
}tjtdd ||k W 5 Q R X t dg| ¡}t dg|d dg ¡}t ||k|¡ t ||k|¡ t |¡}t |dd
¡} t |dd
|dd
¡¡}
|
t |dd
¡} tjtdd || k W 5 Q R X t ||k|¡ t ||
|
k|¡ t |} t | } t |
|
}t | }tjtdd || k W 5 Q R X t || k|¡ t ||k|¡ tjtdd ||k W 5 Q R X tjtdd ||k W 5 Q R X tjtdd || k W 5 Q R X d }tjt|d | |k W 5 Q R X tjtdd | | k W 5 Q R X t
|
|t s¢t dgt |d
|
ddg ¡}|d }t ||k|¡ t | |kt |¡ d S ) Nr éÿÿÿÿéþÿÿÿr zLengths must match)ÚmatchTFz3Can only compare identically-labeled Series objectsé ) ÚlenÚappendÚpytestZraisesÚ
|
ValueErrorr r ÚtmÚassert_numpy_array_equalr Ú
|
isinstancer Úassert_series_equal)r Zindex_aÚnZindex_bZindex_cZindex_dZ expected1Z expected2Zarray_aZarray_bZarray_cZarray_dZseries_aZseries_bZseries_cZseries_dÚmsgZ expected3Úitemr r r Útest_equals_op" sV
|
r# c C sÒ t ddggd ¡} t ddddg¡}| | d k}t ddddg¡}t ||¡ | | d k}t || ¡ | | d k }t ||¡ | | d k}t ||¡ | | d k}t || ¡ | | d k}t || ¡ d S )Nr r Fr T)r Ú from_productr r r r )ÚmiZ all_falseÚresultÚexpectedr r r Útest_compare_tuple_ s r( c C s\ t dddg¡} | dk}t dddg¡}t ||¡ | dk}t dddg¡}t ||¡ d S )N)ÚaÚb)r* Úc)r+ r) FT)r+ )r Ú from_tuplesr r r r )r% r& r' r r r Útest_compare_tuple_strsy s r- c C s | | ¡st| | j¡r t| t| j¡s2t| | ¡s@t| | d d
¡rVt| | d ¡rhtttttdtttdtttdgt ddddddddg¡t ddddddddg¡t ddddddddg¡gd}t|j
|
d d
|j d d
d}| |¡rt| |¡r&ttttd}tttd}t ddddddg¡}t ddddddg¡}t||g||gd}| |¡rt| |¡r¦ttdd d
|
d g}td d g}t ddddddg¡}t ddddddg¡}t||g||gd}| |¡rtd S )Nr é r r r é )ÚlevelsÚcodesZfooÚbarZbazZquxZoneZtwo) r r
|
Úvaluesr Z equal_levelsr r Úranger r r0 r1 )r ÚindexÚindex2Z
|
major_axisZ
|
minor_axisZ major_codesZ minor_codesr r r Útest_equals_multi sF ,ýþ ÿ ÿr7 c C s | ¡ }| ¡ }| |¡s t| ddg¡}| |¡s:t| |¡rHt| ddg¡}| |¡sdtt| ¡ dd}| |¡rt| |¡std S )NZnew1Znew2F)Z tupleize_cols)r Z identicalr
|
Ú set_namesr r Útolist)r r% Úmi2Úmi4r r r Útest_identical¸ s r< c C s | | k ¡ std S ©N)Úallr
|
©r r r r Útest_equals_operatorÉ s r@ c C s^ t dtjfdt d¡fg¡} | dd
| d ¡}|r<t| dd
| d ¡}|rZtd S )Nr Z20130101r r )r r, ÚpdZNaTÚ Timestampr r
|
)Úir& r r r Útest_equals_missing_valuesÎ s
|
rD c C st t dtjftjtjfg¡} t tjtjfdtjfg¡}| |¡rFtt dtjftjtjfg¡}| |¡sptd S )Ng @T@)r r, r Únanr r
|
©Zmi1r: r r r Ú-test_equals_missing_values_differently_sorted× s
|
rG c C s4 t ttdtd¡} | | ¡s&t| | ¡ ¡s8t| | ¡ ¡ ¡ ¡ ¡sVt| ¡ }ddg|_| | ¡svt| |¡st| | ddg¡¡rt| t
|
tdt
|
tdg¡}| |¡rÈt| | ¡sÖt| ¡ }| t
|
tdt
|
tdg¡}| |¡r t| ¡ }| |j ¡}| | ¡r0td S )Né
|
ÚAÚBÚCÚD) r r, Úzipr4 Úis_r
|
ÚviewÚnamesr8 Z
|
set_levelsr r0 )r% r: Zmi3r; Zmi5r r r Útest_is_á s$
|
rQ c C s | j r
|
td S r= )Z _is_all_datesr
|
r? r r r Útest_is_all_datesü s rR c C s t | r td S r= )r r
|
r? r r r Útest_is_numeric s rS c C s\ t ddgg¡} tddg}t| | k}t ||¡ tddg}t| | k}t ||¡ d S )Nr r TF)r r$ r r r )Zmidxr' r& r r r Útest_multiindex_compare s rT c C sT t tddgddddgg¡} t ddgddgg¡}| |¡rBt| | ¡rPtd S )Nr r ZInt64r r/ r. )r Z from_arraysr r r
|
rF r r r Ú test_equals_ea_int_regular_int s rU )Únumpyr r Zpandas.core.dtypes.commonr ZpandasrA r r r Zpandas._testingZ_testingr r r# r( r- r7 r< r@ rD rG rQ rR rS rT rU r r r r Ú<module> s& =1
|
|
