U
|
ý°d@ ã @ s| d Z ddlZddlZddlZddlZddlmZ ej
|
j ddZ G dd dZ G dd dZej
|
jd e d
|
d
|
g¡e d d g¡e d d g¡e e d d¡gd ¡e e d ¡gd ¡e e d¡gd ¡e d d g¡e e dd
|
¡gd ¡ej ddg¡g dd ddd Z dd Z dd Z dd Z ej
|
d ej d d d d!d fej d"d d!d fej d#d$d d d!d%fg¡d&d' Z!d(d) Z"dS )*z1Tests dealing with the NDFrame.allows_duplicates.é NzNot implemented.©Úreasonc @ sF e Zd Zej deje g ¡fejddgfej
|
i fej
|
dddgifg¡dd Z ej de dg¡e d d¡e d
|
ej¡e d
|
d ¡e d ¡ejg¡d d Zej dejddddgdejdddgdg¡edd Zdd Zdd Zej dd dg¡ej dd d g¡ej ddejddgdd g¡d d Zd!d" Zd#d$ Zej d%ejdddgdjd d&ejddd'gdjd d&gi fejdddgdjd d&ejdddgdjd d&gd(d ifejdddgdjd d&ejdddgdjd d&gd)difej
|
dddgiddgdjd d&ej
|
dddgidd'gdjd d&gi fej
|
dddgiddgdjd d&ej
|
dddgiddgdjd d&gd(d ifej
|
dddgiddgdjd d&ej
|
d*ddgiddgdjd d&gd)difej
|
dddgiddgdjd d&ejddgddgd*d+jd d&gd)difg¡d,d- Zej d.ej ej
|
dddgiddgdjd d&ej
|
d*ddgidd'gdjd d&d d d/d ed0ej ej
|
dddgiddgdjd d&ej
|
d*ddgidd'gdd d d/d ed0ej
|
dddgiddgdej
|
d*ddgidd'gdd d d/d fg¡d1d2 Z ed3d4 Z ej dd d g¡ed5d6 Z d7S )8Ú TestPreservesz cls, dataé é ÚAc C s< ||}|j jdkst||jdd}|j jdks8td S )NTF©Úallows_duplicate_labels)Úflagsr ÚAssertionErrorÚ set_flags)ÚselfÚclsÚdataÚresult© r úad:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/generic/test_duplicate_labels.pyÚtest_construction_ok s
|
z"TestPreserves.test_construction_okÚfuncÚaÚaddÚrenameÚnameÚabsc C s6 t jddgddgdjdd}||jjdks2td S ©Nr r r Úb©ÚindexFr )ÚpdÚSeriesr r
|
r r )r r Úsr r r Útest_preserved_series! s z#TestPreserves.test_preserved_seriesÚotherr r Úcr c C sP t jddgddgdjdd}| |¡\}}|jjdks<t|jjdksLtd S r )r r r Zalignr
|
r r )r r" r r r r r r Ú
|
test_align0 s zTestPreserves.test_alignc C sf t jddgddgdddgdjd d
|
}|jdg jjd ks@t|jd d
d d gf jjd ksbtd S ) Nr r é é ©r ÚBr r r Fr r r( )r Ú DataFramer Úlocr
|
r r ©r Údfr r r Útest_preserved_frame; s
|
ÿz"TestPreserves.test_preserved_framec C s, t jtdjdd}| ¡ jjdks(td S )N)ÚdtypeFr )r r Úfloatr Úto_framer
|
r r )r Zserr r r Ú test_to_frameB s zTestPreserves.test_to_frameÚsubÚframeFT©r c C sx t jddgdddgdjdd}|r,| ¡ }t|t jrD|rD| ¡ }t ||¡}|jjdks`t ||jjdkstt d S ) Nr r r r r )r r Fr )
|
r r r r0 Ú
|
isinstanceÚoperatorÚ methodcallerr
|
r r )r r r" r3 r, r r r Ú test_binopsF s ÿ zTestPreserves.test_binopsc C s t dddgi¡jdd}|dg jjdks0t|d jjdksDt|jd jjdksZt|jdg jjdksrt|jddgf jjdkstd S )Nr r r Fr r )r r) r r
|
r r r* r+ r r r Útest_preserve_getitemU s z#TestPreserves.test_preserve_getitemc C sf |s|j tjjdd¡ t ddgi¡jdd}|d jj dksFt
|
d|j_ |d jj dksbt
|
d S )NzUnclear behavior.r r r Fr T) ÚnodeZ
|
add_markerÚpytestÚmarkÚxfailr r) r r
|
r r )r ÚrequestZusing_copy_on_writer, r r r Ú"test_ndframe_getitem_caching_issue] s z0TestPreserves.test_ndframe_getitem_caching_issueú objs, kwargsr ÚdZ ignore_indexÚaxisr( ©r r c C s" t j|f|}|jjdks td S )NF)r Úconcatr
|
r r )r ÚobjsÚkwargsr r r r Ú test_concatg s ZzTestPreserves.test_concatz left, right, kwargs, expected©Z
|
left_indexZ right_index)Zmarksc C s$ t j||f|}|jj|ks td S ©N)r Úmerger
|
r r )r ÚleftÚrightrF Úexpectedr r r r Ú
|
test_mergeÄ s #zTestPreserves.test_mergec C sF t ddddgi¡jdd}| dddg¡ d¡}|jjdksBtd S ) Nr r r r% Fr r Úcount)r r) r ÚgroupbyZaggr
|
r r )r r, r r r r Ú test_groupbyê s zTestPreserves.test_groupbyc C sz t jdt jdddddd}|r(| ¡ }| d¡ ¡ jjdksBt| d¡ ¡ jjdks\t|
|
d¡ ¡ jjdksvtd S ) Nr Ú2000é )Zperiodsr F)r r r r% ) r r Z
|
date_ranger0 ZrollingZmeanr
|
r r ZewmZ expanding)r r3 r, r r r Ú test_windowõ s üzTestPreserves.test_windowN) Ú__name__Ú
|
__module__Ú __qualname__r; r< Ú parametrizer r ÚnpÚarrayr) r r6 Ú
|
itemgetterr7 ÚstrÚupperr r! Únot_implementedr$ r- r1 r8 r9 r? r rG ÚparamrN rQ rT r r r r r s üþ
|
|
|
|
úþ
|
$ÿ
|
ÿÿü÷ ÿÿü÷ ÿÿü÷ÿÿü÷ ÿÿü÷ ÿÿü÷ÿýûü óºþY
|
ÿÿ÷ ÿù üèþ"
|
|
|
r c @ s¶ e Zd Zej dejddgedfej dddgifej ddgddgdfej dddgifg¡dd Z
|
ej d
|
ejd d gedej d d gd ej d d gd g¡dd Z dd Z ej de ddg¡dfe ddg¡dfe e dddgf¡d¡e ddgdf¡dfe d d g¡dfe e d d d gf¡d¡e e d d gd f¡d¡g¡dd Zej dejdd dgddjd d ejd d dgddjd d gd difg¡d d! Zed"d# ZdS )$Ú
|
TestRaisesz cls, axesr )r r. r r )r Úcolumnsra c C sR |f |}|j jdkstd}tjtjj|d |f |jdd W 5 Q R X d S )NTúIndex has duplicates.©ÚmatchFr ) r
|
r r r; Úraisesr ÚerrorsÚDuplicateLabelErrorr )r r Zaxesr Úmsgr r r Ú test_set_flags_with_duplicates
|
s
|
|
|
z)TestRaises.test_set_flags_with_duplicatesr r r ©ra c C s> d}t jtjj|d d|j_W 5 Q R X |jjdks:td S )Nrb rc FT)r; re r rf rg r
|
r r )r r rh r r r Ú+test_setting_allows_duplicate_labels_raises s z6TestRaises.test_setting_allows_duplicate_labels_raisesc C sd t jdddgd}t jddgddgdjdd}d}tjt jj|d t ||g¡ W 5 Q R X d S )
|
Nr r r r r Fr rb rc )r r r r; re rf rg rD ©r r r rh r r r Útest_series_raises* s
|
z TestRaises.test_series_raiseszgetter, targetr Nr* Zilocc C sj t jddgddgdddgdjd d
|
}|r8t||}n|}d }tjt jj|d || W 5 Q R X d S ) Nr r r% r& r' r r r Fr rb rc )r r) r Úgetattrr; re rf rg )r ÚgetterÚtargetr, rh r r r Útest_getitem_raises1 s ÿ z TestRaises.test_getitem_raisesr@ r rC Fr r rB c C s4 d}t jtjj|d tj|f| W 5 Q R X d S )Nrb rc )r; re r rf rg rD )r rE rF rh r r r Útest_concat_raisesM s z TestRaises.test_concat_raisesc C s| t jddddgidddgdjd d
|
}t jd dddgidddgd}d }tjt jj|d t j||ddd W 5 Q R X d S )Nr r r r r r r# r Fr r( rb rc TrH )r r) r r; re rf rg rJ rl r r r Útest_merge_raisesb s ÿ z TestRaises.test_merge_raises)rU rV rW r; r< rX r r r/ r) ri rk rm r6 r[ r_ rq r rr r^ rs r r r r r` s\ üþ
|
ýþ
|
÷þ
|
ÿÿü÷ÿþ
|
r` Úidxr r gñ?rR ÚDr Z1D)r r c C s
|
t | jS rI )ÚtyperU ©Úxr r r Ú<lambda>z ó ry )Úidsc C s² d}t jtjj|d tjd| djdd W 5 Q R X t jtjj|d$ tjdddgi| djdd W 5 Q R X t jtjj|d" tjddgg| d jdd W 5 Q R X d S )
|
Nrb rc r r Fr r r rj )r; re r rf rg r r r) )rt rh r r r Útest_raises_basicm s (r| c C sX t dddddg¡} | ¡ }t jdddgddggit jddgd d
|
d }t ||¡ d S ) Nr r r# Ú positionsr r r r% Úlabelr4 r )r ÚIndexÚ_format_duplicate_messager) ÚtmÚassert_frame_equal©rt r rM r r r Ú$test_format_duplicate_labels_message s ÿr c C sd t j dgdddddgg¡} | ¡ }t jdddgdd ggit j dgddgg¡d
|
}t ||¡ d S ) Nr r r r# r} r r r r% r )r Ú
|
MultiIndexZ from_productr r) r r r r r r Ú*test_format_duplicate_labels_message_multi s þr c C sR t dddgi¡jdd} d}tjt|d | jddd d
|
gd d W 5 Q R X d S ) Nr r r Fr úCannot specifyrc r r% r& T)Zallow_duplicates)r r) r r; re Ú
|
ValueErrorÚinsert)r, rh r r r Ú test_dataframe_insert_raises s r zmethod, frame_onlyZ set_indexr T)ZinplaceZ reset_indexr c C s | S rI r rw r r r ry § rz Fc C s t ddgddgd¡jdd}|d }d|j_d}tjt|d | | W 5 Q R X |s|tjt|d | | W 5 Q R X d S )
|
Nr r r r' Fr r r rc )r r) r r
|
r r; re r )ÚmethodZ
|
frame_onlyr, r rh r r r Útest_inplace_raises¢ s ÿr c C s\ t ddg¡jdd} t | ¡}t | |¡ t dg i¡jdd} t | ¡}t | |¡ d S )Nr r Fr r )r r r r Zround_trip_pickleZassert_series_equalr) r )r r r r r Ú test_pickle¹ s
|
|
r )#Ú__doc__r6 ÚnumpyrY r; Zpandasr Zpandas._testingZ_testingr r< r= r^ r r` rX r Z PeriodIndexZPeriodZ DatetimeIndexÚ TimestampZTimedeltaIndexZ TimedeltaZCategoricalIndexZ IntervalIndexZIntervalr
Ú from_tuplesr| r r r r7 r r r r r r Ú<module> sH zd ÷ ó
|
|
ýþ
|
|
