U
|
ý°dkF ã
|
@ s´ d dl m Z mZ d dlmZmZmZ d dlZd dl Z d dl
|
m Z m Z d dl mZ G dd dZdd Zee d d
|
¡e d d
|
¡e d d
|
¡e d d
|
¡e d d
|
¡e d d
|
¡d d ZdS ) é )ÚdatetimeÚ timedelta)ÚassumeÚgivenÚ
|
strategiesN)ÚIndexÚ
|
RangeIndexc @ sì e Zd Zej ddddg¡dd Zdd Zd d
|
Zd d Z d d Z
|
ejjde ddde ddde ddde dddfe ddde ddde ddde dddfe ddde ddde ddde dddfe ddde ddde ddde dddfe ddde ddde ddde dddfe ddde ddde ddde e eddde eddd fe ddde ddde ddde e eddde eddd fe dd d e d d d e d d de e edd d e ed d d fe ddd e dd d e ddde e eddd e edd d fe dd!de dd!de dd!de dd!dfe dd"d#e dd"de d$dde e edd"d#dg fe dd%de dd&d'e d%dde e edd%ddd%g fe de de de dfe ddd e de ddd e ddd fe dd!de d!d(d)e dd*de dd*dfe dd"d e d"d+d*e d"d de e edd"d d"dg fe dd"de dd,d-e d.dde dd"dfe ddde dd/de dd/de dddfe ddde d#d0de d#dde ddd#gfe dd1de d dde dddd ge dddd gfe ddde g ejd2e ddde dddfe de ddd/ge ddd/ge ddd/gfe dde dde dde ddfgd3d4 d5d6d7 Zd8d9 Zd:d; Zd<d= Zd>d? Zd@dA ZdBdC ZdDdE ZdFdG ZdS )HÚTestRangeIndexSetOpsÚdtypeNÚint64Zuint64c C s t ddddd}t||d}| tj¡}| |¡}tj||dd | |¡}tj||dd | |d d
¡}tj||d d
dd |d d
|¡}tj||d d
dd | |d d
¡}tj||d d
dd |d d
|¡}tj||d d
dd d S )
|
Nr é é Úfoo©ÚstartÚstopÚstepÚname©r
|
T©Úexacté )r r ÚastypeÚnpÚfloat64Ú intersectionÚtmÚassert_index_equal)Úselfr
|
ÚindexZfltÚresult© r! ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/ranges/test_setops.pyÚ"test_intersection_mismatched_dtype s
|
|
z7TestRangeIndexSetOps.test_intersection_mismatched_dtypec C s t ddd|d d}|j|d d
|d ¡|d}tj||d d
|d ¡dd |d d
j| |d ¡|d}tj||d d
|d ¡dd d S ) Nr r r r r ©ÚsortTr )r r Úrenamer r )r r% Únamesr r r! r! r" Útest_intersection_empty3 s
|
" z,TestRangeIndexSetOps.test_intersection_emptyc C sj t dddd}tt dd¡}|j||d}tt t |j|j¡¡}t ||¡ |j||d}tt t
|
t |j|j¡¡¡}t ||¡ t dd}|j||d}tt t |j|j¡¡}tj ||dd t d
|
dd }|j||d}tt t |j|j¡¡}tj ||dd |j||d}tj ||dd t d d d }t d
|
dd }| t ¡j| t ¡|d}|j||d t ¡}t ||¡ |j||d t ¡}t ||¡ t d
|
dd}t d
|
d ddd}|j||d}t ddddd}t ||¡ t d dd }|j||d}t ddd}t ||¡ t ddd}|j||d}t ddd}t ||¡ |j||d}t ||¡ d S )Nr r r ©r r r r é r$ Úequivr é éÿÿÿÿé
|
éþÿÿÿéüÿÿÿr ©r éûÿÿÿ) r r r Zaranger r% Z intersect1dÚvaluesr r Zasarrayr Úint)r r% r Úotherr ÚexpectedÚfirstr! r! r" Útest_intersection? sT ÿ
|
z&TestRangeIndexSetOps.test_intersectionc C sZ t ddd|d d}t ddd|d d}|j||d}t ddd|d d}t ||¡ d S )Nr r. r r r1 é r$ )r r r r )r r% r' r r5 r r6 r! r! r" Ú%test_intersection_non_overlapping_gcd~ s
|
z:TestRangeIndexSetOps.test_intersection_non_overlapping_gcdc C s t dddd}tdd tdD td}|j||d }tt ||f¡}t ||¡ |j||d }tt ||f¡}t ||¡ d S )
|
Nr r r r) c S s g | ]}t ¡ t| qS r! )r Únowr )Ú.0Úir! r! r" Ú
|
<listcomp> s zATestRangeIndexSetOps.test_union_noncomparable.<locals>.<listcomp>r9 r r$ ) r r ÚrangeÚobjectÚunionr Z concatenater r )r r% r r5 r r6 r! r! r" Útest_union_noncomparable s z-TestRangeIndexSetOps.test_union_noncomparablez/idx1, idx2, expected_sorted, expected_notsortedr r. r r, r iöÿÿÿr- éìÿÿÿiíÿÿÿr é é é r9 r/ é iëÿÿÿéd iÿÿÿr2 i¡ÿÿÿiõÿÿÿiôÿÿÿr0 é éÈ éf é2 iÎÿÿÿéýÿÿÿiÿÿÿr* iúÿÿÿé r c C s t | trt| S | S )N)Ú
|
isinstancer Úrepr)Úxr! r! r" Ú<lambda> ó z TestRangeIndexSetOps.<lambda>)Úidsc C s |j |d d}tj||dd |j |dd}tj||dd |j |d d}t|j|jdj |d d}tj||dd tj||dd d S )Nr$ Tr Fr1 r+ )rA r r r Z_valuesr )r Zidx1Zidx2Zexpected_sortedZexpected_notsortedZres1Zres2Zres3r! r! r" Útest_union_sorted s z&TestRangeIndexSetOps.test_union_sortedc C sZ t tddd}t tddd}| |¡}tddddddd d
|
d d g
|
}tj||d d d S )Nr r r9 r rF r, é é rE é é é Tr )r r? rA r r r ©r ÚleftÚrightr r6 r! r! r" Ú test_union_same_step_misaligned* s
|
|
z4TestRangeIndexSetOps.test_union_same_step_misalignedc C sö t jtdddd}| |¡}t jtddd}tj||dd | | d¡¡}tj|| d ¡dd | |d d
¡}tj||d d
dd | |d
|
d
¡}tj||d d
|
dd |d d d
|d
|
d
¡}tj||d d
|
dd |d d d
j|d
|
d
d d }tj||d d
|
d d d
dd |d d d
|d
|
d
d d d
¡}tj||d d
|
dd |d d d
j|d
|
d
d d d
d d }tj||d d
|
d d d
dd | |dd
¡}tdddddgdd}tj||dd d S )Nr r. r r1 r Tr ÚbarrN rM r- Fr$ r r* é rV rW )r Ú
|
from_ranger? Ú
|
differencer r r& r ©r Úobjr r6 r! r! r" Útest_difference3 s*
|
"&*"z$TestRangeIndexSetOps.test_differencec C s° t tdd d d
}t tdd}| |¡}t td}tj||dd |j|dd}|d d d
}tj||dd tdd }|j|d d}|d d d
}tj||dd d S )
|
Nr9 r- rN Tr Fr$ r. rE )r r? rb r r )r Úidxr5 r r6 r! r! r" Útest_difference_sortW s
|
|
z)TestRangeIndexSetOps.test_difference_sortc C sô t jtdddd}| |d d d
¡}|dd d
}tj||dd |d d d
j|d d d
d d
|
}tj||d d d
dd | |dd d
¡}|d d d
}tj||dd |d d d
j|dd d
d d
|
}tj||d d d
dd d S ) Nr r. r r1 r Tr r- Fr$ )r ra r? rb r r rc r! r! r" Ú test_difference_mismatched_stepk s ""z4TestRangeIndexSetOps.test_difference_mismatched_stepc C sX t td}t tdd}| |¡}t ddd}| ¡ ddgksDttj||dd d S )Nr9 r rN r Tr )r r? rb ÚtolistÚAssertionErrorr r r[ r! r! r" Ú4test_difference_interior_overlap_endpoints_preserved| s
|
zITestRangeIndexSetOps.test_difference_interior_overlap_endpoints_preservedc C sV t ddd}t ddd}| |¡}t ddd}| ¡ ddgksBttj||d d
|
d S ) Niøÿÿÿr r` rX i÷ÿÿÿrM r- r* Tr )r rb ri rj r r r[ r! r! r" Ú4test_difference_endpoints_overlap_interior_preserved
s
|
zITestRangeIndexSetOps.test_difference_endpoints_overlap_interior_preservedc
|
C s* t td}|dd
}| |¡}t ddddddd d
|
d g }tj||d d |d d d
}| |¡}t ddddd d
|
g}tj||d d t td}|d dd
}| |¡}t dddd d gttdd }tj||d d |ddd
}| |¡}t ddddd
|
dgttdd }tj||d d d S )Nr. rN r9 r r r r, r* r` rV rW Tr r rD )r r? rb r r Úlist)r rf r5 r r6 rd r! r! r" Ú'test_difference_interior_non_preserving s$
|
|
|
|
"z<TestRangeIndexSetOps.test_difference_interior_non_preservingc C s t jtdddd}| |¡}t jtddd}t ||¡ | | d¡¡}t || d ¡¡ |d d
|dd
¡}tddd d
|
gdd}tj||d d t tdd ¡}| |¡}t tdd ¡}t ||¡ | |dd
¡}tdddddddd d
|
ddddg }tj||d d d S )Nr r. r r1 r r_ r/ r rV rW Tr é rN r9 r, r* r` rD rE rX é )r ra r? Úsymmetric_differencer r r& r )r r\ r r6 r] r! r! r" Útest_symmetric_difference© s
|
|
"z.TestRangeIndexSetOps.test_symmetric_difference)Ú__name__Ú
|
__module__Ú __qualname__ÚpytestÚmarkZ parametrizer# r( r8 r: rB r r rm r? r r rU r^ re rg rh rk rl rn rr r! r! r! r" r s
|
?
|
|
|
|
ü
|
|
|
|
ü
|
|
|
|
ü
|
|
|
|
ü
|
|
|
|
ü
|
|
|
"ü
|
|
|
"ü
|
|
|
"ü
|
|
|
"ü
|
|
|
|
ü
|
|
|
ü
|
|
|
ü
|
|
|
ü
|
|
|
|
ü
|
|
|
ü
|
|
|
|
ü
|
|
|
|
ü
|
|
|
ü
|
|
ü
|
|
|
ü üü ÿ ö
|
$ r c C sF t | tsBt| dkrB| dd
| dd
}||d k ¡ rBtdS )zp
|
Check that we either have a RangeIndex or that this index *cannot*
|
be represented as a RangeIndex.
|
r Nr- r )rO r ÚlenÚallrj )r Zdiffr! r! r" Ú assert_range_or_not_is_rangelikeÄ s rz rC r c C s´ t |dk t |dk t| ||}t|||}|j|d d}t| t| ¡ } t| ¡ }
|
| j|
|
d d} tj|| dd |j|dd}t| | j|
|
dd} tj|| dd d S )Nr r$ r+ r F)r r rb rz r Zto_numpyr r ) Zstart1Zstop1Zstep1Zstart2Zstop2Zstep2r\ r] r Z
|
left_int64Z right_int64Zaltr! r! r" Útest_range_differenceÎ s r{ )r r Z
|
hypothesisr r r ÚstÚnumpyr rv Zpandasr r Zpandas._testingZ_testingr r rz Zintegersr{ r! r! r! r" Ú<module> s& 2
|
|
|
|
|
|
|
ú
|
