U
|
ý°dÆ% ã @ s¤ d Z ddlmZmZ ddlZddlZddlZddlmZm Z m
|
Z
|
m Z m Z ddl mZ dddZG dd d ZG d
|
d d Zd d Zdd Zdd ZG dd dZdS )z, test scalar indexing, including at and iat é )ÚdatetimeÚ timedeltaN)Ú DataFrameÚSeriesÚ TimedeltaÚ TimestampÚ
|
date_rangeFc C s" | j }|rdd |D }tj| S )zf
|
generate the indices
|
if values is True , use the axis values
|
is False, use the range
|
c s s | ]}t tt|V qd S )N)ÚlistÚrangeÚlen)Ú.0Zax© r úXd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexing/test_scalar.pyÚ <genexpr> s z#generate_indices.<locals>.<genexpr>)ÚaxesÚ itertoolsÚproduct)ÚfÚvaluesr r r r Úgenerate_indices s r c @ s e Zd Zej dddg¡ej dddg¡dd Zej dddg¡ej dd d
|
d g¡d d Zej dddg¡ej dddd d
|
d g¡dd ZdS )Ú
|
TestScalarÚkindÚseriesÚframeÚcolÚintsZuintsc C sL | | d| ¡}t|d}|D ]$}d|j|< |j| }t |d¡ q"d S )NÚ_Té )Úgetfixturevaluer Úiatr ÚtmÚassert_almost_equal©Úselfr r Úrequestr ÚindicesÚiÚexpectedr r r Útest_iat_set_ints# s
|
|
|
z TestScalar.test_iat_set_intsÚlabelsÚtsZfloatsc C sN | | d| ¡}d}tjt|d tt|d}d|j|< W 5 Q R X d S )Nr z1iAt based indexing can only have integer indexers©ÚmatchFr )r ÚpytestÚraisesÚ
|
ValueErrorÚnextr r )r# r r r$ r ÚmsgÚidxr r r Útest_iat_set_other- s
|
z TestScalar.test_iat_set_otherc C sL | | d| ¡}t|d}|D ]$}d|j|< |j| }t |d¡ q"d S )Nr Fr )r r ÚatÚlocr r! r" r r r Útest_at_set_ints_other6 s
|
|
|
z!TestScalar.test_at_set_ints_otherN) Ú__name__Ú
|
__module__Ú __qualname__r- ÚmarkÚ parametrizer( r3 r6 r r r r r " s r c @ s e Zd Zdd Zdd Zej deddgdd e dged
|
d gd d e
|
d gg¡d d Z dd Z dd Z dd Zdd Zdd Zdd ZdS ) Ú TestAtAndiATc C sh t dddgdddgd}| ¡ D ]\}}|j| |ks tq tt|D ]}|j| |d ksHtqHd S )Nr é é g¹?gÉ?g333333Ó?©Úindex)r Úitemsr4 ÚAssertionErrorr
|
r r )r# ÚserÚelÚitemr& r r r Útest_float_index_at_iatD s
|
z$TestAtAndiAT.test_float_index_at_iatc C s\ t ddd}ttj dd¡|ddddgd }|d }|j|d
|
}|jd
|
}||ksXtd S ) Nz1/1/2000é ©Zperiodsé ÚAÚBÚCÚD)r@ Úcolumnsé )r r ÚnpÚrandomÚrandnr4 r rB )r# ÚdatesÚdfÚsÚresultZxpr r r Útest_at_iat_coercionK s
|
z!TestAtAndiAT.test_at_iat_coercionz ser, expectedz
|
2014-01-01z
|
2014-02-02zdatetime64[ns]©Údtypez1 daysz2 daysztimedelta64[ns]c C s ||d }||kst d S )Nr )rB )r# Z indexer_ialrC r' rV r r r Ú test_iloc_iat_coercion_datelikeU s z,TestAtAndiAT.test_iloc_iat_coercion_datelikec C s t tddddddgdd}|jd }|dks2t|jd }|dksHtd}tjt|d |jd W 5 Q R X d
|
}tjt|d |jd W 5 Q R X |jddg }t ddgddgdd }t ||¡ |
|
¡ }|jd }t dd gdd}t ||¡ |jd }|dkstd S )NrO r r= r> Úint64)r@ rY z0index 10 is out of bounds for axis 0 with size 5r+ é
|
z1index -10 is out of bounds for axis 0 with size 5iöÿÿÿrX r ©r@ Úname)r= r ) r r
|
ÚilocrB r r- r. Ú
|
IndexErrorr Úassert_series_equalZto_frame)r# rU rV r1 r' rT r r r Útest_imethods_with_dupsh s(
|
|
|
|
z$TestAtAndiAT.test_imethods_with_dupsc C s t j d¡ dd¡}t|ddgd}|jd }|jd }t ||¡ |j jd }t ||¡ d|jd < t
|
d
|
d
|
gddgd d }t |jd |¡ d S ) Né r> r= rJ ©rN ©r rJ r )rJ r )r rJ g @r r] ) rP rQ rR Zreshaper r4 r_ r ra ÚTr )r# ZarrrT rV r' r r r Ú!test_frame_at_with_duplicate_axes
s
|
|
|
z.TestAtAndiAT.test_frame_at_with_duplicate_axesc
|
C s t ddgtddddddtddd d
|
d dgd }|d j d¡|d < tddd}|jd }||ksht|jd }||ks~td S )NZJohnZAndersoniá r> é é é8 r= é é r\ )r^ Údaterm z Asia/Shanghaiz2017-03-13 13:32:56+0800)Útz)r rm )r r ÚdtZ tz_localizer5 rB r4 )r# rT r' rV r r r Ú test_at_getitem_dt64tz_values s þþÿ
|
|
z*TestAtAndiAT.test_at_getitem_dt64tz_valuesc C sà t dddddgdddddgd }| ¡ D ].\}}|j| |j| krP|ks(n tq(tt|D ].}|j| |j| kr|d ksdn tqdt j
|
t d
|
d |jd W 5 Q R X t j
|
t d
|
d |jd W 5 Q R X d S ) Nr r= r> rI rO ÚaÚbÚcr? z^4$r+ ) r rA r4 r5 rB r
|
r r r_ r- r. ÚKeyError)r# rU rD rE r& r r r Ú'test_mixed_index_at_iat_loc_iloc_series¬ s (,z4TestAtAndiAT.test_mixed_index_at_iat_loc_iloc_seriesc C s t dddddgdddd d
|
ggd d d ddgd}| ¡ D ]H\}}| ¡ D ]6\}}|j||f |j||f krv|ksFn tqFq6tdD ]H}tdD ]:}|j||f |j||f krÈ|d | ksn tqqt j
|
t dd |jd W 5 Q R X t j
|
t dd |jd W 5 Q R X d S )Nr r r= r> rI rO rc é rG é rq rr rs rd z^3$r+ )r r> ) r ZiterrowsrA r4 r5 rB r
|
r r_ r- r. rt )r# rT ZrowIdxÚrowrD rE r& r r r Ú*test_mixed_index_at_iat_loc_iloc_dataframe¹ s ÿ2 :z7TestAtAndiAT.test_mixed_index_at_iat_loc_iloc_dataframec C sF t ddgddgd}d |jd< t d dgddgd}t ||¡ d S )Nr r rI rO )rq rr ©r r )r r r Zassert_frame_equal)r# rV r' r r r Ú'test_iat_setter_incompatible_assignmentË s
|
z4TestAtAndiAT.test_iat_setter_incompatible_assignmentN)r7 r8 r9 rF rW r- r: r; r r r rZ rb rg rp ru ry r{ r r r r r< A s&
|
þþûþ
|
r< c C s t ddd} | | }t| ¡ td}t| ¡ td}t||d}|jtk ¡ sTt|j d |j
|
d |j d |j d fD ]0}||d kstt |tstt |trxtqx|j d |j
|
d
|
|j d |j d
|
fD ]0}||d ksâtt |tsðtt |trÎtqÎd S ) Nz
|
2016-01-01r> rH rX )rJ rK re rz r )r rK )r r r )r r Z to_pydatetimeÚobjectZto_pytimedeltar ZdtypesÚallrB r4 r r5 r_ Ú
|
isinstancer r r r )ZdtiZtdirC Zser2rT rV r r r Ú&test_iat_dont_wrap_object_datetimelikeÓ s ((r c C sh t dddgiddgd} | jjdks(t| jd dks:t| d }|jjdksRt|jd dksdtd S )Nrq r r= ©r r= ©r> rI r? ©r rq ©r r@ ÚnlevelsrB r4 ©rT r r r r Ú test_at_with_tuple_index_getè s r c C s| t dddgiddgd} | jjdks(td| jd< | jd dksDt| d }|jjdks\td|jd< |jd dksxtd S ) Nrq r r= r r r? r r> r r
r r r Ú test_at_with_tuple_index_setõ s
|
|
r c @ s$ e Zd Zdd Zdd Zdd ZdS )ÚTestMultiIndexScalarc C s t dddgiddgddggd}|jjdks0t|jd dksBt|jd dksTt|d }|jjdkslt|jd dks~t|jd dkstd S ) Nrq r r= r> rI r? ©©r r> rq r ©r r@ r rB r4 r5 ©r# rT r r r r Útest_multiindex_at_get s z+TestMultiIndexScalar.test_multiindex_at_getc C s¼ t dddgiddgddggd}|jjdks0td|jd< |jd dksLtd|jd< |jd dksht|d }|jjdkstd|jd < |jd dkstd
|
|jd < |jd d
|
ks¸td S ) Nrq r r= r> rI r? r rO r rc r r r r r Útest_multiindex_at_set s
|
|
|
|
z+TestMultiIndexScalar.test_multiindex_at_setc C s, t dddggd}|jd }|dks(td S )N)r r FTr? r )r r4 rB )r# Ús2rV r r r Ú test_multiindex_at_get_one_level% s
|
z5TestMultiIndexScalar.test_multiindex_at_get_one_levelN)r7 r8 r9 r r r r r r r r s r )F)Ú__doc__r r r ÚnumpyrP r- Zpandasr r r r r Zpandas._testingZ_testingr r r r< r r r r r r r r Ú<module> s
|
|
