U
|
ý°d¸# ã @ s^ d dl m Z d dlZd dlZd dlZd dlmZmZ d dlm Z
|
d dl m Z G dd dZ dS )é )ÚdatetimeN)ÚIndexÚSeries)Ú safe_sortc @ sî e Zd Zej dddddg¡dd Zdd Zd
|
d Zej d e j
|
e e g¡d d Z dd Zdd Zdd Zdd Zej d e j
|
e e g¡dd Zdd Zdd Zd d Zej d ed d!gd gfed d!d"gd"d gfg¡d#d$ Zd%d& Zd'd( Zej d)de j
|
d*d+d,d-gd.efd/gd0d1fde j
|
d*d,d+d-gd.efd/gd0d2fde j
|
d*d,d3d+d-d4gd.efd/gd0d2fg¡d5d6 Zej d7d8d9gg g¡ej d:d9d8gg g¡ej d;d<d=d>g¡d?d@ Z ej dAddBd ggddBdCd dDggg¡dEdF Z d2S )GÚTestIndexSetOpsÚmethodÚunionÚ intersectionÚ
|
differenceÚsymmetric_differencec C sZ t ddg}t ddg}tjtdd t|||dd W 5 Q R X t|||dd d S ) NÚaÚbÚcz The 'sort' keyword only takes©Úmatché ©ÚsortT)r ÚpytestÚraisesÚ
|
ValueErrorÚgetattr)Úselfr Úidx1Úidx2© r úbd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/base_class/test_setops.pyÚtest_setops_sort_validation s
|
z+TestIndexSetOps.test_setops_sort_validationc C s t dddgtd}| |dd
¡}|dd
}t ||¡ | |dd
d d d
¡}t ||¡ |j|dd
d d}|}t ||j¡ |j|dd
d d}t ||¡ |j|dd
d d d
d d}t ||j¡ |j|dd
d d d
d d}t ||¡ d S )Né r é ©Zdtypeéÿÿÿÿr ) r Úobjectr ÚtmÚassert_index_equalZ_unionZassert_numpy_array_equalÚvaluesr )r ÚidxÚresultÚexpectedr r r Ú!test_setops_preserve_object_dtype s z1TestIndexSetOps.test_setops_preserve_object_dtypec C sZ t ddddddg}|dd
}|d d
}| |¡}t ddddddg}t ||¡ d S © Nr r r r r r r é )r r r# r$ ©r ÚindexÚfirstÚsecondr' r( r r r Útest_union_base7 s
|
z TestIndexSetOps.test_union_baseÚklassc C sP t ddddddg}|dd
}|d d
}| ||j¡}t ||¡sLtd S r* )r r r% r# Ú equalContentsÚAssertionError)r r1 r- r. r/ r' r r r Ú test_union_different_type_baseA s
|
z.TestIndexSetOps.test_union_different_type_basec C s¤ t dt d¡g}t t¡ | |d d
¡}W 5 Q R X t ||¡ t t¡ |j|d d
d d}W 5 Q R X t ||¡ |j|d d
dd}t ||¡ d S )Nr Ú2000r F)r ÚpdÚ Timestampr# Zassert_produces_warningÚRuntimeWarningr r$ )r r& r' r r r Ú"test_union_sort_other_incomparableL s z2TestIndexSetOps.test_union_sort_other_incomparablec C sF t dt d¡g}tjtdd |j|d d
dd W 5 Q R X d S )Nr r5 z.*r Tr )r r6 r7 r r Ú TypeErrorr ©r r& r r r Ú'test_union_sort_other_incomparable_true^ s z7TestIndexSetOps.test_union_sort_other_incomparable_truec C s6 t dddg}t dddg}t |j|dd|¡ d S )Nr r r Tr ©r r# r$ r )r r& Zsorted_r r r Ú!test_intersection_equal_sort_truec s z1TestIndexSetOps.test_intersection_equal_sort_truec C sn t ddddddg}|d d
}|d d
}|d krBt dddgn t dddg}|j||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 Útest_intersection_baseh s $z&TestIndexSetOps.test_intersection_basec C sT t ddddddg}|d d
}|d d
}|j||j|d }t ||¡sPtd S r? )r r r% r# r2 r3 )r r1 r r- r. r/ r' r r r Ú%test_intersection_different_type_baser s
|
z5TestIndexSetOps.test_intersection_different_type_basec C s4 t dddg ddg¡}t ddg}t ||¡ d S )Nr r r r@ )r r' r( r r r Útest_intersection_nosort| s z(TestIndexSetOps.test_intersection_nosortc C s> t dddg}t |j|dd|¡ t |j|d d|¡ d S )Nr r r Fr r= r; r r r Ú test_intersection_equal_sort s z,TestIndexSetOps.test_intersection_equal_sortc C sT t dddt dddg}t|td}tdgtd}|j||d}t|dksPtd S ) NiÜ r é é r Zaar r )r r r" r Úlenr3 )r r Zdt_datesÚi1Úi2r' r r r Útest_intersection_str_dates s
|
z+TestIndexSetOps.test_intersection_str_dateszindex2,expected_arrÚBÚDÚAc C sJ t ddddg}t |dd}|j||d}|d kr:| ¡ }t ||¡ d S )NrM rK ÚCr" r r )r r Z sort_valuesr# r$ )r Úindex2Z expected_arrr Úindex1r( r' r r r Ú*test_intersection_non_monotonic_non_unique s z:TestIndexSetOps.test_intersection_non_monotonic_non_uniquec C sj t ddddddg}|d d
}|dd
}| ||¡}t dddg}|d krZt t|}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 Útest_difference_base s z$TestIndexSetOps.test_difference_basec C sX t ddddddg}|d d
}|dd
}| |¡}t dddddg}t ||¡ d S rR )r r r# r$ r, r r r Útest_symmetric_difference¨ s
|
z)TestIndexSetOps.test_symmetric_differencezmethod,expected,sort©r rM ©r rM ©r rK ©r rK Únum©ZletZa1r FN©r rN ©r rN c C s t tjddddgdtfdgd}t tjdddddd gdtfdgd}t||||d
|
}|jd kshtt |}t ||¡ d S ) NrV rW rX rY rZ r[ r r\ r] r r ) r ÚnpÚarrayÚintr Úndimr3 r# r$ )r r r( r rP rO r' r r r Útest_tuple_union_bug² s
|
|
þÿ
|
þÿz$TestIndexSetOps.test_tuple_union_bugÚ
|
first_listr r Ú second_listz&first_name, second_name, expected_name)rM rK N)NrK N)rM NNc C s t ||d}t ||d}|j||d} t| |¡}
|
|d krrt|dkrrt|dkrrt t|
|
|d} t | | ¡ nt |
|
|d} t | | ¡ d S )N)Únamer r )r r ÚsetrG Úsortedr# r$ r2 ) r rc rd Z
|
first_nameZ second_nameZ expected_namer r. r/ r Úvalsr( r r r Ú test_union_name_preservationã s z,TestIndexSetOps.test_union_name_preservationzdiff_type, expectedr r rN c C sF t ddddg}t ddddg}t|||}t |}t ||¡ d S )Nr r rM rK r rN )r r r# r$ )r Z diff_typer( r r r' r r r Útest_difference_object_typeù s
|
z+TestIndexSetOps.test_difference_object_type) Ú__name__Ú
|
__module__Ú __qualname__r ÚmarkZ parametrizer r) r0 r^ r_ r Úlistr4 r9 r< r> rA rB rC rD rJ r rQ rT rU r` rb ri rj r r r r r s~
|
ÿ
|
|
|
|
|
|
|
&þ
|
|
|
|
þú
|
|
þú
|
þúïþ
|
þþr )r Únumpyr^ r Zpandasr6 r r Zpandas._testingZ_testingr# Zpandas.core.algorithmsr r r r r r Ú<module> s
|
