U
|
ý°dP ã @ sd d dl Zd dlZd dlmZ d dlZd dlmZmZ d dl m
|
Z d dl m Z eZG dd de ZdS )é N)Úensure_platform_int)ÚIndexÚ
|
RangeIndex)Ú NumericBasec @ s4 e Zd ZeZejdd Zejdddddgdd d
|
Zejd d Z ejed ddddedddddgddgddd Z
|
dd Z d d Z d d Z ej d!ed"d d"d#fed d"d d"d#fed"dd$d d"dfed#d"dd#d"dfg¡d%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Zd3d4 Zd5d6 Zd7d8 Zd9d: Zd;d< Zd=d> Z d?d@ Z dAdB Z dCdD Z dEdF Z dGdH Z!dIdJ Z"dKdL Z#dMdN Z$ej dOdPdQg¡dRdS Z%dTdU Z&dVdW Z'dXdY Z(dZd[ Z)d\d] Z*d^d_ Z+d`da Z,dbdc Z-ddde Z.ej dfe/e0dgdhd h ¡didj Z1eje2d#dkd"ge2d#dkd"fe2d dhdlge2d dhdlfe2d#dme2dmdnge2d#dnfe2d#d"de2d"dhge2d#dhdfe2d#dmde2dldndmge2d#dndmfe2dodmde2dldndge2dodndfe2dodpe2dpdqge2d d fe2dodpe2dmdoge2d d fe2dodpe2dmd"ge2dmd"fe2dode2dmd"ge3dodrdmdlgfe2de2dmd"ge2dmd"fe2de2dge3d d#d d#gfe2de2dd"e2d"dsdlge2d dhfe2de2dmd"e2d"dsdlge3d d#dmdld"gfe2dde2dd"e2d"dsdlge2ddhfe2dme3ddmdtgge3d d#dddmdtgfe2dme3ddudvgge3d d#dddudvgfe2dme3dwdxdygge3d d#ddwdxdygfe2dmd#e3dwdxdygge3dwdxdygfgddzd{ Z4d|d} Z5d~d Z6dd Z7ej ded dded d#d#ed#dmded ddedrdgdg¡dd Z8ej d
ed de3d d#gg¡dd Z9dd Z:dd Z;dxS )ÚTestRangeIndexc C s t jS ©N)ÚnpÚint64©Úself© r ú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/ranges/test_range.pyÚdtype s zTestRangeIndex.dtypeZuint64Úfloat64ÚcategoryZ
|
datetime64Úobject)Úparamsc C s |j S r ©Úparam©r Úrequestr r r Ú invalid_dtype s z TestRangeIndex.invalid_dtypec C s | j ddddS )Nr é é )ÚstartÚstopÚstep)Ú
|
_index_clsr
|
r r r Ú simple_index s zTestRangeIndex.simple_indexr r r Úfoo)r r r Únameé éÿÿÿÿéþÿÿÿÚbarZ index_incZ index_dec)r Úidsc C s |j S r r r r r r Úindex# s zTestRangeIndex.indexc C s0 | dd¡}tjddg|d}t |j|¡ d S )Né é r ©r )r r ÚarrayÚtmÚassert_numpy_array_equalÚ_data)r r ÚresultÚexpectedr r r Ú test_constructor_unwraps_index- s z-TestRangeIndex.test_constructor_unwraps_indexc C s" |}|d }| |¡dks td S )Nr F)Z$_can_hold_identifiers_and_holds_nameÚAssertionError)r r ÚidxÚkeyr r r Útest_can_hold_identifiers2 s z(TestRangeIndex.test_can_hold_identifiersc C s, |}t jtdd ddg|_W 5 Q R X d S )Nz^Length©ÚmatchZrogerZharold)ÚpytestÚraisesÚ
|
ValueErrorÚnames©r r r& r r r Útest_too_many_names7 s z"TestRangeIndex.test_too_many_nameszindex, start, stop, stepé r' )r c C s. |j |kst|j|ks t|j|ks*td S r )r r1 r r )r r& r r r r r r Útest_start_stop_step_attrs< s z)TestRangeIndex.test_start_stop_step_attrsc C sV t ddd}| ¡ }||k s t| |¡s.t|jtdddksDt|jdksRtd S )Nr= ÚFoo©r r r' )r Úcopyr1 Ú identicalÚ_rangeÚranger )r ÚiZi_copyr r r Ú test_copyK s zTestRangeIndex.test_copyc C s| t ddd}t|}d}||ks$tt|}tj||dd t ddd}t|}d }||ks`tt|}tj||dd d S )
|
Nr= r? r@ z/RangeIndex(start=0, stop=5, step=1, name='Foo')T©Úexactr r" z$RangeIndex(start=5, stop=0, step=-1))r Úreprr1 Úevalr+ Úassert_index_equal)r rE r. r/ r r r Ú test_reprS s zTestRangeIndex.test_reprc C sÄ t ddd}|dd
}tj|dd
| d|d ¡dd tdtjdd d
|
dgtjd }tjd tj fD ] }t d d|¡}t ||¡ qft d dtj
|
¡}tdtj
|
dd d
|
dgt d }t ||¡ d S ) Nr= r? r@ r' é r ÚequivrG r r( r) ) r r+ rK Úinsertr r Únanr ÚpdZNAZNaTr )r r2 r. r/ Únar r r Ú test_insertd s $ zTestRangeIndex.test_insertc C sl t tddd}| dd¡}t tddd}tj||dd | dd¡}t tdd d}tj||dd d S )
|
NrM é r r TrG r( é
|
é ©r rD rO r+ rK ©r r2 r. r/ r r r Ú&test_insert_edges_preserves_rangeindexu s z5TestRangeIndex.test_insert_edges_preserves_rangeindexc C sh t tddd}| dd¡}t td}tj||dd |d }| dd¡}|d }tj||dd d S )Nr r( r r' TrG rW rX r r r Ú'test_insert_middle_preserves_rangeindex s z6TestRangeIndex.test_insert_middle_preserves_rangeindexc C s¬ t ddd}|dd
}| d¡}tj||dd |j|jksBt|d d
}| d¡}tj||dd |j|jksxtd }tjtt f|d
|
| t
|
|¡}W 5 Q R X d S ) Nr= r? r@ r' r TrG r" z/index 5 is out of bounds for axis 0 with size 5r5 ) r Údeleter+ rK r r1 r7 r8 Ú
|
IndexErrorr9 Úlen)r r2 r/ r. Úmsgr r r Ú test_delete s
|
|
zTestRangeIndex.test_deletec C sZ t tddd}| dg¡}t tddd}tj||dd | d¡}tj||dd d S )Nr r r@ r' TrG ©r rD r[ r+ rK rX r r r Ú test_delete_preserves_rangeindex s
|
z/TestRangeIndex.test_delete_preserves_rangeindexc C sV t tddd}| d¡}|d d d
}tj||dd | d¡}tj||dd d S ) Nr( r r@ r' r TrG r# r` rX r r r Ú'test_delete_preserves_rangeindex_middle§ s
|
|
z6TestRangeIndex.test_delete_preserves_rangeindex_middlec C sf t ddd}ddddg}| |¡}|d d
}tj||dd | |d d d
|
¡}tj||dd d S © Nr é r' r r( rM r= TrG r" ©r r[ r+ rK ©r r2 Úlocr. r/ r r r Ú,test_delete_preserves_rangeindex_list_at_end° s
|
z;TestRangeIndex.test_delete_preserves_rangeindex_list_at_endc C sf t ddd}ddddg}| |¡}t ddd}tj||dd | |d d d
|
¡}tj||dd d S rc re rf r r r Ú,test_delete_preserves_rangeindex_list_middle» s
|
z;TestRangeIndex.test_delete_preserves_rangeindex_list_middlec C sj t ddd}ddddddg}| |¡}|d d
}tj||dd | |d d d
|
¡}tj||dd d S rc re rf r r r Ú$test_delete_all_preserves_rangeindexÆ s
|
z3TestRangeIndex.test_delete_all_preserves_rangeindexc C sf t ddd}dddg}| |¡}tdddg}tj||dd | |d d d
|
¡}tj||dd d S ) Nr rd r' r( r= r rM TrG r" )r r[ r r+ rK rf r r r Ú%test_delete_not_preserving_rangeindexÑ s
|
|
z4TestRangeIndex.test_delete_not_preserving_rangeindexc C sT t ddd}| ¡ }|jdks"t| d¡}t |j|¡ | t ¡}t ||¡ d S )Nr r? r@ Úi8)r Úviewr r1 r+ r, ÚvaluesrK )r rE Zi_viewr r r Ú test_viewÜ s
|
|
zTestRangeIndex.test_viewc C s |}|j tjkstd S r )r r r r1 r; r r r Ú
|
test_dtypeç s zTestRangeIndex.test_dtypec C s t ddd}|ji kstt| |ji ks0tt| |ji ksFt| d¡ |ji ks^td|k |ji ksttd|k |ji kst| ¡ |ji ks t| ¡ |ji ks¶t|D ]}qº|ji ksÎt| ¡ |ji ksätt j
|
dt di|d}t| |ji kst|j d |ji ks,tt jtd
|
d |j d W 5 Q R X |ji ks`t|j dd
|ji ks~t|jd d
|ji kst|j t|jtjs¶t|j|jksÈtt|jdksÜttjddddd}t |jd |¡ d S )Nr éd rU r éZ é[ Úa)r& é2 Z51r5 é3 r= r' r r) r- )r Ú_cacher1 rI ÚstrÚget_locÚallÚanyÚformatrQ Z DataFramerD rg r7 r8 ÚKeyErrorZilocr- Ú
|
isinstancer Zndarrayr] Úaranger+ r, )r r2 Ú_Zdfr/ r r r Ú
|
test_cacheë sN
|
|
zTestRangeIndex.test_cachec C s t ddd}|jdkst|jdks(t|jdks6t|jdksDt|jdksRtt ddd}|jdkslt|jdkszt|jdkst|jdkstt dd}|jdks®t|jdks¼t|jdksÊt|jdksØt|jdksætt dd}|jdksþt|jdkst|jdks t|jdks.t|jdks>tt dd}|jdksXt|jdksht|jdksxt|jdkst|jdkstd S ) Nr r r TFrM r" r' )r Zis_monotonic_increasingr1 Zis_monotonic_decreasingZ!_is_strictly_monotonic_increasingZ!_is_strictly_monotonic_decreasing©r r& r r r Útest_is_monotonic* s:
|
|
|
z TestRangeIndex.test_is_monotonicc C s t dddt dddft dt dddft dddt dddft dd d
|
t dd d
|
fg}|D ]$\}}| |¡srt| |¡s\tq\d S ) Nr rT r rU r' r" r( rM i÷ÿÿÿr# iöÿÿÿ)r Úequalsr1 )r Z equiv_pairsÚleftÚrightr r r Útest_equals_rangeM s ü z TestRangeIndex.test_equals_rangec C s4 |}| ¡ |j ¡ kst| ¡ |j ¡ ks0td S r )rz rn r1 r{ )r r r2 r r r Útest_logical_compatX s z"TestRangeIndex.test_logical_compatc C sÆ |}t | ¡ }| |¡s tt|tr,d S t |td}| |¡rFt|jtd}| d¡}t |td}| |jtd¡s~t| |¡rtt |dtd |¡s¤t|jtd |jdd¡rÂtd S )Nr) r )r r r )r rA rB r1 r~ r r Úrename)r r r& rE Zsame_values_different_typeZ same_valuesr r r Útest_identical] s
|
|
z TestRangeIndex.test_identicalc C sB t dd}|jt|jjd k s$tt dd}|j|jks>td S )Nr iè rU )r Únbytesr Z_valuesr1 )r r2 Úi2r r r Ú test_nbytess s
|
|
zTestRangeIndex.test_nbyteszstart,stop,step)r r$ Zbaz)Ú0Ú1Ú2c C s> dt | d| }tjt|d t||| W 5 Q R X d S )Nz Wrong type z for value r5 )Útyper7 r8 Ú TypeErrorr )r r r r r^ r r r Útest_cant_or_shouldnt_cast| s
|
z)TestRangeIndex.test_cant_or_shouldnt_castc C s |}| t¡ d S r )rm r r; r r r Útest_view_index s z TestRangeIndex.test_view_indexc C s" |}| d¡}|jtjks td S )NÚO)Zastyper r Zobject_r1 ©r r r& r. r r r Útest_prevent_casting s
|
z#TestRangeIndex.test_prevent_castingc C s |}t tt||¡ d S r )r+ rK rJ rI r; r r r Útest_repr_roundtrip s z"TestRangeIndex.test_repr_roundtripc C s* t dddd}|j|dd
jks&td S )Nr' r Zasdfr@ )r r r1 ©r r2 r r r Útest_slice_keep_name s z#TestRangeIndex.test_slice_keep_namec C s |j s
|
t|jrtd S r )Z is_uniquer1 Zhas_duplicatesr r r r Útest_has_duplicates s
|
z"TestRangeIndex.test_has_duplicatesc C s |}| dd¡}|d |d d |d d ks4td|d ksDt| dd¡}d|d d |d d ksptd|d kstd S )Nrd rU r r' r )Z _extended_gcdr1 r r r r Útest_extended_gcd s $ z TestRangeIndex.test_extended_gcdc C s t ddd d¡}d|ks tt dd d¡}d|ks:tt ddd d¡}d|ksXtt ddd d¡}d|ksvtd
|
}t d|d d |¡}||kstd S ) Nr r r r' rd r! r# r= r" l !Û;o )r Z_min_fitting_elementr1 )r r. Zbig_numr r r Útest_min_fitting_element© s z'TestRangeIndex.test_min_fitting_elementc C s d S r r r
|
r r r Ú test_pickle_compat_construction» s z.TestRangeIndex.test_pickle_compat_constructionc C sì |}d|_ |d }d}||ks"t|d }d}||ks:t|d d
}|}t ||¡ |ddd
}tddgdd }tj||d
|
d |dd d
}tddgdd }tj||d
|
d |ddd
}tddgdd }tj||d
|
d |d d d
}t|jd d d
dd }tj||d
|
d |dd d
}tdddgdd }tj||d
|
d |dd d
}ttjg tjddd }tj||d
|
d |dd d
}t|jdd d
dd }tj||d
|
d |dd d
}t|jd d d
dd }tj||d
|
d d S )Nr r' r r" r! é rU é r@ rN rG éûÿÿÿr# rq rM é éøÿÿÿr iØÿÿÿr) é( ) r r1 r+ rK r rn r r* r )r r r& Úresr/ Z index_slicer r r Útest_slice_specialised¿ sF z%TestRangeIndex.test_slice_specialisedr r¡ rd c C sb |dkr dnd\}}t |||¡}t|||}t|t|ksBtt|||}t|dks^td S )Nr )r r= )r= r )r r r r] r1 )r r r r Zarrr& r r r Útest_len_specialisedö s z#TestRangeIndex.test_len_specialisedr¢ rM r( r éüÿÿÿr£ iôÿÿÿéýÿÿÿé é gÍÌÌÌÌÌ@g .@rt Nr c C s |j S )z6Inputs and expected outputs for RangeIndex.append testr r r r r Úappends s zTestRangeIndex.appendsc C s` |\}}|d |dd
¡}tj||dd t|dkr\|d |d ¡}tj||dd d S )Nr r' TrG r )Úappendr+ rK r] )r r¬ Úindicesr/ r. Zresult2r r r Ú test_append s zTestRangeIndex.test_appendc C sª t ddd}| d¡dks tt | ddg¡tt ddg¡¡ t j
|
t dd | d¡ W 5 Q R X d
|
|j ksttt j
|
t d d | d ¡ W 5 Q R X d
|
|j ks¦td S ) Nr rU r( r= r' rª r Ú3r5 Z_enginez'a'rt ) r ry r1 r+ r, Z get_indexerr r r* r7 r8 r} rw r r r r Útest_engineless_lookup* s ÿz%TestRangeIndex.test_engineless_lookupc C s4 | d¡}| ¡ g kst|jdddgks0td S )Nr Tr@ Ú )r r| r1 )r Z empty_idxr r r Útest_format_empty@ s
|
z TestRangeIndex.test_format_emptyÚRIc C s | g ¡}tj||dd d S )NTrG )r r+ rK )r r´ r. r r r Útest_append_len_oneF s
|
z"TestRangeIndex.test_append_len_oneÚbasec C s2 t dd}| |¡}t ddg¡}t ||¡ d S )Nr r' TF)r Úisinr r* r+ r, )r r¶ rn r. r/ r r r Útest_isin_rangeU s
|
|
z TestRangeIndex.test_isin_rangec sZ dddddd t ddd}|j fddd }td
|
dd ddgd d }tj||dd d S )Nr r rª rU r¢ )rª rd rM r r c s
|
| ¡S r )Úmap)Úx©Z
|
sort_orderr r Ú<lambda>a ó z5TestRangeIndex.test_sort_values_key.<locals>.<lambda>)r3 rM rd r r) T)Z check_exact)r Z sort_valuesr r+ rK )r rn r. r/ r r» r Útest_sort_values_key] s
|
z#TestRangeIndex.test_sort_values_keyc C s t d¡ d S )Nz.casting of strings not relevant for RangeIndex)r7 Úskip)r r r r r Útest_cast_stringe s z TestRangeIndex.test_cast_string)<Ú__name__Ú
|
__module__Ú __qualname__r r r7 Zfixturer r r r& r0 r4 r< ÚmarkZ parametrizer> rF rL rS rY rZ r_ ra rb rh ri rj rk ro rp r r r r r r r r r r r r r r r r¦ ÚsetrD r§ r´ r r¬ r¯ r± r³ rµ r¸ r¾ rÀ r r r r r sÔ
|
ÿ
|
|
þû
|
üþ
|
|
?# üþ
|
|
7
|
"" " $,&((($íÿ
|
|
|
|
|
|
ûþ
|
|
|
r )Únumpyr r7 Zpandas.core.dtypes.commonr ZpandasrQ r r Zpandas._testingZ_testingr+ Zpandas.tests.indexes.commonr r´ r r r r r Ú<module> s
|
