U
|
ý°dÖ ã @ s¨ d dl m Z mZ d dlZd dlZd dlmZ d dlmZm Z m
|
Z
|
m Z m Z m Z mZ d dlmZ dd Zdd ZG d d
|
d
|
ZG d d d ZG d d dZG dd dZdS )é )ÚdatetimeÚtimezoneN)ÚInvalidIndexError)ÚCategoricalDtypeÚCategoricalIndexÚ DataFrameÚ DatetimeIndexÚ
|
MultiIndexÚSeriesÚ Timestampc C s^ t dtdddgi} tdddtjd| jd< t dtdddtjdgitd}t | |¡ d S )NÚfooiÐ é é )Útzinfo)r r ©Údtype)r r r ÚutcÚatÚobjectÚtmÚassert_frame_equal)ÚresultÚexpected© r úTd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexing/test_at.pyÚtest_at_timezone s ÿr c C sè t dddgdddgdd} t d d
|
d d giddd gd}|d | d < d| jd< | }t dddgdddgd dd
|
gd}t ||¡ | jd }|dkst| d }td dd
|
gd d}t ||¡ | d g }t d d dd
|
gi}t ||¡ d S )Nr r é é é é ©ÚaÚb©ÚdataÚcé é é r ©r$ Úindexé )r r% )r! r" r% )Úname)r r r r ÚAssertionErrorr
|
Úassert_series_equal)ÚdfZdf2r r r r r Ú&test_selection_methods_of_assigned_col! s
|
"
|
|
r0 c @ sJ e Zd Zdd Zdd Zdd Zdd Zej d e
|
d
|
d
|
f¡d d Z d S )Ú TestAtSetItemc C s~ t dgd}d|d< d|d< |d |jdg d|jd< d |jd
|
< t dgd d dgd}t ||¡ t |d |d ¡ d S ) Nr ©r* r Úxr Úcostr )r r3 i )r r4 )r3 r4 )r Úlocr r r r. ©Úselfr/ r r r r Ú"test_at_setitem_item_cache_cleared7 s
|
|
z0TestAtSetItem.test_at_setitem_item_cache_clearedc C s\ t dddddgdddddgd }d
|
|jd< |jd d
|
ks<td |jd< |jd d ksXtd S ) Nr r r r r r! r" r% r2 r+ r é )r
|
r Ziatr- ©r7 Úserr r r Ú&test_at_setitem_mixed_index_assignmentN s
|
|
|
z4TestAtSetItem.test_at_setitem_mixed_index_assignmentc C st t tdtdtddgd}d|jd< t tjtjtjgtjdtjgtjtjtjggtddgd}t ||¡ d S )Nr r Úbar)r* Úcolumnsr )r r r )r Úranger r ÚnpÚnanr r r6 r r r Ú#test_at_setitem_categorical_missingV s
|
ÿ
|
ý
|
ú z1TestAtSetItem.test_at_setitem_categorical_missingc C sb t tjdddt ddg¡d}d|jd< t ddgd d gd d ggt ddg¡d}t ||¡ d S )
|
N)r r Úint64r ©r! r )r! r ©r> é
|
)r r! r )r r@ Úzerosr Ú from_tuplesr r r r6 r r r Útest_at_setitem_multiindexg s þ
|
þz(TestAtSetItem.test_at_setitem_multiindexÚrowú
|
2019-01-01c C s^ t dgd gtddgdd}t ddgddggtddgdd}d|j|d f< t ||¡ d S )
|
Nr r rK z
|
2019-01-02r# r) g à?g ð?r )r r r r r )r7 rJ r/ r r r r Útest_at_datetime_indexs s
|
ÿ þz$TestAtSetItem.test_at_datetime_indexN) Ú__name__Ú
|
__module__Ú __qualname__r8 r< rB rI ÚpytestÚmarkZ parametrizer rL r r r r r1 6 s r1 c @ s e Zd Zdd ZdS )ÚTestAtSetItemWithExpansionc C s: t d|d}t|}||jd< t||g}t ||¡ d S )Nz2017-08-05 00:00:00+0100)Útzr )r r
|
r r r. )r7 Ztz_naive_fixtureÚtsr r r r r Ú-test_at_setitem_expansion_series_dt64tz_value s
|
|
zHTestAtSetItemWithExpansion.test_at_setitem_expansion_series_dt64tz_valueN)rM rN rO rU r r r r rR s rR c @ s e Zd Zdd ZdS )ÚTestAtWithDuplicatesc C s( t j d¡ dd¡}t|ddgd}d}tjt|d |jddg W 5 Q R X tjt|d |jddgf W 5 Q R X tjt|d |jd d
df W 5 Q R X tjt|d d|jddg< W 5 Q R X tjt|d d|jddgf< W 5 Q R X tjt|d d|jd d
df< W 5 Q R X d S ) Nr r r ÚArE z Invalid call for scalar access©Úmatchr ) r@ ÚrandomZrandnÚreshaper rP ÚraisesÚ
|
ValueErrorr )r7 Úarrr/ Úmsgr r r Ú2test_at_with_duplicate_axes_requires_scalar_lookup s zGTestAtWithDuplicates.test_at_with_duplicate_axes_requires_scalar_lookupN)rM rN rO r` r r r r rV s rV c @ sL e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dS )Ú TestAtErrorsc C sZ t dddgdddgd}||d }|dks0ttjtdd ||d W 5 Q R X d S )Nr r r r2 r! rX )r
|
r- rP r\ ÚKeyError©r7 Ú
|
indexer_alr; r r r r Ú test_at_series_raises_key_error§ s
|
z,TestAtErrors.test_at_series_raises_key_errorc C s t ddddgidddgd}||d }|dks4ttjtdd ||d W 5 Q R X tjtdd ||d
|
W 5 Q R X d S ) Nr r r r r2 )r r r! rX rD )r r! )r r- rP r\ rb ©r7 rd r/ r r r r Ú test_at_frame_raises_key_error± s z+TestAtErrors.test_at_frame_raises_key_errorc C sX t dddgtdd}||d }|dks.ttjtdd ||d W 5 Q R X d S )
|
Nr r r Úabcr2 r! ú^0$rX r )r
|
Úlistr- rP r\ rb rc r r r Ú test_at_series_raises_key_error2¿ s
|
z-TestAtErrors.test_at_series_raises_key_error2c C s\ t ddddgitdd}||d }|dks2ttjtdd ||d
|
W 5 Q R X d S ) NrW r r r rh r2 )r! rW ri rX rD )r rj r- rP r\ rb rf r r r Ú test_at_frame_raises_key_error2Ê s
|
z,TestAtErrors.test_at_frame_raises_key_error2c C sP t ddgddgd}ddg}tjtdt| d ||jd
|
< W 5 Q R X d S ) Nr r r r r r r& ú*You can only assign a scalar value not a \rX r ©r rP r\ r Útyper ©r7 r/ Znew_rowr r r Ú test_at_frame_multiple_columnsÓ s þz+TestAtErrors.test_at_frame_multiple_columnsc C sl t dddddgdddddgd }tjtd
|
d |jd W 5 Q R X tjtd d |jd W 5 Q R X d S )Nr r r r r r! r" r% r2 ri rX r z^4$)r
|
rP r\ rb r r: r r r Ú'test_at_getitem_mixed_index_no_fallbackÝ s
|
z4TestAtErrors.test_at_getitem_mixed_index_no_fallbackc C st t ddg}t d¡ dd¡}t||d}||jfD ]:}dD ]0}tjtt |d |j
|
||f W 5 Q R X q<q4d S )Nr r r r2 )r r rX ) r r@ Zaranger[ r ÚTrP r\ rb Ústrr )r7 Úcir^ Úframer/ Úkeyr r r Ú test_at_categorical_integerså s z)TestAtErrors.test_at_categorical_integersc C sL t dgddgd}ddg}tjtdt| d ||jd< W 5 Q R X d S ) Nr! Zcol1Zcol2)r* r> é{ é rm rX rn rp r r r Útest_at_applied_for_rowsò s þz%TestAtErrors.test_at_applied_for_rowsN) rM rN rO re rg rk rl rq rr rx r{ r r r r ra £ s
|
|
ra )r r Únumpyr@ rP Z pandas.errorsr Zpandasr r r r r r
|
r Zpandas._testingZ_testingr r r0 r1 rR rV ra r r r r Ú<module> s $
|
K
|
|
