U
|
®ý°d¿ ã @ s d dl Zd dlZd dlZd dlmZ ej dd dddg¡dd Z
|
ej dd d
|
d g¡d d Z dd Z dd Z dd Zdd Zdd ZdS )é NÚ
|
align_axisé ÚindexÚcolumnsc C sº t dddg¡}t dddg¡}|j|| d}| dkr|t dd g¡}t d
|
d g¡}t jddgddgg||d }t ||¡ n:t j dd gd
|
d gg¡}t jddddg|d }t ||¡ d S )NÚaÚbÚcÚxÚz©r )r r r é ÚselfÚother©r r ©r )
|
ÚpdÚSeriesÚcompareÚIndexÚ DataFrameÚtmÚassert_frame_equalÚ
|
MultiIndexÚ from_productÚassert_series_equal)r Ús1Ús2ÚresultÚindicesr Úexpected© r ú_d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/series/methods/test_compare.pyÚtest_compare_axis s ÿr" zkeep_shape, keep_equal)TF)FT)TTc C sê t dddg¡}t dddg¡}|j|| |d}| r¢t ddd g¡}t d
|
d g¡}|rzt jddgddgddgg||d }qÚt jddgtjtjgddgg||d }n8t dd g¡}t d
|
d g¡}t jddgddgg||d }t ||¡ d S ) Nr r r r r
|
)Ú
|
keep_shapeÚ
|
keep_equalr r r r r r ) r r r r r ÚnpÚnanr r )r# r$ r r r r r r r r r! Ú test_compare_various_formats s2
|
ÿý ÿr' c C sV t ddtjg¡} t ddtjg¡}| |¡}t jddggddgd}t ||¡ d S )Nr r r r r )r )r r r% r& r r r r )r r r r r r r! Ú test_compare_with_equal_nullsB s
|
|
r( c C sp t dddg¡} t ddtjg¡}| j|dd}t j ddgdd gg¡}t jdddtjg|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 r! Ú!test_compare_with_non_equal_nullsM s r) c C s¢ t j dddgdddgg¡} t jdddg| d}t jddd g| d}|j|dd
|
}t j ddddgddddgd d d d gg¡}t jdddd g|d}t ||¡ d S ) Nr r r r r r r r r
|
r r r )r r Z from_arraysr r r r )r r r r r r r r r! Útest_compare_multi_indexY s ÿr* c C sª d} t jt| dD tjdddgdddgd }tjdddgddd
|
gd }| |¡ W 5 Q R X d} t jt| d2 t dddg¡}t dddd g¡}| |¡ W 5 Q R X d S ) Nz3Can only compare identically-labeled Series objects)Úmatchr r é r r r r Údé )ÚpytestZraisesÚ
|
ValueErrorr r r )ÚmsgZser1Zser2r r r! Ú test_compare_unaligned_objectsg s r2 c C s ddddddddddd dd
|
d dg} d d d}t j| d |¡}|d |d ¡}|d |d k}|d |d k}t dgd ¡}t dgd ¡}t ||¡ t ||¡ t ||¡ d S )Nz
|
2015-07-01Z 08335394550)r r z
|
2015-07-02z+49 (0) 0345 300033z
|
2015-07-03z+49(0)2598 04457z
|
2015-07-04Z
|
0741470003z
|
2015-07-05z 04181 83668zdatetime64[ns]Ústring)Údatar r Fé T)r r ZastypeÚeqr r r )r4 ZdtypesZdfZ
|
result_eq1Z
|
result_eq2Z
|
result_neqZ expected_eqZ expected_neqr r r! Ú"test_compare_datetime64_and_stringw s û
|
r7 )Únumpyr% r/ Zpandasr Zpandas._testingZ_testingr ÚmarkZ parametrizer" r' r( r) r* r2 r7 r r r r! Ú<module> s$
|
ýþ
|
|
