U
|
ý°d÷ ã @ st d dl mZ d dlmZ d dlZd dlZd dlZd dlm Z m
|
Z
|
m Z m Z d dl mZ G dd dZG dd dZdS ) é )Údeepcopy)Ú methodcallerN)Ú DataFrameÚ
|
MultiIndexÚSeriesÚ
|
date_rangec @ sp e Zd Zej dddg¡dd Zej dddg¡dd Zdd Zd
|
d Z d d Z
|
dd Z dd Z dd Z dS )Ú TestDataFrameÚfuncZ_set_axis_nameÚ rename_axisc C sx t ddgddgg}t|d|}|jjd ks2t|jjdksBtt|ddd|}|jjd ksdt|jjdksttd S )Né é é é ÚfooÚcols©Úaxis)r r ÚindexÚnameÚAssertionErrorÚcolumns)Úselfr ÚdfÚresult© r úVd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/generic/test_frame.pyÚtest_set_axis_name s z TestDataFrame.test_set_axis_namec C s´ t t d¡t dd tdD ¡t dd tdD ¡d}dd g}t|||}|jj|ksbt |j
|
jd d gksvt t||d
|
d |}|j
|
jdd gkst |jjd d gks°t d S ) N)r r c S s g | ] }d |fqS )ÚAr ©Ú.0Úxr r r Ú
|
<listcomp>" s z7TestDataFrame.test_set_axis_name_mi.<locals>.<listcomp>ZaBcc S s g | ] }d |fqS )ÚCr r r r r r! # s Zxyz)r r ZL1ZL2r r ) r ÚnpÚemptyr Ú from_tuplesÚlistr r Únamesr r )r r r Z level_namesr r r r Útest_set_axis_name_mi s ýz#TestDataFrame.test_set_axis_name_mic C s t dgg}| ¡ stt dgg}| ¡ r0tt ddgg}d}tjt|d | ¡ W 5 Q R X tjt|d t| W 5 Q R X d S )NTFz+The truth value of a DataFrame is ambiguous©Úmatch)r Úboolr ÚpytestÚraisesÚ
|
ValueError)r r Úmsgr r r Útest_nonzero_single_element0 s z)TestDataFrame.test_nonzero_single_elementc
|
C s` t ddddddddgddddddddgtj d¡tj d¡d}| d¡ ¡ }t ||¡ d S ) Nr ÚbarZoneZtwoZthreeé )r ÚBr" ÚDr )r r# ÚrandomÚrandnÚgroupbyÚsumÚtmÚassert_metadata_equivalent©r r r r r r Ú'test_metadata_propagation_indiv_groupby? s
|
|
üÿz5TestDataFrame.test_metadata_propagation_indiv_groupbyc C s: t tj dd¡tddddd}| d¡}t ||¡ d S )Niè r Z20130101Ús)ZperiodsÚfreq)r Z1T)r r# r5 r6 r Zresampler9 r: r; r r r Ú(test_metadata_propagation_indiv_resampleL s þ
|
z6TestDataFrame.test_metadata_propagation_indiv_resamplec C sú ddd}| ¡ Þ}| tddg¡ | td|¡ tj d¡ ttj ddd ¡d
|
d gd }ttj ddd ¡d dgd }dgt_d|_d|_|j |d
|
gd gdd}|jdks®t
|
ttj ddd ¡t dd }d|_t ||g¡}|jdksìt
|
W 5 Q R X d S )Nc s | j D ] |dkrL|j|j }}t| dd t| d }t | |¡ q|dkr~d fdd|jD ¡}t | |¡ qt | t| d¡ q| S )NÚmergeÚ ú|Úconcatú+c s" g | ]}t | d rt | qS )N)Úgetattr)r Úo©r r r r! a s zSTestDataFrame.test_metadata_propagation_indiv.<locals>.finalize.<locals>.<listcomp>)Ú _metadataÚleftÚrightrE ÚobjectÚ __setattr__ÚjoinÚobjs)r ÚotherÚmethodÚkwargsrI rJ Úvaluer rG r ÚfinalizeY s
|
ÿz?TestDataFrame.test_metadata_propagation_indiv.<locals>.finalizerH ÚfilenameZ __finalize__é
|
r r )r r ÚaÚb©r ÚcÚdz
|
fname1.csvz
|
fname2.csvÚinner)Zleft_onZright_onÚhowzfname1.csv|fname2.csvÚabr zfoo+foo)N)ÚcontextÚsetattrr r# r5 ÚseedÚrandintrH rT r@ r r& ÚpdrC )r Z monkeypatchrS ÚmZdf1Zdf2r r r r Ú test_metadata_propagation_indivU s
|
|
z-TestDataFrame.test_metadata_propagation_indivc C s\ t ddddgi}d|_dddg|d< d|_|jdks:tt |d tdddgdd ¡ d S )
|
Nr r r r r é Úyé rG )r rf r r9 Zassert_series_equalr )r r r r r Útest_set_attribute s z TestDataFrame.test_set_attributec C s( t g g dgd}t|}t ||¡ d S )Nr )Údatar r )r r r9 Zassert_frame_equal)r Z empty_frameZempty_frame_copyr r r Útest_deepcopy_empty s z!TestDataFrame.test_deepcopy_emptyN)Ú__name__Ú
|
__module__Ú __qualname__r, ÚmarkÚ parametrizer r( r0 r< r? rd rh rj r r r r r s
|
|
* r c @ s8 e Zd Zej ddddddgdg¡dd Zd d
|
Zd S ) ÚTestDataFrame2rR r ÚTruer r g @c C sl t dddgdddgd}d}tjt|d | ¡ jd
|
d dd|d W 5 Q R X tjt|d | ¡ jd d|d W 5 Q R X tjt|d | ¡ jd|d W 5 Q R X tjt|d | ¡ jdd|d W 5 Q R X tjt|d | ¡ j |d W 5 Q R X tjt|d | ¡ j
|
|j dk|d W 5 Q R X tjt|d | ¡ j |j dk|d W 5 Q R X d S )Nr r r r rg re )rV rW z8For argument "inplace" expected type bool, received typer) r rf )Zmapperr ÚinplacerV )r rr r )rR rr é )Z
|
to_replacerR rr ©rr )Zcondrr ) r r, r- r. Úcopyr
|
ÚdropÚfillnaÚreplaceZ interpolateZ_whererV Úmask)r rR r r/ r r r Útest_validate_bool_args s $ "z&TestDataFrame2.test_validate_bool_argsc C sö t tj dd¡ddgd}t dddddtjg¡}|d ¡ }tj|d< d}tj t
|
|d |j dd
|
d d W 5 Q R X tj t
|
|d |j d
|
dgd d W 5 Q R X tj t
|
|d |j dd d W 5 Q R X tj t
|
|d |j dd d W 5 Q R X d S )Nrg r ZjimZjoerX r r zunexpected keywordr) r T)r Úin_placert )r{ )r r# r5 r6 rb Z CategoricalÚnanru r, r- Ú TypeErrorrv Zreindexrw )r r ÚcaÚtsr/ r r r Útest_unexpected_keyword° s
|
z&TestDataFrame2.test_unexpected_keywordN)rk rl rm r, rn ro rz r r r r r rp s
|
rp )ru r Úoperatorr Únumpyr# r, Zpandasrb r r r r Zpandas._testingZ_testingr9 r rp r r r r Ú<module> s
|
