U
|
ý°d7u ã @ sÐ d Z ddlmZ ddlZddlZddlZddlmZ ddlm Z m
|
Z
|
m Z m Z m Z mZ ddlmZ ddlmZmZmZmZ dd Zd d
|
Zej d ejej fej ejfej ej fej ej fg¡d d Z ej ddddddddddddddd d g¡ej d d d g¡d!d" Z ej d#d$d%d%d&gd&d&gg¡d'd( Z!G d)d* d*Z"ej #d+¡ej d,d-d.d/d0g¡d1d2 Z$d3d4 Z%d5d6 Z&d7d8 Z'd9d: Z(d;d< Z)ej d=d$ej*g¡d>d? Z+d@dA Z,dBdC Z-G dDdE dEZ.dS )FzX
|
The tests in this package are to ensure the proper resultant dtypes of
|
set operations.
|
é )ÚdatetimeN)Úfind_common_type)ÚCategoricalIndexÚIndexÚ
|
MultiIndexÚ
|
RangeIndexÚSeriesÚ Timestamp)Ú is_bool_dtypeÚis_datetime64tz_dtypeÚis_signed_integer_dtypeÚ pandas_dtypec C s* | ¡ }| ¡ }| |¡j|jks&td S )N)Ú sort_valuesÚunionÚdtypeÚAssertionError)ÚindexÚidx1Úidx2© r úWd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/test_setops.pyÚtest_union_same_types s r c C s¬ | }|}|j s,|j s,|jjdkr,|jjdksP|j sn|j sn|jjdkrn|jjdkrntjjdtdd}|j |¡ t |j|jg}d }t
|
|rät
|
|snP|jjdkrº|jjdksàt |jt jrà|jjdkrä|jjdksàt |jt jsät }t j|j|jfk}t|j} t|j}
|
| ¡ }| ¡ }tj|dd | |¡} | |¡} W 5 Q R X |r| sZ|
|
r| jt d
|
¡kspt| jt d
|
¡ks¨tn | j|kst| j|ks¨td S ) NÚiÚbzGH#44000 True==1F)ÚreasonÚraisesÚstrictÚc)r ÚuÚfr z'<' not supported between©ÚmatchÚO)Ú is_uniquer ÚkindÚpytestÚmarkZxfailÚ
|
ValueErrorÚnodeZ
|
add_markerr ÚlenÚ
|
isinstanceÚnpÚRuntimeWarningÚuint64r r ÚtmÚassert_produces_warningr r ) Ú
|
index_flatZ index_flat2Úrequestr r r& Z common_dtypeÚwarnZ
|
any_uint64Z idx1_signedZ idx2_signedÚres1Úres2r r r Útest_union_different_types& sl ÿþ
|
ý
|
üúù
|
ø
|
÷ ÿ
|
ÿ
|
ý ü
|
ù
|
÷
|
ö
|
|
|
r5 zidx_fact1,idx_fact2c C sT | d}|d}| |¡}| |¡}|j|j|jfks:t|j|j|jfksPtd S )Né
|
é )r r r )Z idx_fact1Z idx_fact2r r r3 r4 r r r Ú"test_compatible_inconsistent_pairsl s
|
|
r8 zleft, right, expected)Úint64r9 r9 )r9 r- Úobject)r9 Úfloat64r; )r- r; r; )r- r- r- )r; r; r; )údatetime64[ns]r9 r: )r< r- r: )r< r; r: )údatetime64[ns, CET]r9 r: )r= r- r: )r= r; r: )ú Period[D]r9 r: )r> r- r: )r> r; r: Únames)Úfoor@ r@ )r@ ÚbarNc C s~ t | } t |}tg | |d d}tg ||d d}| |¡}|j|ksLt|j|d ks^t| |¡}|j|d ksztd S )Nr )r Únameé é )r r r r r rB Ú intersection)ÚleftÚrightÚexpectedr? Úar Úresultr r r Útest_union_dtypes s
|
|
rK ÚvaluesrC rD é c C s8 t | }t ddg}| |¡}t dg}t ||¡ d S )NrM )r rE r. Úassert_index_equal)rL rI r rJ rH r r r Ú test_intersection_duplicates¥ s
|
|
|
rO c @ s e Zd Zej dddg¡ej dddddg¡d d
|
Zd d Zej d ¡dd Z ej d ¡dd Z
|
ej d ¡dd Z ej ddddddg¡dd Z ej ddddddg¡d d Z ej ddddddg¡d d Zej ddddddg¡d d! Zd"d# Zd$d% Zd&d' Zd(d) Zd*S )+Ú
|
TestSetOpsÚcaseg à?ZxxxÚmethodrE r Ú
|
differenceÚsymmetric_differencec C s0 d}t jt|d t||| W 5 Q R X d S )Nz!Input must be Index or array-liker )r% r Ú TypeErrorÚgetattr)ÚselfrQ rR r Úmsgr r r Útest_set_ops_error_cases± s z#TestSetOps.test_set_ops_error_casesc C sÄ t |trd S |d d
}|d d
}| |¡}t ||¡s@tt|jrNd S | ¡ | ¡ |
|
¡ g}|D ] }| |¡}t ||¡shtqht |t rÀd}t j t|d | dddg¡ W 5 Q R X d S )Né rM ú.other must be a MultiIndex or a list of tuplesr rC rD )r* r rE r. Ú equalContentsr r r Úto_numpyÚ to_seriesÚto_listr r% r rU ) rW r ÚfirstÚsecondÚ intersectÚcasesrQ rJ rX r r r Útest_intersection_base» s
|
|
|
|
|
z!TestSetOps.test_intersection_baseúEignore:Falling back on a non-pyarrow:pandas.errors.PerformanceWarningc
|
C sº |dd
}|d d
}|}| |¡}t ||¡s6tt|jrDd S | ¡ | ¡ | ¡ g}|D ] }| |¡}t ||¡s^tq^t |t
|
r¶d} t j t | d | dddg¡ W 5 Q R X d S )NrM rZ r[ r rC rD )r r. r\ r r r r] r^ r_ r* r r% r rU )
|
rW r r` ra Ú
|
everythingr rc rQ rJ rX r r r Útest_union_baseÔ s
|
|
|
|
zTestSetOps.test_union_basec
|
C sæ |dd
}|d d
}t |r4t| t|¡}n t|trDg }n |dd
}| ||¡}t ||¡slt| ¡ | ¡ |
|
¡ g}|D ] }| ||¡}t ||¡stqt|t râd} t j t| d | dddg|¡ W 5 Q R X d S )NrD é r[ r rC rM )r
|
ÚsetrS r* r r. r\ r r] r^ r_ r r% r rU )
|
rW Úsortr r` ra ÚanswerrJ rc rQ rX r r r Útest_difference_baseï s"
|
|
z TestSetOps.test_difference_basec C sþ t |trd S t|dk r d S |d |dd
ksF|d |d d
krJd S |dd
}|d d
}|ddg }| |¡}t ||¡st| ¡ | ¡ | ¡ g}|D ] }| |¡}t ||¡s¢tq¢t |t
|
rúd}t j t |d | dddg¡ W 5 Q R X d S )NrD r rC éÿÿÿÿr[ r rM )r* r r) rT r. r\ r r] r^ r_ r r% r rU ) rW r r` ra rk rJ rc rQ rX r r r Útest_symmetric_difference s&
|
(
|
|
|
z$TestSetOps.test_symmetric_differencezfname, sname, expected_name©ÚArp rp ©rp ÚBN)rp NN©Nrr N)NNNc
|
C s" |j st d¡ |}| ¡ |¡}| ¡ |¡}| |¡}| ¡ |¡} t || ¡ | ¡ |¡}| |¡ |¡}| |¡}| ¡ |¡} t || ¡ | |¡ |¡}| ¡ |¡}| |¡}| ¡ |¡} t || ¡ | |¡ |¡}| |¡ |¡}| |¡}| |¡ |¡} t || ¡ d S ©Nú-Randomly generated index_flat was not unique.) r# r% ÚskipÚcopyÚ set_namesr r. rN Údrop©
|
rW r0 ÚfnameÚsnameÚ expected_namer r` ra r rH r r r Útest_corner_union) s.
|
|
|
|
|
z TestSetOps.test_corner_unionc
|
C s` |j st d¡ |}| ¡ |¡}|dd
|¡}| |¡ ¡ }| |¡ ¡ } t || ¡ d S ©Nru rC ) r# r% rv rw rx r r r. rN rz r r r Útest_union_unequalW s
|
z TestSetOps.test_union_unequalc
|
C s& |j st d¡ |}| ¡ |¡}| ¡ |¡}| |¡}| ¡ |¡} t || ¡ | ¡ |¡}| |¡ |¡}| |¡}| |¡ |¡} t || ¡ | |¡ |¡}| ¡ |¡}| |¡}| |¡ |¡} t || ¡ | |¡ |¡}| |¡ |¡}| |¡}| |¡ |¡} t || ¡ d S rt ) r# r% rv rw rx rE r. rN ry ©
|
rW r0 r{ r| r} r r` ra rb rH r r r Útest_corner_intersectm s.
|
|
|
|
|
z TestSetOps.test_corner_intersectc
|
C sh |j st d¡ |}| ¡ |¡}|dd
|¡}| |¡ ¡ }|dd
|¡ ¡ } t || ¡ d S r ) r# r% rv rw rx rE r r. rN r r r r Útest_intersect_unequal s
|
z!TestSetOps.test_intersect_unequalc C s t |tr | tt|j¡}n
|
| d¡}t |¡}| |¡}|j |j ksNt
|
| |d d
¡}|j |j kspt
|
|d d
|¡}|j |j kst
|
d S )Nr@ r ) r* r ÚrenameÚlistÚrangeÚnlevelsr+ ZasarrayrE rB r )rW r ÚotherrJ r r r Ú.test_intersection_name_retention_with_nameless° s
|
|
|
|
z9TestSetOps.test_intersection_name_retention_with_namelessc C s8 |j s
|
d S |j||d}|d d
}tj||dd d S )N©rj r T©Úexact)r# rS r. rN )rW r rj rJ rH r r r Ú$test_difference_preserves_type_emptyà s
|
z/TestSetOps.test_difference_preserves_type_emptyc sv t tr fdd|D } |d ¡ |d ¡} |¡sFt |¡} d d
|d ¡}t ||¡ d S )Nc s g | ]}|g j qS r )r )Ú.0Úx©r r r Ú
|
<listcomp>Ï s zDTestSetOps.test_difference_name_retention_equals.<locals>.<listcomp>r rC rD )r* r r Úequalsr rS r. rN )rW r r? r rJ rH r r r Ú%test_difference_name_retention_equalsÍ s
|
|
z0TestSetOps.test_difference_name_retention_equalsc C s> |j s
|
d S | |d d
¡}|j||d}tj||dd d S )Nr r Tr )r# rE rS r. rN )rW r rj ÚinterZdiffr r r Ú(test_intersection_difference_match_emptyÙ s
|
z3TestSetOps.test_intersection_difference_match_emptyN)Ú__name__Ú
|
__module__Ú __qualname__r% r& Ú parametrizerY rd Úfilterwarningsrg rl rn r~ r r r r r r r r r r r rP ¯ s~
|
ÿ ÿ
|
ÿ
|
ÿ
|
ûþ
|
|
$ûþ
|
|
ûþ
|
|
#ûþ
|
|
|
rP re rR rE r rS rT c C s | }| d¡}t|tr dnd}t||||d}t||||d}tj|||d t|||d d
|d}t|||d d
|d}tj|||d d S )NÚcategoryÚequivTr r rZ )Zastyper* r rV r. rN )r0 rj rR r r r rJ rH r r r Útest_setop_with_categoricalå s
|
r c C sL | j r
|
d S | }|ddddg }| |¡ | |¡¡s8t| |¡jsHtd S ©Nr rC rD )ÚemptyrE r r r# )r ÚidxZidx_non_uniquer r r Ú(test_intersection_duplicates_all_indexesû s r¡ c C s | }t ddddg|d}t dddg|d}t ddddddg|d}t|tr`t ddddddg}| |¡}t ||¡ |j|dd}t ||¡ d S )NrC rD rM ©r rh Fr ©r r* r r r. rN ©Z'any_dtype_for_small_pos_integer_indexesr rI r rH rJ r r r Ú0test_union_duplicate_index_subsets_of_each_other s
|
|
r¥ c C sh | }t dddg|d}t ddg|d}t dddg|d}| |¡}t ||¡ | |¡}t ||¡ d S )NrC r r¢ ©r r r. rN r¤ r r r Ú1test_union_with_duplicate_index_and_non_monotonic s
|
|
r§ c C sR t ddddg} t dddg}t dddddddg}| j|dd}t ||¡ d S )NrC rD rM Ú1Ú0Fr r¦ )rI r rH rJ r r r Ú+test_union_duplicate_index_different_dtypes) s
|
rª c C sH t dddg} t ddddg}| |¡}t ddddg}t ||¡ d S r r¦ )rI r rJ rH r r r Ú(test_union_same_value_duplicated_in_both2 s
|
|
r« Údupc C sV t tjdddg}t tj| ddg}|j|dd}t tj| dddg}t ||¡ d S )NrC rD Fr g ð?ç @)r r+ Únanr r. rN )r¬ rI r rJ rH r r r Útest_union_nan_in_both; s
|
r¯ c C s | }t dddg|d}t dddg|d}t ddddg|d}t|trVt ddddg}| |¡}t ||¡ | |¡}t ||¡ d S )NrC r rD r¢ r£ r¤ r r r Ú<test_union_with_duplicate_index_not_subset_and_non_monotonicE s
|
|
|
r° c C st t ddtjg} | jjjdks"ttddg}| | ¡}tddtjgtj d}t
|
||¡ | |¡}t
|
||¡ d S )NrC rD r r¢ ) r r+ r® Ú
|
categoriesr r$ r r r r; r. rN )Úcir rJ rH r r r Ú#test_union_int_categorical_with_nanW s
|
|
r³ c @ s4 e Zd Zdd Zejjddgdddd Zej d ed
|
d d d dgdddfed
|
d d d dgdddfed
|
d d d dgdfg¡dd Z ejjddgddej ddddg¡dd Z
|
dd Z ejjddgddd d Z ej d e jeeg¡ejjddgddd d Zejjddgddd!d" Zejjddgddej d#d$d%g¡d&d' Zd(d) Zejjddgddd*d+ Zejjddgddd,d- Zej d.d/d0g¡d1d2 Zej d.d/d0g¡d3d4 Zd5d6 Zej d7ed8d9e jged:d;d<gfed8d9gee jd:d;d<gfg¡d=d> Zd?d@ Z dAdB Z dCS )DÚTestSetOpsUnsortedc C sX t dddt dddg}t|td}tdgtd}| |¡}tg td}t ||¡ d S )NiÜ rD é é r¢ Zaa)r r r: rE r. rN )rW Zdt_datesÚindex1Úindex2rJ rH r r r Útest_intersect_str_datesh s
|
z+TestSetOpsUnsorted.test_intersect_str_datesr ÚstringT)Zindirectc C sl |d d
}|d d
}|j ||d}|d kr>t || ¡ ¡ t ||¡sNt|j ||d}||kshtd S )Nr7 r6 r )rE r. rN r r\ r )rW r rj r` ra rb r r r r Útest_intersectionr s z$TestSetOpsUnsorted.test_intersectionzindex2,keeps_namerM rh rZ é é ©rB r Fc C sZ t dddddgdd}t dddg}| ||¡}|r:d|_|j|jksJtt ||¡ d S )NrC rD rM rh rZ r r¾ )r rE rB r r. rN )rW r¸ Z
|
keeps_namerj r· rH rJ r r r Ú#test_intersection_name_preservation s z6TestSetOpsUnsorted.test_intersection_name_preservationz$first_name,second_name,expected_namero rq rs c C sD |dd
}|d d
}||_ ||_ |j||d}|j |ks@td S ©NrZ r7 r6 r )rB rE r ) rW r Z
|
first_nameÚ second_namer} rj r` ra rb r r r Ú$test_intersection_name_preservation2 s z7TestSetOpsUnsorted.test_intersection_name_preservation2c
|
C sÔ t ddgdd}t ddgdd}t dd gd
|
d}|j|j||d |d }|j||d j||d }t ||¡ t ddgd d}t g d d}t g dd} |j|j| |d |d }|j||d j| |d }t ||¡ d S )NrC rD Úi1r¾ rZ r¼ Úi2rM rh Úi3r Új1Új2Új3r¦ )
|
rW rj rà rÄ rÅ r rH rÆ rÇ rÈ r r r Útest_chained_union¡ s z%TestSetOpsUnsorted.test_chained_unionc C s^ |dd
}|d d
}|d d
}|j ||d}|d krJt || ¡ ¡ t ||¡sZtd S rÀ )r r. rN r r\ r )rW r rj r` ra rf r r r r Ú
|
test_union± s z TestSetOpsUnsorted.test_unionÚklassc C sh |dd
}|d d
}|d d
}||j }|j||d}|d krTt || ¡ ¡ t ||¡sdtd S rÀ )rL r r. rN r r\ r ) rW r rË rj r` ra rf rQ rJ r r r Útest_union_from_iterables¼ s
|
z,TestSetOpsUnsorted.test_union_from_iterablesc C st |dd
}|j ||d}||k| ks,t|j g |d}||k| ksLttg j ||d}||k| ksptd S )NrZ r7 r )r r r )rW r rj r` r r r r Útest_union_identityÊ s z&TestSetOpsUnsorted.test_union_identityzsecond_name,expected)NN)rB rB c C sx |dd
}|d d
}|dd
}d|_ ||_ |j||d}t ||¡sNt|d krf|j d ksttn|j |ksttd S )NrZ r7 r6 rB r )rB rS r. r\ r ) rW r rÁ rH rj r` ra rk rJ r r r Ú!test_difference_name_preservationÚ s z4TestSetOpsUnsorted.test_difference_name_preservationc C s. |dd
}d|_ | g |¡}t ||¡ d S )NrZ r7 rB )rB rS r. rN ©rW r rj r` rJ r r r Útest_difference_empty_argì s z,TestSetOpsUnsorted.test_difference_empty_argc C sB |dd
}d|_ | ||¡}t|dks.t|j |j ks>td S )NrZ r7 rB r )rB rS r) r rÏ r r r Útest_difference_identityó s
|
z+TestSetOpsUnsorted.test_difference_identityc C sP |dd
}|d d
}| ||¡}|dd
}|d kr@| ¡ }t ||¡ d S )NrZ r7 r6 )rS r r. rN )rW r rj r` ra rJ rH r r r Útest_difference_sortü s z'TestSetOpsUnsorted.test_difference_sortÚopnamerS rT c C s® t dtddg}t dtddg}t ||¡}t t¡ ||}W 5 Q R X t dtddtdg}|dkrz|d d
}t ||¡ tj||dd}||}t ||¡ d S ) NrM Ú2000rC rD Ú1999rS Fr )r r ÚoperatorÚ methodcallerr. r/ r, rN )rW rÓ rI r ÚoprJ rH r r r Ú test_difference_incomparable s z/TestSetOpsUnsorted.test_difference_incomparablec C s^ t dtddg}t dtddg}tj||dd}d}tjt|d || W 5 Q R X d S )
|
NrM rÔ rC rD rÕ Tr z<'<' not supported between instances of 'Timestamp' and 'int'r )r r rÖ r× r% r rU )rW rÓ rI r rØ rX r r r Ú!test_difference_incomparable_true s z4TestSetOpsUnsorted.test_difference_incomparable_truec C sx t tdddgdddg¡}t ddg¡}|j||d }t d
|
d dg¡}|d krX| ¡ }t ||¡ t ||¡sttd S ) Nr@ rA ÚbazrC rD rM )r@ rC )rA rM r )rA rD )rÛ rM ) r Ú from_tuplesÚziprT r r. rN r\ r )rW rj r· r¸ rJ rH r r r Ú test_symmetric_difference_mi& s z/TestSetOpsUnsorted.test_symmetric_difference_mizindex2,expectedr rC r g @g c C s@ t dtjddg}|j||d}|d kr0| ¡ }t ||¡ d S )NrC rD rM r )r r+ r® rT r r. rN )rW r¸ rH rj r· rJ r r r Ú!test_symmetric_difference_missing0 s
|
|
z4TestSetOpsUnsorted.test_symmetric_difference_missingc C s t ddddgdd}t ddddg¡}t ddg}|j||d}t ||¡sPt|jdks^t|j|d |d
|
}t ||¡s~t|jd kstd S ) NrC rD rM rh r· r¾ rZ r Únew_name)Z result_namerj )r r+ ÚarrayrT r. r\ r rB )rW rj r· r¸ rH rJ r r r Ú#test_symmetric_difference_non_indexA s z6TestSetOpsUnsorted.test_symmetric_difference_non_indexc C sT t dddg|d}t dddg|d}| |¡}t dddddg|d}t ||¡ d S )NrC rD rM r¢ rh rZ r¦ )rW Z any_numeric_ea_and_arrow_dtyper r rJ rH r r r Útest_union_ea_dtypesM s
|
|
z'TestSetOpsUnsorted.test_union_ea_dtypesN) 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Þ r® rß râ rã r r r r r´ e s^
|
|
ýþ
|
þ
|
|
|
|
|
|
|
|
|
þþ
|
|
r´ )/Ú__doc__r rÖ Únumpyr+ r% Zpandas.core.dtypes.castr Zpandasr r r r r r Zpandas._testingZ_testingr. Zpandas.api.typesr
|
r r r r r5 r& r Z makeIntIndexZmakeRangeIndexZmakeFloatIndexZ makeUIntIndexr8 rK rO rP r r r¡ r¥ r§ rª r« r® r¯ r° r³ r´ r r r r Ú<module> s~ F
|
|
|
|
üþ
|
ñþ
|
8ÿ
|
ÿ
|
|
