U
|
ý°d¯J ã @ sÎ d dl Zd dlZd dlZd dlmZmZ d dlmZ d dl
|
m Z G dd de Z G dd de Z G dd d e ZG d
|
d d ZG d d d e Zej dedd dd g¡dd Zdd Zdd Zdd ZdS )é N)ÚIndexÚSeries)Ú NumericBasec @ sv e Zd ZeZejejej gddd Z
|
ejdd Z ejdddd d
|
gd d d ddgd
|
d dddgddd d d ggddddgddd Z ejdd Z ejdd Zdd ZdBd d Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zd,d- Zej d.ed/dgejd0ed1d2ged0ed/dged0f¡d3d4 Z ej d5e j d6dd7e j d8dd7g¡d9d: Z d;d< Z!d=d> Z"d?d@ Z#dAS )CÚTestFloatNumericIndex©Úparamsc C s |j S ©N©Úparam©ÚselfÚrequest© r ú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/numeric/test_numeric.pyÚdtype s zTestFloatNumericIndex.dtypec C s t jd|d}| |¡S ©Né ©r )ÚnpÚarangeÚ
|
_index_cls)r r Úvaluesr r r Ú simple_index s z"TestFloatNumericIndex.simple_indexç ø?é é é r ç ç @ç @ç @ç $@ÚmixedÚfloatZ mixed_decZ float_dec©r Úidsc C s | j |j|dS ©Nr ©r r
|
©r r r r r r Úindex s
|
zTestFloatNumericIndex.indexc C s | j dddddg|dS )Nr r r r r r ©r ©r r r r r Ú mixed_index% s z!TestFloatNumericIndex.mixed_indexc C s | j dddddg|dS )Nr r r r r! r r* r+ r r r Ú float_index) s z!TestFloatNumericIndex.float_indexc C s t jtt||dd d S )NT©Úexact)ÚtmÚassert_index_equalÚevalÚrepr©r r) r r r Útest_repr_roundtrip- s z)TestFloatNumericIndex.test_repr_roundtripTc C sH | |¡sttj||dd |r4t|| jsDtnt|tksDtd S )NFr. )ÚequalsÚAssertionErrorr0 r1 Ú
|
isinstancer Útyper )r ÚaÚbÚis_float_indexr r r Ú check_coerce0 s
|
z"TestFloatNumericIndex.check_coercec C s$ | dddg¡}|jtjks td S )Nr r ç @)r r r Úfloat64r7 r4 r r r Ú#test_constructor_from_list_no_dtype8 s z9TestFloatNumericIndex.test_constructor_from_list_no_dtypec C sÎ | j }|dddddg|d}t||s*t|j|ks8ttjdddddg|d}t |j|¡ |t dddddg¡|d}t||st|j|kst|dddddg|d}t||sºt|j|ksÈt|t dddddg¡|d}t||sòt|j|kst|dddddg|d}t||s(t|j|ks8t|t dddddg¡|d}t||sdt|j|kstt|tj tj g|d}t
|
|j¡ ¡ st|t tj g¡|d}t
|
|j¡ ¡ sÊtd S )Né r r r r r ç ð?) r r8 r7 r r Úarrayr0 Úassert_numpy_array_equalr ÚnanÚpdZisnaÚall)r r Ú index_clsr) ÚexpectedÚresultr r r Útest_constructor< s2 z&TestFloatNumericIndex.test_constructorc C s< | j }|j}| d}tjt|d |d W 5 Q R X d S )NzI\(\.\.\.\) must be called with a collection of some kind, 0\.0 was passed©Úmatchr )r Ú__name__ÚpytestÚraisesÚ TypeError)r rH Zcls_nameÚmsgr r r Útest_constructor_invalidc s ÿz.TestFloatNumericIndex.test_constructor_invalidc C st | |tdddddg¡ | |tt d¡d ¡ ttjt d¡d td}|jtks^t| || d¡¡ d S ) Nr r r r r r r r? ) r= r r r rC Úobjectr r7 Úastype)r r, r- rJ r r r Útest_constructor_coercen s
|
z-TestFloatNumericIndex.test_constructor_coercec C sH | j |tt d¡d tddd | j |tdddd dgtddd d S )
|
Nr r r F)r< r r r r )r= r r r rT )r r, r- r r r Útest_constructor_explicitv s ÿ ÿz/TestFloatNumericIndex.test_constructor_explicitc C s4 d}t jt|d tdddg|d W 5 Q R X d S )Nú)Trying to coerce float values to integersrL rA r r> r )rO rP Ú
|
ValueErrorr )r Zany_int_numpy_dtyperR r r r Útest_type_coercion_fail s z-TestFloatNumericIndex.test_type_coercion_failc C s | j }|ddg}| |¡s t| |¡s.t|ddg}| |¡sHt|dtjg}| |¡sdt| |¡srt|dtjg}| |¡std S ©NrB ç @)r r6 r7 Ú identicalr rE )r rH ÚidxZidx2r r r Útest_equals_numeric
s z)TestFloatNumericIndex.test_equals_numericÚotherrA r rB r\ c C s. | ddg¡}| |¡s t| |¡s*td S r[ )r r6 r7 )r r` r^ r r r Ú$test_equals_numeric_other_index_type s z:TestFloatNumericIndex.test_equals_numeric_other_index_typeÚvalsz
|
2016-01-01)Zperiodsz1 Dayc C sP t |tddd}|j |¡|_|d }|d }t|t|rF||ksJt|d }t|t|rh||kslt|jd }t|t|r||kst|jd }t|t|r°||ks´t|jd }t|t|rÔ||ksØt|jd }t|t|rø||ksüt|j d }t|t|r ||ks$t|j
|
d }t|t|rH||ksLtd S )Nr é ©r) rA g @r ) r Úranger) rU r8 r9 r7 ÚlocÚatÚilocZiat)r rb r ZserrI rJ r r r Ú test_lookups_datetimelike_values£ s&
|
|
|
|
|
|
|
z6TestFloatNumericIndex.test_lookups_datetimelike_valuesc C sP | tjg¡}| dg¡ ¡ r"t| dg¡ ¡ r6t| tjg¡ ¡ sLtd S )Nr rA )r r rE ÚisinÚitemr7 )r r^ r r r Ú"test_doesnt_contain_all_the_thingsÈ s z8TestFloatNumericIndex.test_doesnt_contain_all_the_thingsc C sÆ | j }|dtjg}t | dg¡t ddg¡¡ t | dtjg¡t ddg¡¡ t | tjg¡t ddg¡¡ t | dtjg¡t ddg¡¡ |ddg}t | tjg¡t ddg¡¡ d S )NrB TFr\ )r r rE r0 rD rj rC Úpi)r rH r^ r r r Ú test_nan_multiple_containmentÎ s " " z3TestFloatNumericIndex.test_nan_multiple_containmentc C s | j }tdtjdgtdd}tdddgdd}tj| d¡|dd |dd dgdd}t | d
|
¡|¡ tdd dgdd}tj| d ¡|dd d S ) NrB g @Úx)r Únameg¹?)rp Tr. r\ r Úobj)r r r rE r# r0 r1 Zfillna)r rH r^ Úexpr r r Útest_fillna_float64Ù s z)TestFloatNumericIndex.test_fillna_float64N)T)$rN Ú
|
__module__Ú __qualname__r r rO Úfixturer r? Úfloat32r r r) r, r- r5 r= r@ rK rS rV rW rZ r_ ÚmarkÚ parametrizeÚint64rT ra rF Z
|
date_rangeZtimedelta_rangeri rl rn rs r r r r r sZ
|
|
ü
|
ù
|
|
|
|
' ýþ
|
þþ
|
r c @ sH e Zd ZeZdd Zdd Zdd Zdd Zd d
|
Z d d Z
|
d d Z dS )Ú
|
NumericIntc C sê | j }|ddddg}|jdks$t|jdks2t|jdks@t|jdksNt|jdks\t|ddddg}|jdkszt|jdkst|jdkst|dg}|jdks®t|jdks¼t|jdksÊt|jdksØt|jdksætd S )NrA r r r TF©r Zis_monotonic_increasingr7 Z!_is_strictly_monotonic_increasingZis_monotonic_decreasingZ!_is_strictly_monotonic_decreasing©r rH r) r r r Útest_is_monotonicí s"
|
z NumericInt.test_is_monotonicc C s | j }|ddddg}|jdks$t|jdks2t|ddddg}|jdksPt|jdks^t|ddg}|jstt|js~t|jrt|jrtd S )NrA r r TFr| r} r r r Útest_is_strictly_monotonic s
|
|
|
z%NumericInt.test_is_strictly_monotonicc C s4 |}| ¡ |j ¡ kst| ¡ |j ¡ ks0td S r )rG r r7 Úany)r r r^ r r r Útest_logical_compat s z NumericInt.test_logical_compatc C s° |}t | ¡ }| |¡s tt |td}| |¡r8t|jtd}| d¡}t |td}| |¡sht| |¡rvtt |dtd |¡st|jtd |j|d¡r¬td S )Nr Úfoo)rp r )r Úcopyr] r7 rT rU Úrename)r r r r) r^ Zsame_values_different_typeZ same_valuesr r r Útest_identical s
|
zNumericInt.test_identicalc C s: d}dddg}t jt|d | j||d W 5 Q R X d S )Nz/invalid literal for int\(\) with base 10: 'foo'r ÚbarZbazrL r )rO rP rY r )r r rR Údatar r r Útest_cant_or_shouldnt_cast, s
|
z%NumericInt.test_cant_or_shouldnt_castc C s |}| t¡ d S r )Úviewr )r r r) r r r Útest_view_index4 s zNumericInt.test_view_indexc C s" |}| d¡}|jtjks td S )NÚO)rU r r Zobject_r7 )r r r) rJ r r r Útest_prevent_casting8 s
|
z NumericInt.test_prevent_castingN) rN rt ru r r r~ r r r
r r r r r r r r{ ê s r{ c @ s¤ e Zd Zejejejejej gddd Z
|
ejdd Z eje ddde d d
|
d
|
gd d gd dd Z dd Zdd Zdd Zdd Zdd Zdd Zd d Zd S ) ÚTestIntNumericIndexr c C s |j S r r r r r r r ? s zTestIntNumericIndex.dtypec C s | j tddd|dS )Nr é r r )r re r+ r r r r C s z TestIntNumericIndex.simple_indexr r r é éÿÿÿÿÚ index_incÚ index_decr$ c C s | j |j|dS r& r' r( r r r r) G s zTestIntNumericIndex.indexc C s$ | dddg¡}|jtjks td S )NrA r r )r r r rz r7 r4 r r r r@ M s z7TestIntNumericIndex.test_constructor_from_list_no_dtypec C sz | j }|j d}tjt|d |d W 5 Q R X |ddddg|d}|j}||d d
|
}tj||d d |d d }||d< |d |kst|t j
|
krv|ddddg|d}t ddddg|d}t ||¡ |t ddddg|d}|ddddg|d}tj||d d |ddg|d}t |fD ]R} | ddg|d| t ddg¡|d| tddg|dfD ]}
|
t |
|
|¡ q^q"d S ) NzF\(\.\.\.\) must be called with a collection of some kind, 5 was passedrL r éûÿÿÿr rA r r T)r r. é¸ )r rN rO rP rQ r r0 r1 r7 r rz r ÚiterrC r ) r r rH rR r) ÚarrZ new_indexÚvalrI Úclsr^ r r r rK Q s4
|
ÿ ýz$TestIntNumericIndex.test_constructorc C s² | j }tjddddgtd}|||d}|jj|jks:t|tjkrnt|}|jtksZtt
|
|| tj¡¡ tjddddgtd}d}t j t|d |||d W 5 Q R X d S )
|
NrA r r r r Ú2Ú4rX rL )r r rC rT r r r7 rz r r0 r1 rU rO rP rY )r r rH r r) Z without_dtyperR r r r Útest_constructor_corner} s
|
z+TestIntNumericIndex.test_constructor_cornerc C s0 d}t jt|d tdg|d W 5 Q R X d S )Nz5Trying to coerce negative values to unsigned integersrL r r )rO rP Ú OverflowErrorr )r Ú any_unsigned_int_numpy_dtyperR r r r Ú,test_constructor_coercion_signed_to_unsigned s z@TestIntNumericIndex.test_constructor_coercion_signed_to_unsignedc C s< t |¡ d¡}t|g}tdg|d}tj||dd d S ©NrA r Tr. ©r r r9 r r0 r1 )r Zany_signed_int_numpy_dtypeÚscalarrJ rI r r r Útest_constructor_np_signed s
|
z.TestIntNumericIndex.test_constructor_np_signedc C s< t |¡ d¡}t|g}tdg|d}tj||dd d S r r )r r r¡ rJ rI r r r Ú test_constructor_np_unsigned£ s
|
z0TestIntNumericIndex.test_constructor_np_unsignedc C sH t ddddg}t|| js tt ddddgtd}t|t ksDtd S )NrA r r r r )r r8 r r7 rT r9 )r r r r r Útest_coerce_listª s z$TestIntNumericIndex.test_coerce_listN)rN rt ru rO rv r rz Úint32Úint16Úint8r r re r) r@ rK r r r¢ r£ r¤ r r r r r > s
|
|
ÿ
|
,
|
r c @ s e Zd ZeZdd ZdS )ÚTestFloat16Indexc C s¤ | j }tj}d}tjt|d |dddddg|d W 5 Q R X tjt|d" |t dddddg¡|d W 5 Q R X tjt|d |d ddddg|d W 5 Q R X tjt|d" |t d ddddg¡|d W 5 Q R X tjt|d |d ddddg|d W 5 Q R X tjt|d" |t d ddddg¡|d W 5 Q R X tjt|d |tjtjg|d W 5 Q R X tjt|d |t tjg¡|d W 5 Q R X d S )
|
Nz!float16 indexes are not supportedrL rA r r r r r rB )r r Úfloat16rO rP ÚNotImplementedErrorrC rE )r rH r rR r r r rK ¹ s& & & & z!TestFloat16Index.test_constructorN)rN rt ru r r rK r r r r r¨ ´ s r¨ c @ sd e Zd Zejejgddd Zejdd Zejdddd d
|
gd
|
d dddggd d gd dd Z dS )ÚTestUIntNumericIndexr c C s |j S r r r r r r r Û s zTestUIntNumericIndex.dtypec C s | tjd|d¡S r )r r r r+ r r r r ß s z!TestUIntNumericIndex.simple_indexl l
|
l l l r r r$ c C s | j |jtjdS r& )r r
|
r Úuint64r r r r r) ä s zTestUIntNumericIndex.indexN)
|
rN rt ru rO rv r r¬ r r r) r r r r r« Ú s
|
|
þûr« Úboxc C s t j| tdS r& )r rC rT ©ro r r r Ú<lambda>ñ ó r¯ c C s t | tdS r& )r rT r® r r r r¯ ñ r° c C sp t ddddddgdddd d
|
d gd }|j| ddg }tdddgd d}t |j|¡ t ||jd d
¡ d S )Nr rA r r r r l È@ï>"NL l '2w»@ l :Ûe/ l Ol_)#;f l Pl_)#;f rd r¬ r )r rf r r0 r1 r) Z assert_equalrh )r ZseriesrJ rI r r r Ú+test_uint_index_does_not_convert_to_float64ï s" úþ þr± c C sL t dddg} t dddg}| |¡}|dks2t| | ¡}|dksHtd S )NrB r r Ú1r Ú3F)r r6 r7 )r- Z string_indexrJ r r r Útest_float64_index_equals s
|
|
r´ c C sF t dddgtjd} | dd ¡}t ddd gtjd}t ||¡ d S )
|
NrA r r r c S s | S r r r® r r r r¯ r° z=test_map_dtype_inference_unsigned_to_signed.<locals>.<lambda>r éþÿÿÿéýÿÿÿ)r r r¬ Úmaprz r0 r1 ©r^ rJ rI r r r Ú+test_map_dtype_inference_unsigned_to_signed s r¹ c C sL t tjdddgtjd} | dd ¡}t ddd gtjd}t ||¡ d S )
|
NrA r r r c S s | d S )Néè r r® r r r r¯ $ r° z4test_map_dtype_inference_overflows.<locals>.<lambda>rº iÐ r )r r rC r§ r· rz r0 r1 r¸ r r r Ú"test_map_dtype_inference_overflows! s r» )Únumpyr rO ZpandasrF r r Zpandas._testingZ_testingr0 Zpandas.tests.indexes.commonr r r{ r r¨ r« rx ry Úlistr± r´ r¹ r» r r r r Ú<module> s& ^Tv&þ
|
|
