U
|
ý°d ã @ sh d dl Zd dlZd dlmZ d dlmZ d dlm Z dd Z
|
dd Z dd Z d
|
d Z d d Zdd ZdS )é N©Ú DataFrame)Ú get_arrayc C s\ t ddddgi}| ¡ }| r.|j d¡s.tt|d}d|jd< t |t|d¡sXtd S )NÚaé é é éd )r r ) r Z reset_indexÚ_mgrZ_has_no_referenceÚAssertionErrorr ÚilocÚnpÚ shares_memory©Úusing_copy_on_writeÚdfÚarr© r úgd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/copy_view/test_core_functionalities.pyÚ2test_assigning_to_same_variable_removes_references s
|
|
r c C sL t dddgddd}d|d< t|d}d|jd< t |t|d¡sHtd S © Nr r r ©r ÚbÚcr r r ©r r )r r r r r r r r r r Ú.test_setitem_dont_track_unnecessary_references s
|
|
|
r c C sp t dddgddd}|d d
}| ¡ }d|d< t|d}d|jd< | rlt |t|d¡r`tt ||¡ d S r ) r Úcopyr r r r r ÚtmZassert_frame_equal)r r ÚviewÚexpectedr r r r Ú test_setitem_with_view_copies s
|
|
r c C sz t dddgddd}|d d
}d|d< t|d}d }d|jd< | rvtjjd d
|
}|j |¡ t |t|d¡svt
|
d S ) Nr r r r r r r r z.blk.delete does not track references correctly)Úreason) r r r ÚpytestÚmarkZxfailÚnodeZ
|
add_markerr r r )r Úrequestr r r r# r r r Ú0test_setitem_with_view_invalidated_does_not_copy, s
|
|
ÿ r& c C sB dd }| }| r>|j jd j ¡ r(t|j jd j ¡ r>td S )Nc S s$ t ddgddd} | ddg }|S )Nr r g ø?r r r r )r Úresultr r r ÚfuncA s z test_out_of_scope.<locals>.funcr r )r
|
ÚblocksÚrefsÚ has_referencer )r r( r' r r r Útest_out_of_scope@ s
|
r, c C sz t tj dd¡dddgd}|d= | rR|jjd j ¡ s<t|jjd j ¡ sRt|dg }| rv|jjd j ¡ rvtd S ) Né r r r r )Úcolumnsr r ) r r ÚrandomZrandnr
|
r) r* r+ r )r r r r r Ú test_deleteM s
|
r0 )Únumpyr r" Zpandasr Zpandas._testingZ_testingr Zpandas.tests.copy_view.utilr r r r r& r, r0 r r r r Ú<module> s
|
