U
|
ý°d ã @ sè d dl mZ d dlZd dlZd dlmZ d dlmZ d dl m
|
Z dd Z ej ddd gd
|
d d Zej e d e ddgd
|
dd Zej ejdddejdd dejdddejd d dddejgddgd
|
dd ZG dd dZdS )é )Ú permutationsN)Ú IntervalTree)ÚIS64c C s t jjt dd}t j| |dS )z
|
Skip parameters in a parametrize on 32bit systems. Specifically used
|
here to skip leaf_size parameters related to GH 23440.
|
z$GH 23440: int type mismatch on 32bit©Úreason)Úmarks)ÚpytestÚmarkÚskipifr Úparam)r r © r úgd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/interval/test_interval_tree.pyÚ skipif_32bit s r Úint64Úfloat64Úuint64)Úparamsc C s | j S ©N©r ©Úrequestr r r Údtype s r é é é
|
c C s | j S )zd
|
Fixture to specify IntervalTree leaf_size parameter; to be used with the
|
tree fixture.
|
r r r r r Ú leaf_size s r é ©r é é c C s | j }t||d |dS )Nr )r )r r )r r Úleftr r r Útree# s r! c @ s¬ e Zd Zdd Zej dddg¡dd Zdd Zej dddg¡d
|
d Z d d Z
|
ej de de de ddg¡dd Z ej de jdddgdde ddd g¡d fe jdddgdde d ddg¡d fe dde jg¡e d de jg¡d fe jdddgdde ddd g¡d fe dde jg¡e dde jg¡d fg¡ej d d d! eedD ¡d"d# Zej d d$d! eedD ¡d%d& Zej d'e jg dde jg ddfe jdgdde jdgddfe e jg¡e e jg¡fe e jgd ¡e e jgd ¡fg¡d(d) Zejje d*d+d,d- Zejje d*d+ej de j d.gd.d/gd0fe j d1gd1d2gd1fd1d2gd2e jgd0fd.d/gd/e jgd/fg¡d3d4 Zd5S )6ÚTestIntervalTreec C sf | t dddg¡¡}tjdddgdd}t ||¡ tjtd d
|
| t d g¡¡ W 5 Q R X d S ) Nç ð?g @ç @r r éÿÿÿÿÚintpr ú6'indexer does not intersect a unique set of intervals'©Úmatchg @)Ú get_indexerÚnpÚarrayÚtmÚassert_numpy_array_equalr ÚraisesÚKeyError)Úselfr! ÚresultÚexpectedr r r Útest_get_indexer1 s ÿz!TestIntervalTree.test_get_indexerz!dtype, target_value, target_dtype)r l r )r r% r c C sf t jddg|dt jddg|d }}t||}| t j|g|d¡}t jdgdd}t ||¡ d S )Nr r r r r% r& )r+ r, r r* r- r. ) r1 r Ú target_valueÚ target_dtyper Úrightr! r2 r3 r r r Útest_get_indexer_overflow; s
|
&
|
z*TestIntervalTree.test_get_indexer_overflowc C sÄ | t dddg¡¡\}}|d d
}tjdgdd}t ||¡ t |dd
¡}tjddgdd}t ||¡ t |dd
¡}tjd gdd}t ||¡ |}tjd
|
gdd}t ||¡ d S ) Nr# ç @r$ r r r& r r r% r )Úget_indexer_non_uniquer+ r, r- r. Úsort)r1 r! ÚindexerÚmissingr2 r3 r r r Útest_get_indexer_non_uniqueG s z,TestIntervalTree.test_get_indexer_non_uniquec C s t jddg|dt jddg|d }}t||}t j|g|d}| |¡\}} t jdgdd}
|
t ||
|
¡ t jdgdd} t | | ¡ d S )Nr r r r r r% r& )r+ r, r r: r- r. ) r1 r r5 r6 r r7 r! ÚtargetZresult_indexerZresult_missingZexpected_indexerZexpected_missingr r r Ú$test_get_indexer_non_unique_overflowZ s &
|
z5TestIntervalTree.test_get_indexer_non_unique_overflowc C s° t jdddg|d}t||d }tjtdd | t dg¡¡ W 5 Q R X | t dg¡¡\}}t |¡}t jdddgdd}t
|
||¡ |}t jg dd}t
|
||¡ d S ) Nr r r r' r( ç à?r r& ) r+ r, r r r/ r0 r* r: r; r- r. )r1 r r r! r< r= r2 r3 r r r Útest_duplicatesj s ÿ
|
z TestIntervalTree.test_duplicatesr r r éd i' c C s¢ t jddd}| d¡}dt d¡ d¡}t||d ||d}t || |d ¡¡ |jr`|n|}t || |d ¡¡ |j r|n|}t || |d ¡¡ d S )
|
Niè r r r& r% rA )Úclosedr g Ð?ç )
|
r+ ÚarangeZastypeZonesr r- r. r* Z closed_leftZ closed_right)r1 rD r ÚxÚfoundÚ not_foundr! r3 r r r Útest_get_indexer_closed| s
|
z(TestIntervalTree.test_get_indexer_closedzleft, right, expectedr r r r r r r TFÚorderc c s | ]}t |V qd S r ©Úlist©Ú.0rG r r r Ú <genexpr> s zTestIntervalTree.<genexpr>c C s, t || || |d}|j}||ks(td S )N©rD ©r Úis_overlappingÚAssertionError)r1 rD rK r r7 r3 r! r2 r r r Útest_is_overlapping s z$TestIntervalTree.test_is_overlappingc c s | ]}t |V qd S r rL rN r r r rP s c C sP t jdddt dd¡ }}t|| || |d}|j}|dk}||ksLtdS ) z*shared endpoints are marked as overlappingr r r r r rQ ZbothN)r+ rF r rS rT )r1 rD rK r r7 r! r2 r3 r r r Ú test_is_overlapping_endpoints s
|
z.TestIntervalTree.test_is_overlapping_endpointsz left, rightc C s t |||d}|jdks td S )NrQ FrR )r1 rD r r7 r! r r r Útest_is_overlapping_trivial¨ s z,TestIntervalTree.test_is_overlapping_trivialzGH 23440r c C s\ t jdddt t j¡jgd }}t||}|jj}dt t j¡j d }||ksXtd S )Née r r é2 r ) r+ rF Ziinfor Úmaxr ÚrootÚpivotrT )r1 r r7 r! r2 r3 r r r Útest_construction_overflow¶ s
|
$
|
z+TestIntervalTree.test_construction_overflowr# r9 rE g Àg ð¿c C s* t |d |d }|jj}||ks&td S )NrX )r r[ r\ rT )r1 r r7 r3 r! r2 r r r Ú!test_inf_bound_infinite_recursionÁ s z2TestIntervalTree.test_inf_bound_infinite_recursionN)Ú__name__Ú
|
__module__Ú __qualname__r4 r r Z parametrizer8 r> r@ rB r rJ r+ r, Únanr ÚrangerU rV rW r
|
r r] ZxfailÚinfr^ r r r r r" 0 sh
|
þ
|
þ
|
ÿ
|
$$$$$ûþ
|
|
"üþ
|
|
|
üþ r" )Ú itertoolsr Únumpyr+ r Zpandas._libs.intervalr Z pandas.compatr Zpandas._testingZ_testingr- r Zfixturer r rF r, rb r! r" r r r r Ú<module> s&
|
|
üÿ
|
|
