U
|
ý°d_Z ã @ s¾ d dl Z d dlZd dlZd dlmZ d dlmZmZm Z m
|
Z
|
m Z m Z m Z mZmZmZmZmZmZmZmZ d dlmZ G dd dZG dd dZG dd d ZG d
|
d d Z G d d d Z dS )é N)ÚInvalidIndexError)ÚNAÚCategoricalIndexÚ DatetimeIndexÚIndexÚIntervalÚ IntervalIndexÚ
|
MultiIndexÚNaTÚ TimedeltaÚ TimestampÚarrayÚ
|
date_rangeÚinterval_rangeÚ period_rangeÚtimedelta_rangec @ sÌ e Zd Zej dddddg¡dd Zej dd d
|
d d d ddddg ¡dd Zej ddd
|
d ddddg¡dd Zej dddddg¡ej dd d d d g¡d d! Z ejjd"e
|
d#d$d%e
|
d#d$d&d'e d(d$d%gd)d* d+d,d- Z ejjd.e
|
d#d$d%e
|
d/d$d%fe
|
d#d$d&d'e
|
d/d$d&d'fe d(d$d%e d0d$d%fgd1d* d+d2d3 Z ejjd4e
|
d5d$d6d7e
|
d5d$d6d&d8e d9d$d6d7e d:d;d;¡e ddd¡gd<d* d+d=d> Zej d?dgd@g¡dAdB ZdCdD ZdES )FÚ
|
TestGetLocÚsideÚrightÚleftÚbothÚneitherc
|
C s* t jddg|d}ddgddgddgddgddgd dgd
|
dgfD ]ä}t d |d d |d d | d¡}||krö|ddgkr| tdd|d¡dksôtnX|ddgkrÆ| tdd|d¡dksôtn.tjt |d | t|d|i¡ W 5 Q R X q@tjt |d | t|d|i¡ W 5 Q R X q@d S )N©r é ©é é ©Úclosedr r r r é ç @éÿÿÿÿú Interval(ú, ú
|
, closed='ú')©Úmatchr )
|
r Ú from_tuplesÚreÚescapeÚget_locr ÚAssertionErrorÚpytestÚraisesÚKeyError)Úselfr r ÚidxÚboundÚmsg© r4 úbd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/interval/test_indexing.pyÚtest_get_loc_interval s 2& z TestGetLoc.test_get_loc_intervalÚscalarç à¿r ç à?r ç ø?r r r ç @c C s ddddddddddddddddddddd}t jdd g|d
|
}||| ¡ krt| |¡|| | kstn(tjtt|d | |¡ W 5 Q R X d S ) Nr r )r9 r r r )r r9 r r )r r9 r r r r )r9 r )r r r r r r r r& ) r r( Úkeysr+ r, r- r. r/ Ústr)r0 r r7 Úcorrectr1 r4 r4 r5 Útest_get_loc_scalar1 s ü z TestGetLoc.test_get_loc_scalarr! ç @é é c C s` t jdg|d}||d kr4| |¡}|dks\tn(tjtt|d | |¡ W 5 Q R X d S )N©r rA r r r& )r r( r+ r, r- r. r/ r= )r0 r7 r ÚindexÚresultr4 r4 r5 Ú test_get_loc_length_one_scalarF s
|
z)TestGetLoc.test_get_loc_length_one_scalarÚ other_closedz left, rightrC )r! r )r! rB ©rB é c C s t jdg|d}t|||d}||d krB| |¡}|dkstn>tjtt d| d| d| d¡d | |¡ W 5 Q R X d S ) NrC r r r" r# r$ r% r& )
|
r r( r r+ r, r- r. r/ r) r* )r0 r r r rG rD ÚintervalrE r4 r4 r5 Ú test_get_loc_length_one_intervalQ s
|
þz+TestGetLoc.test_get_loc_length_one_intervalÚbreaksZ20180101r ©Úperiodsz
|
US/Eastern©rN Útzz0 daysc C s
|
t | jS ©N©r= Údtype©Úxr4 r4 r5 Ú<lambda>i ó zTestGetLoc.<lambda>)Úidsc C sb t |¡}|d j}| |¡}d}||ks.tt|d j|d j}| |¡}d}||ks^td S ©Nr )r Ú from_breaksÚmidr+ r, r r r )r0 rL rD ÚvaluerE ÚexpectedrJ r4 r4 r5 Ú(test_get_loc_datetimelike_nonoverlappingb s
|
|
|
|
z3TestGetLoc.test_get_loc_datetimelike_nonoverlappingÚarraysZ20180103z2 daysc C s t | d jS rY rR rT r4 r4 r5 rV rW c C sr t j| }|d jtd }| |¡}tddd }||ks>tt|d j|d j }| |¡}d}||ksntd S )Nr z12 hoursr )
|
r Ú from_arraysr[ r r+ Úslicer, r r r )r0 r_ rD r\ rE r] rJ r4 r4 r5 Ú%test_get_loc_datetimelike_overlappingz s
|
|
|
z0TestGetLoc.test_get_loc_datetimelike_overlappingÚvaluesz
|
2018-01-04z-1D)rN Úfreq)rN rd rP z3 daysç @g ð¿c C s
|
t | jS rQ rR rT r4 r4 r5 rV rW c C s> t |dd
|d d
¡}| |d ¡}d}||ks:td S )Nr r! r )r r` r+ r, )r0 rc rD rE r] r4 r4 r5 Útest_get_loc_decreasing s z"TestGetLoc.test_get_loc_decreasingÚkeyr c C sD t dddddg¡}t|}tjt|d | |¡ W 5 Q R X d S )N©r r ©r r )r rA )rI é
|
)r rj r& )r r( r= r- r. r r+ )r0 rg r1 r3 r4 r4 r5 Ú test_get_loc_non_scalar_errors© s z)TestGetLoc.test_get_loc_non_scalar_errorsc
|
C s¨ t tjtddtjg}t dddg¡}d tjtfD ]&}||ksDt| |¡}t ||¡ q4t
|
t dd¡t dd¡fD ],}t jtt|d | |¡ W 5 Q R X qvd S )Nr r TFr
|
Únsr& )r ÚnpÚnanr r r r, r+ ÚtmÚassert_numpy_array_equalr
|
Z timedelta64Z
|
datetime64r- r. r/ r= )r0 rD r] rg rE r4 r4 r5 Útest_get_indexer_with_nans² s
|
z%TestGetLoc.test_get_indexer_with_nansN)Ú__name__Ú
|
__module__Ú __qualname__r- ÚmarkÚ parametrizer6 r? rF rK r r r^ rb rm Úarangerf rk rq r4 r4 r4 r5 r sZ
|
|
|
|
|
|
ýù
|
þ
|
|
þú ó
|
û÷
|
|
r c @ s e Zd Zej deddddgdgfeddddgdgfeddd dgdgfeddd
|
dgdgfeddddgdgfed dddgdgfed d ddgdgfedddded dddgddgfeddddeddddgddgfeddddeddddgddgfeddddeddddgddgfg ¡dd Zej ddgdgfd gdgfd gd gfdgd gfd gdgfdgdgfdgdgfdgdgfdgdgfdgdgfdgdgfddgd dgfdddgd ddgfddddgd dddgfdddddgd ddddgfg¡dd Zej ddge
|
d dd ¡g¡dd Z ej d ddg¡d d Z ej d e d d!d"d#g¡e d#d!d"e jg¡e jd#d!d"gd ddd d ddde jgd$d$d%d&gg¡d'd( Zd)d* Zd+d, Zej d-d.d/d"gd
|
fd0d1d2gdfd#d#d!gdfd#d3d"gd fg¡d4d5 Zej ddgdgd gffd gd gg ffd gd gg ffdgd dgg ffd gd dgg ffdgd ddgg ffdgddgg ffdgdgg ffdgdgg ffdgdgd gffdgdgd gffddgd dd ddgg ffdddgd dd dddgg ffddddgd dd ddddgdgffdddddgd dd ddddd ddg
|
dgffg¡d6d7 Zd8d9 Zd:d; Zd<d= Zd>d? Zej d@e eeg¡dAdB ZdCS )DÚTestGetIndexerzquery, expectedr r r r r r r! r r r r9 r: rA rI c C s@ dddg}t j|dd}| |¡}tj|dd}t ||¡ d S )N©r r ri )rA rI r r Úintp©rS ©r r( Ú get_indexerrm r ro rp ©r0 Úqueryr] ÚtuplesrD rE r4 r4 r5 Ú test_get_indexer_with_interval s
|
|
|
z-TestGetIndexer.test_get_indexer_with_intervalr8 r r r; r@ c C s@ dddg}t j|dd}| |¡}tj|dd}t ||¡ d S )Nr ©r r ©r r r r rz r{ r| r~ r4 r4 r5 Ú#test_get_indexer_with_int_and_floatÚ s
|
|
|
z2TestGetIndexer.test_get_indexer_with_int_and_floatÚitemc C sB t jdg|d}| |¡}tjdgt| dd}t ||¡ d S )NrC r r rz r{ )r r( r} rm r Úlenro rp )r0 r
r rD rE r] r4 r4 r5 Útest_get_indexer_length_oneö s
|
z*TestGetIndexer.test_get_indexer_length_oneÚsizec C sL t jdg|d}| tdd|g| ¡}tjdg| dd}t ||¡ d S )NrC r r rA rz r{ )r r( r} r rm r ro rp )r0 r r rD rE r] r4 r4 r5 Ú$test_get_indexer_length_one_intervalþ s z3TestGetIndexer.test_get_indexer_length_one_intervalÚtarget)rI é r r r ÚfooÚbarZbazc C s@ t dddg¡}t||d}| |¡}| |¡}t ||¡ d S )Nr r r )Úordered)r r( r r} ro rp )r0 r r rD Zcategorical_targetrE r] r4 r4 r5 Ú test_get_indexer_categorical s
|
|
|
z+TestGetIndexer.test_get_indexer_categoricalc C sÔ t td¡}| t tjg¡}t|}| |¡}tjdtj d}t
|
||¡ |dd
|d d d
¡}tj dddddgtj d}t
|
||¡ | | |¡¡}tj ddddddddddg
|
tj d}t
|
||¡ d S ) NrA r{ r r! r r r r ) r rZ ÚrangeÚappendrm rn r r} rw rz ro rp r )r0 ÚiiZii2Zci2rE r] r4 r4 r5 Ú&test_get_indexer_categorical_with_nans s
|
$z5TestGetIndexer.test_get_indexer_categorical_with_nansc C s t tddd¡}| tdg¡}tjdgtjd}t ||¡ | tdg
|
t ¡¡}t ||¡ | tdgj ¡}t ||¡ d S )Nz
|
2018-01-01r rM z
|
2018-01-02r r{ ) r rZ r r} r rm r rz ro rp Zastyper= Zasi8)r0 r rE r] r4 r4 r5 Útest_get_indexer_datetime- s z(TestGetIndexer.test_get_indexer_datetimeztuples, closedry rh rC )r r rH r c C s> t j||d}d}tjt|d | ddg¡ W 5 Q R X d S )Nr zKcannot handle overlapping indices; use IntervalIndex.get_indexer_non_uniquer& r r )r r( r- r. r r} )r0 r r rD r3 r4 r4 r5 Útest_get_indexer_errors; s
|
ÿz&TestGetIndexer.test_get_indexer_errorsc C sf dddg}t j|dd}| |¡\}}tj|d dd}tj|d dd}t ||¡ t ||¡ d S )
|
N)r r rh ri r r r rz r{ r )r r( Úget_indexer_non_uniquerm r ro rp ) r0 r r] r rD Zresult_indexerZresult_missingZexpected_indexerZexpected_missingr4 r4 r5 Ú.test_get_indexer_non_unique_with_int_and_floatO s
|
z=TestGetIndexer.test_get_indexer_non_unique_with_int_and_floatc C s t dddg¡}t ddddg¡}| |¡}tjddddgtjd }t ||¡ | |d
|
d
¡}tjd
|
dgtjd }t ||¡ d S ) Nr )r rA r rH )r é r r r! r{ r )r r( r} rm r rz ro rp )r0 Zidx1Zidx2rE r] r4 r4 r5 Ú test_get_indexer_non_monotonicq s
|
z-TestGetIndexer.test_get_indexer_non_monotonicc C sT t tjtjg}t tjg}|jr&t| |¡}tjddgtjd}t ||¡ d S )Nr r r{ )
|
r rm rn Z_index_as_uniquer, Úget_indexer_forr rz ro rp )r0 rD ÚotherrE r] r4 r4 r5 rq } s
|
|
z)TestGetIndexer.test_get_indexer_with_nansc C sL t ddddg¡}| tddg¡\}}tjddgtjd}t ||¡ d S )N©ç ç ð?©r ç @r r r r r{ ) r r( r r rm r rz ro rp )r0 rD rE Ú_r] r4 r4 r5 Ú'test_get_index_non_unique_non_monotonic s
|
ÿz6TestGetIndexer.test_get_index_non_unique_non_monotonicc C sr t jdddgdd}tdddgd d}t ||g¡}| d¡ td
|
d g¡}tj dd d gtj
|
d}t ||¡ d S )N)r re r r rJ )Únamer r r r r r r rI r{ ) r r( r r Z from_productZget_level_valuesr r rm r rz ro rp )r0 Zinterval_indexZ foo_indexZ multi_indexrE r] r4 r4 r5 Ú*test_get_indexer_multiindex_with_intervals s ÿ
|
|
ÿz9TestGetIndexer.test_get_indexer_multiindex_with_intervalsÚboxc C sT t dddd}|ttdddd}| |¡}tjdddgtjd}t ||¡ d S )Nz
|
2022-07-01ÚDr )rd rN Z3Dr! r{ ) r r r r} rm r rz ro rp )r0 r¥ Úrngr1 Úactualr] r4 r4 r5 Ú test_get_indexer_interval_index¡ s
|
|
z.TestGetIndexer.test_get_indexer_interval_indexN) rr rs rt r- ru rv r r r rm rw r r r r( rn r r r r r r rq r¢ r¤ r Úlistr© r4 r4 r4 r5 rx Á s® """"õþ
|
|
|
|
|
|
|
|
|
|
|
|
ñþ
|
|
|
|
ûþ
|
|
üþ
|
"*ñþ
|
|
rx c @ s e Zd Zdd Zdd Zej dddgddgdd gdd
|
gg¡ej d d d dgdd d gd d dgd dd gd d dd gg¡dd ZdS )Ú TestSliceLocsc C s¬ t dddg¡}|jtddtddddks2t|jtddd dksLt|jtddd
|
dksft|jtddd
|
d kst|jtddtdddd ks¢tt dddg¡}|jtddtdddd ksÔt|jtddd d ksît|jtddd
|
d ks
|
t|jtddd
|
dks&t|jtddtddddksJtt dddg¡}|jtddtddddks~t|jtddd dkst|jtddd
|
dks¶t|jtddd
|
dksÒt|jtddtddddksött dddg¡}tjtt d¡d |jtddtddd W 5 Q R X tjtt d¡d |jtddd W 5 Q R X |jtddd
|
dksttjtt d¡d |jtddd
|
W 5 Q R X tjtt d¡d |jtddtddd W 5 Q R X t ddddg¡}|jtddtddddks0t|jtddd dksLt|jtddd
|
dksht|jtddd
|
dkst|jtddtddddks¨td S )Nry rh ri r r r )ÚstartÚend©r r )r¬ )r r ©r r r )r r zR"Cannot get left slice bound for non-unique label: Interval(0, 2, closed='right')"r& zS"Cannot get right slice bound for non-unique label: Interval(0, 2, closed='right')")r r )
|
r r( Ú
|
slice_locsr r, r- r. r/ r) r* ©r0 rD r4 r4 r5 Ú test_slice_locs_with_interval sl """ $$ $ÿþ$ÿþ ÿþ ÿþ$$ z+TestSliceLocs.test_slice_locs_with_intervalc C s t dddg¡}| dd¡dks$t| dd¡dks8t| dd¡dksLt| dd¡d ks`t| dd
|
¡d kstt| dd
|
¡d kstt dddg¡}| dd¡d ks¬t| dd¡dksÀt| dd¡dksÔt| dd¡dksèt| dd
|
¡dksüt| dd
|
¡dkstd S )Nr r r r r r ry r r¯ r r r® )r r )r r )r r rh )r r )r r( r° r, r± r4 r4 r5 Ú-test_slice_locs_with_ints_and_floats_succeedsû s z;TestSliceLocs.test_slice_locs_with_ints_and_floats_succeedsr r r r r r r ry rh ri c C s< |\}}t |¡}tjtdd | ||¡ W 5 Q R X d S )Nzt'can only get slices from an IntervalIndex if bounds are non-overlapping and all monotonic increasing or decreasing'r& )r r( r- r. r/ r° )r0 r r r¬ ÚstoprD r4 r4 r5 Ú+test_slice_locs_with_ints_and_floats_errors s
|
ýz9TestSliceLocs.test_slice_locs_with_ints_and_floats_errorsN) rr rs rt r² r³ r- ru rv rµ r4 r4 r4 r5 r« ¬ s N$
|
ûþ
|
r« c @ s. e Zd Zej dddg¡dd Zdd ZdS )Ú TestPutmaskrP z
|
US/PacificNc C sv t dd|d}t |¡}tj|jtd}d|dd
< | ||d ¡}t|d gd t|dd
}t
|
||¡ d S ) Nú
|
2016-01-01r rO r{ Tr r r! © r r rZ rm ÚzerosÚshapeÚboolZputmaskrª ro Zassert_index_equal)r0 rP Údtir1 ÚmaskrE r] r4 r4 r5 Útest_putmask_dt64( s
|
"z TestPutmask.test_putmask_dt64c C s t ddd}||d }t |¡}tj|jtd}d|dd
< | ||d ¡}t|d gd t|dd
}t
|
||¡ d S ) Nr· r rM r r{ Tr r! r¸ )r0 r¼ Ztdir1 r½ rE r] r4 r4 r5 Útest_putmask_td644 s
|
"z TestPutmask.test_putmask_td64)rr rs rt r- ru rv r¾ r¿ r4 r4 r4 r5 r¶ ' s
|
r¶ c @ s e Zd Zdd ZdS )Ú TestContainsc C sÚ t jddgddgdd}d|ks$td|ks0td|ks<ttdddd|ksRttdddd|kshttdddd|ks~ttdddd|ksttd dd
|
d|ksªttddd
|
d|ksÀttddd d|ksÖtd S ) Nr r r r r r9 r rA r! r r )r r` r, r r± r4 r4 r5 Útest_contains_dunderD s z!TestContains.test_contains_dunderN)rr rs rt rÁ r4 r4 r4 r5 rÀ A s rÀ ) r) Únumpyrm r- Z pandas.errorsr Zpandasr r r r r r r r
|
r r r r r r r Zpandas._testingZ_testingro r rx r« r¶ rÀ r4 r4 r4 r5 Ú<module> s D & l{
|
