a
    ds                    @   sB	  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"ejjd9dHd dId dJd dKd gg dLddMdN Z#dOdP Z$dQdR Z%ejdSdTdUgidVdUidWdUigdXdY Z&dZd[ Z'd\d] Z(d^d_ Z)d`da Z*dbdc Z+ejddded dfd dgd gdhdi Z,djdk Z-dldm Z.dndo Z/dpdq Z0ejdrd9dsgdtdu Z1dvdw Z2dxdy Z3ejjdzd{d d fd|d d:fd}d d fd~d d fgg dddd Z4ejded;d:gdUdedUd;d:gigdd Z5ejded;d:gdUdedUd;d:gigdd Z6dd Z7dd Z8dd Z9dd Z:e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 ZAejddUdd;dd d;d dgdd ZBejdddgdd ZCejded;d:gedUd;d:gigdd ZDejded;d:gedUd;d:gigdd ZEdd ZFdd ZGejdeg ddUdi fedUg diddUifgdd ZHejdeg ddUdi fedUg diddUifgdd ZIejdg ddd ZJdd ZKddĄ ZLejdeg dedUg digddƄ ZMddȄ ZNddʄ ZOdd̄ ZPejdddid9digddЄ ZQejdddgddՄ ZRddׄ ZSddل ZTddۄ ZUejdddgdd ZVejdddgdd ZWejdddgdd ZXejdddeYfdgdd ZZejdddgejdddd dd gdd Z[ejdddgejdddd dd gdd Z\ejdddgejdddd dd gdd Z]ejddddgdd Z^dd  Z_dd Z`dd Zadd Zbejjdddgddgdd	d
 Zcdd ZdejddUdUggdd Zeejdddgejjdddgddgddd Zfejdd d;gejdddgdd Zgdd Zhdd Zidd  Zjd!d" Zkd#d$ Zlejd	dd%gd&d' Zmd(d) Znd*d+ Zod,d- Zpd.d/ Zqd0d1 Zrd2d3 Zsd4d5 Ztd6d7 Zud8d9 Zvd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}| }|j|jus4J |j|jusDJ |j|jsVJ |j|jshJ tt|dt|drJ | r|jj	d j
 rJ |jj	d j
 rJ d|jd< |jd dksJ d S )	N                  g?g?g333333?abcr   r   r   r   r   r   copyindexcolumnsis_npshares_memoryr   _mgrblocksrefsZhas_referenceilocusing_copy_on_writedfZdf_copy r(   l/var/www/html/stable-diffusion-webui/venv/lib/python3.9/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}| rr|j|jus<J |j|jusLJ |j|js^J |j|jsJ n |j|ju sJ |j|ju sJ tt|dt|dsJ | r|jj	d j
 sJ |jj	d	 j
 sJ | r@d|jd
< |jd
 d	ksJ tt|dt|dr J tt|dt|ds|J n<d|jd
< |jd
 dks^J tt|dt|ds|J 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r   r   )renamestrlowerr'   r   r(   r(   r)   <lambda>R       r4   c                 C   s   | j ddg|dS )Nr   r   r/   )reindexr3   r(   r(   r)   r4   S   r5   c                 C   s   | j | |dS Nr   reindex_liker3   r(   r(   r)   r4   T   r5   c                 C   s   | j | |dd S Nr8   r   alignr3   r(   r(   r)   r4   U   r5   c                 C   s   | j g dd|dS Nr   r   )axisr   set_axisr3   r(   r(   r)   r4   V   r5   c                 C   s   | j d|dS Ntestr   r   rename_axisr3   r(   r(   r)   r4   W   r5   c                 C   s   | j d|dS )NrC   r/   rE   r3   r(   r(   r)   r4   X   r5   c                 C   s   | j ddi|dS )Nr   int64r8   astyper3   r(   r(   r)   r4   Y   r5   c                 C   s   | j dd|dS Nr   r8   swapaxesr3   r(   r(   r)   r4   [   r5   c                 C   s   | j dd|dS Nr   r   r8   truncater3   r(   r(   r)   r4   \   r5   c                 C   s   | j |dS r7   infer_objectsr3   r(   r(   r)   r4   ]   r5   c                 C   s   | j |dS r7   to_timestampr3   r(   r(   r)   r4   ^   r5   c                 C   s   | j d|dS ND)freqr   	to_periodr3   r(   r(   r)   r4   _   r5   c                 C   s   | j d|dS N
US/Centralr8   tz_localizer3   r(   r(   r)   r4   `   r5   c                 C   s   | j d|dS rY   
tz_convertr3   r(   r(   r)   r4   a   r5   c                 C   s   | j d|dS NF)allows_duplicate_labelsr   	set_flagsr3   r(   r(   r)   r4   b   r5   )r0   r6   r:   r=   rA   rename_axis0Zrename_axis1rI   rL   rO   rQ   rS   rX   r\   r^   rb   )idsc           
      C   sx  d }d| j jjv r"tdddd}nZd| 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| j jjv rd}tjt|d |||d}W d    q1 s0    Y  n|||d}|p|du }	| j jj	dr0|s0|s0|du r0d}	|	rVt
t|dt|dstJ nt
t|dt|drtJ d S )NrS   
2012-01-01rU   r   rV   periodsrX   r\   r^   Europe/BrusselsrV   rg   tzr   r   r   r   r   rL   "'DataFrame.swapaxes' is deprecatedmatchr8   Fzreindex-r   )nodecallspecidr
   r	   r   tmassert_produces_warningFutureWarning
startswithr   r    r   )
requestr.   r   r&   using_array_managerr   r'   msgdf2share_memoryr(   r(   r)   test_methods_copy_keywordN   s,    - , r{   c                 C   s   | j ddi|dS )Nr   d   rD   r0   serr   r(   r(   r)   r4      r5   c                 C   s   | j d |dS r7   r}   r~   r(   r(   r)   r4      r5   c                 C   s   | j | j|dS )NrD   )r6   r   r~   r(   r(   r)   r4      r5   c                 C   s   | j | |dS r7   r9   r~   r(   r(   r)   r4      r5   c                 C   s   | j | |dd S r;   r<   r~   r(   r(   r)   r4      r5   c                 C   s   | j g dd|dS r>   r@   r~   r(   r(   r)   r4      r5   c                 C   s   | j d|dS rB   rE   r~   r(   r(   r)   r4      r5   c                 C   s   | j d|dS )NrG   r8   rH   r~   r(   r(   r)   r4      r5   c                 C   s   | j dd|dS )Nr   r   r8   )	swaplevelr~   r(   r(   r)   r4      r5   c                 C   s   | j dd|dS rJ   rK   r~   r(   r(   r)   r4      r5   c                 C   s   | j dd|dS rM   rN   r~   r(   r(   r)   r4      r5   c                 C   s   | j |dS r7   rP   r~   r(   r(   r)   r4      r5   c                 C   s   | j |dS r7   rR   r~   r(   r(   r)   r4      r5   c                 C   s   | j d|dS rT   rW   r~   r(   r(   r)   r4      r5   c                 C   s   | j d|dS rY   r[   r~   r(   r(   r)   r4      r5   c                 C   s   | j d|dS rY   r]   r~   r(   r(   r)   r4      r5   c                 C   s   | j d|dS r_   ra   r~   r(   r(   r)   r4      r5   )zrename (dict)r0   r6   r:   r=   rA   rc   rI   r   rL   rO   rQ   rS   rX   r\   r^   rb   c           	      C   s\  d }d| j jjv r"tdddd}nd| j jjv r@tdddd}nbd| j jjv r^tdddd}nDd| j jjv r~tdddd	d
}n$d| j jjv rtg dg dg}tg d|d}d| j jjv rd}tj	t
|d |||d}W d    n1 s0    Y  n|||d}|p|du }|r>tt|t|sXJ ntt|t|rXJ d S )NrS   re   rU   r   rf   rX   r\   r^   rh   ri   r   r   r   rk   rL   z'Series.swapaxes' is deprecatedrm   r8   F)ro   rp   rq   r
   r	   r   Zfrom_arraysr   rr   rs   rt   r   r    r   )	rv   r.   r   r&   r   r   rx   ser2rz   r(   r(   r)    test_methods_series_copy_keyword   s*    +,r   c                 C   s   t g dg dd}|j|d}| p4|du p4|d u }|o>| }|rbtt|dt|ds~J ntt|dt|dr~J d S )Nr   r   r   r   r8   Fr   r   )r   	transposer   r    r   )r&   r   rw   r'   resultrz   r(   r(   r)   test_transpose_copy_keyword   s    
r   c                 C   s   t g dg dg ddg dd}| }| }|j  | rztt|dt|ds^J tt|dt|dszJ d	|jd
< tt|dt|drJ | rtt|dt|dsJ t	
|| d S )Nr   r   r   r   )
         rk   r   r   r   )r   r   )r   r   reset_indexr!   _verify_integrityr   r    r   r$   rr   assert_frame_equalr&   r'   df_origry   r(   r(   r)   test_reset_index   s    

r   r   r   r   c                 C   s   t ddg|d}| }|jdd}| rRtt|t|s@J |jdrjJ ntt|t|rjJ d|jd< t	
|| d S )Nr   r   rk   T)dropr   r|   )r   r   r   r   r    r   r!   _has_no_referencer$   rr   assert_series_equalr&   r   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}| rRtt|dt|dsRJ d|jd	< tt|dt|drxJ | rtt|d
t|dsJ 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$   rr   r   r&   r'   r   ry   expectedr(   r(   r)   test_rename_columns  s    
r   c                 C   s   t g dg dg dd}|jtjd}| }| rTtt|dt|dspJ ntt|dt|drpJ d|jd	< tt|dt|drJ | rtt|d
t|dsJ 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$   rr   r   )r&   r'   ry   Z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sHJ d|jd< | rt|| tt|dt|drJ n:t g d	dd}t|| tt|dt|dsJ tt|d
t|d
sJ d S )Nr         ?r   c                 S   s   | S Nr(   r'   r(   r(   r)   testfuncC  s    ztest_pipe.<locals>.testfuncr   r   r   r   r   )	r   r   piper   r    r   r$   rr   r   r&   r'   r   r   ry   r   r(   r(   r)   	test_pipe?  s    

r   c                 C   s   t g ddd}| }dd }||}tt|dt|dsHJ | rvt|| tt|dt|drJ n:t g ddd}t|| tt|dt|dsJ tt|dt|dsJ 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   rr   r   r   r(   r(   r)   test_pipe_modify_dfW  s    
r   c                 C   s   t g dg dg dd}| }|jddgd}| rVtt|dt|dsrJ ntt|dt|drrJ d|jd	< tt|dt|drJ | rtt|dt|dsJ t|| d S )
Nr   r   r   r   r   r   r   r   r   )	r   r   r6   r   r    r   r$   rr   r   r   r(   r(   r)   test_reindex_columnsn  s    
r   c                 C   s   | S r   r(   idxr(   r(   r)   r4     r5   c                 C   s   |   S r   )viewr   r(   r(   r)   r4     r5   c                 C   s   |   S r   r8   r   r(   r(   r)   r4     r5   c                 C   s   t | S r   )listr   r(   r(   r)   r4     r5   )Z	identicalr   r   valuesc                 C   s   t g dg dg dd}| }|j| |jd}|rXtt|dt|dstJ ntt|dt|drtJ d|jd< tt|dt|drJ |rtt|d	t|d	sJ t	|| d S )
Nr   r   r   r   rk   r   r   r   r   )
r   r   r6   r   r   r    r   r$   rr   r   )r   r&   r'   r   ry   r(   r(   r)   test_reindex_rows  s    
r   c                 C   s  t g dg dg dd}| }|jdd}|j  | rxtt|dt|dsZJ tt|dt|dsJ n8tt|dt|drJ tt|dt|drJ d	|jd
< tt|dt|drJ | rtt|dt|dsJ t	
|| d S )Nr   r   r   r   r   r   r   r   r   r   )r   r   r   r!   r   r   r    r   r$   rr   r   r   r(   r(   r)   test_drop_on_column  s    

r   c                 C   s   t g dg dg dd}| }|d}|j  | rZtt|dt|dsvJ ntt|dt|drvJ d|jd< | rtt|dt|drJ t	
|| d S )	Nr   r   r   r   rG   r   r   r   )r   r   select_dtypesr!   r   r   r    r   r$   rr   r   r   r(   r(   r)   test_select_dtypes  s    


r   filter_kwargsitemsr   likeregexc                 C   s   t g dg dg dd}| }|jf i |}| rVtt|dt|dsrJ ntt|dt|drrJ | rd|jd< tt|dt|drJ t|| d S )Nr   r   r   r   r   r   r   )	r   r   filterr   r    r   r$   rr   r   )r&   r   r'   r   ry   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}| r`tt|d	t|d	s|J ntt|d	t|d	r|J d|jd< | rtt|d	t|d	rJ tt|d
t|d
sJ t	|| d S )Nr   r   r   r   r   r   
2020-01-01
2020-01-03r   r   r   r   r   )rg   r   )
r   r	   r   shiftr   r    r   r$   rr   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	rTJ d S )Nr   r   r   r   r   r   r   r   r   r   r   r   rg   r?   )r   r	   r   r   r    r   )r&   r'   ry   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}| rntt|d	t|d	sJ ntt|d	t|d	rJ d|jd< | rtt|d	t|d	rJ t	
|| d S )Nr   r   r   r   r   r   r   r   r   r   r   
2020-01-02z
2020-01-04Z1Drg   rV   r   r   )r   r	   r   r   r   r   r    r   r$   rr   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sNJ d|jd< | rtt|dt|drxJ 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$   nanrr   r   )r&   r'   ry   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sRJ tt|dt|dsnJ | rd|jd< t|jt|drJ d|jd< | rtt|dt|drJ t	|| n(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    r   r   r$   rr   r   )r&   r'   r   Zview_originalr   r   r(   r(   r)   test_pop$  s    


r   funcc                 C   s
   |  |S r   r<   xyr(   r(   r)   r4   <  r5   c                 C   s   | j |jddS )Nr   r?   )r=   r   r   r(   r(   r)   r4   =  r5   c                 C   s   | j |jjtdd ddS )Nr   r   r   )r=   r   r$   slicer   r(   r(   r)   r4   >  r5   c                 C   s   t g ddd}| }|ddg  }|||\}}| rZtt|dt|dsvJ ntt|dt|drvJ d|jd< | rtt|dt|drJ t|| d S )Nr   r   r   r   r   r   r   r   r   r    r   r$   rr   r   )r&   r   r'   r   Z
df_changedry   _r(   r(   r)   test_align_frame9  s    	
r   c                 C   s   t ddg}| }| }||\}}| rXt|j|jsBJ t|j|jsJ n(t|j|jrlJ t|j|jrJ d|jd< d|jd< | rt|j|jrJ t|j|jrJ t|| t|| d S )Nr   r   r   )	r   r   r=   r   r    r   r$   rr   r   )r&   r   r   Z	ser_otherr   Zser_other_resultr(   r(   r)   test_align_seriesR  s     

r   c                 C   s   t g dg dd}| }|j|dd\}}tt|dt|dsLJ tt|dt|dshJ | rd|jd	< t|| d|jd	< t|| d S )
Nr   r   r   Fr8   r   r   r   r   r   )	r   r   r=   r   r    r   locrr   r   )r&   r'   r   ry   Zdf3r(   r(   r)   test_align_copy_falseh  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	sfJ tt|d
t|d
sJ tt|dt|dsJ | rd|jd< t	|| d|jd< t
|| d S )Nr   r   r   r   nameFr   )r   r?   r   r   r   )r   r   r   r=   r   r    r   r   rr   r   r   )r&   r'   r   r   r   ry   r   r(   r(   r)   !test_align_with_series_copy_falsex  s    

r   c                 C   s   t g d}| }|d d   }t|jt|ds<J d|jd< | rpt|jt|drbJ t	|| n| }d|jd< t	|| |d d   }d|jd< | rt
||  n"|  }d|jd< t
|| d S )Nr   r   r   )r   r   to_framer   r    r   r   r$   rr   r   r   )r&   r   r   r'   r   r(   r(   r)   test_to_frame  s$    



r   axr   c                 C   s   t g dg dd}| }d}tjt|d |||}W d    n1 sR0    Y  | r~tt|dt|dsJ ntt|dt|drJ d|j	d< | rtt|dt|drJ t
|| d S )	Nr   r   r   rl   rm   r   r   r   r   r   rr   rs   rt   rL   r   r    r   r$   r   )r&   r   r'   r   rx   ry   r(   r(   r)   test_swapaxes_noop  s    *
r   c                 C   s   t g dg ddg dd}| }d}tjt|d |dd	}W d    n1 sZ0    Y  | rtt|d
t|dsJ ntt|d
t|drJ d|j	d< | rtt|d
t|drJ t
|| d S )Nr   r   r   )r   r   zrk   rl   rm   r   r   r   r   r   r   r   )r&   r'   r   rx   ry   r(   r(   r)   test_swapaxes_single_block  s    *
r   c                  C   s   t ddgdd} d}tjt|d | jddd	} W d    n1 sH0    Y  d
| jd< t d
dgddgdddgd}t| | d S )Nr   r   r   r   rl   rm   r   r   )Zaxis1Zaxis2r|   r   r   r   r   r   rk   )r   rr   rs   rt   rL   r$   r   )r'   rx   r   r(   r(   r)   test_swapaxes_read_only_array  s    ,
r   zmethod, idxc                 C   s   | j ddj ddS )NFr+   r8   r   r(   r(   r)   r4     r5   c                 C   s   |     S r   )r   r   r(   r(   r)   r4     r5   c                 C   s   | j tjdj tjdS )Nr   )r0   r1   r   r2   r   r(   r(   r)   r4     r5   c                 C   s   | j ddjddS )NFr+   number)include)r   r   r   r(   r(   r)   r4     r5   )shallow-copyr   r0   r   c                 C   s   t g dg dg dd}| }| o6| jjjdk}||}d|jd|f< |s^t|| ||}d|jd< |st|jd d |d f | d S )Nr   r   r   r   r   r   r   )r   r   ro   rp   rq   r$   rr   r   )rv   r.   r   r&   r'   r   Zdf2_is_viewry   r(   r(   r)   test_chained_methods  s    
r   objr   c                 C   s   t tdddtdddg|_| }| }| rPtt|dt|dslJ ntt|dt|drlJ d|jd< tt|dt|drJ t	
|| d S )Nz2012-1-1rU   rV   z2012-1-2r   r   )r   r   r   r   rS   r   r    r   r$   rr   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}| rLtt|dt|dshJ ntt|dt|drhJ d|jd< tt|dt|drJ t	
|| d S )N
2019-12-31
2020-12-31Yr   r   r   )r   r   r   r   rX   r   r    r   r$   rr   r   r   r(   r(   r)   test_to_period  s    
r   c                 C   s   t g dg dg dd}| }|d}| rPtt|dt|dslJ ntt|dt|drlJ d|jd< tt|d	t|d	rJ t|| d S )
Nr   r   r   r   r   r   r   r   r   )	r   r   	set_indexr   r    r   r$   rr   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$   rr   r   r'   r   r   r(   r(   r)   4test_set_index_mutating_parent_does_not_mutate_index1  s
    

r   c                 C   s   t g dg dg dd}| }|d}| rNtt|dt|dsNJ d|jd	< tt|dt|drtJ | rtt|d
t|dsJ t g dg dg dd}t|| t|| d S )Nr   r   r   r   ZCoW_CoW_ar   r   r   CoW_cr   r   )r   ZCoW_br   )	r   r   Z
add_prefixr   r    r   r$   rr   r   r   r(   r(   r)   test_add_prefix:  s    

r   c                 C   s   t g dg dg dd}| }|d}| rNtt|dt|dsNJ d|jd	< tt|dt|drtJ | rtt|d
t|dsJ t g dg dg dd}t|| t|| d S )Nr   r   r   r   Z_CoWa_CoWr   r   r   c_CoWr   r   )r   Zb_CoWr   )	r   r   Z
add_suffixr   r    r   r$   rr   r   r   r(   r(   r)   test_add_suffixO  s    

r   z	axis, val)r         @c                 C   s   t g dd|dgdd}| }|j|d}| rPtt|dt|dslJ ntt|dt|drlJ d|jd	< | rtt|dt|drJ t|| d S )
Nr   r   r   dr   r   r   r   r   )	r   r   dropnar   r    r   r$   rr   r   )r&   r?   valr'   r   ry   r(   r(   r)   test_dropnaa  s    
r   r   r   r   c                 C   s~   t d|dg}| }| }| r8t|j|jsLJ nt|j|jrLJ d|jd< | rnt|j|jrnJ t|| d S )Nr   r   r   )	r   r   r   r   r    r   r$   rr   r   )r&   r   r   r   r   r(   r(   r)   test_dropna_seriesr  s    
r   c                 C   s   |   S r   headr   r(   r(   r)   r4     r5   c                 C   s
   |  dS )Nr   r   r   r(   r(   r)   r4     r5   c                 C   s   |   S r   tailr   r(   r(   r)   r4     r5   c                 C   s
   |  dS )Nr   r  r   r(   r(   r)   r4     r5   c                 C   s   t g dg dd}| }| |}|j  |rltt|dt|drPJ tt|dt|drlJ d|jd< |rtt|dt|drJ tt|dt|drJ n&tt|dt|dsJ d|jd< t	|| d S )	Nr   r   r   r   r   r   r   r   )
r   r   r!   r   r   r    r   r$   rr   r   )r.   r&   r'   r   ry   r(   r(   r)   test_head_tail  s    



r  c                 C   s   t ddgdddd}| }| }| rdtt|dt|dsFJ tt|dt|dsJ n8tt|dt|drJ tt|dt|drJ d|jd	< d
|jd< | rtt|dt|drJ tt|dt|drJ t|| d S )Nr   r   r   r   r   r   r   r   r   r   r   r   r   r   )	r   r   rQ   r   r    r   r$   rr   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< | rt|t|d
sJ t|t|drJ t|t|dsJ 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   rQ   r   r$   r   r    )r&   r'   arr_aarr_barr_dr(   r(   r)   test_infer_objects_no_reference  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< | rt|t|d	rJ t|t|d
rJ t|t|dsJ 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_infer_objects_reference  s*    





r  kwargsr   )beforeafterr?   c                 C   s   t g dddd}| }|jf i |}|j  | rXtt|dt|dstJ ntt|dt|drtJ d|jd< | rtt|dt|drJ t	
|| d S )Nr   r   r   r   r   r   r   )r   r   rO   r!   r   r   r    r   r$   rr   r   r&   r  r'   r   ry   r(   r(   r)   test_truncate  s    

r  ZassignZdrop_duplicatesc                 C   s   t dg di}| }t|| }|j  | rPtt|dt|dslJ ntt|dt|drlJ d|jd< | rtt|dt|drJ t	
|| d S )Nr   r   r   r   )r   r   getattrr!   r   r   r    r   r$   rr   r   )r&   r.   r'   r   ry   r(   r(   r)   test_assign_drop_duplicates  s    

r  c                 C   sv   |  }|ddg}| r0t|j|jsDJ nt|j|jrDJ d|jd< | rft|j|jrfJ t|| d S )Nr   r   )r   Ztaker   r    r   r$   rr   r   r   r(   r(   r)   	test_take  s    
r  c                 C   s   t dddd|_| }|dd}| r>t|j|jsRJ nt|j|jrRJ d|jd< | rtt|j|jrtJ t	|| d S )Nz
2018-04-09r   Z1D20minr   z0:00z1:00r   )
r	   r   r   Zbetween_timer   r    r   r$   rr   r   r   r(   r(   r)   test_between_time+  s    
r  c                 C   s   t ddgdd}t dddgd}| }||}| rXtt|dt|dstJ ntt|dt|drtJ d|jd< | rtt|dt|drJ t|| d S )Nr   r   r   r   )r   r   r   r   )	r   r   r:   r   r    r   r$   rr   r   )r&   r'   otherr   ry   r(   r(   r)   test_reindex_like<  s    

r  c                 C   sx   t g d}| }| }| r6t|j|jsJJ nt|j|jrJJ d|jd< t|j|jrhJ t|| d S )Nr   r   )	r   r   Z
sort_indexr   r    r   r$   rr   r   r&   r   r   r   r(   r(   r)   test_sort_indexN  s    
r  zobj, kwargsr   Zbyc                 C   s   |  }|jf i |}| r:tt|dt|dsVJ ntt|dt|drVJ d|jd< tt|dt|dr|J t|| d S )Nr   r   r   Zsort_valuesr   r    r   r$   rr   r   )r&   r   r  r   r   r(   r(   r)   test_sort_values_  s    
r!  c                 C   s   |  }|d d  }|jf ddi| tt|dt|dsDJ d|jd< | r|tt|dt|drnJ t|| ntt|dt|dsJ d S )NinplaceTr   r   r   )r&   r   r  rw   r   r   r(   r(   r)   test_sort_values_inplacer  s    
r#  decimals)r   r   c                 C   s  t ddgdd}| }|j|d}| rtt|dt|dsFJ |dkrltt|dt|dsJ qtt|dt|drJ ntt|dt|drJ d	|jd
< d|jd< | rtt|dt|drJ tt|dt|drJ t|| d S )Nr   r   r   r   )r$  r   r   r   r   r   r   r   )	r   r   roundr   r    r   r$   rr   r   )r&   r$  r'   r   ry   r(   r(   r)   
test_round  s    

r'  c                 C   s   t jg dddgd}tdg di|d}| }|jddgd}| rdtt|dt|dsJ ntt|dt|drJ d	|jd
< | rtt|dt|drJ t	
|| d S )Nr   r   r   r   r   r   )r   r   onetwonamesr   r   r   r   r   rk   orderr   r   )r   from_tuplesr   r   reorder_levelsr   r    r   r$   rr   r   r&   r   r'   r   ry   r(   r(   r)   test_reorder_levels  s    
r6  c                 C   s   t jg dddgd}tg d|d}| }|jddgd}| rXt|j|jslJ nt|j|jrlJ d|jd< | rt|j|jrJ t	
|| d S )	Nr(  r,  r-  r.  r0  rk   r1  r   )r   r3  r   r   r4  r   r    r   r$   rr   r   r   r(   r(   r)   test_series_reorder_levels  s    
r7  c                 C   s   t jg dddgd}||_| }| }| rFt|j|jsZJ nt|j|jrZJ d|jd< | r|t|j|jr|J t	
|| d S )Nr)  r*  r+  r,  r-  r.  r   )r   r3  r   r   r   r   r    r   r$   rr   r   )r&   r   r   r   r   r(   r(   r)   test_swaplevel  s    
r9  c                 C   s   t g dg dg dd}| }|jg ddd}| rXtt|dt|dstJ ntt|dt|drtJ d|jd	< tt|dt|drJ t|| d S )
Nr   r   r   r   r   r   r   r   r   )	r   r   rA   r   r    r   r$   rr   r   r   r(   r(   r)   test_frame_set_axis  s    
r:  c                 C   sv   t g d}| }|jg ddd}| r<t||sLJ nt||rLJ d|jd< t||rfJ t|| d S )Nr   r   r   r   r   )r   r   rA   r   r    r$   rr   r   r  r(   r(   r)   test_series_set_axis  s    
r;  c                 C   s   t g d}| }|jdd}t||s0J d|jd< | r\t||rNJ t|| n(t||slJ t g d}t|| d S )Nr   F)r`   r   r   )r   r   rb   r   r    r$   rr   r   r&   r   r   r   r   r(   r(   r)   test_set_flags  s    
r=  ZmapperrC   c                 C   s   t dg ditg dddd}| }|jf i |}| rZtt|dt|dsvJ ntt|dt|drvJ d|jd< | rtt|dt|drJ t	|| d S )Nr   r0  r   rk   r   r   )
r   r   r   rF   r   r    r   r$   rr   r   r  r(   r(   r)   test_rename_axis  s     
r>  zfunc, tz)r^   zEurope/Berlin)r\   Nc                 C   s   t ddgtddd|dd}| }t||d}| rLt|j|js`J nt|j|jr`J d|jd< t|j|jr~J t	|| d S )	Nr   r   z2014-08-01 09:00H)startrV   rg   rj   rk   rZ   r   )
r   r	   r   r  r   r    r   r$   rr   r   )r&   r   rj   r   r   r   r(   r(   r)   test_tz_convert_localize  s    
rA  c                 C   s  t jg dddgd}tg dg dg dd|d	}| }|d
}| rtt|dt|dshJ tt|dt|dsJ n8tt|dt|drJ tt|dt|drJ d
|jd< tt|dt|drJ | rtt|dt|dsJ t	
|| d S )Nr8  r,  r-  r.  r   r         	   r   rk   r   r   r   r   r   )r   r3  r   r   Z	droplevelr   r    r   r$   rr   r   r5  r(   r(   r)   test_droplevel2  s     

rF  c                 C   s   t dg di}| }| }t|jt|ds8J d|jd< | rlt|jt|dr^J t	|| n*t|jt|dsJ |j
d dksJ d S )Nr   r   r   r   )r   r   Zsqueezer   r    r   r   r$   rr   r   r   )r&   r'   r   Zseriesr(   r(   r)   test_squeezeJ  s    
rG  c                 C   s   t g dg dg dd}| }tdD ]}| D ]r\}}tt||t||s\J d|jd< | rtt||t||rJ t	|| q8|j
d|f dks8J q8q,d S )Nr   r   rB  r   r   r   )r   r   ranger   r   r    r   r$   rr   r   r   )r&   r'   r   ir   r   r(   r(   r)   
test_items]  s    
rJ  r	  rG   Int64c                 C   s   t ddgddd|d}|d d  }| }d|||k< | rftt|dt|drXJ t|| n.tt|dt|dsJ |jd dksJ d S )Nr   r   r   r  r   r   r   )r   r   r   r    r   rr   r   r$   )r&   r	  r'   r   r   r(   r(   r)   test_putmaskr  s    rL  c                 C   sL   t ddgddd|d}t|d}d|||k< | rHt|t|dsHJ d S )Nr   r   r   r  r   r   r   r   r   r    r&   r	  r'   r  r(   r(   r)   test_putmask_no_reference  s
    
rO  float64ZFloat64c                 C   sV   t ddgdd|d}t|d}t dddgi|||k< | rRt|t|dsRJ d S )Nr   r   r   r  r   r   r   rM  rN  r(   r(   r)   $test_putmask_aligns_rhs_no_reference  s
    
rQ  zval, exp, warnT)r   FNc                 C   sh  t ddgddd}|d d  }| }t g dg dgtdd}tj|dd	 |||< W d    n1 sp0    Y  | r
tt|d
t|d
rJ tt|dt|d|u sJ tt|dt|dsJ |j	d|usJ |j	drJ t
|| nZ|dkrdtt|d
t|d
s2J tt|dt|dsPJ |jd dksdJ d S )Nr   r   r   r   )TFFabcr   zincompatible dtyperm   r   r   r   r   r   )r   r   r   rr   rs   r   r    r   r!   r   r   r$   )r&   r   expwarnr'   r   r   Zindexerr(   r(   r)   "test_putmask_dont_copy_some_blocks  s&    & 
rU  c                 C   s   |  | dkdS Nr   r   wherer   r(   r(   r)   r4     r5   c                 C   s   |  | dkdS rV  maskrY  r(   r(   r)   r4     r5   c                 C   s   t g d|d}| }||}| r>tt|t|sVJ ntt|t|rVJ d|jd< | r|tt|t|r|J t|| d S )Nr   r  r   r   )r   r   r   r    r   r$   rr   r   r&   r	  r   r   r   r   r(   r(   r)   test_where_mask_noop  s    	
r]  c                 C   s   |  | dk dS rV  rW  rY  r(   r(   r)   r4     r5   c                 C   s   |  | dkdS rV  rZ  rY  r(   r(   r)   r4     r5   c                 C   sH   t g d|d}| }||}tt|t|r8J t|| d S )Nr   r  r   r   r   r    r   rr   r   r\  r(   r(   r)   test_where_mask  s
    	r_  z
dtype, val)rG   g      %@)rK  r   c                 C   s   |  | dk |S Nr   rW  r'   r   r(   r(   r)   r4     r5   c                 C   s   |  | dk|S r`  rZ  ra  r(   r(   r)   r4     r5   c                 C   s   t g dg dd|d}| }|||}| rjtt|dt|dsLJ tt|dt|drJ ntt|dt|drJ d|jd< | rtt|dt|drJ t|| d S )	Nr   )ir   r  r   r   r   r   r   )r&   r	  r   r   r'   r   r   r(   r(   r)   %test_where_mask_noop_on_single_column  s    	

rd  r[  rX  c                 C   s   t g ddd}| }| rt 0 t|d ||d dkddd W d    n1 s\0    Y  t|| t 2 t|dg ||d dkddd W d    n1 s0    Y  t|| d S )	N)r   r   r   r   r   r   r   r   Tr"  )r   r   rr   raises_chained_assignment_errorr  r   )r&   r   r'   r   r(   r(   r)   test_chained_where_mask  s    
>
@rg  c                 C   s   t dg ditddddd}| }|jdd}| rTtt|dt|dspJ ntt|dt|drpJ d	|jd
< tt|dt|drJ t	|| d S )Nr   )g        N       @      @z1/1/2000r   Tr   rk   r   r   r   )
r   r	   r   Zasfreqr   r    r   r$   rr   r   r   r(   r(   r)   test_asfreq_noop  s    

rk  c                 C   sN   t dddg dd}| }| D ]\}}d|jd< q&| rJt|| d S )Nr   r   r   r   rk   r|   )r   r   Ziterrowsr$   rr   r   )r&   r'   r   r   subr(   r(   r)   test_iterrows  s    rm  c                 C   sp   t ddtjdgi}|d d  }| }|jdd d|jd< | rPt|| nt dg di}t|| d S )	Nr   r   r   Tre        Y@r   )rn  r   r   )r   r   r   r   Zffillr$   rr   r   )r&   r'   r   r   r(   r(   r)   test_interpolate_creates_copy#  s    
ro  c                 C   s  t g dg dg dd}| }|jd d}|dtg d | rtt|dt|dsfJ tt|d	t|d	sJ n8tt|dt|drJ tt|d	t|d	rJ d
|jd< t	|| | rtt|dt|dsJ ntt|dt|drJ d S )Nr   r   rB  r   r+   r   )r%  r   r   r   r   )
r   r   isetitemr   arrayr    r   r   rr   r   r   r(   r(   r)   test_isetitem3  s    
rt  zsingle-blockzmixed-blockc                 C   s   t g dtjg d|dd}tg d}| }|d| | rntt|dt|s^J |j	drnJ 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   rB  r   r   r   )r   r   rs  r   r   rr  r    r   r!   r   r   rr   r   r   )r&   r	  r'   r   r   r   r(   r(   r)   test_isetitem_seriesI  s     
 
 ru  c                 C   s   t g dddd}t g ddd}|ddg| | rtt|dt|dsVJ tt|d	t|d	srJ |jdrJ n8tt|dt|drJ tt|d	t|d	rJ | }d
|jd< d
|jd< t	
|| d S )Nr   r   r   r   r   r   r   r   r   r|   r   r   )r   rr  r   r    r   r!   r   r   r$   rr   r   )r&   r'   rhsr   r(   r(   r)   test_isetitem_framed  s    

rw  keyc              	   C   s   t g dg dd}| }||}| r|tt|dt|dsHJ d|jd< tt|dt|drnJ t|| nt	|t
rtnd }tddD t| d|jd< W d    n1 s0    Y  W d    n1 s0    Y  t	|t
rt|| n|jd dksJ d S )	Nr   r   r   r   r   chained_assignmentrT  r   )r   r   getr   r    r   r$   rr   r   
isinstancer   r   pdoption_contextrs   )r&   rx  r'   r   r   rT  r(   r(   r)   test_getu  s    

Fr~  z	axis, keyr   )r   r   c           
   	   C   sH  |dko| }|p|o|dk}t g dg dtjg d|dd}| }|j||d}	|dksf|rtt|d	t|	sJ n| r|	jd
sJ | s|rd
|	j	d
< nbt
ddD tt d
|	j	d
< W d    n1 s0    Y  W d    n1 s0    Y  | s"|s0|d
kr0t|| n|j	d d
ksDJ d S )NrG   r   r   r   rB  r  r   r   r   r   ry  rT  r   )r   r   rs  r   xsr    r   r!   r   r$   r|  r}  rr   rs   r   r   )
r&   rw   r?   rx  r	  Zsingle_blockZis_viewr'   r   r   r(   r(   r)   test_xs  s&     Hr  r?   z
key, level)l1r   r+  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rV|  }| }|j	|||d}	|dkrt 
t||jd t|	|	jd sJ | s|stnd }
tddD t|
 d|	jd< W d    n1 s0    Y  W d    n1 s0    Y  t|| d S )N   r   r   r  l2r   Zlev1Zlev2r.  rR  r   r   )levelr?   r   ry  rT  r   )r   ZarangeZreshaper   Zfrom_productr   r   r   r   r  r    r   r   r   r|  r}  rr   rs   r$   r   )r&   rw   rx  r  r?   Zarrr   r'   r   r   rT  r(   r(   r)   test_xs_multiindex  s*    H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|| | rt|| tt|d	t|d	sJ tt|dt|drJ nt|| d S )
N      ?rh  ri  )      @g      @      @r   r         Y@r   rk   )r  r  r  r   )r   r   updaterr   r   r   r    r   )r&   Zdf1ry   Zdf1_origr   r   r(   r(   r)   test_update_frame  s    
r  c                 C   st   t g d}t dgdgd}| }|d d  }|| t g d}t|| | rdt|| nt|| d S )Nr  r  r   rk   )r  r  ri  )r   r   r  rr   r   )r&   Zser1r   Z	ser1_origr   r   r(   r(   r)   test_update_series  s    
r  c                 C   s   t dg di}tdgdgd}| }| rt  |d | W d    n1 sX0    Y  t|| t $ |dg |  W d    n1 s0    Y  t|| d S )Nr   r   r  r   rk   )r   r   r   rr   rf  r  r   r   )r&   r'   r   r   r(   r(   r)   test_update_chained_assignment  s    
,
2r  c                  C   sD   t g d} t| }| d9 } tt| |s0J t|t|  d S Nr   r   )r   r   r   r    rr   Zassert_numpy_array_equal)r   datar(   r(   r)   test_inplace_arithmetic_series  s
    r  c                 C   sn   t g d}| }|d d  }|d9 }| rRtt|t|rDJ t|| ntt|t|sjJ d S r  r^  )r&   r   r   r   r(   r(   r)   -test_inplace_arithmetic_series_with_reference  s    r  Fc                 C   s   t g ddd}| }|j|d}|s.|r2| rPtt|dt|dslJ ntt|dt|drlJ d|jd< | rt|| d S )	Nr   r   r   r8   r   r   r|   r   )	r   r   r   r   r    r   r$   rr   r   )r&   r   rw   r'   r   r   r(   r(   r)   test_transpose  s    
r  c                 C   sZ   t g ddd}| }|j}tt|dt|dr<J d|jd< | rVt|| d S )Nr   r   r   r   r   r|   r   )	r   r   rj  r   r    r   r$   rr   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r6J d S )Nr   r   rK  r  r   )r   rj  r   r    r   )r&   r'   r   r(   r(   r)   test_transpose_ea_single_column-  s    r  c                 C   s@   t g ddd}| }dd }|| | r<t|| d S )Nr   r   r   c                 S   s   d| j d< | S Nr|   r   r   rY  r(   r(   r)   r   8  s    
z"test_transform_frame.<locals>.func)r   r   	transformrr   r   )r&   r'   r   r   r(   r(   r)   test_transform_frame4  s    
r  c                 C   s:   t g d}| }dd }|| | r6t|| d S )Nr   c                 S   s   d| j d< | S r  r   rY  r(   r(   r)   r   E  s    
z#test_transform_series.<locals>.func)r   r   r  rr   r   )r&   r   r   r   r(   r(   r)   test_transform_seriesA  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   rk   )r   countr$   r   rr   r   r   r(   r(   r)   test_count_read_only_arrayN  s
    
r  c                 C   s   t g d}| }| }tt|t|s4J | rH|jdrHJ d|jd< | rdt	
|| nt g d}t	
|| d S )Nr   r   r|   r   )r   r   r   r   r    r   r!   r   r$   rr   r   r<  r(   r(   r)   test_series_viewV  s    
r  c                 C   s   t dg di}tg d}| }|jd|dd | rdtt|t|dsRJ |jdr~J ntt|t|dr~J d|j	d< t
|| d S )Nr   r   r   r   )r   valuecolumnr|   r   )r   r   r   insertr   r    r   r!   r   r$   rr   r   )r&   r'   r   r   r(   r(   r)   test_insert_seriesh  s    
r  c                 C   s|   t g ddd}| }|d}| rFtt|dt|dsbJ ntt|dt|drbJ d|jd< t|| d S )Nr   r   r   c = a+br   r|   r   	r   r   evalr   r    r   r$   rr   r   r  r(   r(   r)   	test_evalw  s    

r  c                 C   sn   t g ddd}| }|d d  }|jddd tt|dt|dsPJ d|jd	< | rjt|| d S )
Nr   r   r   r  Tre  r   r|   r   r  )r&   r'   r   Zdf_viewr(   r(   r)   test_eval_inplace  s    
r  )wnumpyr   ZpytestZpandas.errorsr   Zpandasr|  r   r   r   r   r   r   r	   r
   Zpandas._testingZ_testingrr   Zpandas.tests.copy_view.utilr   r*   r-   markZparametrizer{   r   r   r   Z
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  r  r  r  r  r  r  r  r!  r#  r'  r6  r7  r9  r:  r;  r=  r>  rA  rF  rG  rJ  rL  rO  rQ  rt   rU  r]  r_  rd  rg  rk  rm  ro  rt  ru  rw  r~  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r(   r(   r(   r)   <module>   s  (
$)#)
 




"







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