
    d&                        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 Zd Zej                            dg d          ej                            d	d
 d d d d d d d d d d d d d d d gg d          d                         Zej                            dg d          ej                            d	d d d d  d! d" d# d$ d% d& d' d( d) d* d+ d, d- gg d.          d/                         Zej                            dg d          d0             Zd1 Zej                            d2 ej        d d3           ed4d3g          g          d5             Zd6 Zd7 Zd8 Z d9 Z!d: Z"d; Z#d< Z$ej                            d=d>d?gid@d?idAd?ig          dB             Z%dC Z&dD Z'dE Z(dF Z)dG Z*ej                            dHdI dJ dK g          dL             Z+dM Z,dN Z-dO Z.dP Z/ej                            dQd2dRg          dS             Z0dT Z1dU Z2ej                            dVdW d fdX d3fdY d fdZ d fgg d[          d\             Z3ej                            d] ed4d3gd?^           ed?d4d3gi          g          d_             Z4ej                            d] ed4d3gd?^           ed?d4d3gi          g          d`             Z5da Z6db Z7dc Z8dd Z9ej                            dedfd4ej:        fg          dg             Z;ej                            dhdidjg          dk             Z<ej                            d	dl dm dn do g          dp             Z=dq Z>dr Z?ds Z@ej                            dtd?dud4dvd d4d dvg          dw             ZAej                            d	dxdyg          dz             ZBej                            d] ed4d3g           ed?d4d3gi          g          d{             ZCej                            d] ed4d3g           ed?d4d3gi          g          d|             ZDd} ZEd~ ZFej                            d eg dd?^          i f ed?g di          dd?ifg          d             ZGej                            d eg dd?^          i f ed?g di          dd?ifg          d             ZHej                            dg d          d             ZId ZJd ZKej                            d] eg d           ed?g di          g          d             ZLd ZMd ZNd ZOej                            dtddid2dig          d             ZPej                            dddg          d             ZQd ZRd ZSd ZTej                            dddg          d             ZUej                            dddg          d             ZVej                            dddg          d             ZWej                            dddg          d             ZXej                            dddg          d             ZYej                            dddg          d             ZZej                            dddg          d             Z[d Z\d Z]d Z^d Z_ej                            dddgddg          d             Z`ej                            dd?d?gg          d             Zaej                            dddg          ej                            dddgddg          d                         Zbej                            dd d4g          ej                            dddg          d                         Zcd Zdd Zed Zfd Zgej                            dddg          d             Zhd ZidÄ ZjdĄ Zkdń ZldS )    N)SettingWithCopyWarning)	DataFrameIndex
MultiIndexPeriodSeries	Timestamp
date_rangeperiod_range)	get_arrayc                    t          g dg dg dd          }|                                }t          j        t	          |d          t	          |d                    rJ | rV|j        j        d         j                                        rJ |j        j        d         j                                        rJ d|j	        d<   |j	        d         dk    sJ 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_copys      _/var/www/html/t/fyr/venv311/lib/python3.11/site-packages/pandas/tests/copy_view/test_methods.py	test_copyr*      s    	II	J	JBggiiG 	'3 7 72s9K9KLLLLL ?<&q).<<>>>>><&q).<<>>>>> GL74=A    c                 (   t          g dg dg dd          }|                    d          }t          j        t	          |d          t	          |d                    sJ | rV|j        j        d         j                                        sJ |j        j        d	         j                                        sJ | rd|j	        d
<   |j	        d
         d	k    sJ t          j        t	          |d          t	          |d                    rJ t          j        t	          |d          t	          |d                    sJ d S d|j	        d
<   |j	        d
         dk    sJ t          j        t	          |d          t	          |d                    sJ d S )Nr   r   r   r   Fdeepr   r   r   r   r   r   r%   s      r)   test_copy_shallowr/   $   s   	II	J	JBgg5g!!G Igs33Yr35G5GHHHHH ;|"1%*88:::::|"1%*88::::: MTwt}!!!!#Igs$;$;Yr3=O=OPPPPP	'3 7 72s9K9KLLLLLLL Twt}!!!!	'3 7 72s9K9KLLLLLLLr+   r   )TNFmethodc                 D    |                      t          j        |          S )Ncolumnsr   )renamestrlowerr'   r   s     r)   <lambda>r8   B   s    394@@ r+   c                 4    |                      ddg|          S )Nr   r   r2   )reindexr7   s     r)   r8   r8   C   s    S#JTBB r+   c                 0    |                      | |          S Nr   reindex_liker7   s     r)   r8   r8   D   s    $77 r+   c                 <    |                      | |          d         S Nr=   r   alignr7   s     r)   r8   r8   E   s    "4003 r+   c                 6    |                      g dd|          S Nr   index)axisr   set_axisr7   s     r)   r8   r8   F   s    ___7NN r+   c                 0    |                      d|          S NtestrF   r   rename_axisr7   s     r)   r8   r8   G   s    f4@@ r+   c                 0    |                      d|          S )NrL   r2   rN   r7   s     r)   r8   r8   H   s    TBB r+   c                 4    |                      ddi|          S )Nr   int64r=   astyper7   s     r)   r8   r8   I   s    C>== r+   c                 2    |                      dd|          S Nr   r=   swapaxesr7   s     r)   r8   r8   K       Q55 r+   c                 2    |                      dd|          S Nr   r   r=   truncater7   s     r)   r8   r8   L   rY   r+   c                 .    |                      |          S r<   infer_objectsr7   s     r)   r8   r8   M   s    ))t)44 r+   c                 .    |                      |          S r<   to_timestampr7   s     r)   r8   r8   N   s    d33 r+   c                 0    |                      d|          S ND)freqr   	to_periodr7   s     r)   r8   r8   O   s    3T:: r+   c                 0    |                      d|          S N
US/Centralr=   tz_localizer7   s     r)   r8   r8   P   s    4@@ r+   c                 0    |                      d|          S rk   
tz_convertr7   s     r)   r8   r8   Q   s    |$?? r+   c                 0    |                      d|          S NF)allows_duplicate_labelsr   	set_flagsr7   s     r)   r8   r8   R   s    e$OO r+   )r4   r:   r?   rC   rI   rename_axis0rename_axis1rT   rX   r]   r`   rc   ri   rn   rq   rv   )idsc                    d }d| j         j        j        v rt          ddd          }nrd| j         j        j        v rt	          ddd          }nLd| j         j        j        v rt	          ddd          }n&d| j         j        j        v rt	          dddd	
          }t          g dg dg dd|          } |||          }|p|du }| j         j        j                            d          r
|s|s|du rd}|r5t          j        t          |d          t          |d                    sJ d S t          j        t          |d          t          |d                    rJ d S )Nrc   
2012-01-01rf   r   rg   periodsri   rn   rq   Europe/Brusselsrg   r}   tzr   r   r   r   rF   r=   Fzreindex-r   )
nodecallspecidr   r
   r   
startswithr   r   r   )	requestr0   r   r&   using_array_managerrF   r'   df2share_memorys	            r)   test_methods_copy_keywordr   >   s   Z E.111\Q???	-0	0	0<c1===	',/2	2	2<c1===	.1	1	1<c1ARSSS	IIQV	W	W	WB
&$


C&7$%-L|**:66 !" 	!+> 	!45== L M	#s 3 3Yr35G5GHHHHHHH#Ic3$7$72s9K9KLLLLLLLr+   c                 4    |                      ddi|          S )Nr   d   rM   r4   serr   s     r)   r8   r8      s    #**As8$*?? r+   c                 0    |                      d |          S r<   r   r   s     r)   r8   r8      s    #**T*55 r+   c                 :    |                      | j        |          S )NrM   )r:   rF   r   s     r)   r8   r8      s    #++CID+AA r+   c                 0    |                      | |          S r<   r>   r   s     r)   r8   r8      s    #**3T*:: r+   c                 <    |                      | |          d         S rA   rB   r   s     r)   r8   r8      s    #))Cd)33A6 r+   c                 6    |                      g dd|          S rE   rH   r   s     r)   r8   r8      s    #,,W4,PP r+   c                 0    |                      d|          S rK   rN   r   s     r)   r8   r8      s    #//T/BB r+   c                 0    |                      d|          S )NrR   r=   rS   r   s     r)   r8   r8      s    #**W4*88 r+   c                 2    |                      dd|          S )Nr   r   r=   )	swaplevelr   s     r)   r8   r8      s    #--14-88 r+   c                 2    |                      dd|          S rV   rW   r   s     r)   r8   r8          #,,q!$,77 r+   c                 2    |                      dd|          S r[   r\   r   s     r)   r8   r8      r   r+   c                 .    |                      |          S r<   r_   r   s     r)   r8   r8      s    #+++66 r+   c                 .    |                      |          S r<   rb   r   s     r)   r8   r8      s    #***55 r+   c                 0    |                      d|          S re   rh   r   s     r)   r8   r8      s    #--St-<< r+   c                 0    |                      d|          S rk   rm   r   s     r)   r8   r8      s    #//,T/BB r+   c                 0    |                      d|          S rk   rp   r   s     r)   r8   r8      s    #..D.AA r+   c                 0    |                      d|          S rs   ru   r   s     r)   r8   r8      s    #--D-QQ r+   )zrename (dict)r4   r:   r?   rC   rI   rw   rT   r   rX   r]   r`   rc   ri   rn   rq   rv   c                    d }d| j         j        j        v rt          ddd          }nd| j         j        j        v rt	          ddd          }nzd| j         j        j        v rt	          ddd          }nTd| j         j        j        v rt	          dddd	
          }n-d| j         j        j        v rt          j        g dg dg          }t          g d|          } |||          }|p|du }|r3t          j	        t          |          t          |                    sJ d S t          j	        t          |          t          |                    rJ d S )Nrc   r{   rf   r   r|   ri   rn   rq   r~   r   r   r   r   r   r=   F)r   r   r   r   r
   r   from_arraysr   r   r   r   )r   r0   r   r&   rF   r   ser2r   s           r)    test_methods_series_copy_keywordr      sx   V E.111\Q???	-0	0	0<c1===	',/2	2	2<c1===	.1	1	1<c1ARSSS	-0	0	0&			999'=>>
%
(
(
(C6#D!!!D&7$%-L E	$3@@@@@@@#IdOOYs^^DDDDDDDr+   c                 P   t          g dg dd          }|                    |          }| p|du p|d u }|o| }|r5t          j        t	          |d          t	          |d                    sJ d S t          j        t	          |d          t	          |d                    rJ d S )Nr   r   r   r   r=   Fr   r   )r   	transposer   r   r   )r&   r   r   r'   resultr   s         r)   test_transpose_copy_keywordr      s    	33	4	4B\\t\$$F&G$%-G44<L;(;$;L N	"c 2 2Ifa4H4HIIIIIII#Ib#$6$6	&!8L8LMMMMMMMr+   c                    t          g dg dg ddg d          }|                                }|                                }|j                                         | rft          j        t          |d          t          |d                    sJ t          j        t          |d          t          |d                    sJ d	|j        d
<   t          j        t          |d          t          |d                    rJ | r3t          j        t          |d          t          |d                    sJ t          j
        ||           d S )Nr   r   r   r   )
         r   r   r   r   )r   r   )r   r   reset_indexr    _verify_integrityr   r   r   r$   tmassert_frame_equalr&   r'   df_origr   s       r)   test_reset_indexr      sS    
iiiiiooo>>lll
 
 
B ggiiG
..

CH    I	#s 3 3Yr35G5GHHHHH	#s 3 3Yr35G5GHHHHHCHTN	#s 3 3Yr35G5GHHHHH I	#s 3 3Yr35G5GHHHHH"g&&&&&r+   rF   r   r   c                    t          ddg|          }|                                }|                    d          }| rNt          j        t          |          t          |                    sJ |j                            d          rJ n1t          j        t          |          t          |                    rJ d|j        d<   t          j
        ||           d S )Nr   r   r   T)dropr   r   )r   r   r   r   r   r   r    _has_no_referencer$   r   assert_series_equalr&   rF   r   ser_origr   s        r)   test_reset_index_series_dropr      s    
!Qu
%
%
%CxxzzH???%%D E	#	$@@@@@8--a000000#IcNNIdOODDDDDDIaL3)))))r+   c                 t   t          g dg dg dd          }|                                }|                    t          j                  }| r3t          j        t          |d          t          |d                    sJ d|j        d	<   t          j        t          |d          t          |d                    rJ | r3t          j        t          |d
          t          |d                    sJ t          g dg dg dd          }t          j
        ||           t          j
        ||           d S )Nr   r   r   r   r3   Ar   r   r   Cr   r   r   r   )r   Br   )r   r   r4   r5   upperr   r   r   r$   r   r   r&   r'   r   r   expecteds        r)   test_rename_columnsr      s<    
II	J	JBggiiG
))CI)
&
&C I	#s 3 3Yr35G5GHHHHHCHTN	#s 3 3Yr35G5GHHHHH I	#s 3 3Yr35G5GHHHHHyyyyyyOOPPH#x((("g&&&&&r+   c                    t          g dg dg dd          }|                    t          j                  }|                                }| r4t          j        t          |d          t          |d                    sJ n3t          j        t          |d          t          |d                    rJ d|j        d	<   t          j        t          |d          t          |d                    rJ | r3t          j        t          |d
          t          |d                    sJ t          g dg dg dd          }t          j
        ||           t          j
        ||           d S )Nr   r   r   r   r   r   r   r   r   r   r   r   )r   r4   r5   r   r   r   r   r   r$   r   r   )r&   r'   r   df2_origr   s        r)   !test_rename_columns_modify_parentr     sl    
II	J	JB
))CI)
&
&CxxzzH M	#s 3 3Yr35G5GHHHHHH#Ic3$7$72s9K9KLLLLLBGDM	#s 3 3Yr35G5GHHHHH I	#s 3 3Yr35G5GHHHHHyyyyyyOOPPH"h'''#x(((((r+   c                    t          g ddd          }|                                }d }|                    |          }t          j        t          |d          t          |d                    sJ d|j        d<   | rIt          j        ||           t          j        t          |d          t          |d                    rJ n\t          g ddd          }t          j        ||           t          j        t          |d          t          |d                    sJ t          j        t          |d	          t          |d	                    sJ d S )
Nr         ?r   c                     | S N r'   s    r)   testfuncztest_pipe.<locals>.testfunc'  s    	r+   r   r   r   r   r   )	r   r   piper   r   r   r$   r   r   r&   r'   r   r   r   r   s         r)   	test_piper   #  s\   	--	.	.BggiiG   ''(

CIc3//2s1C1CDDDDD CHTN I
b'***#Ic3$7$72s9K9KLLLLLL99937788
b(+++	#s 3 3Yr35G5GHHHHHIc3//2s1C1CDDDDDDDr+   c                    t          g ddd          }|                                }d }|                    |          }t          j        t          |d          t          |d                    sJ | rIt          j        ||           t          j        t          |d          t          |d                    rJ n\t          g ddd          }t          j        ||           t          j        t          |d          t          |d                    sJ t          j        t          |d          t          |d                    sJ d S )Nr   r   r   c                     d| j         d<   | S )Nr   r   )r$   r   s    r)   r   z%test_pipe_modify_df.<locals>.testfunc?  s    	r+   r   r   r   r   r   )r   r   r   r   r   r   r   r   r   s         r)   test_pipe_modify_dfr   ;  sO   	--	.	.BggiiG   ''(

CIc3//2s1C1CDDDDD I
b'***#Ic3$7$72s9K9KLLLLLL;;;S99::
b(+++	#s 3 3Yr35G5GHHHHHIc3//2s1C1CDDDDDDDr+   c                 p   t          g dg dg dd          }|                                }|                    ddg          }| r4t          j        t          |d          t          |d                    sJ n3t          j        t          |d          t          |d                    rJ d|j        d	<   t          j        t          |d          t          |d                    rJ | r3t          j        t          |d          t          |d                    sJ t          j        ||           d S )
Nr   r   r   r   r   r   r   r   r   )	r   r   r:   r   r   r   r$   r   r   r   s       r)   test_reindex_columnsr   R  s8    
II	J	JBggiiG
**c3Z*
(
(C M	#s 3 3Yr35G5GHHHHHH#Ic3$7$72s9K9KLLLLLCHTN	#s 3 3Yr35G5GHHHHH I	#s 3 3Yr35G5GHHHHH"g&&&&&r+   c                 j   t          g dg dg dd          }|                                }|                    d          }|j                                         | rgt          j        t          |d          t          |d                    sJ t          j        t          |d          t          |d                    sJ nft          j        t          |d          t          |d                    rJ t          j        t          |d          t          |d                    rJ d	|j        d
<   t          j        t          |d          t          |d                    rJ | r3t          j        t          |d          t          |d                    sJ t          j
        ||           d S )Nr   r   r   r   r   r   r   r   r   r   )r   r   r   r    r   r   r   r   r$   r   r   r   s       r)   test_drop_on_columnr   f  s   	II	J	JBggiiG
''#'

CH    M	#s 3 3Yr35G5GHHHHH	#s 3 3Yr35G5GHHHHHH#Ic3$7$72s9K9KLLLLL#Ic3$7$72s9K9KLLLLLCHTN	#s 3 3Yr35G5GHHHHH I	#s 3 3Yr35G5GHHHHH"g&&&&&r+   c                 6   t          g dg dg dd          }|                                }|                    d          }|j                                         | r4t          j        t          |d          t          |d                    sJ n3t          j        t          |d          t          |d                    rJ d|j        d<   | r3t          j        t          |d          t          |d                    rJ t          j
        ||           d S )	Nr   r   r   r   rR   r   r   r   )r   r   select_dtypesr    r   r   r   r   r$   r   r   r   s       r)   test_select_dtypesr   y  s    
II	J	JBggiiG


7
#
#CH    M	#s 3 3Yr35G5GHHHHHH#Ic3$7$72s9K9KLLLLL CHTN M#Ic3$7$72s9K9KLLLLL"g&&&&&r+   filter_kwargsitemsr   likeregexc                    t          g dg dg dd          }|                                } |j        di |}| r4t          j        t          |d          t          |d                    sJ n3t          j        t          |d          t          |d                    rJ | r=d|j        d<   t          j        t          |d          t          |d                    rJ t          j        ||           d S )	Nr   r   r   r   r   r   r   r   )	r   r   filterr   r   r   r$   r   r   )r&   r   r'   r   r   s        r)   test_filterr     s    
II	J	JBggiiG
")
$
$m
$
$C M	#s 3 3Yr35G5GHHHHHH#Ic3$7$72s9K9KLLLLL  M#Ic3$7$72s9K9KLLLLL"g&&&&&r+   c                 *   t          ddgddgddggt          dd          d	d
g          }|                                }|                    d          }| r4t	          j        t          |d	          t          |d	                    sJ n3t	          j        t          |d	          t          |d	                    rJ d|j        d<   | r3t	          j        t          |d
          t          |d
                    rJ t          j	        ||           d S )Nr   r   r   r   r   r   
2020-01-01
2020-01-03r   r   rF   r3   r   )r}   r   )
r   r
   r   shiftr   r   r   r$   r   r   r   s       r)   test_shift_no_opr     s"   	
Q!Q!Q |44c

 
 
B
 ggiiG
((1(

C M	#s 3 3Yr35G5GHHHHHH#Ic3$7$72s9K9KLLLLLBGDM M#Ib#$6$6	#s8K8KLLLLL#w'''''r+   c                     t          ddgddgddggt          dd          d	d
g          }|                    dd          }t          j        t          |d	          t          |d	                    rJ d S )Nr   r   r   r   r   r   r   r   r   r   r   r   r}   rG   )r   r
   r   r   r   r   )r&   r'   r   s      r)   test_shift_indexr     s    	
Q!Q!Q |44c

 
 
B
 ((11(
%
%C	#s 3 3Yr35G5GHHHHHHHr+   c                 V   t          ddgddgddggt          dd          d	d
g          }|                                }t          dd          |_        |                    dd          }| r4t          j        t          |d	          t          |d	                    sJ n3t          j        t          |d	          t          |d	                    rJ d|j        d<   | r3t          j        t          |d	          t          |d	                    rJ t          j
        ||           d S )Nr   r   r   r   r   r   r   r   r   r   r   
2020-01-02z
2020-01-041Dr}   rg   r   r   )r   r
   r   rF   r   r   r   r   r$   r   r   r   s       r)   test_shift_rows_freqr     s5   	
Q!Q!Q |44c

 
 
B
 ggiiG|\::GM
((14(
(
(C M	#s 3 3Yr35G5GHHHHHH#Ic3$7$72s9K9KLLLLLBGDM M#Ib#$6$6	#s8K8KLLLLL#w'''''r+   c                    t          ddgddgddggt          dd          	          }|                    dd
          }t          j        t          |d          t          |d                    sJ d|j        d<   | r3t          j        t          |d          t          |d                    rJ t          t          j        dgt          j        dgt          j        dggt          dd          	          }t          j	        ||           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   s       r)   test_shift_columnsr     s!   	
Q!Q!Q *\<*P*P
 
 
B ((11(
%
%CIc<88)B:U:UVVVVVBGDM 
#c<(()B*E*E
 
 	
 	
 	
 
&!rvqkBFA;/<66  H #x(((((r+   c                    t          g dg dg dd          }|                                }|d d          }|                    d          }t          j        |j        t          |d                    sJ t          j        t          |d          t          |d                    sJ | r4d|j        d<   t          j        |j        t          |d                    rJ d|j        d<   | rJt          j        t          |d          t          |d                    rJ t          j	        ||           d S t          g dg d	g dd          }t          j	        ||           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   s         r)   test_popr     so   	II	J	JBggiiGqqqEMVVC[[FFM9]C+H+HIIIIIIb#..	-0M0MNNNNN RA#FM9]C3P3PQQQQQBGDM 7#Ib#$6$6	-QT8U8UVVVVV
mW55555999999???SSTT
mX66666r+   funcc                 ,    |                      |          S r   rB   xys     r)   r8   r8     s    QWWQZZ r+   c                 :    |                      |j        d          S )Nr   rG   )rC   r   r   s     r)   r8   r8     s    QWWQSqW)) r+   c                 l    |                      |j        j        t          dd                   d          S )Nr   r   r  )rC   r   r$   slicer   s     r)   r8   r8     s'    QWWQSXeAqkk2W;; r+   c                 &   t          g ddd          }|                                }|ddg                                         } |||          \  }}| r4t          j        t	          |d          t	          |d                    sJ n3t          j        t	          |d          t	          |d                    rJ d|j        d<   | r3t          j        t	          |d          t	          |d                    rJ t          j        ||           d S )Nr   r   r   r   r   r   )r   r   r   r   r   r$   r   r   )r&   r   r'   r   
df_changedr   _s          r)   test_align_framer	    s    
--	.	.BggiiGS#J$$&&JT"j!!FC M	#s 3 3Yr35G5GHHHHHH#Ic3$7$72s9K9KLLLLLCHTN M#Ic3$7$72s9K9KLLLLL"g&&&&&r+   c                    t          ddg          }|                                }|                                }|                    |          \  }}| rCt          j        |j        |j                  sJ t          j        |j        |j                  sJ nBt          j        |j        |j                  rJ t          j        |j        |j                  rJ d|j        d<   d|j        d<   | rBt          j        |j        |j                  rJ t          j        |j        |j                  rJ t          j        ||           t          j        ||           d S )Nr   r   r   )	r   r   rC   r   r   r   r$   r   r   )r&   r   r   	ser_otherr   ser_other_results         r)   test_align_seriesr    sS   
!Q..CxxzzH

I YYy11D
 OSZ88888 0 79IJJJJJJ#DK<<<<<#$4$;Y=MNNNNNDIaL ! O#DK<<<<<#$4$;Y=MNNNNN3)))9h/////r+   c                    t          g dg dd          }|                                }|                    |d          \  }}t          j        t          |d          t          |d                    sJ t          j        t          |d          t          |d                    sJ | r@d|j        d	<   t          j        ||           d|j        d	<   t          j        ||           d S d S )
Nr   r   r   Fr=   r   r   r   r   r   )	r   r   rC   r   r   r   locr   r   )r&   r'   r   r   df3s        r)   test_align_copy_falser  -  s    	33	4	4BggiiGxxx''HCIb#..	#s0C0CDDDDDIb#..	#s0C0CDDDDD +
b'***
b'*****+ +r+   c                    t          g dg dd          }t          g dd          }|                                }|                                }|                    |dd          \  }}t	          j        t          |d	          t          |d	                    sJ t	          j        t          |d
          t          |d
                    sJ t	          j        t          |d          t          |d                    sJ | r@d|j        d<   t          j	        ||           d|j        d<   t          j
        ||           d S d S )Nr   r   r   r   nameFr   )r   rG   r   r   r  )r   r   r   rC   r   r   r   r  r   r   r   )r&   r'   r   r   r   r   r   s          r)   !test_align_with_series_copy_falser  =  sE   	33	4	4B

%
%
%CxxzzHggiiG5q11ICIb#..	#s0C0CDDDDDIb#..	#s0C0CDDDDDIc3//41E1EFFFFF .
b'***
sH-----. .r+   c                    t          g d          }|                                }|d d                                          }t          j        |j        t          |d                    sJ d|j        d<   | r@t          j        |j        t          |d                    rJ t          j	        ||           n3|                                }d|j        d<   t          j	        ||           |d d                                          }d|j        d<   | r)t          j
        ||                                           d S |                                                                }d|j        d<   t          j
        ||           d S )Nr   r   r   )r   r   to_framer   r   r   r   r$   r   r   r   )r&   r   r   r'   r   s        r)   test_to_framer  P  si   


CxxzzH	QQQ		B CJ	"a(8(899999BGDM .#CJ	"a0@0@AAAAA
sH---- ==??a
sH--- 
QQQ		BCHQK ,
b("3"3"5"566666==??++--d
b(+++++r+   axr3   c                     t          g dg dd          }|                                }|                    ||          }| r4t          j        t          |d          t          |d                    sJ n3t          j        t          |d          t          |d                    rJ d|j        d<   | r3t          j        t          |d          t          |d                    rJ t          j        ||           d S )Nr   r   r   r   r   r   	r   r   rX   r   r   r   r$   r   r   )r&   r  r'   r   r   s        r)   test_swapaxes_noopr  r  s   	33	4	4BggiiG
++b"

C M	#s 3 3Yr35G5GHHHHHH#Ic3$7$72s9K9KLLLLL CHTN M#Ic3$7$72s9K9KLLLLL"g&&&&&r+   c                    t          g dg ddg d          }|                                }|                    dd          }| r4t          j        t          |d          t          |d	                    sJ n3t          j        t          |d          t          |d	                    rJ d
|j        d<   | r3t          j        t          |d          t          |d	                    rJ t          j        ||           d S )Nr   r   r   )r   r  zr   rF   r3   r   r   r   r   r  r   s       r)   test_swapaxes_single_blockr     s
   	33???	K	K	KBggiiG
++gy
)
)C M	#s 3 3Yr35G5GHHHHHH#Ic3$7$72s9K9KLLLLL CHTN M#Ic3$7$72s9K9KLLLLL"g&&&&&r+   c                      t          ddgdd          } |                     dd          } d| j        d	<   t          ddgddgd
ddg          }t          j        | |           d S )Nr   r   r   r   rF   r3   )axis1axis2r   r   r   r   r   r   )r   rX   r$   r   r   )r'   r   s     r)   test_swapaxes_read_only_arrayr$    sz    	!Qa((	)	)B	7)	4	4BBGDMc1X1a&11#sDDDH"h'''''r+   zmethod, idxc                 V    |                      d                               d          S )NFr-   r=   r   s    r)   r8   r8     s%    BGGG'',,%,88 r+   c                 N    |                                                                   S r   )r   r   s    r)   r8   r8     s    BNN$$0022 r+   c                 ~    |                      t          j                                       t          j                  S )Nr   )r4   r5   r   r6   r   s    r)   r8   r8     s)    BIIciI0077	7JJ r+   c                 V    |                      d                              d          S )NFr-   number)include)r   r   r   s    r)   r8   r8     s%    BGGG''55h5GG r+   )shallow-copyr   r4   r   c                 j   t          g dg dg dd          }|                                }| o| j        j        j        dk    } ||          }d|j        d|f<   |st          j        ||            ||          }d|j        d<   |s(t          j        |j        d d |d f         |           d S d S )Nr   r   r   r   r+  r   r   )r   r   r   r   r   r$   r   r   )r   r0   idxr&   r'   r   df2_is_viewr   s           r)   test_chained_methodsr/    s     
II	J	JBggiiG *)Xgl.C.F..XK &**CCHQV +
b'*** &**CBGDM :
chqqq#$$w/99999: :r+   objr  c                 4   t          t          dd          t          dd          g          |_        |                                }|                                }| r4t          j        t          |d          t          |d                    sJ n3t          j        t          |d          t          |d                    rJ d|j        d<   t          j        t          |d          t          |d                    rJ t          j
        ||           d S )Nz2012-1-1rf   rg   z2012-1-2r   r   )r   r   rF   r   rc   r   r   r   r$   r   assert_equalr&   r0  obj_origobj2s       r)   test_to_timestampr7    s   vjs333VJS5Q5Q5QRSSCIxxzzHD O	$ 4 4iS6I6IJJJJJJ#IdC$8$8)C:M:MNNNNN DIaL	$ 4 4iS6I6IJJJJJOC"""""r+   c                 0   t          t          d          t          d          g          |_        |                                }|                    d          }| r4t          j        t          |d          t          |d                    sJ n3t          j        t          |d          t          |d                    rJ d|j        d<   t          j        t          |d          t          |d                    rJ t          j
        ||           d S )N
2019-12-31
2020-12-31Yr2  r   r   )r   r	   rF   r   ri   r   r   r   r$   r   r3  r4  s       r)   test_to_periodr<    s    y..	,0G0GHIICIxxzzH==c=""D O	$ 4 4iS6I6IJJJJJJ#IdC$8$8)C:M:MNNNNN DIaL	$ 4 4iS6I6IJJJJJOC"""""r+   c                     t          g dg dg dd          }|                                }|                    d          }| r4t          j        t          |d          t          |d                    sJ n3t          j        t          |d          t          |d                    rJ d|j        d<   t          j        t          |d	          t          |d	                    rJ t          j        ||           d S )
Nr   r   r   r   r   r   r   r   r   )	r   r   	set_indexr   r   r   r$   r   r   r   s       r)   test_set_indexr?    s    	II	J	JBggiiG
,,s

C M	#s 3 3Yr35G5GHHHHHH#Ic3$7$72s9K9KLLLLL CHTN	#s 3 3Yr35G5GHHHHH"g&&&&&r+   c                      t          g ddd          } |                     d          }|                                }d| j        d<   t	          j        ||           d S )Nr   r   r   r   r   r   )r   r>  r   r$   r   r   r'   r   r   s      r)   4test_set_index_mutating_parent_does_not_mutate_indexrB    s]    	++	,	,B\\#F{{}}HBGDM&(+++++r+   c                 ^   t          g dg dg dd          }|                                }|                    d          }| r3t          j        t          |d          t          |d                    sJ d|j        d	<   t          j        t          |d          t          |d                    rJ | r3t          j        t          |d
          t          |d                    sJ t          g dg dg dd          }t          j        ||           t          j        ||           d S )Nr   r   r   r   CoW_CoW_ar   r   r   CoW_cr   r   )rE  CoW_brF  )	r   r   
add_prefixr   r   r   r$   r   r   r   s        r)   test_add_prefixrI    s=   	II	J	JBggiiG
--

C M	#w 7 72s9K9KLLLLLCHTN	#w 7 72s9K9KLLLLL M	#w 7 72s9K9KLLLLL))iii///JJ H #x((("g&&&&&r+   c                 ^   t          g dg dg dd          }|                                }|                    d          }| r3t          j        t          |d          t          |d                    sJ d|j        d	<   t          j        t          |d          t          |d                    rJ | r3t          j        t          |d
          t          |d                    sJ t          g dg dg dd          }t          j        ||           t          j        ||           d S )Nr   r   r   r   _CoWa_CoWr   r   r   c_CoWr   r   )rL  b_CoWrM  )	r   r   
add_suffixr   r   r   r$   r   r   r   s        r)   test_add_suffixrP    s=   	II	J	JBggiiG
--

C M	#w 7 72s9K9KLLLLLCHTN	#w 7 72s9K9KLLLLL M	#w 7 72s9K9KLLLLL))iii///JJ H #x((("g&&&&&r+   z	axis, val)r         @c                    t          g dd|dgdd          }|                                }|                    |          }| r4t          j        t          |d          t          |d                    sJ n3t          j        t          |d          t          |d                    rJ d|j        d	<   | r3t          j        t          |d          t          |d                    rJ t          j        ||           d S )
Nr   r   r   dr   r  r   r   r   )	r   r   dropnar   r   r   r$   r   r   )r&   rG   valr'   r   r   s         r)   test_dropnarV     s   	!S!3??	@	@BggiiG
)))

C M	#s 3 3Yr35G5GHHHHHH#Ic3$7$72s9K9KLLLLLCHTN M#Ic3$7$72s9K9KLLLLL"g&&&&&r+   rU  r   rQ  c                    t          d|dg          }|                                }|                                }| r"t          j        |j        |j                  sJ n!t          j        |j        |j                  rJ d|j        d<   | r!t          j        |j        |j                  rJ t          j        ||           d S )Nr   r   r   )	r   r   rT  r   r   r   r$   r   r   )r&   rU  r   r   r   s        r)   test_dropna_seriesrX  1  s    
!S!

CxxzzH::<<D =SZ888888#DK<<<<<DIaL =#DK<<<<<3)))))r+   c                 *    |                                  S r   headr   s    r)   r8   r8   E      27799 r+   c                 ,    |                      d          S )Nr   rZ  r   s    r)   r8   r8   F      2771:: r+   c                 *    |                                  S r   tailr   s    r)   r8   r8   G  r\  r+   c                 ,    |                      d          S )Nr   r`  r   s    r)   r8   r8   H  r^  r+   c                    t          g dg dd          }|                                } | |          }|j                                         |rft	          j        t          |d          t          |d                    sJ t	          j        t          |d          t          |d                    sJ d|j        d<   t	          j        t          |d          t          |d                    sJ |r4t	          j        t          |d          t          |d                    rJ n
d|j        d<   t          j	        ||           d S )	Nr   r   r   r   r   r   r   r   )
r   r   r    r   r   r   r   r$   r   r   )r0   r&   r'   r   r   s        r)   test_head_tailrd  B  sJ    
99	:	:BggiiG
&**CH    I	#s 3 3Yr35G5GHHHHH	#s 3 3Yr35G5GHHHHH CHTNIc3//2s1C1CDDDDD #Ic3$7$72s9K9KLLLLLL "g&&&&&r+   c                 B   t          ddgdddd          }|                                }|                                }| rgt          j        t          |d          t          |d                    sJ t          j        t          |d          t          |d                    sJ nft          j        t          |d          t          |d                    rJ t          j        t          |d          t          |d                    rJ d|j        d	<   d
|j        d<   | rft          j        t          |d          t          |d                    rJ t          j        t          |d          t          |d                    rJ t          j        ||           d S )Nr   r   r   r   r   r   r   rS  r   r   r   r   rS  r   )	r   r   r`   r   r   r   r$   r   r   r   s       r)   test_infer_objectsrg  `  s   	!Qc<<	=	=BggiiG




C M	#s 3 3Yr35G5GHHHHH	#s 3 3Yr35G5GHHHHHH #Ic3$7$72s9K9KLLLLL#Ic3$7$72s9K9KLLLLLCHTNCHTN M#Ic3$7$72s9K9KLLLLL#Ic3$7$72s9K9KLLLLL"g&&&&&r+   c                 N   t          ddgddt          t          d          t          d          gd          dd	          }|                                }t	          |d
          }t	          |d          }t	          |d          }d|j        d<   d|j        d<   t          d          |j        d<   | rot          j        |t	          |d
                    sJ t          j        |t	          |d                    rJ t          j        |t	          |d                    sJ d S d S )Nr   r   r   r9  r:  objectdtyper   )r   r   r   rS  er   rS  r   r   r   
2018-12-31r   r   r   r   r	   r`   r   r$   r   r   )r&   r'   arr_aarr_barr_ds        r)   test_infer_objects_no_referencers  u  sE   	Q<(()L*A*AB(   	
 	


 

B 
				Bb#Eb#Eb#EBGDMBGDMl++BGDM ;yS'9'9:::::#E9R+=+=>>>>>yS'9'9:::::	; ; 	;:r+   c                 `   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<   | rot          j        |t	          |d	                    rJ t          j        |t	          |d
                    rJ t          j        |t	          |d                    sJ d S d S )Nr   r   r   r9  r:  ri  rj  rf  r   r   rS  r   r   r   rm  rn  ro  )r&   r'   viewrp  rq  rr  s         r)   test_infer_objects_referencerv    sM   	Q<(()L*A*AB(  		
 	
	
 	
B aaa5D					Bb#Eb#Eb#EBGDMBGDMl++BGDM ;#E9R+=+=>>>>>#E9R+=+=>>>>>yS'9'9:::::; ; 	;:r+   kwargsr   )beforeafterrG   c                    t          g dddd          }|                                } |j        di |}|j                                         | r4t          j        t          |d          t          |d                    sJ n3t          j        t          |d          t          |d                    rJ d|j        d<   | r3t          j        t          |d          t          |d                    rJ t          j
        ||           d S )	Nr   r   r   r   r   r   r   r   )r   r   r]   r    r   r   r   r   r$   r   r   r&   rw  r'   r   r   s        r)   test_truncater|    s    
33	4	4BggiiG
"+



CH    M	#s 3 3Yr35G5GHHHHHH#Ic3$7$72s9K9KLLLLLCHTN M#Ic3$7$72s9K9KLLLLL"g&&&&&r+   assigndrop_duplicatesc                 0   t          dg di          }|                                } t          ||                      }|j                                         | r4t          j        t          |d          t          |d                    sJ n3t          j        t          |d          t          |d                    rJ d|j        d<   | r3t          j        t          |d          t          |d                    rJ t          j
        ||           d S )Nr   r   r   r   )r   r   getattrr    r   r   r   r   r$   r   r   )r&   r0   r'   r   r   s        r)   test_assign_drop_duplicatesr    s   	C#	$	$BggiiG
'"f



CH    M	#s 3 3Yr35G5GHHHHHH#Ic3$7$72s9K9KLLLLLCHTN M#Ic3$7$72s9K9KLLLLL"g&&&&&r+   c                 j   |                                 }|                    ddg          }| r"t          j        |j        |j                  sJ n!t          j        |j        |j                  rJ d|j        d<   | r!t          j        |j        |j                  rJ t          j        ||           d S )Nr   r   )r   taker   r   r   r$   r   r3  r4  s       r)   	test_taker    s     xxzzH88QFD =SZ888888#DK<<<<<DIaL =#DK<<<<<OC"""""r+   c                    t          ddd          |_        |                                }|                    dd          }| r"t	          j        |j        |j                  sJ n!t	          j        |j        |j                  rJ d|j        d<   | r!t	          j        |j        |j                  rJ t          j	        ||           d S )Nz
2018-04-09r   1D20minr   z0:00z1:00r   )
r
   rF   r   between_timer   r   r   r$   r   r3  r4  s       r)   test_between_timer    s    <CCCCIxxzzHFF++D =SZ888888#DK<<<<<DIaL =#DK<<<<<OC"""""r+   c                 "   t          ddgdd          }t          dddgd          }|                                }|                    |          }| r4t          j        t          |d          t          |d                    sJ n3t          j        t          |d          t          |d                    rJ d|j        d<   | r3t          j        t          |d          t          |d                    rJ t          j        ||           d S )Nr   r   r   r   )r   r   r   r   )	r   r   r?   r   r   r   r$   r   r   )r&   r'   otherr   r   s        r)   test_reindex_liker    s   	!Qc**	+	+BCq!f--..EggiiG
//%
 
 C M	#s 3 3Yr35G5GHHHHHH#Ic3$7$72s9K9KLLLLLCHTN M#Ic3$7$72s9K9KLLLLL"g&&&&&r+   c                    t          g d          }|                                }|                                }| r"t          j        |j        |j                  sJ n!t          j        |j        |j                  rJ d|j        d<   t          j        |j        |j                  rJ t          j        ||           d S )Nr   r   )	r   r   
sort_indexr   r   r   r$   r   r   r&   r   r   r   s       r)   test_sort_indexr  
  s    


CxxzzH>>D =
DK888888#CJ<<<<< DIaLSZ888883)))))r+   zobj, kwargsr   byc                    |                                 } |j        di |}| r4t          j        t	          |d          t	          |d                    sJ n3t          j        t	          |d          t	          |d                    rJ d|j        d<   t          j        t	          |d          t	          |d                    rJ t          j        ||           d S )Nr   r   r   r   sort_valuesr   r   r   r$   r   r3  )r&   r0  rw  r5  r6  s        r)   test_sort_valuesr    s    
 xxzzH3?$$V$$D O	$ 4 4iS6I6IJJJJJJ#IdC$8$8)C:M:MNNNNN DIaL	$ 4 4iS6I6IJJJJJOC"""""r+   c                    |                                 }|d d          } |j        dddi| t          j        t	          |d          t	          |d                    sJ d|j        d<   | rJt          j        t	          |d          t	          |d                    rJ t          j        ||           d S t          j        t	          |d          t	          |d                    sJ d S )NinplaceTr   r   r   r  )r&   r0  rw  r   r5  ru  s         r)   test_sort_values_inplacer  .  s    
 xxzzHqqq6DCO++D+F+++Ic3//41E1EFFFFF CHQK K#Ic3$7$749M9MNNNNN
h'''''	#s 3 3YtS5I5IJJJJJJJr+   decimals)r   r   c                 P   t          ddgdd          }|                                }|                    |          }| rt          j        t          |d          t          |d                    sJ |dk    r4t          j        t          |d          t          |d                    sJ ngt          j        t          |d          t          |d                    rJ n3t          j        t          |d          t          |d                    rJ d	|j        d
<   d|j        d<   | rft          j        t          |d          t          |d                    rJ t          j        t          |d          t          |d                    rJ t          j        ||           d S )Nr   r   r   r   )r  r   r   r   rS  r   r   r   )	r   r   roundr   r   r   r$   r   r   )r&   r  r'   r   r   s        r)   
test_roundr  B  s   	!Qc**	+	+BggiiG
((H(
%
%C 	M	#s 3 3Yr35G5GHHHHHq==#Ic3$7$72s9K9KLLLLLL'	#s(;(;Yr3=O=OPPPPPP#Ic3$7$72s9K9KLLLLLCHTNCHTN M#Ic3$7$72s9K9KLLLLL#Ic3$7$72s9K9KLLLLL"g&&&&&r+   c                 6   t          j        g dddg          }t          dg di|          }|                                }|                    ddg          }| r4t          j        t          |d          t          |d                    sJ n3t          j        t          |d          t          |d                    rJ d	|j        d
<   | r3t          j        t          |d          t          |d                    rJ t          j
        ||           d S )Nr   r   r   r   r   r   )r   r   onetwonamesr   r   r   r   r   r   orderr   r   )r   from_tuplesr   r   reorder_levelsr   r   r   r$   r   r   r&   rF   r'   r   r   s        r)   test_reorder_levelsr  [  s,   "(((  E 
C&e	4	4	4BggiiG


5%.

1
1C M	#s 3 3Yr35G5GHHHHHH#Ic3$7$72s9K9KLLLLLCHTN M#Ic3$7$72s9K9KLLLLL"g&&&&&r+   c                    t          j        g dddg          }t          g d|          }|                                }|                    ddg          }| r"t          j        |j        |j                  sJ n!t          j        |j        |j                  rJ d|j        d<   | r!t          j        |j        |j                  rJ t          j
        ||           d S )	Nr  r  r  r  r  r   r  r   )r   r  r   r   r  r   r   r   r$   r   r   r   s        r)   test_series_reorder_levelsr  n  s    "(((  E U
+
+
+CxxzzHUEN33D =SZ888888#DK<<<<<DIaL =#DK<<<<<3)))))r+   c                    t          j        g dddg          }||_        |                                }|                                }| r"t          j        |j        |j                  sJ n!t          j        |j        |j                  rJ d|j        d<   | r!t          j        |j        |j                  rJ t          j
        ||           d S )Nr  r  r  r  r  r  r   )r   r  rF   r   r   r   r   r   r$   r   r3  )r&   r0  rF   r5  r6  s        r)   test_swaplevelr    s    "#;#;#;E5>RRRECIxxzzH==??D =SZ888888#DK<<<<<DIaL =#DK<<<<<OC"""""r+   c                    t          g dg dg dd          }|                                }|                    g dd          }| r4t          j        t          |d          t          |d                    sJ n3t          j        t          |d          t          |d                    rJ d|j        d	<   t          j        t          |d          t          |d                    rJ t          j        ||           d S )
Nr   r   r   r   rF   r  r   r   r   )	r   r   rI   r   r   r   r$   r   r   r   s       r)   test_frame_set_axisr    s   	II	J	JBggiiG
++oooG+
4
4C M	#s 3 3Yr35G5GHHHHHH#Ic3$7$72s9K9KLLLLL CHTN	#s 3 3Yr35G5GHHHHH"g&&&&&r+   c                 P   t          g d          }|                                }|                    g dd          }| rt          j        ||          sJ nt          j        ||          rJ d|j        d<   t          j        ||          rJ t          j        ||           d S )Nr   r   rF   r  r   )r   r   rI   r   r   r$   r   r   r  s       r)   test_series_set_axisr    s    


CxxzzH<<g<66D /T******#C..... DIaLc*****3)))))r+   c                    t          g d          }|                                }|                    d          }t          j        ||          sJ d|j        d<   | r.t          j        ||          rJ t          j        ||           d S t          j        ||          sJ t          g d          }t          j        ||           d S )Nr   F)rt   r   r   )r   r   rv   r   r   r$   r   r   r&   r   r   r   r   s        r)   test_set_flagsr    s    


CxxzzH===77DC&&&&& DIaL .#D#.....
sH-----c*****)))$$
sH-----r+   mapperrL   c                    t          dg dit          g dd                    }|                                } |j        di |}| r4t	          j        t          |d          t          |d                    sJ n3t	          j        t          |d          t          |d                    rJ d|j        d<   | r3t	          j        t          |d          t          |d                    rJ t          j	        ||           d S )Nr   r  r  r   r   r   r   )
r   r   r   rO   r   r   r   r$   r   r   r{  s        r)   test_rename_axisr    s   	C&eLLLs.K.K.K	L	L	LBggiiG
".
"
"6
"
"C M	#s 3 3Yr35G5GHHHHHH#Ic3$7$72s9K9KLLLLLCHTN M#Ic3$7$72s9K9KLLLLL"g&&&&&r+   zfunc, tz)rq   zEurope/Berlin)rn   Nc           	         t          ddgt          ddd|                    }|                                } t          ||          d          }| r"t	          j        |j        |j                  sJ n!t	          j        |j        |j                  rJ d|j        d<   t	          j        |j        |j                  rJ t          j	        ||           d S )	Nr   r   z2014-08-01 09:00H)startrg   r}   r   r   rl   r   )
r   r
   r   r  r   r   r   r$   r   r   )r&   r   r   r   r   r   s         r)   test_tz_convert_localizer    s    
 	
Aj'9QSUVVV  C xxzzH73l++D =
DK888888#CJ<<<<< DIaLSZ888883)))))r+   c                 8   t          j        g dddg          }t          g dg dg dd|	          }|                                }|                    d
          }| r4t          j        t          |d          t          |d                    sJ n3t          j        t          |d          t          |d                    rJ d
|j        d<   t          j        t          |d          t          |d                    rJ t          j
        ||           d S )Nr  r  r  r  r   r         	   r   r   r   r   r   )r   r  r   r   	droplevelr   r   r   r$   r   r   r  s        r)   test_droplevelr    s   "#;#;#;E5>RRRE	CC5	Q	Q	QBggiiG
,,q//C M	#s 3 3Yr35G5GHHHHHH#Ic3$7$72s9K9KLLLLL CHTN	#s 3 3Yr35G5GHHHHH"g&&&&&r+   c                    t          dg di          }|                                }|                                }t          j        |j        t          |d                    sJ d|j        d<   | rAt          j        |j        t          |d                    rJ t          j	        ||           d S t          j        |j        t          |d                    sJ |j
        d         dk    sJ d S )Nr   r   r   r  )r   r   squeezer   r   r   r   r$   r   r   r  )r&   r'   r   seriess       r)   test_squeezer    s    	C#	$	$BggiiGZZ\\F FM9R+=+=>>>>> FKN ##FM9R3E3EFFFFF
b'***** yS/A/ABBBBBvf~""""""r+   c           	         t          g dg dg dd          }|                                }t          d          D ]}|                                D ]\  }}t	          j        t          ||          t          ||                    sJ d|j        d<   | rIt	          j        t          ||          t          ||                    rJ t          j	        ||           |j
        d|f         dk    sJ d S )Nr   r   r  r   r   r   )r   r   ranger   r   r   r   r$   r   r   r  )r&   r'   r   ir  r   s         r)   
test_itemsr    s   	CC	D	DBggiiG 1XX , , 	, 	,ID##Ic4$8$8)B:M:MNNNNN CHQK" ,+Ic4,@,@)BPTBUBUVVVVV%b'2222 vag!+++++	,, ,r+   rk  rR   Int64c                    t          ddgddd|          }|d d          }|                                }d|||k    <   | rJt          j        t	          |d          t	          |d                    rJ t          j        ||           d S t          j        t	          |d          t	          |d                    sJ |j        d         dk    sJ d S )Nr   r   r   rj  r   r   r   )r   r   r   r   r   r   r   r$   )r&   rk  r'   ru  r   s        r)   test_putmaskr  )  s    	!Qaa00	>	>	>Baaa5DggiiGBrRxL $#IdC$8$8)B:L:LMMMMM
dG,,,,, 	$ 4 4iC6H6HIIIIIy!######r+   c                     t          ddgddd|          }t          |d          }d|||k    <   | r%t          j        |t          |d                    sJ d S d S )Nr   r   r   rj  r   r   r   r   r   r   r&   rk  r'   rp  s       r)   test_putmask_no_referencer  9  sy    	!Qaa00	>	>	>Bb#EBrRxL ;yS'9'9:::::; ;::r+   float64Float64c                     t          ddgdd|          }t          |d          }t          dddgi          |||k    <   | r%t          j        |t          |d                    sJ d S d S )Nr   r   r   rj  r   rQ  r   r  r  s       r)   $test_putmask_aligns_rhs_no_referencer  C  s    	#q,,E	:	:	:Bb#EcC8_--BrRxL ;yS'9'9:::::; ;::r+   zval, exp)rQ  T)r   Fc                    t          ddgddd          }|d d          }|                                }t          g dg dgt          d                    }|||<   | rt          j        t          |d          t          |d                    rJ t          j        t          |d	          t          |d	                    |u sJ t          j        t          |d
          t          |d
                    sJ |j                            d          |usJ |j                            d          rJ t          j	        ||           d S |dk    ryt          j        t          |d          t          |d                    sJ t          j        t          |d
          t          |d
                    sJ |j
        d         dk    sJ 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&   rU  expr'   ru  r   indexers          r)   "test_putmask_dont_copy_some_blocksr  M  s   	!Qac22	3	3Baaa5DggiiG			3334d5kk  G BwK $#IdC$8$8)B:L:LMMMMM	$ 4 4iC6H6HIISPPPP	$ 4 4iC6H6HIIIIIw((++366667,,Q/////
dG,,,,,		$ 4 4iC6H6HIIIII	$ 4 4iC6H6HIIIIIy!####	 
 $#r+   c                    t          g d|          }|                                }|                    |dk    d          }| r2t          j        t          |          t          |                    sJ n1t          j        t          |          t          |                    rJ d|j        d<   | r1t          j        t          |          t          |                    rJ t          j        ||           d S Nr   rj  r   r   )	r   r   wherer   r   r   r$   r   r   r&   rk  r   r   r   s        r)   test_where_noopr  f  s    
%
(
(
(CxxzzHYYsQw##F G	#	&0A0ABBBBBB#IcNNIf4E4EFFFFFFKN G#IcNNIf4E4EFFFFF3)))))r+   c                    t          g d|          }|                                }|                    |dk     d          }t          j        t          |          t          |                    rJ t          j        ||           d S r  )r   r   r  r   r   r   r   r   r  s        r)   
test_wherer  x  s|    
%
(
(
(CxxzzHYYsQw##F	#	&0A0ABBBBB3)))))r+   z
dtype, val)rR   g      %@)r  r   c                 r   t          g dg dd|          }|                                }|                    |dk     |          }| rgt          j        t          |d          t          |d                    sJ t          j        t          |d          t          |d                    rJ n3t          j        t          |d          t          |d                    rJ d|j        d	<   | r3t          j        t          |d          t          |d                    rJ t          j        ||           d S )
Nr   )ir   rj  r   r   r   r   r   )	r   r   r  r   r   r   r$   r   r   )r&   rk  rU  r'   r   r   s         r)    test_where_noop_on_single_columnr    s6   	66e	D	D	DBggiiGXXb1fc""F P	"c 2 2Ifc4J4JKKKKK#Ib#$6$6	&#8N8NOOOOOO#Ib#$6$6	&#8N8NOOOOOFK P#Ib#$6$6	&#8N8NOOOOO"g&&&&&r+   c                    t          dg dit          ddd                    }|                                }|                    d          }| r4t	          j        t          |d          t          |d                    sJ n3t	          j        t          |d          t          |d                    rJ d	|j        d
<   t	          j        t          |d          t          |d                    rJ t          j	        ||           d S )Nr   )g        N       @      @z1/1/2000r   Tr   r   r2  r   r   )
r   r
   r   asfreqr   r   r   r$   r   r   r   s       r)   test_asfreq_noopr    s   		###$QS999
 
 
B ggiiG
)))

C M	#s 3 3Yr35G5GHHHHHH#Ic3$7$72s9K9KLLLLL CHTN	#s 3 3Yr35G5GHHHHH"g&&&&&r+   c                     t          dddg d          }|                                }|                                D ]\  }}d|j        d<   | rt	          j        ||           d S d S )Nr   r   r   r   r   r   )r   r   iterrowsr$   r   r   )r&   r'   r   r  subs        r)   test_iterrowsr    s    	##999	5	5	5BggiiG++--  3 +
b'*****+ +r+   c                 @   t          ddt          j        dgi          }|d d          }|                                }|                    d           d|j        d<   | rt          j        ||           d S t          dg di          }t          j        ||           d S )	Nr   r   r   T)r        Y@r   )r  r   r   )r   r   r   r   ffillr$   r   r   )r&   r'   ru  r   s       r)   test_interpolate_creates_copyr    s    	C#rvq)*	+	+Baaa5DwwyyHHHTHBGDM .
dH-----c???344
dH-----r+   c                    t          g dg dg dd          }|                                }|                    d           }|                    dt          j        g d                     | rgt          j        t          |d          t          |d                    sJ t          j        t          |d	          t          |d	                    sJ nft          j        t          |d          t          |d                    rJ t          j        t          |d	          t          |d	                    rJ d
|j        d<   t          j	        ||           | r5t          j        t          |d          t          |d                    sJ d S t          j        t          |d          t          |d                    rJ d S )Nr   r   r  r   r-   r   )r  r   r   r   r  )
r   r   isetitemr   arrayr   r   r  r   r   r   s       r)   test_isetitemr    s   	CC	D	DBggiiG
''t'

CLLBH\\\**+++ M	"c 2 2Ic34G4GHHHHH	"c 2 2Ic34G4GHHHHHH#Ib#$6$6	#s8K8KLLLLL#Ib#$6$6	#s8K8KLLLLLCGFO"g&&& M	"c 2 2Ic34G4GHHHHHHH#Ib#$6$6	#s8K8KLLLLLLLr+   zsingle-blockzmixed-blockc                    t          g dt          j        g d|          d          }t          g d          }|                                }|                    d|           | r2t          j        t          |d          t          |                    rJ d|j        d<   t          j
        ||           t          g dt          j        g d|          d          }t          g d          }|                    d|           d|j        d<   t          g dt          j        g d|          d          }t          j        ||           d S )	Nr   r   rj  r   r  r   r   r  )r   r   r  r   r   r  r   r   r  r   r   r   )r&   rk  r'   r   r   r   s         r)   test_isetitem_seriesr     sW    
)))5)I)I)IJJ	K	KB


CxxzzHKK3 H#Ib#$6$6	#GGGGG BF6N3))) 
)))5)I)I)IJJ	K	KB


CKK3CGAJyyyrx			/O/O/OPPQQH"h'''''r+   keyc                    t          g dg dd          }|                                }|                    |          }| rt          j        t          |d          t          |d                    sJ d|j        d<   t          j        t          |d          t          |d                    rJ t          j        ||           d S t          |t                    rt          nd }t          j        dd          5  t          j        |          5  d|j        d<   d d d            n# 1 swxY w Y   d d d            n# 1 swxY w Y   t          |t                    rt          j        ||           d S |j        d         dk    sJ d S )	Nr   r   r   r   r   chained_assignmentwarnr   )r   r   getr   r   r   r$   r   r   
isinstancer  r   pdoption_contextassert_produces_warning)r&   r  r'   r   r   r  s         r)   test_getr
    s   	33	4	4BggiiGVVC[[F &	&# 6 6	"c8J8JKKKKKA#Ifc$:$:Ib#<N<NOOOOO
b'***** *4C)>)>H%%D3V<< 	# 	#+D11 # #!"A# # # # # # # # # # # # # # #	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# c4   	&!"g.....74=A%%%%%%s6   <D?D(D?(D,	,D?/D,	0D??EEz	axis, keyr   )r   r   c                    |dk    o| }|p|o|dk    }t          g dg dt          j        g d|          d          }|                                }|                    ||          }	|dk    s|r3t          j        t          |d	          t          |	                    sJ n| r|	j                            d
          sJ | s|rd
|	j	        d
<   nht          j        dd          5  t          j        t                    5  d
|	j	        d
<   d d d            n# 1 swxY w Y   d d d            n# 1 swxY w Y   | s|s|d
k    rt          j        ||           d S |j	        d         d
k    sJ d S )NrR   r   r   r   r  rj  r   r  r   r   r  r  r   )r   r   r  r   xsr   r   r    r   r$   r  r  r   r	  r   r   )
r&   r   rG   r  rk  single_blockis_viewr'   r   r   s
             r)   test_xsr    s+   
 W$A.A*ALA2@tqyG	iiiiibhyyy.N.N.NOO
 
B ggiiGUU3TU""FqyyLy	"c 2 2If4E4EFFFFFF	 0{,,Q///// #g #A3V<< 	# 	#+,BCC # #!"A# # # # # # # # # # # # # # #	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	#  "< "DAII
b'*****wt}!!!!!!s6   )D1DD1D	D1!D	"D11D58D5rG   z
key, level)l1r   r  c                 (   t          j        d                              dd          }t          j        ddgg dgddg	          }t          ||t          d
                    }|dk    r&|                                                                }|                                }|	                    |||          }	|dk    rIt          j
        t          ||j        d                   t          |	|	j        d                             sJ | s	|st          nd }
t          j        dd          5  t!          j        |
          5  d|	j        d<   d d d            n# 1 swxY w Y   d d d            n# 1 swxY w Y   t!          j        ||           d S )N   r   r   r  l2r   lev1lev2r  r  r   r   )levelrG   r   r  r  r   )r   arangereshaper   from_productr   r  r   r   r  r   r   r3   r   r  r  r   r	  r$   r   )r&   r   r  r  rG   arrrF   r'   r   r   r  s              r)   test_xs_multiindexr  /  s    )B--

1
%
%C#dD\999$=ffEUVVVE	3eT%[[	9	9	9Bqyy\\^^  ""ggiiGUU3e$U//Fzzb"*Q-(()FFN1<M*N*N
 
 	
 	
 	
 #	+>	 	
 
	/	8	8 " "'-- 	" 	" !FK	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	"" " " " " " " " " " " " " " " "g&&&&&s6   /E2EE2E	E2"E	#E22E69E6c                 @   t          g dg dd          }t          ddgidg          }|                                }|d d          }|                    |           t          g dg dd          }t          j        ||           | r}t          j        ||           t          j        t          |d	          t          |d	                    sJ t          j        t          |d          t          |d                    rJ d S t          j        ||           d S )
N      ?r  r  )      @g      @      @r   r         Y@r   r   )r  r!  r   r   )r   r   updater   r   r   r   r   )r&   df1r   df1_origru  r   s         r)   test_update_framer%  L  s)   
///@@
A
AC
S5'N1#
.
.
.CxxzzHqqq6DJJsOOO5F5F5FGGHHH#x((( .
dH---	#s 3 3YtS5I5IJJJJJ#Ic3$7$749M9MNNNNNNN
dH-----r+   c                 \   t          g d          }t          dgdg          }|                                }|d d          }|                    |           t          g d          }t          j        ||           | rt          j        ||           d S t          j        ||           d S )Nr  r!  r   r   )r  r!  r  )r   r   r"  r   r   )r&   ser1r   	ser1_origru  r   s         r)   test_update_seriesr)  _  s    ///""D5'!%%%D		I7DKK'''((H4*** /
tY/////
tX.....r+   c                      t          g d          } t          |           }| dz  } t          j        t          |           |          sJ t	          j        |t          |                      d S Nr   r   )r   r   r   r   r   assert_numpy_array_equal)r   datas     r)   test_inplace_arithmetic_seriesr.  p  se    


CS>>D1HCIcNND11111inn55555r+   c                 d   t          g d          }|                                }|d d          }|dz  }| rHt          j        t	          |          t	          |                    rJ t          j        ||           d S t          j        t	          |          t	          |                    sJ d S r+  )r   r   r   r   r   r   r   )r&   r   r   ru  s       r)   -test_inplace_arithmetic_series_with_referencer0  x  s    


CxxzzHqqq6D1HC A#IcNNIdOODDDDD
x.....	#	$@@@@@@@r+   TFc                    t          g ddd          }|                                }|                    |          }|s|r| r4t          j        t          |d          t          |d                    sJ n3t          j        t          |d          t          |d                    rJ d|j        d<   | rt          j        ||           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   s         r)   test_transposer2    s    	++	,	,BggiiG\\t\$$F N+ N/B N	"c 2 2Ifa4H4HIIIIII#Ib#$6$6	&!8L8LMMMMMFK +
b'*****+ +r+   c                    t          g ddd          }|                                }|j        }t          j        t          |d          t          |d                    rJ d|j        d<   | rt          j        ||           d S d S )Nr   r   r   r   r   r   r   )	r   r   r  r   r   r   r$   r   r   )r&   r'   r   r   s       r)   test_transpose_different_dtypesr4    s    	--	.	.BggiiGTF	"c 2 2Ifa4H4HIIIIIFK +
b'*****+ +r+   c                     t          dg did          }|j        }t          j        t	          |d          t	          |d                    rJ d S )Nr   r   r  rj  r   )r   r  r   r   r   )r&   r'   r   s      r)   test_transpose_ea_single_columnr6    sW    	C#7	3	3	3BTF	"c 2 2Ifa4H4HIIIIIIIr+   c                      t          ddgdd          } |                                 }d|j        d<   t          ddgddg	          }t	          j        ||           d S )
Nr   r   r   r   r   r   r   r   r   )r   countr$   r   r   r   rA  s      r)   test_count_read_only_arrayr9    sf    	!Qa((	)	)BXXZZFFKNsAhsCj111H68,,,,,r+   c                    t          g d          }|                                }|                                }t          j        t          |          t          |                    sJ | r|j                            d          rJ d|j        d<   | rt          j
        ||           d S t          g d          }t          j
        ||           d S )Nr   r   r   r   )r   r   ru  r   r   r   r    r   r$   r   r   r  s        r)   test_series_viewr;    s    


CxxzzH88::DIcNNIdOO<<<<< 29..q11111DIaL .
x-----+++&&
sH-----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/  r7  r<  r?  rB  rI  rP  r   rV  rX  rd  rg  rs  rv  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.  r0  r2  r4  r6  r9  r;  r   r+   r)   <module>rF     s        0 0 0 0 0 0    	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	       1 1 1 1 1 1  M M M4 !4!4!455@@BB7733NN@@BB==55554433::@@??OO#&	 	 	+  ( (RM MS( ( 65TM: !4!4!455??55AA::66PPBB888877776655<<BBAAQQ#&	 	 	+  ( (RE ES( ( 65TE0 !4!4!455	N 	N 65	N ' ' ', =2=A#6#6q!f"FGG* * HG*' ' '$) ) )(E E E0E E E.' ' '(' ' '&' ' '( w&~F ' ' '$( ( ((I I I( ( (*) ) )&7 7 7* 
));; ' ' '"0 0 0,+ + + . . .&, , ,D 344' ' 54'"' ' '"( ( ( 	8	8!<	2	2A6	J	JAN	G	GK	 	CBB  	 	: :	 	:( AS!9!9!999cAq6];S;S TUU# # VU#" AS!9!9!999cAq6];S;S TUU# # VU#"' ' '", , ,' ' '*' ' '$ x!RV&=>>' ' ?>'  C))* * *)*  	 ' ' '*' ' '*; ; ;8; ; ;6 a00q!,, ' ' '" H.?#@AA' ' BA'" AC!Q=1I1I JKK# # LK#  AC!Q=1I1I JKK# # LK# ' ' '$* * *" fYYYS!!!2&C3C)D)DtSk(RS # #	 # fYYYS!!!2&C3C)D)DtSk(RS K K	 K  ZZZ00' ' 10'0' ' '&* * *& 			!2!2IIsIII>N4O4O PQQ# # RQ#"' ' '"* * *". . .$ Xv$6&8I#JKK' ' LK'  02GH * * *&' ' '&# # #&, , ,* 7G"455$ $ 65$ 7G"455; ; 65; 9i"899; ; :9; k:%>??$ $ @?$0 7G"455* * 65*" 7G"455* * 65* 'GHH' ' IH'$' ' '(+ + +. . . M M M, gy!'F   ( ( (0 se--& & .-&2 vx&899gy!'F   " "  :9"8 !Q((	6':;;' ' <; )('6. . .&/ / /"6 6 6	A 	A 	A $//+ + 0/++ + +J J J- - -. . . . .r+   