U
|
ý°dé ã @ sJ d dl Zd dlZd dlZd dlmZmZmZ d dlm Z
|
G dd dZ dS )é N)Ú DataFrameÚSeriesÚ
|
date_rangec @ sT e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dS )ÚTestDataFrameRoundc C s t }t || ¡ ¡ t dddgdddgd}t dd d
|
gdd d
|
gd}t | ¡ |¡ d }t d d dgdddgd}t | |¡|¡ t t ||¡|¡ dd g}d}tjt|d | |¡ W 5 Q R X t dddgdddgd}dd d}t | |¡|¡ t dddgdddgd}ddi}t | |¡|¡ d dd } t | | ¡|¡ dd d}
|
d }tjt|d | |
|
¡ W 5 Q R X dd d}
|
tjt|d | |
|
¡ W 5 Q R X t|
|
} tjt|d | | ¡ W 5 Q R X ddd gd}
|
tjt|d | |
|
¡ W 5 Q R X t|
|
} tjt|d | | ¡ W 5 Q R X t|
|
} tjt|d | | ¡ W 5 Q R X t|
|
} tjt|d | | ¡ W 5 Q R X d d!d} |d" } t d#d$d%gd&d'd(gd}t | | ¡|¡ ttj dd}tjt|d | |¡ W 5 Q R X t
|
|d) d¡|d) ¡ d }t d d dgdddgd}d*|j _ d*|j _ t | |¡|¡ t
|
|d) |¡|d) ¡ t
|
| |¡d) |d) ¡ d S )+Nç+ÙÎ÷ñ?çClçû @çClçû@çX9´Èv¾ó?ç¬ Zd;ß@ç¬ Zd;ß @©Úcol1Úcol2ç ð?ç @ç @é gìQ¸
ëñ?gö(\Âõ @gö(\Âõ@g®Gáz®ó?g×£p=
|
×@g×£p=
|
× @é z4decimals must be an integer, a dict-like or a Series©Úmatchçñ?gÍÌÌÌÌÌ @gÍÌÌÌÌÌ@g333333ó?ç@g @r )Úcol3r g à?z#Values in decimals must be integersZfooéÿÿÿÿéþÿÿÿéd g [@éÒ i6 g Y@éÈ i, r Úcols) r ÚtmÚassert_frame_equalÚroundÚnpÚpytestÚraisesÚ TypeErrorr ÚnanZassert_series_equalÚcolumnsÚname)ÚselfÚdfÚexpected_roundedÚdecimalsZ
|
round_listÚmsgZ
|
round_dictZexpected_partially_roundedZpartial_round_dictZwrong_round_dictZnon_int_round_dictZnon_int_round_SeriesZnegative_round_dictZbig_dfZexpected_neg_roundedZnan_round_Series© r. ú\d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_round.pyÚ
|
test_round s ÿÿ
|
ÿ
|
|
|
|
ÿÿz TestDataFrameRound.test_roundc C st t ddgddgg}tj|dd}t ddgd d
|
gg}t ||¡ d }tjt|d tj|d|d W 5 Q R X d S )Nç{®Gázø?gÃõ(\Âõ?ç¸
ëQ¸®?g
|
×£p=
|
@r )r, r r ç g @z$the 'out' parameter is not supportedr )r, Úout)r r" r! r r r# r$ Ú
|
ValueError)r) r* r4 Úexpectedr- r. r. r/ Útest_round_numpy s z#TestDataFrameRound.test_round_numpyc C sV t dtjdg ¡ }t d ¡ | ¡ }W 5 Q R X t dtjdg ¡ }t ||¡ d S )Nr1 r2 r r3 )r r" r& Zto_framer Zassert_produces_warningr! r )r) r* Úresultr6 r. r. r/ Útest_round_numpy_with_nan s
|
z,TestDataFrameRound.test_round_numpy_with_nanc C sÐ t ddddgddddgtd d
|
d d }t d dddgddddgtd d
|
d d }t | ¡ |¡ t | d¡|¡ t | ddi¡|¡ t | ddi¡|¡ t | ddd¡|¡ t | ddi¡|¡ d S )Nr r gffffff
|
@g@Ú1ÚaÚcÚfZ20111111é )Zperiods)r r r r r r g @r r r r r )r r r r r! )r) r* Zround_0r. r. r/ Útest_round_mixed_type s$
|
|
|
ýÿ
|
|
|
ýÿz(TestDataFrameRound.test_round_mixed_typec C s t tj ddg¡dddgdddgd}tj||fd d
|
}| ¡ }t |j|j¡ t d d d gdddgd }d}t
|
j t |d | |¡ W 5 Q R X d S )Né ÚAÚBÚCÚfirstÚsecondÚthird)r' Úindexr )Zaxisr r )rG z Index of decimals must be uniquer ) r r" ÚrandomÚpdÚconcatr! r Zassert_index_equalrG r r# r$ r5 )r) r* ÚdfsZroundedr, r- r. r. r/ Ú!test_round_with_duplicate_columns© s ýz4TestDataFrameRound.test_round_with_duplicate_columnsc C sH t dddgdddgd}t dd d
|
gdd d
|
gd}t t||¡ d S ) Nr r r r r
|
r r r r r )r r r r! )r) r* r+ r. r. r/ Útest_round_builtin» s z%TestDataFrameRound.test_round_builtinc C s t dgd dgd ¡}ttj dd¡tdd}| d¡}||_| ¡
|
|¡}|j dks`t | d¡}|j dksxt t ||¡ d S )NÚlowr@ Úhié Úabc©r' )rP r@ )rI ÚCategoricalIndexr r" rH ZrandÚlistr! rG ÚcopyZ set_indexÚshapeÚAssertionErrorr r )r) Úidxr* r6 Zdf_categoricalr8 r. r. r/ Ú test_round_nonunique_categoricalÄ s
|
|
z3TestDataFrameRound.test_round_nonunique_categoricalc C sZ t t dd¡¡}tddgddgg|d}| ¡ }tddgd d gg|d}t ||¡ d S )
|
Nr r g
ëQ¸ å?r g333333Ó?g Ð?rR r r3 )rI rS Zinterval_ranger r! r r )r) r' r* r8 r6 r. r. r/ Ú$test_round_interval_category_columnsÓ s
|
z7TestDataFrameRound.test_round_interval_category_columnsc C s* t }| ¡ }t ||¡ ||k s&td S )N)r r! r r rW )r) r* r8 r. r. r/ Útest_round_empty_not_inputÜ s z-TestDataFrameRound.test_round_empty_not_inputN) Ú__name__Ú
|
__module__Ú __qualname__r0 r7 r9 r? rL rM rY rZ r[ r. r. r. r/ r s q r ) Únumpyr" r# ZpandasrI r r r Zpandas._testingZ_testingr r r. r. r. r/ Ú<module> s
|
|
