U
|
ý°d= ã @ sr d dl m Z mZ d dlZd dlZd dlmZ d dlmZm Z d dl
|
m Z d ddZ G dd dZG d d
|
d
|
ZdS ) é )ÚcopyÚdeepcopyN)Ú is_scalar)Ú DataFrameÚSeriesc K sÔ t |trt|g| j }|dk r´t|rf|dkrLd}tj}| | jd¡ q²tj ||d}|
|
|¡ qÀt |¡}| ¡ }||j d }||j d dkr tdt ||¡ |¡}n tjj| }| |fd|i|S )z
|
construct an object for the given shape
|
if value is specified use that if its a scalar
|
if value is an array, repeat it as needed
|
NÚempty©Údtyper z!invalid value passed in constructr )Ú
|
isinstanceÚintÚtupleÚ _AXIS_LENr ÚnpÚfloat64ÚpopZ_info_axis_namer ÚfillÚprodZravelÚshapeÚ ExceptionÚrepeatZreshapeÚrandomZrandn)Úboxr Úvaluer ÚkwargsÚarrZfshapeZ new_shape© r úXd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/generic/test_generic.pyÚ construct s$
|
|
r c @ sê e Zd Zej dejdd edD e dd edD g¡dd Z
|
dd Z d
|
d Z d d Z dd Zdd Zdd Zdd Zej dddddg¡dd Zd d Zd d Zej deed d! d"d! g¡ej d#d$d%d&g¡d'd( Zd)S )*Ú TestGenericÚfuncc C s i | ]}|| ¡ qS r ©Úlower©Ú.0Úxr r r Ú
|
<dictcomp>: s zTestGeneric.<dictcomp>ÚABCDc C s i | ]}|| ¡ qS r r r" r r r r% ; s c C sb t d}|jD ]N}||i}t|df|}|jf ||i}| ¡ }t||t d t ||¡ qd S )Nr& é Zabcd)ÚlistZ _AXIS_ORDERSr Úrenamer ÚsetattrÚtmÚ assert_equal) ÚselfÚframe_or_seriesr ÚidxÚaxisr ÚobjÚresultÚexpectedr r r Ú test_rename6 s
|
|
zTestGeneric.test_renamec sÌ d fddt jD }t f|}| ¡ }t ||¡ | ¡ }t fddi|}t|trx|j d d
|_ t |j
|
dd|¡ t ddd dg¡}t fd|i|}| ¡ }t ||¡ d S )
|
Nr' c s i | ]} |¡ttqS r )Ú_get_axis_namer( Úrange)r# Úi©r. Únr r r% N s ÿ z5TestGeneric.test_get_numeric_data.<locals>.<dictcomp>r r r T)ZdropF) r6 r r Z_get_numeric_datar+ r, Z_get_bool_datar
|
r ÚcolumnsZ reset_indexr Úarray)r- r. r Úor2 r3 r r r8 r Útest_get_numeric_dataL s þ
|
z!TestGeneric.test_get_numeric_datac C sX t |dd}d|j d}tjt|d t|dk W 5 Q R X tjt|d t|dk W 5 Q R X tjt|d t| W 5 Q R X t |ddd}tjt|d t|dk W 5 Q R X tjt|d t|dk W 5 Q R X tjt|d t| W 5 Q R X t |dtjd}tjt|d t|dk W 5 Q R X tjt|d t|dk W 5 Q R X tjt|d t| W 5 Q R X t |dd}tjt|d t| W 5 Q R X t |ddd}t |ddd}tjt|d |râW 5 Q R X tjt|d |o| W 5 Q R X tjt|d |p(| W 5 Q R X tjt|d | W 5 Q R X d S ) Nr' ©r zThe truth value of a z is ambiguous©Úmatchr é )r r )r Ú__name__ÚpytestÚraisesÚ
|
ValueErrorÚboolr Únan)r- r. r1 ÚmsgZobj1Úobj2r r r Ú test_nonzerog sH
|
zTestGeneric.test_nonzeroc s^ fdd}d j d}tjt|d |dddg W 5 Q R X |d |d
|
|d d S ) Nc s t dd| dS )Né rA )r r r )r r ©r. r r Úf s z;TestGeneric.test_frame_or_series_compound_dtypes.<locals>.fz+compound dtypes are not implemented in the z constructorr? )ÚAz datetime64[h])ÚBÚstr)ÚCÚint32Úint64r zM8[ns])rB rC rD ÚNotImplementedError)r- r. rM rH r rL r Ú$test_frame_or_series_compound_dtypes s ÿz0TestGeneric.test_frame_or_series_compound_dtypesc C s( t |dd}d|_t |dd}d|_dD ] }t||d}t ||¡ q(dD ] }t|||}t ||¡ qLdD ]>}t|||}t ||¡ t |||@ ¡ t |||B ¡ qp| |¡}t ||¡ || }t |¡ dD ]F}t|||}t|||}t |¡ t ||@ ¡ t ||B ¡ qÜd S )NrK r> ÚfooÚbar)Ú__add__Ú__sub__Ú __truediv__Ú__mul__rA )Ú__eq__Ú__le__Ú__ge__)r ÚnameÚgetattrr+ Zassert_metadata_equivalentZ combine_first)r- r. r< Zo2Úopr2 Zv1Zv2r r r Útest_metadata_propagation¬ s2
|
|
|
z%TestGeneric.test_metadata_propagationc C s> t |dd}|jt |j¡ks"t|jdt|j ks:td S )Né
|
r> )r Úsizer r r ÚAssertionErrorÚlenÚaxes©r- r. r< r r r Útest_size_compatÞ s z TestGeneric.test_size_compatc C s@ t |dd}tt |d¡dks$ttt |d¡dks<td S )Nrc r> é é )r rf r Z array_splitre rh r r r Útest_split_compatæ s z TestGeneric.test_split_compatc C s® t |d}d}d}tjt|d |j|d W 5 Q R X tjt|d |j|d W 5 Q R X tjt|d |j|d W 5 Q R X tjt|d |j|d W 5 Q R X d S )Nrj z Star Warszunexpected keywordr? )Zepic)r rC rD Ú TypeErrorÚmaxÚvarÚsumÚany)r- r. r1 ZstarwarsÚerrmsgr r r Ú test_stat_unexpected_keywordí s
|
z(TestGeneric.test_stat_unexpected_keywordrp Zcumsumrq ro c C s6 t |d}t||}|j|ks"t|j |¡s2td S )Nrj )r r` rB re Ú __qualname__Úendswith)r- r r. r1 rM r r r Útest_api_compatû s
|
|
zTestGeneric.test_api_compatc C s¶ t |d}t dg¡}d}tjt|d |j|d W 5 Q R X tjt|d |j|d W 5 Q R X tjt|d |j|d W 5 Q R X tjt|d |j |d W 5 Q R X d S )Nrj r ú$the 'out' parameter is not supportedr? ©Úout)
|
r r r; rC rD rE rn ro rp rq )r- r. r1 ry rr r r r Útest_stat_non_defaults_args s
|
z'TestGeneric.test_stat_non_defaults_argsc C sÈ dgdg|j d }t||ddd}t | ¡ |¡ t |jddd|¡ t |jdd d|¡ d
|
gdg|j d }t||ddd}t | ¡ |¡ t |jdd d|¡ t |jdd d|¡ d S ) NiÐ rA Zint8)r r r g p§@)ZbeforeÚafteréÿÿÿÿg @@i g `ãFAg >A)r r r+ r, Útruncate)r- r. r ÚsmallÚbigr r r Útest_truncate_out_of_bounds s z'TestGeneric.test_truncate_out_of_boundsc C s | j ddS )NF©Údeep©r ©r$ r r r Ú<lambda>& ó zTestGeneric.<lambda>c C s | j ddS )NTr r r r r r r
& r r r rA rk c C s. t ||}||}||k s tt ||¡ d S ©N)r re r+ r, )r- r. r r r1 Zobj_copyr r r Útest_copy_and_deepcopy$ s
|
z"TestGeneric.test_copy_and_deepcopyN)rB Ú
|
__module__rt rC ÚmarkÚ parametrizerP r! r( r r4 r= rJ rU rb ri rl rs rv rz r r r r r r r r r 5 s2 ýþ
|
12
|
þr c @ sä e Zd Zej de ¡ e ¡ e ¡ g¡dd Z
|
dd Z dd Z dd Z d
|
d Zd d Zdd Zej de ¡ e ¡ e ¡ g¡dd Zdd Zdd Zej de ¡ e ¡ e ¡ g¡dd Zdd Zdd Zd d Zd d Zd S )!Ú TestNDFrameÚserc C s t | ¡ |¡ d S r )r+ Úassert_series_equalÚsqueeze©r- r r r r Útest_squeeze_series_noop4 s z$TestNDFrame.test_squeeze_series_noopc C s t ¡ }t | ¡ |¡ d S r ©r+ ÚmakeTimeDataFrameÚassert_frame_equalr ©r- Údfr r r Útest_squeeze_frame_noop; s z#TestNDFrame.test_squeeze_frame_noopc C s* t ¡ jdgd}t | ¡ |d ¡ d S ©NrN )r: )r+ r Úreindexr r r r r r Útest_squeeze_frame_reindex@ s z&TestNDFrame.test_squeeze_frame_reindexc C s> t g dtjd}t|g}t || ¡ ¡ t || ¡ ¡ d S )NZfive)r_ r )r r r r r+ r r )r- Z empty_seriesZ empty_framer r r Útest_squeeze_0_len_dimE s
|
z"TestNDFrame.test_squeeze_0_len_dimc C s t jddjd d
d d
f }|jdks,tt |jdd|jd ¡ t |jdd|jd ¡ t |jdd|jd d
df ¡ t |jdd|jd d
df ¡ | ¡ |jd ksºtd }tjt |d
|
|jd d W 5 Q R X d }tjt |d
|
|jd d W 5 Q R X d S )NrA )Znper)rA rA r ©r0 Úindexr: )r r z)No axis named 2 for object type DataFramer? rk z)No axis named x for object type DataFramer$ )
|
r+ r Zilocr re r r rC rD rE )r- r rH r r r Útest_squeeze_axisL s ""z TestNDFrame.test_squeeze_axisc C s" t d¡}t |jdd|¡ d S )NrK r r r r r r r Útest_squeeze_axis_len_3\ s
|
z#TestNDFrame.test_squeeze_axis_len_3c C sF t ¡ }t t |¡|¡ t ¡ jdgd}t t |¡|d ¡ d S r )r+ ÚmakeFloatSeriesr r r r r )r- Úsr r r r Útest_numpy_squeeze` s z TestNDFrame.test_numpy_squeezec C s t | ¡ |¡ d S r )r+ r Ú transposer r r r Útest_transpose_seriesg s z!TestNDFrame.test_transpose_seriesc C s t ¡ }t | ¡ ¡ |¡ d S r )r+ r r r£ r r r r Útest_transpose_framen s z TestNDFrame.test_transpose_framec C sv t ¡ }t ||¡}|tkr.t t |¡|¡ t t t |¡¡|¡ d}tj t
|
|d tj|dd W 5 Q R X d S )Nz%the 'axes' parameter is not supportedr? rA )rg ) r+ r Úget_objr r r r£ r, rC rD rE )r- r. r1 rH r r r Útest_numpy_transposer s z TestNDFrame.test_numpy_transposec C sJ ddddddg}| |¡}t|j |¡|j |¡|jd}t ||¡ d S )NrA rj éþÿÿÿé rK r| )Údatar r )Útaker Úvaluesr r r+ r )r- r Úindicesry r3 r r r Útest_take_series s
|
|
|
ýz TestNDFrame.test_take_seriesc C sV ddddddg}t ¡ }| |¡}t|jj|dd|j |¡|jd }t ||¡ d S )
|
NrA rj r¨ r© rK r| r r )rª r r: )r+ r r« r r¬ r r: r )r- r r ry r3 r r r Útest_take_frame s
|
|
ýzTestNDFrame.test_take_framec C s¨ ddddg}t ¡ }t ||¡}d}tjt|d |j|dd W 5 Q R X d}tjt|d |j||d W 5 Q R X d
|
}tjt|d |j|d d W 5 Q R X d S ) Néýÿÿÿrk r rA z1take\(\) got an unexpected keyword argument 'foo'r? )rV rw rx z%the 'mode' parameter is not supportedZclip)Úmode)r+ r r¦ rC rD rm r« rE )r- r. r r1 rH r r r Útest_take_invalid_kwargs s z$TestNDFrame.test_take_invalid_kwargsc C sn |}|t d}|j ¡ }|D ]L}| |¡| |¡ks8t| |¡| |¡ksPt| |¡| |¡ks tq d S )Nr )ÚobjectZ_AXIS_TO_AXIS_NUMBERÚkeysZ_get_axis_numberre r5 Z_get_block_manager_axis)r- r. r r1 r¬ Úvr r r Útest_axis_classmethods« s
|
|
z"TestNDFrame.test_axis_classmethodsc C sH t ddg}|tkr | ¡ }|j|jks,t| ¡ }|j|jk sDtd S )NrA rk )r r Zto_frameÚflagsre r )r- r. r1 rI r r r Útest_flags_identity´ s z TestNDFrame.test_flags_identityN)rB r rt rC r r r+ r ZmakeStringSeriesZmakeObjectSeriesr r r r r r r¢ r¤ r¥ r§ r® r¯ r² r¶ r¸ r r r r r 1 s6 ÿ
|
ÿ
|
ÿ
|
|
r )NN)r r Únumpyr rC Zpandas.core.dtypes.commonr Zpandasr r Zpandas._testingZ_testingr+ r r r r r r r Ú<module> s
|
}
|
