U
|
ý°d/ ã @ s d dl mZ d dlZd dlZd dlZd dlZd dlmZ d dl m
|
Z
|
m Z d dl Z d dl mZmZmZmZ d dlmZ G dd dZdS )é )ÚdeepcopyN)Úoption_context)Ú
|
async_markÚ
|
skip_if_no)Ú DataFrameÚSeriesÚ
|
date_rangeÚtimedelta_rangec @ s e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd Zdd Zd d Zej d e ed gd!ed gd"ed g ig¡d#d$ Zd%d& Zd'd( Zd)d* Ze d+d, Zd-d. Z ej d/d0d1d2g¡d3d4 Z d5d6 Z e d7d8d9 Z d2S ):ÚTestDataFrameMiscc C sn |d }|j dkst| d¡}|j dks.t|jd d
df }|j dksNt|jd d
}|j dksjtd S )NÚAÚB)ÚnameÚAssertionErrorÚpopÚloc)ÚselfÚ float_frameÚsÚs2© r úRd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/test_api.pyÚ test_getitem_pop_assign_name s
|
z.TestDataFrameMisc.test_getitem_pop_assign_namec C st |}| d¡dkst| d¡dks(t| d¡dks:t| d¡dksLt| d¡dks^t| d¡dkspt| d¡dkst| d¡dkst| d¡dks¦t| d¡dks¸t| d¡|jksÌt| d¡|jksàttjtdd | d¡ W 5 Q R X tjtd d | d
|
¡ W 5 Q R X tjtd d | d ¡ W 5 Q R X tjtdd | d ¡ W 5 Q R X d S ) Nr é ÚindexÚrowsÚcolumnsz No axis named©Úmatché z No axis.*fooÚfooz No axis.*None) Z_get_axis_numberr Z_get_axis_nameZ _get_axisr r ÚpytestÚraisesÚ
|
ValueError)r r Úfr r r Ú test_get_axis' s* z TestDataFrameMisc.test_get_axisc C s( t jtdd |j|k W 5 Q R X d S )Nzunhashable type: 'Index'r )r r! Ú TypeErrorr )r r r r r Útest_column_contains_raisesD s z-TestDataFrameMisc.test_column_contains_raisesc C sÌ t tdtdgtdd}tdD ]}|t|ks$tq$t| d¡tsNtt tdtdgtj tt
|
dd¡d}tdD ]}|t|kstqtdD ]}|t|kstqt| d¡t sÈtd S )NZabcdZefghÚABCD©r r ZEFGH) r ÚlistÚdirr Ú
|
isinstanceÚ __getitem__r ÚpdZ
|
MultiIndexÚ from_tuplesÚzip)r ÚdfÚkeyr r r Útest_tab_completionH s þ z%TestDataFrameMisc.test_tab_completionc C sÆ dd t dD }t dt dg}t||d}dt|ks>tdt|ksNttdd2 t||d}d t|ksvtd
|
t|kstW 5 Q R X tdd " t||d}d t|ks¸tW 5 Q R X d S ) Nc S s g | ]}d t | qS )Úa)Ústr)Ú.0Úir r r Ú
|
<listcomp>] s z@TestDataFrameMisc.test_display_max_dir_items.<locals>.<listcomp>i¤ r( Za99Za100zdisplay.max_dir_itemsi, Za299Za300Za419)Úranger r* r r )r r Úvaluesr0 r r r Útest_display_max_dir_items[ s z,TestDataFrameMisc.test_display_max_dir_itemsc C s\ t }t dg}d}tjt|d t| W 5 Q R X tjt|d t| W 5 Q R X d S )Nr z unhashable type: 'DataFrame'r )r r r! r% Úhash)r Ú empty_framer0 Úmsgr r r Útest_not_hashablen s
|
z#TestDataFrameMisc.test_not_hashablec C s6 d}t |g i}|t|ks t|jd |ks2td S )Nu í ½r )r r* r r )r Zcolnamer0 r r r Ú+test_column_name_contains_unicode_surrogatex s z=TestDataFrameMisc.test_column_name_contains_unicode_surrogatec C s@ t tj dd¡}t tj dd¡}d|j_|jjd ks<td S )Nr é r )r ÚnpÚrandomZrandnr r r )r Zdf1Zdf2r r r Útest_new_empty_index s z&TestDataFrameMisc.test_new_empty_indexc C s\ | d¡}||jkst| d¡}||jks0td}tjt|d | d¡ W 5 Q R X d S )Nr r z Axis must be 0 or 1 \(got 2\)r r )Z _get_agg_axisr r r r r! r" )r r ÚcolsÚidxr= r r r Útest_get_agg_axis s
|
|
z#TestDataFrameMisc.test_get_agg_axisc C s\ t }|jst|jrt|jr$tt dddgdddgdt d¡d }|d
|
= |jrXtd S ) Ng ð?g @g @r3 ÚbÚc©r r r@ ©r r )r Úemptyr rA Úarange)r r Úfloat_string_framer< r0 r r r Ú
|
test_empty s
|
|
|
$z TestDataFrameMisc.test_emptyc C sF t |t |jkst|ddg j}|jddgdj}t ||¡ d S )Nr r r( )Úlenr r r9 ZreindexÚtmZassert_almost_equal)r r ZarrÚexpectedr r r Útest_len s zTestDataFrameMisc.test_lenc C sP |}|j dd}|j dd}t ||¡ |j dd}|j dd}t ||¡ d S )Nr )Zaxisr r r )ÚsumrP Úassert_series_equal)r r r# rQ Úresultr r r Útest_axis_aliases¥ s z#TestDataFrameMisc.test_axis_aliasesc C s$ t tj¡stt tj¡s td S ©N)ÚpydocÚgetdocr r r r )r r r r Útest_class_axis± s z!TestDataFrameMisc.test_class_axisc C s* |j }| ¡ D ]\}}|j|kstqd S rW )Z_seriesÚitemsr r )r rM ÚseriesÚkÚvr r r Útest_series_put_names· s z'TestDataFrameMisc.test_series_put_namesc C sd t dddg}|jrtt dgdgd}|jr2tt ddgddgd ¡ }|jsTt|jjs`td S ) Nr r r@ )r r r3 rG rH Úd)r rK r ZdropnaÚT©r r0 r r r Útest_empty_nonzero¼ s
|
|
|
z$TestDataFrameMisc.test_empty_nonzeror0 r rJ r( c C s |j s
|
t|jj std S rW )rK r ra rb r r r Útest_empty_likeÅ s
|
|
z!TestDataFrameMisc.test_empty_likec C sT t tdddtdddd}|j}|j ¡ }tt d¡didd}t
|
||¡ d S ) NZ20130101é
|
)Zperiodsz1 dayrI ÚobjectÚcount)r ) r r r ra ZdtypesZ value_countsr rA ZdtyperP rT )r r0 ÚtrU rQ r r r Útest_with_datetimelikesÒ s
|
|
þÿ
|
z)TestDataFrameMisc.test_with_datetimelikesc C sF t |}|d }d|d d
< | ¡ D ] \}}|d | |ks$tq$d S )Nr re )r r[ r )r r Úcpr\ rE Úvaluer r r Ú test_deepcopyß s
|
z TestDataFrameMisc.test_deepcopyc C sP t ddddgddddgdddd gd
|
}d d }d d }|| ¡ | dd }|| d¡| dd }|| ¡ | dd }|| ¡ | dd }|| ¡ | dd }|| ¡ | dd }|| ¡ | dd }|| ¡ | | ¡ d }dd }|| d¡d | dd }|| ¡ | dd }|| ¡ | dd }|| ¡ | d S ) Nr ÚbarZbazZquxr r r r@ é )r3 rG rH c S s || }|d kst d S rW )r )Úbaser# rU r r r Ú_check_fí s z<TestDataFrameMisc.test_inplace_return_self.<locals>._check_fc S s | j dddS )Nr3 T©Úinplace)Ú set_index©Úxr r r Ú<lambda>ô ó z<TestDataFrameMisc.test_inplace_return_self.<locals>.<lambda>c S s | j ddS ©NTrq ©Z reset_indexrt r r r rv ø rw r3 c S s | j ddS rx )Zdrop_duplicatesrt r r r rv ü rw c S s | j dddS )NrG Trq )Z sort_valuesrt r r r rv rw c S s | j ddS rx )Z
|
sort_indexrt r r r rv rw c S s | j dddS ©Nr Trq ©Zfillnart r r r rv rw c S s | j ddddS ©Nr r Trq ©Úreplacert r r r rv rw c S s | j ddiddS ©Nr r Trq ©Úrenamert r r r rv rw rH c S s | j dddS )NT)rr Zdropry rt r r r rv rw c S s | j dddS rz r{ rt r r r rv rw c S s | j ddddS r| r} rt r r r rv rw c S s | j ddiddS r r rt r r r rv # rw )r Úcopyrs )r Údatarp r# r` r r r Útest_inplace_return_selfæ s: "ÿ z*TestDataFrameMisc.test_inplace_return_selfc
|
à s| t jddd ddlm} |tkr(d}nd}| |¡I d H t d ¡, |d t|j
|
d d
|
¡ W 5 Q R X W 5 Q R X d S ) NZIPythonz6.0.0)Z
|
minversionr )Úprovisionalcompleterz/from pandas import DataFrame; obj = DataFrame()z5from pandas import Series; obj = Series(dtype=object)Úignorezobj.r ) r Z importorskipZIPython.core.completerr
r Zrun_coderP Úassert_produces_warningr) Z CompleterZ completions)r ÚipÚframe_or_seriesr
Úcoder r r Útest_tab_complete_warning& s
|
z+TestDataFrameMisc.test_tab_complete_warningc C sJ t dddgi}|ji ks td|jd< |jtd}|jddiksFtd S )Nr r r@ r Úversionr( )r Úattrsr r r4 )r r0 rU r r r Ú
|
test_attrs9 s
|
|
z TestDataFrameMisc.test_attrsÚallows_duplicate_labelsTFNc C s" t dddgi}d}|tkr(|d }d}|j|d}|d krN|jjdks^tn|jj|ks^t||k sjt|jjdkszt|tkrt |j|j¡s´tn t |d j|d j¡s´td|j |< |rÖ|j | dksòtn |j | dksètd|j |< |jd|d}d |j |< |j | dks td S )
|
Nr r r )r r r )r T)r r re )
|
r r Z set_flagsÚflagsr r rA Zmay_share_memoryr9 Ziloc)r r r Zusing_copy_on_writeÚobjr1 rU r r r Útest_set_flagsA s2
|
|
ÿ
|
z TestDataFrameMisc.test_set_flagsc C sB t }d}tjt|d | t d¡ ddd¡¡ W 5 Q R X d S )Nz<'DataFrame' object has no attribute '_constructor_expanddim'r é r@ )r r r! ÚAttributeErrorZ_constructor_expanddimrA rL Zreshape)r r0 r= r r r Útest_constructor_expanddimn s z,TestDataFrameMisc.test_constructor_expanddimZjinja2c C s* t }t d ¡ t |¡ W 5 Q R X d S rW )r rP r ÚinspectÚ
|
getmembersrb r r r Útest_inspect_getmembersx s z)TestDataFrameMisc.test_inspect_getmembers)!Ú__name__Ú
|
__module__Ú __qualname__r r$ r& r2 r: r> r? rC rF rN rR rV rZ r_ rc r ÚmarkZ parametrizer rd ri rl r r r r r r r r r r r r r
|
sF
|
|
|
|
üþ
|
@
|
|
,
|
r
|
)r r r rX ÚnumpyrA r Zpandas._config.configr Z pandas.util._test_decoratorsr r Zpandasr- r r r r Zpandas._testingZ_testingrP r
|
r r r r Ú<module> s
|
