U
|
ý°d7X ã @ sÄ d dl Zd dlZd dlmZ d dlmZmZmZm Z m
|
Z
|
d dl m Z d dlmZ ejdd ZG dd dZG d d
|
d
|
ZG d d d ZG d d dZG dd dZG dd dZG dd dZdS )é N)ÚInvalidIndexError)ÚNAÚIndexÚ
|
RangeIndexÚSeriesÚ Timestamp)Ú FloatingArrayc C s dddddg} t | tjdS )Nì l
|
l l l ©Údtype)r ÚnpÚuint64)Zlarge© r úad:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/numeric/test_indexing.pyÚ index_large s r c @ s e Zd Zdd Zdd Zdd Zdd Zd d
|
Zej
|
d d gd ge dgdgg¡dd Z ej
|
ddddg¡dd Z dd ZdS )Ú
|
TestGetLocc C s$ t dddg}| d¡dks td S )Nr é é )r Úget_locÚAssertionError©ÚselfÚindexr r r Ú test_get_loc s zTestGetLoc.test_get_locc C s: t dddg}tjtdd | ddg¡ W 5 Q R X d S )Nr r r z\[1, 2\]©Úmatch)r ÚpytestÚraisesr r r r r r Ú test_get_loc_raises_bad_label s z(TestGetLoc.test_get_loc_raises_bad_labelc C s¨ t dddgtjd}tjtdd | d¡ W 5 Q R X tjtdd | d ¡ W 5 Q R X tjtd
|
d | d ¡ W 5 Q R X tjtd d | d ¡ W 5 Q R X d S )Nç ç ð?ç @r
|
z^'foo'$r Úfooz^1\.5$ç ø?z^True$Tz^False$F)r r Úfloat64r r ÚKeyErrorr ©r Úidxr r r Útest_get_loc_float64" s z TestGetLoc.test_get_loc_float64c C sì t tjddgtjd}| d¡dks(t| tj¡dks<tt tjdtjgtjd}| d¡dksftd}tjt|d | tj¡ W 5 Q R X t tjdtjtjgtjd}| d¡dks¾td}tjt|d | tj¡ W 5 Q R X d S )Nr r r
|
r z7'Cannot get left slice bound for non-unique label: nan'r z6'Cannot get left slice bound for non-unique label: nan)
|
r r Únanr$ r r r r r% Ú
|
slice_locs)r r' Úmsgr r r Útest_get_loc_na. s zTestGetLoc.test_get_loc_nac C s t ddgtjd}| d¡dks$ttjtdd | d¡ W 5 Q R X tjtdd | tj¡ W 5 Q R X tjt d d | tjg¡ W 5 Q R X d S )
|
Nr r r
|
r z^3$r é z^nan$z\[nan\])
|
r r r$ r r r r r% r) r r& r r r Útest_get_loc_missing_nanA s z#TestGetLoc.test_get_loc_missing_nanÚvalsr r z
|
2019-12-31Útestc C s2 t |}tjtdd | tj¡ W 5 Q R X d S )Nr) r )r r r r% r r r) )r r/ r' r r r Ú(test_get_loc_float_index_nan_with_methodM s z3TestGetLoc.test_get_loc_float_index_nan_with_methodr Zf8Úi8Úu8c C s> t jd|d}t|}tjtdd | d ¡ W 5 Q R X d S )Ni r
|
ÚNoner )r Úaranger r r r% r )r r Zarrr' r r r Ú%test_get_loc_numericindex_none_raisesT s z0TestGetLoc.test_get_loc_numericindex_none_raisesc C sv t dddg}t tj¡jd }tjtt|d | |¡ W 5 Q R X tjtt|d |j
|
|¡ W 5 Q R X d S )Nr r r r ) r r ZiinfoÚint64Úmaxr r r% Ústrr Z_engine)r r' Úvalr r r Útest_get_loc_overflows\ s z!TestGetLoc.test_get_loc_overflowsN)Ú__name__Ú
|
__module__Ú __qualname__r r r( r, r. r ÚmarkÚ parametrizer r1 r6 r; r r r r r s
|
|
r c @ s e Zd Zdd Zej dddg¡ej dejdddd d gej d
|
d fejdddd d gej d
|
d fejddd d d gej d
|
dfejddd d d gej d
|
dfg¡dd Z
|
dd Z ej dd ddddgdddgfdddddgdddgfdddddgdddgfd ddddgdddgfdddddgdddgfdddddgdddgfd ddddgdd d gfdddddgd d dgfdddddgdd dgfd d dddgdd d gfdd dddgd d dgfdd dddgdd dgfd ddddgdddgfdddddgdd dgfdddddgdd dgfg¡d d Z ej d e eeejg¡ej d e ed!d!d"gdd"d"gd"d#d#ggdd dgdddgdd dgg¡d$d% Zd&d' Zej d(d d d)dgfddd d gfddd)dgfg¡d*d+ Zej d,d-d.d/d0g¡ej d1d2d3g¡d4d5 Zej d1d ddg¡d6d7 Zd8d9 Zd:d; Zd<d= Zd>d? Zd@dA Zej dBdCdDdEefeefg¡dFdG Z dHdI Z dJdK Z ej dLdEd g¡dMdN Z ej dOdPdQg¡dRdS Z dS )TÚTestGetIndexerc C sP t dddddg}t dddg}| |¡}tjdddgtjd}t ||¡ d S ) Nr r r- é é é éÿÿÿÿr
|
)r Ú get_indexerr ÚarrayÚintpÚtmÚassert_almost_equal)r Úindex1Úindex2Zr1Úe1r r r Útest_get_indexeri s
|
|
z TestGetIndexer.test_get_indexerÚreverseTFzexpected,methodrE r r r
|
ÚpadZffillr ÚbackfillZbfillc C s^ t dddddg}t dddg}|r@|d d d
}|d d d
}|j||d}t ||¡ d S ) Nr r r- rB rC rD rE ©Úmethod)r rF rI rJ )r rO ÚexpectedrS rK rL Úresultr r r Útest_get_indexer_methodsq s z'TestGetIndexer.test_get_indexer_methodsc C sj t t d¡}tjtdd |jddgdd W 5 Q R X tjtdd |jddgdd W 5 Q R X d S ) Né
|
ztolerance argumentr r r )Ú toleranceúlimit argument)Úlimit©r r r5 r r Ú
|
ValueErrorrF r r r r Útest_get_indexer_invalid s
|
z'TestGetIndexer.test_get_indexer_invalidz$method, tolerance, indexer, expectedNrC é ÚnearestçÉ?çÍÌÌÌÌÌü?ç !@é c C s: t t d¡}|j|||d}t |tj|tjd¡ d S )NrW ©rS rX r
|
©r r r5 rF rI Úassert_numpy_array_equalrG rH )r rS rX ÚindexerrT r Úactualr r r Útest_get_indexer_nearest s z'TestGetIndexer.test_get_indexer_nearestÚlisttypeztolerance, expectedg333333Ó?g¹?ç à?c C sD t t d¡}|jdddgd||d}t |tj|tjd¡ d S )NrW r` ra rb r_ rd r
|
re )r rX rT rj r rh r r r Ú+test_get_indexer_nearest_listlike_toleranceª s ÿz:TestGetIndexer.test_get_indexer_nearest_listlike_tolerancec C st t t d¡}tjtdd |jddgddd W 5 Q R X tjtdd |jddgddd d
|
gd W 5 Q R X d S ) NrW rY r r r r_ )rS rZ ztolerance size must matchr r- rd r[ r r r r Ú test_get_indexer_nearest_error¾ s
|
z-TestGetIndexer.test_get_indexer_nearest_errorzmethod,expectedé c C sz t t d¡d d d
}|jdddg|d}t |tjdddgtjd¡ |jd d
|
d g|d}t |tj|tjd¡ d S ) NrW rE r rC r^ rR rB r
|
r` ra rb re )r rS rT r rh r r r Ú#test_get_indexer_nearest_decreasingÆ s
|
z2TestGetIndexer.test_get_indexer_nearest_decreasingÚ idx_dtyper7 r$ r ÚrangerS rF Úget_indexer_non_uniquec C s |dkrt d}nttjd|d}tdddg}t|||}tjdddgtjd}|dkrlt ||¡ n0tjdtjd}t |d |¡ t |d
|
|¡ d S ) Nrq rB r
|
TFrE rF r- r r ) r r r r5 ÚgetattrrG rH rI rf )r rS rp Z numeric_indexÚotherrU rT Úmissingr r r Ú-test_get_indexer_numeric_index_boolean_targetÓ s
|
z<TestGetIndexer.test_get_indexer_numeric_index_boolean_targetc C sn t dddg}t ddg}tjtdd |j||d W 5 Q R X tjtdd |j||d W 5 Q R X d S ) Nr r r- TFzCannot comparer rR )r r r Ú TypeErrorrF )r rS ÚleftÚrightr r r Ú,test_get_indexer_with_method_numeric_vs_boolè s z;TestGetIndexer.test_get_indexer_with_method_numeric_vs_boolc C sÞ t dddg}t ddg}| |¡}dtjt|tjd }t ||¡ | |¡}dtjt|tjd }t ||¡ | |¡d }dtjt|tjd }t ||¡ | |¡d }dtjt|tjd }t ||¡ d S ) Nr r r- TFrE r
|
r ) r rF r ZonesÚlenrH rI rf rr )r rx ry ÚresrT r r r Ú test_get_indexer_numeric_vs_booló s
|
|
z/TestGetIndexer.test_get_indexer_numeric_vs_boolc C s¸ t dddgtjd}t | |¡tjdddgtjd¡ dd d
|
g}t | |d ¡tjd ddgtjd¡ t | |d ¡tjdddgtjd¡ t | |d¡tjdddgtjd¡ d S )Nr r r! r
|
r r r g¹¿rk gñ?rP rE rQ r_ )r r r$ rI rf rF rG rH )r r' Útargetr r r Útest_get_indexer_float64 s$ ÿ
|
|
ÿ
|
ÿ
|
ÿz'TestGetIndexer.test_get_indexer_float64c C sB t ddtjgtjd tjg¡}tjdgtjd}t ||¡ d S )Nr r r
|
) r r r) r$ rF rG rH rI rf )r rU rT r r r Útest_get_indexer_nan s z#TestGetIndexer.test_get_indexer_nanc C s t tdddtjd}t t d¡tjd}| |¡}tjddddddddd dg
|
tjd}t ||¡ t t d¡tjd}|j|d
|
d }tjddddddddd d g
|
tjd}t ||¡ t t d¡tjd}|j|d d }tjdddddddd d d g
|
tjd}t ||¡ d S )Nr é r r
|
rW rE r r- rB rP rR rQ rC )
|
r rq r r7 r5 rF rG rH rI rf )r r r~ rg rT r r r Útest_get_indexer_int64 s
|
$ $ $z%TestGetIndexer.test_get_indexer_int64c C s t t d¡ d¡d d }| |¡}tjddddd d
|
ddddg
|
tjd }t ||¡ t t d¡ d¡d d }|j|d d }tjddddd d
|
d
|
d
|
d
|
d
|
g
|
tjd }t ||¡ t t d¡ d¡d d }|j|dd }tjddddd d
|
ddddg
|
tjd }t ||¡ d S )NrW r rC r r rE r r r- rB r
|
rP rR rQ ) r r r5 ZastyperF rG rH rI rf )r r r~ rg rT r r r Útest_get_indexer_uint64/ s
|
$ $ $z&TestGetIndexer.test_get_indexer_uint64z val, val2)rB rC )rB rB rB c C sT t ddd||g|d}| d¡}|dks,ttjtdd | d¡ W 5 Q R X d S )Nr r r- r
|
Ú9r r^ )r r r r r r% )r r: Zval2Ú any_numeric_ea_and_arrow_dtyper' rU r r r Útest_get_loc_masked? s
|
|
z"TestGetIndexer.test_get_loc_maskedc C s t ddtg|d}| t¡}|dks(tt ddttg|d}| t¡}t |t ddddg¡¡ t dddg|d}tj t
|
dd | t¡ W 5 Q R X d S ) Nr r r
|
FTr- r r ) r r r r rI rf r rG r r r% )r r
r' rU r r r Útest_get_loc_masked_naI s
|
|
z%TestGetIndexer.test_get_loc_masked_nac C s, t tt dddtjg¡t ddddg¡d}| t¡}|dksDt| tj¡}|dks\tt tt dddg¡t dddg¡d}| t¡}|dksttj t
|
dd | tj¡ W 5 Q R X t tt ddtjg¡t dddg¡d}| tj¡}|dksttj t
|
d
|
d | t¡ W 5 Q R X d S ) Nr r FT)Úmaskr- r r) r r ) r r r rG r) r r r r r r% )r r' rU r r r Ú test_get_loc_masked_na_and_nanW s6 ÿÿ
|
"ÿ
|
ÿÿ z-TestGetIndexer.test_get_loc_masked_na_and_nanr: c C sJ t ddtd|g|d}| dtdg¡}t dddg¡}tj||dd d S )
|
Nr r r- r
|
rC r rE F)Z check_dtype)r r Zget_indexer_forr rG rI rf )r r
r: r' rU rT r r r Útest_get_indexer_masked_nau s z)TestGetIndexer.test_get_indexer_masked_nar Úbooleanú bool[pyarrow]c C sT |dkrt d¡ tddtg|d}| d¡}|dks:t| t¡}|dksPtd S )Nr ZpyarrowTFr
|
r r )r Z importorskipr r r r )r r r' rU r r r Ú"test_get_indexer_masked_na_boolean} s
|
|
|
z1TestGetIndexer.test_get_indexer_masked_na_boolean)!r< r= r> rN r r? r@ r rG rH rV r] ri ÚlistÚtupler Úziprl rm ro rv rz r} r r r r r r r r r r r r r r rA h s üþ
|
ñþ
|
þÿþ
|
&þ
|
|
|
|
|
rA c
|
@ sf e Zd Zej deejdddee dddej
|
deejdddg¡d d
|
Z d d Z d d Z dS )Ú TestWherer rC r$ r
|
r r r r c C sx dgt | }|}| ||¡}dgdgt |d }t|jg|dd
¡ tjd}| ||¡}t ||¡ d S )NTFr r
|
) r{ Úwherer Z _na_valueÚtolistr r$ rI Úassert_index_equal)r Z listlike_boxr ÚcondrT rU r r r Ú
|
test_where s "zTestWhere.test_wherec C s t dddgtjd}t dddg¡}tjdgtjd}t dddgtjd}| ||¡}t ||¡ | | |¡}t ||¡ d S )Nr rD r r
|
FTr ) r r r rG r7 r rI r Zputmask)r r' r rt rT rU r r r Útest_where_uint64 s zTestWhere.test_where_uint64c C sN t dtjg}| ¡ }t ddgdd}t ddg}| ||¡}t ||¡ d S )Nr ÚaÚbÚstringr
|
r )r r r) Znotnar rI r )r r r rt rT rU r r r ÚCtest_where_infers_type_instead_of_trying_to_convert_string_to_float© s zMTestWhere.test_where_infers_type_instead_of_trying_to_convert_string_to_floatN)r< r= r> r r? r@ r r r5 rq r7 r r r r r r r r s ýþ
|
|
r c @ sT e Zd Zej dejejej g¡dd Z
|
dd Z ej dejej g¡dd Z d S )
|
ÚTestTakerp c C s: t ddddg|dd}| dddg¡}|j|jks6td S )Nr r r- rB r" ©r Únamer )r Útaker r )r rp r Ztakenr r r Útest_take_preserve_name¶ s z TestTake.test_take_preserve_namec C sl t dddgdtjd}| t dddg¡¡}t dddgtjdd }t ||¡ |jt dddg¡d
|
d }t ddtjgtjdd }t ||¡ |jt dddg¡d d
|
d }t dddgtjdd }t ||¡ d}tj t
|
|d |jt dddg¡d
|
d W 5 Q R X tj t
|
|d |jt dddg¡d
|
d W 5 Q R X d}tj t |d | t ddg¡¡ W 5 Q R X d S )Nr r! g @Úxxx)r r r r rE r T©Ú
|
fill_valueF©Z
|
allow_fillr£ zJWhen allow_fill=True and fill_value is not None, all indices must be >= -1r éþÿÿÿéûÿÿÿú3index -5 is out of bounds for (axis 0 with )?size 3) r r r$ r rG rI r r) r r r\ Ú
|
IndexError)r r' rU rT r+ r r r Ú test_take_fill_value_float64¼ s& ÿ$$z%TestTake.test_take_fill_value_float64r c C sn t dddg|dd}| t dddg¡¡}t dddg|dd}t ||¡ t|j}d| d }tj t
|
|d
|
|jt dddg¡d d W 5 Q R X |jt dddg¡d d d}t dddg|dd}t ||¡ tj t
|
|d
|
|jt dddg¡d d W 5 Q R X tj t
|
|d
|
|jt dddg¡d d W 5 Q R X d}tj t |d
|
| t ddg¡¡ W 5 Q R X d S )Nr r r- r¡ r r rE z Unable to fill values because z cannot contain NAr Tr¢ Fr¤ r¥ r¦ r§ ) r r r rG rI r Útyper< r r r\ r¨ )r r r' rU rT r r+ r r r Útest_take_fill_value_intsÚ s$
|
$ $$z"TestTake.test_take_fill_value_intsN) r< r= r> r r? r@ r r$ r7 r r r© r« r r r r r µ s
|
|
r c @ s> e Zd Zej dejejej g¡dd Z
|
dd Z dd Z dS ) Ú TestContainsr c C s& t dddddg|d}d |ks"td S )Nr r r r- rB r
|
)r r )r r r r r r Útest_contains_noneù s z TestContains.test_contains_nonec C s( t ddtjgtjd}tj|ks$td S ©Nr r! r
|
©r r r) r$ r r r r r Útest_contains_float64_nansÿ s z'TestContains.test_contains_float64_nansc C s& t ddtjgtjd}d|ks"td S r® r¯ r r r r Ú test_contains_float64_not_nans s z+TestContains.test_contains_float64_not_nansN) r< r= r> r r? r@ r r$ r7 r r r° r± r r r r r¬ ø s
|
r¬ c @ sj e Zd Zej deeg¡dd Zej deeg¡dd Z ej deeg¡dd Z
|
dd Z d
|
d Z d S ) Ú TestSliceLocsr c
|
C sè t tjddddddddg|d }t|}|jdd
|
d|fksBt|jd d
|
d |fksZt| d d ¡d ksnt| dd¡d |fkst|jd ddkst|jdddks®t|d d d
}| d d¡dksÐt| dd ¡dksätd S )Nr r r rC rD rn r^ rW r
|
©Ústartr- rc )r- rD ©Úend)r rD )r rn rE ©r rD )r rC ©r r rG r{ r* r ©r r r ÚnrL r r r Útest_slice_locs s "z TestSliceLocs.test_slice_locsc
|
C s t tjddddddddg|d }t|}| d
|
d ¡d |fksBt| d d¡dksVt|d d d
}| dd¡dksxt| dd¡d|fkstd S )Nr r r rC rD rn r^ rW r
|
g @g $@r- g @g %@)r- rc rE rb r# r· r¸ r¹ r r r Útest_slice_locs_float_locs s "z(TestSliceLocs.test_slice_locs_float_locsc C s| t tjddddg|d}| dd¡dks.t| dd¡dksBt|d d d
}| dd¡dksdt| dd¡dksxtd S ) NrW é é r
|
©r r- é é rE )r r rG r* r )r r r rL r r r Útest_slice_locs_dup_numeric% s z)TestSliceLocs.test_slice_locs_dup_numericc C sd t tjddg}| d¡dks"t| tj¡dks6tt dtjtjddg}| tj¡dks`td S )Nr r r¿ )r r- r )r rC )r r r) r* r r r r r Útest_slice_locs_na/ s
|
z TestSliceLocs.test_slice_locs_nac C s` t tjddg}tjtdd |jdd W 5 Q R X tjtdd |jdd W 5 Q R X d S )Nr r Ú r r# r³ rµ )r r r) r r r% r* r r r r Útest_slice_locs_na_raises7 s
|
z'TestSliceLocs.test_slice_locs_na_raisesN) r< r= r> r r? r@ ÚintÚfloatr» r¼ r rà rÅ r r r r r² s
|
|
|
|
r² c @ sR e Zd Zej dddg¡dd Zej dddg¡ej d d
|
d g¡d d ZdS )ÚTestGetSliceBoundszside, expected)rx rB )ry rC c C s* t td}|jd|d}||ks&td S )NrD rB ©Úside©r rq Zget_slice_boundr )r rÊ rT r rU r r r Ú test_get_slice_bounds_withinA s z/TestGetSliceBounds.test_get_slice_bounds_withinrÊ rx ry zbound, expected)rE r )rW rD c C s* t td}|j||d}||ks&td S )NrD rÉ rË )r rÊ rT Úboundr rU r r r Ú test_get_slice_bounds_outsideG s z0TestGetSliceBounds.test_get_slice_bounds_outsideN)r< r= r> r r? r@ rÌ rÎ r r r r rÈ @ s
|
|
rÈ )Únumpyr r Z pandas.errorsr Zpandasr r r r r Zpandas._testingZ_testingrI Zpandas.core.arraysr Zfixturer r rA r r r¬ r² rÈ r r r r Ú<module> s
|
P #,C8
|
