U
|
ý°dP ã @ sN d dl Zd dlZd dlmZmZmZmZ d dlm Z
|
dd Z G dd dZ dS )é N)Ú DataFrameÚIndexÚ
|
RangeIndexÚSeriesc C sD | t kr t t t|¡|d}n ttj t|t|¡||d}|S )N©Úindex)r Úcolumns)r ÚnpÚarangeÚlenr ÚrandomÚrandn)Úklassr Úobj© r úXd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexing/test_floats.pyÚgen_obj s ÿr c @ sÞ e Zd Zdd Zej dejej ej
|
ej ej g¡dd Z ej dejej ej
|
ej ej g¡dd Zdd Zej dejejg¡d
|
d Zej dejejg¡d d Zdd Zej dejej
|
ej ej g¡ej deddeddeddg¡dd Zdd Zej deddeddeddg¡dd Zej dejejg¡d d Zej deddeddeddg¡ej dejejg¡d d Zej deddeddeddg¡d!d" Zd#d$ Z d%d& Z d'd( Z d)S )*ÚTestFloatIndexersc C sH t |tr|j| }n"|r.|jdd
|f }n
|
|j| }t ||¡ dS )zw
|
comparator for results
|
we need to take care if we are indexing on a
|
Series or a frame
|
N)Ú
|
isinstancer ÚilocÚtmZassert_almost_equal)ÚselfÚresultÚoriginalÚindexerÚgetitemÚexpectedr r r Úcheck s
|
|
zTestFloatIndexers.checkÚ
|
index_funcc C s¼ |d}t ||}tjtdd ||d W 5 Q R X d|ksDt| ¡ }d||d< |tjkrvd|jd ks¸tnB|tj krd|jd ks¸tn$d|jd ks¦td|jd ks¸td S )Né ú^3.0$©Úmatchç @é
|
éÿÿÿÿr )
|
r ÚpytestÚraisesÚKeyErrorÚAssertionErrorÚcopyr ÚsetitemZaxesÚloc)r r Úframe_or_seriesÚ
|
indexer_slÚiÚsÚs2r r r Útest_scalar_non_numeric( s
|
|
|
z)TestFloatIndexers.test_scalar_non_numericc C sL |d}t t t|¡|d}|d tjtdd |d W 5 Q R X d S )Nr r é r r! r# )r r r
|
r r& r' r( )r r r/ r0 r r r Ú'test_scalar_non_numeric_series_fallbackL s
|
z9TestFloatIndexers.test_scalar_non_numeric_series_fallbackc C s( t dddgdddgd}t dddgdddgd}tjtd d
|
||d W 5 Q R X tjtd d
|
||d W 5 Q R X ||d }d}||ksttjtd d
|
||d W 5 Q R X |tjk rà|d }d}||ksàttjtd d
|
||d W 5 Q R X ||d }d}||ks$td S ) Né é r3 ÚaÚbÚcr ç ø?z^1.0$r! ç ð?z^1\.0$)r r& r' r( r) r r, )r r. r1 Zs3r r r r r Útest_scalar_with_mixed^ s(
|
z(TestFloatIndexers.test_scalar_with_mixedc C sÌ |t jk }|d}t||}||d }| ||d|¡ t|trPdd }d} n8t j}|rrtdtt|dd} ntdtt|dd} | ¡ }
|
d||
|
d< ||
|
d }||| ||
|
d }||| d S ) Nr r# r3 c S s | |ks t d S )N)r) )ÚxÚyr r r Úcompare s z6TestFloatIndexers.test_scalar_integer.<locals>.compareéd )r Únameg Y@)
|
r r, r r r r Úassert_series_equalÚranger r* ) r r r- r. r r/ r r r? r r1 r r r Útest_scalar_integer s$
|
|
|
|
z%TestFloatIndexers.test_scalar_integerc C s" |d}t ||}d|ks td S )Nr r# )r r) )r r r- r r r r r Ú"test_scalar_integer_contains_float¥ s
|
z4TestFloatIndexers.test_scalar_integer_contains_floatc
|
|
C sè t t d¡}t||}|d }tjtjfD ]t}|tjk }||| }| ||d|¡ | ¡ }||| }| ||d|¡ t j
|
t dd ||d W 5 Q R X q,d|ks®t |j d } | ¡ }| |j d< |j d }| ||dd¡ d S )Nç @r3 z^3\.5$r! ç @r# F)r r r
|
r r r, r+ r r* r& r' r( r) r )
|
r r- r r0 r Zidxrr r r1 r r r r Útest_scalar_float¯ s$
|
|
|
|
|
z#TestFloatIndexers.test_scalar_floatÚidxr# é r3 ç @c C s |d}t ||}|tjkr0dt|j d}ndt|j d}tjt|d ||| W 5 Q R X |tjkrvd}tjt|d d|||< W 5 Q R X d S ) Nr z!cannot do positional indexing on ú/ with these indexers \[(3|4)\.0\] of type floatú cannot do slice indexing on z8 with these indexers \[(3|4)(\.0)?\] of type (float|int)r! zBslice indices must be integers or None or have an __index__ methodr )r r r ÚtypeÚ__name__r& r' Ú TypeError)r r rI r- Z indexer_slir r0 Úmsgr r r Útest_slice_non_numericÒ s
|
|
ÿÿ
|
z(TestFloatIndexers.test_slice_non_numericc
|
C sæ t tjdtjddftddft tjdtjdd dffD ] \}}ttd|d}tddtd d
|
tdd
|
fD ]8}|j| }|rtd d }n
|
td d}| |||d¡ qttd d tddfD ]8}|j| }|ràtd d }n
|
td d }| |||d¡ qÂdt
|
|j d}t j t|d |tdd W 5 Q R X tddtd dftddtddftddtd dffD ]n\}}|j| }|rtd d } n|} | ||| d¡ dt
|
|j d}t j t|d || W 5 Q R X qpq>d S )Nr ©ÚdtypeFr$ Tr r# rJ r3 rK r iúÿÿÿé g Àg @rM z, with these indexers \[-6\.0\] of type floatr! ç @r6 rG z/ with these indexers \[(2|3)\.5\] of type float)r r r
|
Úint64r r rC Úslicer, r rN rO r& r' rP )
|
r r Zoobr0 rI r r rQ Zres1Úresr r r Útest_slice_integerü sD
|
ý
|
|
|
|
ÿý
|
ÿz$TestFloatIndexers.test_slice_integerr6 ç @c C s t tddtddd}|dd
}|jdd
}t ||¡ t}d|j d}tjt |d || W 5 Q R X tjt |d |j| W 5 Q R X dS ) z]make sure that we are raising on positional indexing
|
w.r.t. an integer index
|
r6 rU r rJ z)cannot do (slice|positional) indexing on z/ with these indexers \[(2|4)\.0\] of type floatr! N)
|
r rC r r rB r rO r& r' rP )r rI r0 r r r rQ r r r Ú test_integer_positional_indexingC s ÿz2TestFloatIndexers.test_integer_positional_indexingc
|
C s¤ |d}t tj dd¡|d}tddtddtddfD ]\}|j| }tdd}| |||d¡ d t|j d
|
}t j
|
t |d || W 5 Q R X q:td d tddfD ]$}|j| }| ||td d d¡ q¬d t|j d}t j
|
t |d |tdd W 5 Q R X tddtddftddtddftddtddffD ]X\}}|j| }| |||d¡ d t|j d}t j
|
t |d || W 5 Q R X qFd S )Nr r6 r ç r5 r r; FrM z/ with these indexers \[(0|1)\.0\] of type floatr! iöÿÿÿr$ g $Àç $@Tz- with these indexers \[-10\.0\] of type floatg à?r: z+ with these indexers \[0\.5\] of type float) r r r r rX r, r rN rO r& r' rP ) r r r r0 rI r r rQ rY r r r Ú test_slice_integer_frame_getitemY s6
|
|
ÿ
|
ÿý
|
ÿz2TestFloatIndexers.test_slice_integer_frame_getitemc C sª |d}t tj dd¡|d}| ¡ }d|j|< |j| j ¡ }|dk ¡ sPt dt
|
|j d}t j t|d d||< W 5 Q R X t j t|d || W 5 Q R X d S )Nr r6 r r rM rL r! )r r r r r* r, ÚvaluesÚravelÚallr) rN rO r& r' rP )r rI r r r0 Zscr rQ r r r Ú2test_float_slice_getitem_with_integer_index_raises s
|
ÿzDTestFloatIndexers.test_float_slice_getitem_with_integer_index_raisesc C s t t d¡d }t||}|jdd
}||| }t|t|sHtt ||¡ |
|
¡ }d|||< ||| j ¡ }|dk ¡ std S )NrF g¹?r3 rJ r )r r r
|
r r r rN r) r Z assert_equalr* r` ra rb ) r rI r- r. r r0 r r r1 r r r Útest_slice_float© s
|
z"TestFloatIndexers.test_slice_floatc C sZ t dddddg}ttd|d}|d dks2t|jd dksDt|jd dksVtd S )Nr: r6 r3 g @r r )r r rC r) r, r )r r r0 r r r Ú test_floating_index_doc_example¼ s
|
z1TestFloatIndexers.test_floating_index_doc_examplec
|
C s¬ t t d¡t d¡d tjd}||dd
}t ddgd}t ||¡ ||d }|d ksbt||d }|d kszttjt d
|
d ||d W 5 Q R X t d d gt
|
ddgtj dd}ddgt ddg¡fD ]}t ||| |¡ qÔt d d gt
|
dd gddd}dd gt dd g¡fD ]}t ||| |¡ q ||d d
}||dd
}||dd
}||dd
} t ||¡ t ||¡ t || ¡ t dd gddgd}||d d
}t ||¡ ||dddg }|j d d d g }t ||¡ tjt dd ||dddg W 5 Q R X tjt dd ||d dd g W 5 Q R X ||ddg }t |t dd gddgd¡ ||dg }t |t dgdgd¡ d S )Nr rV )r rT r; r# r5 r rF r6 z^4$r! rJ r r] rS Úfloat64r[ gÍÌÌÌÌÌ @r$ z not in indexgù?)r r r
|
rW r rB r) r& r' r( r rf Úarrayr )
|
r r. r0 r r Z fancy_idxZresult1Zresult2Zresult3Zresult4r r r Útest_floating_miscà sJ z$TestFloatIndexers.test_floating_miscc C sf |}ddddddddddddddd dd
|
dd dddd dd dddddddddddddddddddddddddddddddd dd dddd d d dd!dd"dd#dddd$dd%dd&dd'dddd(dd)d*d+dd,d-ddd.dd/dd0dd1dd2dddddddddddddddddddddddd3d4d5dd6ddddd7dd8d9dddd:d;dd<dd=dd>}t ||d?}| ¡ }|jj|ksZtt| d S )@Ng "¢@g S@g 8¥@r] g w@g Ðp@g «@g 8@g @g @Y@g ÕÂ@g "ï@g à{@g úà@g È@g \ª@g ¼@g v@g .î@g º@g @g ¥@g ü«@g ð@g @g Q@g 0¤@g @t@g è@g *¯@r^ g E@g @@g V@g ì¿@g À°ï@g °u@g ßá@g °w@g r§@g x@g @g Àv@g Àtð@g "@g ¾@g È@g ¦@g &§@g °@g ð@g 6@g p@g h@g ¸@g ¤@g 8@g `d@g +Ä@g 0@g ¢¥@)é r5 r6 r3 rJ r rU é é é r$ é é é é é é é é é é é é é é é é é é é é é é é! é" é# é$ é% é& é' é( é) é* é+ é, é- é. é/ é0 é1 é2 é3 é4 é5 é6 é7 é8 é9 é: é; é< é= é> é? é@ éA éB éC éD éE éF éG éH éI éJ éK éL éM éN éO éP éQ éR éS éT éU éV éW éX éY éZ é[ é\ é] é^ é_ é` éa éb éc r@ ée éf ég éh éi éj ék él ém én éo ép éq ér és ét éu év éw éx éy éz é{ é| é} é~ é é é i i i i i i
|
i i i i i i i éú éû éü éý éþ éÿ rS )r Z value_countsr rT r) Ústr)r Zfloat_numpy_dtyperT Zserr0 r r r r Útest_floatindex_slicing_bug s: ë z-TestFloatIndexers.test_floatindex_slicing_bugN) rO Ú
|
__module__Ú __qualname__r r& ÚmarkZ parametrizer ZmakeStringIndexZmakeCategoricalIndexZ makeDateIndexZmakeTimedeltaIndexZmakePeriodIndexr2 r4 r< Z makeIntIndexZmakeRangeIndexrD rE rH rX rR rZ r\ r_ rc rd re rh rê r r r r r sd ûþ
|
|
ûþ
|
|
"
|
$
|
#üþ $ G$
|
|
5$ $
|
Ar ) Únumpyr r& Zpandasr r r r Zpandas._testingZ_testingr r r r r r r Ú<module> s
|
|
