U
|
ý°d. ã @ s d dl Zd dlZd dlmZmZmZ d dlmZ d dl
|
m Z ddgfddZ dd Z d
|
d Zd d Zdd Zdd Zdd Zdd Zdd Zejjddd dd d d d d d d d d d d d!d d"d d#d g
|
d$d%d&d'd(d)d*d+d,d-g
|
d.d/d0 Zd1d2 ZdS )3é N)Ú DataFrameÚIndexÚSeries)Ú get_arrayé é c C s6 t | dd}|d d
}|jddd}|j}||fS )Nç ø?©ÚaÚbr
|
T©Zdrop)r Ú set_indexÚindex)Z
|
index_dataÚdfÚviewÚidx© r ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/copy_view/index/test_index.pyÚ
|
index_view s
|
r c C sj t ddgdd}|jddd}|jjdd}d |jd
|
< | rLt |j|¡ nt |jtd dgdd ¡ d S ) Nr r r r
|
Fr T©Údeepéd ©r r )Úname)r r r ÚcopyÚilocÚtmÚassert_index_equalr )Úusing_copy_on_writer Úexpectedr r r Ú test_set_index_update_column s
|
r c C sV t ddgdd}|d d
}|jddd}|jjdd}d |jd
|
< t |j|¡ d S ) Nr r r r r
|
Tr r r r )r r r r r r r )r r r r r r r Ú!test_set_index_drop_update_column! s
|
r! c C sn t ddgdd}tddg}| |¡}|jjdd}d |jd
|
< | rTt |j|¡ nt |jtd dg¡ d S © Nr r r r é
|
é Tr r r ) r r r r r r r r r ©r r Úserr r r r Útest_set_index_series* s
|
|
r' c C sj t ddgdd}tddg}||_|jjdd}d |jd
|
< | rPt |j|¡ nt |jtd dg¡ d S r" )r r r r r r r r r% r r r Útest_assign_index_as_series6 s
|
r( c C sv t ddgdd}tddg}t|}||_d }|jjdd}d |jd
|
< | r\t |j|¡ nt |jtd dg¡ d S r" )r r r r r r r r )r r r& Z rhs_indexr r r r Útest_assign_index_as_indexB s
|
r) c C sT t ddg}t|}|jdd}d|jd< | r<t ||¡ nt |tddg¡ d S ©Nr r Tr r r ©r r r r r r ©r r& r r r r r Útest_index_from_seriesP s
|
r- c C sB t ddg}t|dd}t|}d|jd< t t||¡s>td S )Nr r T©r r r )r r r r ÚnpZ shares_memoryÚAssertionError)r r& r Zarrr r r Útest_index_from_series_copy[ s
|
|
r1 c C s\ t ddg}t|}t|}|jdd}d|jd< | rDt ||¡ nt |tddg¡ d S r* r+ r, r r r Útest_index_from_indexc s
|
r2 Úfuncc C s | | j¡S ©N)Ú _shallow_copyZ_values©Úxr r r Ú<lambda>r ó r8 c C s | ¡ S r4 )r r6 r r r r8 s r9 c C s | ddg¡S ©Nr r )Útaker6 r r r r8 t r9 c C s | ddg¡S )Nr )Úrepeatr6 r r r r8 u r9 c C s | t dd S ©Nr r )Úslicer6 r r r r8 v r9 c C s | ddg S r: r r6 r r r r8 w r9 c C s | tdd¡S r= )Ú_getitem_slicer> r6 r r r r8 x r9 c C s
|
| g ¡S r4 )Údeleter6 r r r r8 y r9 c C s
|
| d¡S )Nr )Úrenamer6 r r r r8 z r9 c C s | j dddS )NÚInt64Fr. )Úastyper6 r r r r8 { r9 r5 r r; r< Z getitem_sliceZ getitem_listr? r@ rA rC )Úidsc C sX t \}}|jdd}d|jjjkr.| d¡}||}d|jd< | rTtj||dd d S ) NTr rC rB r r F©Z check_names) r r ÚnodeZcallspecÚidrC r r r )r r3 Úrequestr Úview_r r r r Útest_index_opso s
|
|
|
rJ c C sJ t ddg\}}|jdd}|jdd}d|jd< | rFtj||dd d S )
|
Nr
|
r Tr Fr. Zaaaar rE )r r Z infer_objectsr r r )r r rI r r r r Útest_infer_objects s
|
rK )Únumpyr/ ZpytestZpandasr r r Zpandas._testingZ_testingr Zpandas.tests.copy_view.utilr r r r! r' r( r) r- r1 r2 ÚmarkZ parametrizerJ rK r r r r Ú<module> sP ö öò
|
|
