U
|
ý°d ã @ sN d Z ddlZddlZddlmZ ddlmZmZ ddl m
|
Z G dd dZ dS )z
|
test_insert is specifically for the DataFrame.insert method; not to be
|
confused with tests with "insert" in their names that are really testing
|
__setitem__.
|
é N)ÚPerformanceWarning)Ú DataFrameÚIndexc @ s< e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd S )ÚTestDataFrameInsertc C s: t tj dd¡t d¡dddgd}| dd|d ¡ t |jt ddddg¡ tj
|
|d |d d d
|
| d d |d ¡ t |jt ddd ddg¡ tj |d |d d d
|
t j td d | dd|d ¡ W 5 Q R X d}t j t|d | dd|d ¡ W 5 Q R X d|j_| dd|d ¡ |jjdks6td S )Né é ÚcÚbÚa)ÚindexÚcolumnsr ZfooF)Z check_namesé Úbarzalready exists©Úmatché z cannot insert c, already existsZ some_nameZbaz)r ÚnpÚrandomÚrandnZarangeÚinsertÚtmZassert_index_equalr r Zassert_series_equalZassert_almost_equalÚpytestÚraisesÚ
|
ValueErrorÚnameÚAssertionError©ÚselfÚdfÚmsg© r ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/indexing/test_insert.pyÚ test_insert s& ÿ z TestDataFrameInsert.test_insertc C s t dddgi}|ji d}| ddddg¡ |ji d}t| t ddgddggddgd}t ||¡ | dd d
|
d g¡ |ji d}t| t d
|
ddgd ddggd ddgd}t ||¡ d S ) Nr gñ?g@©r r r
|
r r r gÍÌÌÌÌÌô?gffffff@)r Úrenamer Ústrr Úassert_frame_equal)r r ÚresultÚexpectedr r r! Útest_insert_column_bug_4032, s "z/TestDataFrameInsert.test_insert_column_bug_4032c C s t }|jdddddgdd |jdddd d
|
gdd |jddd d d gdd t d ddgd d dgd d
|
dggdddgd}t ||¡ d S )Nr ÚAÚgÚhÚiT)Zallow_duplicatesÚdÚeÚfr
|
r r r# )r r r r& )r r Úexpr r r! Ú test_insert_with_columns_dups? s ÿz1TestDataFrameInsert.test_insert_with_columns_dupsc C sÖ t tj dd¡}|d }|r$d }nt}t |¡( tdD ]}|d | ||d < q<W 5 Q R X |rd|jd< |jd |d d kst |jd dksÒt n6d|j
|
d< |jd |d d ksÀt |jd dksÒt d S )Né r r éd r éc )r r ) r r r r r r Úassert_produces_warningÚrangeZilocr Úvalues)r Zusing_array_managerZusing_copy_on_writer ZserZexpected_warningÚnr r r! Útest_insert_item_cacheJ s
|
|
z*TestDataFrameInsert.test_insert_item_cachec C sH t tjjdddddd}t d ¡ t ddd g¡|d
|
< W 5 Q R X d S ) Nr r4 )r r4 )ÚsizeZInt64)Zdtyper r r r
|
)r r r Úrandintr r6 Úarray)r r r r r! Útest_insert_EA_no_warninga s z-TestDataFrameInsert.test_insert_EA_no_warningc C sF t ddgddgd}d}tjt|d | dd|¡ W 5 Q R X d S ) Nr r r r3 )Zcol1Zcol2z5Expected a 1D array, got an array with shape \(2, 2\)r Znewcol)r r r r r r r r r! Útest_insert_frameh s z%TestDataFrameInsert.test_insert_frameN) Ú__name__Ú
|
__module__Ú __qualname__r" r) r2 r: r> r? r r r r! r s r ) Ú__doc__Únumpyr r Z pandas.errorsr Zpandasr r Zpandas._testingZ_testingr r r r r r! Ú<module> s
|
