o
    d&                     @   s&  d dl Zd dlZd dlmZ d dlZd dlmZmZm	Z	m
Z
mZmZmZmZ d dlmZ d dlmZ dd Zdd Zejd	g d
ejjddd dd dd dd dd dd dd dd dd dd dd dd dd dd dd dd gg dddd  Zejd	g d
ejjdd!d d"d d#d d$d d%d d&d d'd d(d d)d d*d d+d d,d d-d d.d d/d d0d d1d gg d2dd3d4 Zejd	g d
d5d6 Zd7d8 Zejd9ed d:ed;d:ggd<d= Zd>d? Zd@dA ZdBdC Z dDdE Z!dFdG Z"dHdI Z#dJdK Z$ejdLdMdNgidOdNidPdNigdQdR Z%dSdT Z&dUdV Z'dWdX Z(dYdZ Z)d[d\ Z*ejd]d^d d_d d`d gdadb Z+dcdd Z,dedf Z-dgdh Z.didj Z/ejdkd9dlgdmdn Z0dodp Z1dqdr Z2ejjdsdtd d fdud d:fdvd d fdwd d fgg dxddydz Z3ejd{ed;d:gdNd|edNd;d:gigd}d~ Z4ejd{ed;d:gdNd|edNd;d:gigdd Z5dd Z6dd Z7dd Z8dd Z9ejddd;ej:fgdd Z;ejdddgdd Z<ejddd dd dd dd gdd Z=dd Z>dd Z?dd Z@ejddNdd;dd d;d dgdd ZAejdddgdd ZBejd{ed;d:gedNd;d:gigdd ZCejd{ed;d:gedNd;d:gigdd ZDdd ZEdd ZFejdeg ddNd|i fedNg diddNifgdd ZGejdeg ddNd|i fedNg diddNifgdd ZHejdg ddd ZIdd ZJdd ZKejd{eg dedNg digdd ZLdd ZMddÄ ZNddń ZOejdddid9digddɄ ZPejdddgdd΄ ZQddЄ ZRdd҄ ZSddԄ ZTejdddgddل ZUejdddgddۄ ZVejdddgdd߄ ZWejdddgdd ZXejdddgdd ZYejdddgdd ZZejdddgdd Z[dd Z\dd Z]dd Z^dd Z_ejjdddgddgddd Z`ejddNdNggdd Zaejdddgejjdddgddgdd d Zbejdd d;gejdddgdd Zcdd	 Zdd
d Zedd Zfdd Zgejd	ddgdd Zhdd Zidd Zjdd Zkdd ZldS (      N)SettingWithCopyWarning)	DataFrameIndex
MultiIndexPeriodSeries	Timestamp
date_rangeperiod_range)	get_arrayc                 C   s   t g dg dg dd}| }tt|dt|dr J | r8|jjd j r-J |jjd j r8J d|j	d< |j	d dksFJ d S )	N                  g?g?g333333?abcr   r   r   r   r   
r   copynpshares_memoryr   _mgrblocksrefshas_referenceilocusing_copy_on_writedfdf_copy r'   h/var/www/html/visualizacion-main/env/lib/python3.10/site-packages/pandas/tests/copy_view/test_methods.py	test_copy   s   
r)   c                 C   s  t g dg dg dd}|jdd}tt|dt|ds"J | r:|jjd j s/J |jjd	 j s:J | rhd|j	d
< |j	d
 d	ksJJ tt|dt|drXJ tt|dt|dsfJ d S d|j	d
< |j	d
 dksvJ tt|dt|dsJ d S )Nr   r   r   r   Fdeepr   r   r   r   r   r   r#   r'   r'   r(   test_copy_shallow$   s   
 
 r,   r   )TNFmethodc                 C   s   | j tj|dS )Ncolumnsr   )renamestrlowerr%   r   r'   r'   r(   <lambda>B       r4   c                 C   s   | j ddg|dS )Nr   r   r.   )reindexr3   r'   r'   r(   r4   C       c                 C      | j | |dS Nr   reindex_liker3   r'   r'   r(   r4   D       c                 C      | j | |dd S Nr:   r   alignr3   r'   r'   r(   r4   E   r7   c                 C      | j g dd|dS Nr   index)axisr   set_axisr3   r'   r'   r(   r4   F       c                 C      | j d|dS NtestrD   r   rename_axisr3   r'   r'   r(   r4   G   r=   c                 C   rI   )NrK   r.   rM   r3   r'   r'   r(   r4   H   r=   c                 C      | j ddi|dS )Nr   int64r:   astyper3   r'   r'   r(   r4   I   r7   c                 C      | j dd|dS Nr   r:   swapaxesr3   r'   r'   r(   r4   K   r5   c                 C      | j dd|dS Nr   r   r:   truncater3   r'   r'   r(   r4   L   r5   c                 C      | j |dS r9   infer_objectsr3   r'   r'   r(   r4   M       c                 C   r[   r9   to_timestampr3   r'   r'   r(   r4   N   r^   c                 C   rI   ND)freqr   	to_periodr3   r'   r'   r(   r4   O   r=   c                 C   rI   N
US/Centralr:   tz_localizer3   r'   r'   r(   r4   P   r=   c                 C   rI   rf   
tz_convertr3   r'   r'   r(   r4   Q   r=   c                 C   rI   NF)allows_duplicate_labelsr   	set_flagsr3   r'   r'   r(   r4   R   r=   )r0   r6   r<   rA   rG   rename_axis0rename_axis1rR   rV   rZ   r]   r`   re   ri   rk   ro   )idsc           	      C   s  d }d| j jjv rtdddd}n-d| j jjv r tdddd}nd| j jjv r/tdddd}nd| j jjv r>tdddd	d
}tg dg dg dd|d}|||d}|pY|du }| j jjdrl|sl|sl|du rld}|r~tt	|dt	|ds|J d S tt	|dt	|drJ d S )Nr`   
2012-01-01rb   r   rc   periodsre   ri   rk   Europe/Brusselsrc   ru   tzr   r   r   r   rD   r:   Fzreindex-r   )
nodecallspecidr
   r	   r   
startswithr   r   r   )	requestr-   r   r$   using_array_managerrD   r%   df2share_memoryr'   r'   r(   test_methods_copy_keyword>   s$   -   r   c                 C   rO   )Nr   d   rL   r0   serr   r'   r'   r(   r4      r7   c                 C   s   | j d |dS r9   r   r   r'   r'   r(   r4      r=   c                 C   s   | j | j|dS )NrL   )r6   rD   r   r'   r'   r(   r4      r5   c                 C   r8   r9   r;   r   r'   r'   r(   r4      r=   c                 C   r>   r?   r@   r   r'   r'   r(   r4      r7   c                 C   rB   rC   rF   r   r'   r'   r(   r4      rH   c                 C   rI   rJ   rM   r   r'   r'   r(   r4      r=   c                 C   rI   )NrP   r:   rQ   r   r'   r'   r(   r4      r=   c                 C   rW   )Nr   r   r:   )	swaplevelr   r'   r'   r(   r4      r5   c                 C   rS   rT   rU   r   r'   r'   r(   r4      r5   c                 C   rW   rX   rY   r   r'   r'   r(   r4      r5   c                 C   r[   r9   r\   r   r'   r'   r(   r4      r^   c                 C   r[   r9   r_   r   r'   r'   r(   r4      r^   c                 C   rI   ra   rd   r   r'   r'   r(   r4      r=   c                 C   rI   rf   rh   r   r'   r'   r(   r4      r=   c                 C   rI   rf   rj   r   r'   r'   r(   r4      r=   c                 C   rI   rl   rn   r   r'   r'   r(   r4      r=   )zrename (dict)r0   r6   r<   rA   rG   rp   rR   r   rV   rZ   r]   r`   re   ri   rk   ro   c                 C   s  d }d| j jjv rtdddd}n@d| j jjv r tdddd}n1d| j jjv r/tdddd}n"d| j jjv r?tdddd	d
}nd| j jjv rQtg dg dg}tg d|d}|||d}|pd|du }|rut	t
|t
|ssJ d S t	t
|t
|rJ d S )Nr`   rs   rb   r   rt   re   ri   rk   rv   rw   r   r   r   ry   r:   F)rz   r{   r|   r
   r	   r   from_arraysr   r   r   r   )r~   r-   r   r$   rD   r   ser2r   r'   r'   r(    test_methods_series_copy_keyword   s"   +r   c                 C   s   t g dg dd}|j|d}| p|du p|d u }|o| }|r2tt|dt|ds0J d S tt|dt|dr@J d S )Nr   r   r   r   r:   Fr   r   )r   	transposer   r   r   )r$   r   r   r%   resultr   r'   r'   r(   test_transpose_copy_keyword   s   
  r   c                 C   s   t g dg dg ddg dd}| }| }|j  | r=tt|dt|ds/J tt|dt|ds=J d	|jd
< tt|dt|drPJ | r`tt|dt|ds`J t	
|| d S )Nr   r   r   r   )
         ry   r   r   r   )r   r   )r   r   reset_indexr   _verify_integrityr   r   r   r"   tmassert_frame_equalr$   r%   df_origr   r'   r'   r(   test_reset_index   s   

r   rD   r   r   c                 C   s   t ddg|d}| }|jdd}| r)tt|t|s J |jdr(J ntt|t|r5J d|jd< t	
|| d S )Nr   r   ry   T)dropr   r   )r   r   r   r   r   r   r   _has_no_referencer"   r   assert_series_equalr$   rD   r   ser_origr   r'   r'   r(   test_reset_index_series_drop   s   
r   c                 C   s   t g dg dg dd}| }|jtjd}| r)tt|dt|ds)J d|jd	< tt|dt|dr<J | rLtt|d
t|dsLJ t g dg dg dd}t	
|| t	
|| d S )Nr   r   r   r   r/   Ar   r   r   Cr   r   r   r   )r   Br   )r   r   r0   r1   upperr   r   r   r"   r   r   r$   r%   r   r   expectedr'   r'   r(   test_rename_columns   s   
r   c                 C   s   t g dg dg dd}|jtjd}| }| r*tt|dt|ds)J ntt|dt|dr8J d|jd	< tt|dt|drKJ | r[tt|d
t|ds[J t g dg dg dd}t	
|| t	
|| d S )Nr   r   r   r   r   r   r   r   r   r   r   r   )r   r0   r1   r   r   r   r   r   r"   r   r   )r$   r%   r   df2_origr   r'   r'   r(   !test_rename_columns_modify_parent  s   
r   c                 C   s   t g ddd}| }dd }||}tt|dt|ds$J d|jd< | r@t|| tt|dt|dr?J nt g d	dd}t|| tt|dt|ds]J tt|d
t|d
skJ d S )Nr         ?r   c                 S   s   | S Nr'   r%   r'   r'   r(   testfunc'  s   ztest_pipe.<locals>.testfuncr   r   r   r   r   )	r   r   piper   r   r   r"   r   r   r$   r%   r   r   r   r   r'   r'   r(   	test_pipe#  s   

 r   c                 C   s   t g ddd}| }dd }||}tt|dt|ds$J | r;t|| tt|dt|dr:J nt g ddd}t|| tt|dt|dsXJ tt|dt|dsfJ d S )	Nr   r   r   c                 S   s   d| j d< | S )Nr   r   )r"   r   r'   r'   r(   r   ?  s   
z%test_pipe_modify_df.<locals>.testfuncr   r   r   r   r   )r   r   r   r   r   r   r   r   r   r'   r'   r(   test_pipe_modify_df;  s   
 r   c                 C   s   t g dg dg dd}| }|jddgd}| r+tt|dt|ds*J ntt|dt|dr9J d|jd	< tt|dt|drLJ | r\tt|dt|ds\J t|| d S )
Nr   r   r   r   r   r   r   r   r   )	r   r   r6   r   r   r   r"   r   r   r   r'   r'   r(   test_reindex_columnsR  s   
r   c                 C   s  t g dg dg dd}| }|jdd}|j  | r<tt|dt|ds-J tt|dt|ds;J ntt|dt|drJJ tt|dt|drXJ d	|jd
< tt|dt|drkJ | r{tt|dt|ds{J t	
|| d S )Nr   r   r   r   r   r   r   r   r   r   )r   r   r   r   r   r   r   r   r"   r   r   r   r'   r'   r(   test_drop_on_columnf  s   

r   c                 C   s   t g dg dg dd}| }|d}|j  | r-tt|dt|ds,J ntt|dt|dr;J d|jd< | rPtt|dt|drPJ t	
|| d S )	Nr   r   r   r   rP   r   r   r   )r   r   select_dtypesr   r   r   r   r   r"   r   r   r   r'   r'   r(   test_select_dtypesy  s   


r   filter_kwargsitemsr   likeregexc                 C   s   t g dg dg dd}| }|jdi |}| r+tt|dt|ds*J ntt|dt|dr9J | rNd|jd< tt|dt|drNJ t|| d S )	Nr   r   r   r   r   r   r   r'   )	r   r   filterr   r   r   r"   r   r   )r$   r   r%   r   r   r'   r'   r(   test_filter  s   
r   c                 C   s   t ddgddgddggtddd	d
gd}| }|jdd}| r0tt|d	t|d	s/J ntt|d	t|d	r>J d|jd< | rStt|d
t|d
rSJ t	|| d S )Nr   r   r   r   r   r   
2020-01-01
2020-01-03r   r   rD   r/   r   )ru   r   )
r   r	   r   shiftr   r   r   r"   r   r   r   r'   r'   r(   test_shift_no_op  s   
r   c                 C   sX   t ddgddgddggtddd	d
gd}|jddd}tt|d	t|d	r*J d S )Nr   r   r   r   r   r   r   r   r   r   r   r   ru   rE   )r   r	   r   r   r   r   )r$   r%   r   r'   r'   r(   test_shift_index  s    r   c                 C   s   t ddgddgddggtddd	d
gd}| }tdd|_|jddd}| r7tt|d	t|d	s6J ntt|d	t|d	rEJ d|jd< | rZtt|d	t|d	rZJ t	
|| d S )Nr   r   r   r   r   r   r   r   r   r   r   
2020-01-02z
2020-01-041Dru   rc   r   r   )r   r	   r   rD   r   r   r   r   r"   r   r   r   r'   r'   r(   test_shift_rows_freq  s   
r   c                 C   s   t ddgddgddggtddd	}|jddd
}tt|dt|ds'J d|jd< | r<tt|dt|dr<J t tjdgtjdgtjdggtddd	}t	|| d S )Nr   r   r   r   r   r   r   r   r   r   r   r   r   )
r   r	   r   r   r   r   r"   nanr   r   )r$   r%   r   r   r'   r'   r(   test_shift_columns  s   
r   c                 C   s   t g dg dg dd}| }|d d  }|d}t|jt|ds)J tt|dt|ds7J | rJd|jd< t|jt|drJJ d|jd< | rgtt|dt|dr_J t	|| d S t g dg d	g dd}t	|| d S )
Nr   r   r   r   r   r   r   r   )r   r   r   )
r   r   popr   r   valuesr   r"   r   r   )r$   r%   r   view_originalr   r   r'   r'   r(   test_pop  s   


r   funcc                 C   s
   |  |S r   r@   xyr'   r'   r(   r4        
 c                 C   s   | j |jddS )Nr   rE   )rA   r   r   r'   r'   r(   r4     r5   c                 C   s   | j |jjtdd ddS )Nr   r   r   )rA   r   r"   slicer   r'   r'   r(   r4     s    c                 C   s   t g ddd}| }|ddg  }|||\}}| r-tt|dt|ds,J ntt|dt|dr;J d|jd< | rPtt|dt|drPJ t|| d S )Nr   r   r   r   r   r   )r   r   r   r   r   r"   r   r   )r$   r   r%   r   
df_changedr   _r'   r'   r(   test_align_frame  s   	
r   c                 C   s   t ddg}| }| }||\}}| r,t|j|js!J t|j|js+J nt|j|jr6J t|j|jr@J d|jd< d|jd< | r`t|j|jrVJ t|j|jr`J t|| t|| d S )Nr   r   r   )	r   r   rA   r   r   r   r"   r   r   )r$   r   r   	ser_otherr   ser_other_resultr'   r'   r(   test_align_series  s    

r   c                 C   s   t g dg dd}| }|j|dd\}}tt|dt|ds&J tt|dt|ds4J | rNd|jd	< t|| d|jd	< t|| d S d S )
Nr   r   r   Fr:   r   r   r   r   r   )	r   r   rA   r   r   r   locr   r   )r$   r%   r   r   df3r'   r'   r(   test_align_copy_false-  s   

r   c                 C   s   t g dg dd}tg ddd}| }| }|j|ddd\}}tt|d	t|d	s3J tt|d
t|d
sAJ tt|dt|dsOJ | rid|jd< t	|| d|jd< t
|| d S d S )Nr   r   r   r   nameFr   )r   rE   r   r   r   )r   r   r   rA   r   r   r   r   r   r   r   )r$   r%   r   r   r   r   r   r'   r'   r(   !test_align_with_series_copy_false=  s   

r   c                 C   s   t g d}| }|d d   }t|jt|dsJ d|jd< | r8t|jt|dr1J t	|| n| }d|jd< t	|| |d d   }d|jd< | r`t
||  d S |  }d|jd< t
|| d S )Nr   r   r   )r   r   to_framer   r   r   r   r"   r   r   r   )r$   r   r   r%   r   r'   r'   r(   test_to_frameP  s$   



r   axr/   c                 C   s   t g dg dd}| }|||}| r&tt|dt|ds%J ntt|dt|dr4J d|jd< | rItt|dt|drIJ t|| d S )Nr   r   r   r   r   r   	r   r   rV   r   r   r   r"   r   r   )r$   r   r%   r   r   r'   r'   r(   test_swapaxes_noopr  s   
r   c                 C   s   t g dg ddg dd}| }|dd}| r*tt|dt|d	s)J ntt|dt|d	r8J d
|jd< | rMtt|dt|d	rMJ t|| d S )Nr   r   r   )r   r   zry   rD   r/   r   r   r   r   r   r   r'   r'   r(   test_swapaxes_single_block  s   
r   c                  C   sX   t ddgdd} | jddd} d| jd	< t ddgddgd
ddgd}t| | d S )Nr   r   r   r   rD   r/   )axis1axis2r   r   r   r   r   ry   )r   rV   r"   r   r   )r%   r   r'   r'   r(   test_swapaxes_read_only_array  s
   
r   zmethod, idxc                 C   s   | j ddj ddS )NFr*   r:   r   r'   r'   r(   r4     rH   c                 C   s   |     S r   )r   r   r'   r'   r(   r4     r^   c                 C   s   | j tjdj tjdS )Nr   )r0   r1   r   r2   r   r'   r'   r(   r4     s    c                 C   s   | j ddjddS )NFr*   number)include)r   r   r   r'   r'   r(   r4     rH   )shallow-copyr   r0   r   c                 C   s   t g dg dg dd}| }| o| jjjdk}||}d|jd|f< |s/t|| ||}d|jd< |sKt|jd d |d f | d S d S )Nr   r   r   r   r   r   r   )r   r   rz   r{   r|   r"   r   r   )r~   r-   idxr$   r%   r   df2_is_viewr   r'   r'   r(   test_chained_methods  s   
"r   objr   c                 C   s   t tdddtdddg|_| }| }| r(tt|dt|ds'J ntt|dt|dr6J d|jd< tt|dt|drIJ t	
|| d S )Nz2012-1-1rb   rc   z2012-1-2r   r   )r   r   rD   r   r`   r   r   r   r"   r   assert_equalr$   r   obj_origobj2r'   r'   r(   test_to_timestamp  s   
r   c                 C   s   t tdtdg|_| }|jdd}| r&tt|dt|ds%J ntt|dt|dr4J d|jd< tt|dt|drGJ t	
|| d S )N
2019-12-31
2020-12-31Yr   r   r   )r   r   rD   r   re   r   r   r   r"   r   r   r   r'   r'   r(   test_to_period  s   
r   c                 C   s   t g dg dg dd}| }|d}| r(tt|dt|ds'J ntt|dt|dr6J d|jd< tt|d	t|d	rIJ t|| d S )
Nr   r   r   r   r   r   r   r   r   )	r   r   	set_indexr   r   r   r"   r   r   r   r'   r'   r(   test_set_index  s   

r   c                  C   s>   t g ddd} | d}| }d| jd< t|| d S )Nr   r   r   r   r   r   )r   r   r   r"   r   r   r%   r   r   r'   r'   r(   4test_set_index_mutating_parent_does_not_mutate_index  s
   

r  c                 C      t g dg dg dd}| }|d}| r'tt|dt|ds'J d|jd	< tt|dt|dr:J | rJtt|d
t|dsJJ t g dg dg dd}t|| t|| d S )Nr   r   r   r   CoW_CoW_ar   r   r   CoW_cr   r   )r  CoW_br  )	r   r   
add_prefixr   r   r   r"   r   r   r   r'   r'   r(   test_add_prefix  s   

r  c                 C   r  )Nr   r   r   r   _CoWa_CoWr   r   r   c_CoWr   r   )r
  b_CoWr  )	r   r   
add_suffixr   r   r   r"   r   r   r   r'   r'   r(   test_add_suffix  s   

r  z	axis, val)r         @c                 C   s   t g dd|dgdd}| }|j|d}| r(tt|dt|ds'J ntt|dt|dr6J d|jd	< | rKtt|dt|drKJ t|| d S )
Nr   r   r   dr   r   r   r   r   )	r   r   dropnar   r   r   r"   r   r   )r$   rE   valr%   r   r   r'   r'   r(   test_dropna   s   
r  r  r   r  c                 C   s~   t d|dg}| }| }| rt|j|jsJ n
t|j|jr&J d|jd< | r7t|j|jr7J t|| d S )Nr   r   r   )	r   r   r  r   r   r   r"   r   r   )r$   r  r   r   r   r'   r'   r(   test_dropna_series1  s   
r  c                 C      |   S r   headr   r'   r'   r(   r4   E      c                 C   
   |  dS )Nr   r  r   r'   r'   r(   r4   F  r   c                 C   r  r   tailr   r'   r'   r(   r4   G  r  c                 C   r  )Nr   r  r   r'   r'   r(   r4   H  r   c                 C   s   t g dg dd}| }| |}|j  |r6tt|dt|ds(J tt|dt|ds6J d|jd< tt|dt|dsIJ |rZtt|dt|drYJ nd|jd< t	|| d S )	Nr   r   r   r   r   r   r   r   )
r   r   r   r   r   r   r   r"   r   r   )r-   r$   r%   r   r   r'   r'   r(   test_head_tailB  s   



r  c                 C   s   t ddgdddd}| }| }| r2tt|dt|ds#J tt|dt|ds1J ntt|dt|dr@J tt|dt|drNJ d|jd	< d
|jd< | rvtt|dt|drhJ tt|dt|drvJ t|| d S )Nr   r   r   r   r   r   r   r  r   r   r   r   r  r   )	r   r   r]   r   r   r   r"   r   r   r   r'   r'   r(   test_infer_objects`  s   

r  c                 C   s   t ddgddttdtdgdddd	}| }t|d
}t|d}t|d}d|jd< d|jd< td|jd< | r]t|t|d
sGJ t|t|drRJ t|t|ds_J d S d S )Nr   r   r   r   r   objectdtyper   )r   r   r   r  er   r  r   r   r   
2018-12-31r   r   r   r   r   r]   r   r"   r   r   )r$   r%   arr_aarr_barr_dr'   r'   r(   test_infer_objects_no_referenceu  s.   




r)  c                 C   s   t ddgddttdtdgddd}|d d  }| }t|d	}t|d
}t|d}d|jd< d|jd< td|jd< | rbt|t|d	rLJ t|t|d
rWJ t|t|dsdJ d S d S )Nr   r   r   r   r   r  r   r  r   r   r  r   r   r   r#  r$  r%  )r$   r%   viewr&  r'  r(  r'   r'   r(   test_infer_objects_reference  s.   





r+  kwargsr   )beforeafterrE   c                 C   s   t g dddd}| }|jdi |}|j  | r,tt|dt|ds+J ntt|dt|dr:J d|jd< | rOtt|dt|drOJ t	
|| d S )	Nr   r   r   r   r   r   r   r'   )r   r   rZ   r   r   r   r   r   r"   r   r   r$   r,  r%   r   r   r'   r'   r(   test_truncate  s   

r0  assigndrop_duplicatesc                 C   s   t dg di}| }t|| }|j  | r(tt|dt|ds'J ntt|dt|dr6J d|jd< | rKtt|dt|drKJ t	
|| d S )Nr   r   r   r   )r   r   getattrr   r   r   r   r   r"   r   r   )r$   r-   r%   r   r   r'   r'   r(   test_assign_drop_duplicates  s   

r4  c                 C   sv   |  }|ddg}| rt|j|jsJ n
t|j|jr"J d|jd< | r3t|j|jr3J t|| d S )Nr   r   )r   taker   r   r   r"   r   r   r   r'   r'   r(   	test_take  s   
r6  c                 C   s   t dddd|_| }|dd}| rt|j|jsJ n
t|j|jr)J d|jd< | r:t|j|jr:J t	|| d S )Nz
2018-04-09r   1D20minr   z0:00z1:00r   )
r	   rD   r   between_timer   r   r   r"   r   r   r   r'   r'   r(   test_between_time  s   
r9  c                 C   s   t ddgdd}t dddgd}| }||}| r,tt|dt|ds+J ntt|dt|dr:J d|jd< | rOtt|dt|drOJ t|| d S )Nr   r   r   r   )r   r   r   r   )	r   r   r<   r   r   r   r"   r   r   )r$   r%   otherr   r   r'   r'   r(   test_reindex_like  s   

r;  c                 C   sx   t g d}| }| }| rt|j|jsJ n
t|j|jr%J d|jd< t|j|jr4J t|| d S )Nr   r   )	r   r   
sort_indexr   r   r   r"   r   r   r$   r   r   r   r'   r'   r(   test_sort_index
  s   
r>  zobj, kwargsr   byc                 C   s   |  }|jdi |}| rtt|dt|dsJ ntt|dt|dr+J d|jd< tt|dt|dr>J t|| d S )Nr   r   r'   r   sort_valuesr   r   r   r"   r   r   )r$   r   r,  r   r   r'   r'   r(   test_sort_values  s   
rB  c                 C   s   |  }|d d  }|jdddi| tt|dt|ds"J d|jd< | r?tt|dt|dr7J t|| d S tt|dt|dsMJ d S )NinplaceTr   r   r'   r@  )r$   r   r,  r   r   r*  r'   r'   r(   test_sort_values_inplace.  s   
 rD  decimals)r   r   c                 C   s  t ddgdd}| }|j|d}| rEtt|dt|ds#J |dkr6tt|dt|ds5J ntt|dt|drDJ ntt|dt|drSJ d	|jd
< d|jd< | r{tt|dt|drmJ tt|dt|dr{J t|| d S )Nr   r   r   r   )rE  r   r   r   r  r   r   r   )	r   r   roundr   r   r   r"   r   r   )r$   rE  r%   r   r   r'   r'   r(   
test_roundB  s   

rH  c                 C   s   t jg dddgd}tdg di|d}| }|jddgd}| r2tt|dt|ds1J ntt|dt|dr@J d	|jd
< | rUtt|dt|drUJ t	
|| d S )Nr   r   r   r   r   r   )r   r   onetwonamesr   r   r   r   r   ry   orderr   r   )r   from_tuplesr   r   reorder_levelsr   r   r   r"   r   r   r$   rD   r%   r   r   r'   r'   r(   test_reorder_levels[  s   
rW  c                 C   s   t jg dddgd}tg d|d}| }|jddgd}| r,t|j|js+J n
t|j|jr6J d|jd< | rGt|j|jrGJ t	
|| d S )	NrI  rM  rN  rO  rQ  ry   rR  r   )r   rT  r   r   rU  r   r   r   r"   r   r   r   r'   r'   r(   test_series_reorder_levelsn  s   
rX  c                 C   s   t jg dddgd}||_| }| }| r#t|j|js"J n
t|j|jr-J d|jd< | r>t|j|jr>J t	
|| d S )NrJ  rK  rL  rM  rN  rO  r   )r   rT  rD   r   r   r   r   r   r"   r   r   )r$   r   rD   r   r   r'   r'   r(   test_swaplevel  s   
rZ  c                 C   s   t g dg dg dd}| }|jg ddd}| r,tt|dt|ds+J ntt|dt|dr:J d|jd	< tt|dt|drMJ t|| d S )
Nr   r   r   r   rD   r   r   r   r   )	r   r   rG   r   r   r   r"   r   r   r   r'   r'   r(   test_frame_set_axis  s   
r[  c                 C   sv   t g d}| }|jg ddd}| rt||sJ nt||r&J d|jd< t||r3J t|| d S )Nr   r   rD   r   r   )r   r   rG   r   r   r"   r   r   r=  r'   r'   r(   test_series_set_axis  s   
r\  c                 C   s   t g d}| }|jdd}t||sJ d|jd< | r/t||r'J t|| d S t||s7J t g d}t|| d S )Nr   F)rm   r   r   )r   r   ro   r   r   r"   r   r   r$   r   r   r   r   r'   r'   r(   test_set_flags  s   
r^  mapperrK   c                 C   s   t dg ditg dddd}| }|jdi |}| r-tt|dt|ds,J ntt|dt|dr;J d|jd< | rPtt|dt|drPJ t	|| d S )Nr   rQ  r   ry   r   r   r'   )
r   r   r   rN   r   r   r   r"   r   r   r/  r'   r'   r(   test_rename_axis  s    
r`  zfunc, tz)rk   zEurope/Berlin)ri   Nc                 C   s   t ddgtddd|dd}| }t||d}| r&t|j|js%J n
t|j|jr0J d|jd< t|j|jr?J t	|| d S )	Nr   r   z2014-08-01 09:00H)startrc   ru   rx   ry   rg   r   )
r   r	   r   r3  r   r   r   r"   r   r   )r$   r   rx   r   r   r   r'   r'   r(   test_tz_convert_localize  s   
rc  c                 C   s   t jg dddgd}tg dg dg dd|d	}| }|d
}| r5tt|dt|ds4J ntt|dt|drCJ d
|jd< tt|dt|drVJ t	
|| d S )NrY  rM  rN  rO  r   r         	   r   ry   r   r   r   )r   rT  r   r   	droplevelr   r   r   r"   r   r   rV  r'   r'   r(   test_droplevel  s    

ri  c                 C   s   t dg di}| }| }t|jt|dsJ d|jd< | r7t|jt|dr/J t	|| d S t|jt|dsCJ |j
d dksLJ d S )Nr   r   r   r   )r   r   squeezer   r   r   r   r"   r   r   r   )r$   r%   r   seriesr'   r'   r(   test_squeeze  s   
rl  c                 C   s   t g dg dg dd}| }tdD ]@}| D ]9\}}tt||t||s.J d|jd< | rJtt||t||rCJ t	|| q|j
d|f dksUJ qqd S )Nr   r   rd  r   r   r   )r   r   ranger   r   r   r   r"   r   r   r   )r$   r%   r   ir   r   r'   r'   r(   
test_items  s   
ro  r!  rP   Int64c                 C   s   t ddgddd|d}|d d  }| }d|||k< | r4tt|dt|dr,J t|| d S tt|dt|dsBJ |jd dksKJ d S )Nr   r   r   r   r   r   r   )r   r   r   r   r   r   r   r"   )r$   r!  r%   r*  r   r'   r'   r(   test_putmask)  s   rq  c                 C   sP   t ddgddd|d}t|d}d|||k< | r$t|t|ds&J d S d S )Nr   r   r   r   r   r   r   r   r   r   r$   r!  r%   r&  r'   r'   r(   test_putmask_no_reference9  s   
rt  float64Float64c                 C   sZ   t ddgdd|d}t|d}t dddgi|||k< | r)t|t|ds+J d S d S )Nr   r   r   r   r   r  r   rr  rs  r'   r'   r(   $test_putmask_aligns_rhs_no_referenceC  s   
rw  zval, exp)r  T)r   Fc                 C   s6  t ddgddd}|d d  }| }t g dg dgtdd}|||< | rntt|dt|dr6J tt|d	t|d	|u sFJ tt|d
t|d
sTJ |jd|us^J |jdrfJ t	|| d S |dkrtt|dt|dsJ tt|d
t|d
sJ |j
d dksJ d S d S )Nr   r   r   r   )TFFabcr   r   r   r   r   r   )r   r   listr   r   r   r   r   r   r   r"   )r$   r  expr%   r*  r   indexerr'   r'   r(   "test_putmask_dont_copy_some_blocksM  s(    r|  c                 C   s   t g d|d}| }||dkd}| r#tt|t|s"J ntt|t|r/J d|jd< | rBtt|t|rBJ t|| d S Nr   r   r   r   )	r   r   wherer   r   r   r"   r   r   r$   r!  r   r   r   r'   r'   r(   test_where_noopf  s   
r  c                 C   sP   t g d|d}| }||dk d}tt|t|r J t|| d S r}  )r   r   r~  r   r   r   r   r   r  r'   r'   r(   
test_wherex  s
   r  z
dtype, val)rP   g      %@)rp  r   c                 C   s   t g dg dd|d}| }||dk |}| r8tt|dt|ds)J tt|dt|dr7J ntt|dt|drFJ d|jd	< | r[tt|dt|dr[J t|| d S )
Nr   )ir   r   r   r   r   r   r   )	r   r   r~  r   r   r   r"   r   r   )r$   r!  r  r%   r   r   r'   r'   r(    test_where_noop_on_single_column  s   
r  c                 C   s   t dg ditddddd}| }|jdd}| r*tt|dt|ds)J ntt|dt|dr8J d	|jd
< tt|dt|drKJ t	|| d S )Nr   )g        N       @      @z1/1/2000r   Tr   ry   r   r   r   )
r   r	   r   asfreqr   r   r   r"   r   r   r   r'   r'   r(   test_asfreq_noop  s   

r  c                 C   sR   t dddg dd}| }| D ]	\}}d|jd< q| r't|| d S d S )Nr   r   r   r   ry   r   )r   r   iterrowsr"   r   r   )r$   r%   r   r   subr'   r'   r(   test_iterrows  s   r  c                 C   sr   t ddtjdgi}|d d  }| }|jdd d|jd< | r)t|| d S t dg di}t|| d S )	Nr   r   r   T)rC        Y@r   )r  r   r   )r   r   r   r   ffillr"   r   r   )r$   r%   r*  r   r'   r'   r(   test_interpolate_creates_copy  s   
r  c                 C   s  t g dg dg dd}| }|jd d}|dtg d | rBtt|dt|ds3J tt|d	t|d	sAJ ntt|dt|drPJ tt|d	t|d	r^J d
|jd< t	|| | r{tt|dt|dsyJ d S tt|dt|drJ d S )Nr   r   rd  r   r*   r   )rF  r   r   r   r   )
r   r   isetitemr   arrayr   r   r   r   r   r   r'   r'   r(   test_isetitem  s   
  r  zsingle-blockzmixed-blockc                 C   s   t g dtjg d|dd}tg d}| }|d| | r/tt|dt|r/J d|jd< t	
|| t g dtjg d|dd}tg d}|d| d|jd< t g dtjg d|dd}t	|| d S )	Nr   r   r   r   rd  r   r   r   )r   r   r  r   r   r  r   r   r   r   r   r   )r$   r!  r%   r   r   r   r'   r'   r(   test_isetitem_series  s    
 
 r  keyc              	   C   s   t g dg dd}| }||}| r?tt|dt|ds$J d|jd< tt|dt|dr7J t|| d S t	|t
rFtnd }tdd" t| d|jd< W d    n1 sdw   Y  W d    n1 ssw   Y  t	|t
rt|| d S |jd dksJ d S )	Nr   r   r   r   r   chained_assignmentwarnr   )r   r   getr   r   r   r"   r   r   
isinstancery  r   pdoption_contextassert_produces_warning)r$   r  r%   r   r   r  r'   r'   r(   test_get  s$   


r  z	axis, keyr   )r   r   c           
   	   C   s@  |dko| }|p|o|dk}t g dg dtjg d|dd}| }|j||d}	|dks3|rAtt|d	t|	s@J n
| rK|	jd
sKJ | sO|rUd
|	j	d
< n0t
dd" tt d
|	j	d
< W d    n1 sqw   Y  W d    n1 sw   Y  | s|s|d
krt|| d S |j	d d
ksJ d S )NrP   r   r   r   rd  r   r   r   r   r   r  r  r   )r   r   r  r   xsr   r   r   r   r"   r  r  r   r  r   r   )
r$   r   rE   r  r!  single_blockis_viewr%   r   r   r'   r'   r(   test_xs  s,    r  rE   z
key, level)l1r   rL  c              	   C   s  t ddd}tjddgg dgddgd	}t||td
d}|dkr+|  }| }|j	|||d}	|dkrOt 
t||jd t|	|	jd sOJ | sU|sUtnd }
tdd" t|
 d|	jd< W d    n1 ssw   Y  W d    n1 sw   Y  t|| d S )N   r   r   r  l2r   lev1lev2rO  rx  r   r   )levelrE   r   r  r  r   )r   arangereshaper   from_productr   ry  r   r   r  r   r   r/   r   r  r  r   r  r"   r   )r$   r   r  r  rE   arrrD   r%   r   r   r  r'   r'   r(   test_xs_multiindex/  s0   r  c                 C   s   t g dg dd}t ddgidgd}| }|d d  }|| t g dg dd}t|| | r[t|| tt|d	t|d	sKJ tt|dt|drYJ d S t|| d S )
N      ?r  r  )      @g      @      @r   r         Y@r   ry   )r  r  r  r   )r   r   updater   r   r   r   r   )r$   df1r   df1_origr*  r   r'   r'   r(   test_update_frameL  s   
 r  c                 C   sv   t g d}t dgdgd}| }|d d  }|| t g d}t|| | r3t|| d S t|| d S )Nr  r  r   ry   )r  r  r  )r   r   r  r   r   )r$   ser1r   	ser1_origr*  r   r'   r'   r(   test_update_series_  s   
r  c                  C   sD   t g d} t| }| d9 } tt| |sJ t|t|  d S Nr   r   )r   r   r   r   r   assert_numpy_array_equal)r   datar'   r'   r(   test_inplace_arithmetic_seriesp  s
   r  c                 C   sp   t g d}| }|d d  }|d9 }| r*tt|t|r"J t|| d S tt|t|s6J d S r  )r   r   r   r   r   r   r   )r$   r   r   r*  r'   r'   r(   -test_inplace_arithmetic_series_with_referencex  s   r  TFc                 C   s   t g ddd}| }|j|d}|s|r| r(tt|dt|ds'J ntt|dt|dr6J d|jd< | rEt|| d S d S )	Nr   r   r   r:   r   r   r   r   )	r   r   r   r   r   r   r"   r   r   )r$   r   r   r%   r   r   r'   r'   r(   test_transpose  s   
r  c                 C   s^   t g ddd}| }|j}tt|dt|drJ d|jd< | r-t|| d S d S )Nr   r   r   r   r   r   r   )	r   r   r  r   r   r   r"   r   r   )r$   r%   r   r   r'   r'   r(   test_transpose_different_dtypes  s   
r  c                 C   s:   t dg didd}|j}tt|dt|drJ d S )Nr   r   rp  r   r   )r   r  r   r   r   )r$   r%   r   r'   r'   r(   test_transpose_ea_single_column  s    r  c                  C   sH   t ddgdd} |  }d|jd< tddgddgd	}t|| d S )
Nr   r   r   r   r   r   r   r   ry   )r   countr"   r   r   r   r   r'   r'   r(   test_count_read_only_array  s
   
r  c                 C   s   t g d}| }| }tt|t|sJ | r$|jdr$J d|jd< | r3t	
|| d S t g d}t	
|| d S )Nr   r   r   r   )r   r   r*  r   r   r   r   r   r"   r   r   r]  r'   r'   r(   test_series_view  s   
r  )mnumpyr   pytestpandas.errorsr   pandasr  r   r   r   r   r   r   r	   r
   pandas._testing_testingr   pandas.tests.copy_view.utilr   r)   r,   markparametrizer   r   r   r   
RangeIndexr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r  r  r  r   r  r  r  r  r)  r+  r0  r4  r6  r9  r;  r>  rB  rD  rH  rW  rX  rZ  r[  r\  r^  r`  rc  ri  rl  ro  rq  rt  rw  r|  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r'   r'   r'   r(   <module>   s   (
))
 


"






(
(
	


	



$
$
*
*

$




	
	







