
    Pi                     6   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lmZ d Zd Zej                            d          ej                            d	g d
          ej                            dd d d d d d d d d d d d d d d gg d          d                                     Zej                            d          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d0 Zej                            d1 ej        d d2           ed3d2g          g          d4             Zd5 Z d6 Z!d7 Z"d8 Z#d9 Z$d: Z%d; Z&ej                            d1d< d= d> d? gg d@          dA             Z'dB Z(dC Z)ej                            dDdEdFgidGdFidHdFig          dI             Z*dJ Z+dK Z,dL Z-dM Z.dN Z/ej                            dOdP dQ dR g          dS             Z0dT Z1dU Z2dV Z3dW Z4ej                            dXdY d fdZ d2fd[ d fd\ d fgg d]          d^             Z5ej                            d_ ed3d2gdF`           edFd3d2gi          g          da             Z6ej                            d_ ed3d2gdF`           edFd3d2gi          g          db             Z7dc Z8dd Z9de Z:df Z;ej                            dgdhd3ej<        fg          di             Z=ej                            djdkdlg          dm             Z>ej                            ddn do dp dq g          dr             Z?ds Z@dt ZAdu ZBej                            dvdFdwd3dxd d3d dxg          dy             ZCej                            ddzd{g          d|             ZDej                            d_ ed3d2g           edFd3d2gi          g          d}             ZEej                            d_ ed3d2g           edFd3d2gi          g          d~             ZFd ZGd ZHej                            d eg ddF`          i f edFg di          ddFifg          d             ZIej                            d eg ddF`          i f edFg di          ddFifg          d             ZJej                            dg d          d             ZKd ZLd ZMej                            d_ eg d           edFg di          g          d             ZNd ZOd ZPd ZQej                            dvddid1dig          d             ZRej                            dddg          d             ZSd ZTd ZUd ZVej                            dddg          d             ZWej                            dddg          d             ZXej                            dddg          d             ZYej                            dddg          deZfd            Z[ej                            dddg          ej                            dOd d g          d                         Z\ej                            dddg          ej                            dOd d g          d                         Z]ej                            dddg          ej                            dOd d g          d                         Z^ej                            dOddg          d             Z_d Z`d Zad Zbd Zcej                            dddgddg          d             Zdd Zeej                            ddFdFgg          d             Zfej                            dddg          ej                            dddgddg          d                         Zgej                            dd d3g          ej                            dddg          dǄ                         ZhdȄ ZidɄ Zjdʄ Zkd˄ Zld̄ Zmd̈́ Znd΄ Zodτ ZpdЄ Zqdф Zrd҄ Zsdӄ ZtdԄ ZudՄ Zvdք Zwdׄ Zxd؄ ZydS )    N)HAS_PYARROW)	DataFrameIndex
MultiIndexPeriodSeries	Timestamp
date_rangeperiod_range)	get_array)Versionc                     t          g dg dg dd          } |                                 }|j        | j        usJ |j        | j        usJ |j                            | j                  sJ |j                            | j                  sJ t          j        t          |d          t          | d                    rJ |j        j	        d         j
                                        rJ |j        j	        d         j
                                        rJ |j        | j        usJ |j        | j        us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indexcolumnsis_npshares_memoryr   _mgrblocksrefshas_referenceilocdfdf_copys     w/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pandas/tests/copy_view/test_methods.py	test_copyr-      sh   	II	J	JBggiiG =((((?"*,,,,=RX&&&&&?rz***** 	'3 7 72s9K9KLLLLL|"1%*88:::::|"1%*88:::::=((((?"*,,,, GL74=A    c                  @   t          g dg dg dd          } |                     d          }|j        | j        usJ |j        | j        usJ |j                            | j                  sJ |j                            | j                  sJ t          j        t          |d          t          | d                    sJ |j        j	        d         j
                                        sJ |j        j	        d	         j
                                        sJ d|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 )Nr   r   r   r   Fdeepr   r   r   r   r   r   r)   s     r,   test_copy_shallowr2   -   s   	II	J	JBgg5g!!G =((((?"*,,,,=RX&&&&&?rz***** Igs33Yr35G5GHHHHH<q!&4466666<q!&4466666 GL74=A	'3 7 72s9K9KLLLLLIgs33Yr35G5GHHHHHHHr.   zignore::DeprecationWarningr   )TNFmethodc                 D    |                      t          j        |          S )Nr    r   )renamestrlowerr*   r   s     r,   <lambda>r:   J   s    394@@ r.   c                 4    |                      ddg|          S )Nr   r   r5   )reindexr9   s     r,   r:   r:   K   s    S#JTBB r.   c                 0    |                      | |          S Nr   reindex_liker9   s     r,   r:   r:   L   s    $77 r.   c                 <    |                      | |          d         S Nr?   r   alignr9   s     r,   r:   r:   M   s    "4003 r.   c                 6    |                      g dd|          S Nr   r   )axisr   set_axisr9   s     r,   r:   r:   N   s    ___7NN r.   c                 0    |                      d|          S Ntestr   r   rename_axisr9   s     r,   r:   r:   O   s    f4@@ r.   c                 0    |                      d|          S )NrM   r5   rO   r9   s     r,   r:   r:   P   s    TBB r.   c                 4    |                      ddi|          S )Nr   int64r?   astyper9   s     r,   r:   r:   Q   s    C>== r.   c                 2    |                      dd|          S Nr   r   r?   truncater9   s     r,   r:   r:   S   s    Q55 r.   c                 .    |                      |          S r>   infer_objectsr9   s     r,   r:   r:   T   s    ))t)44 r.   c                 .    |                      |          S r>   to_timestampr9   s     r,   r:   r:   U   s    d33 r.   c                 0    |                      d|          S ND)freqr   	to_periodr9   s     r,   r:   r:   V   s    3T:: r.   c                 0    |                      d|          S N
US/Centralr?   tz_localizer9   s     r,   r:   r:   W   s    4@@ r.   c                 0    |                      d|          S rg   
tz_convertr9   s     r,   r:   r:   X   s    |$?? r.   c                 0    |                      d|          S NF)allows_duplicate_labelsr   	set_flagsr9   s     r,   r:   r:   Y   s    e$OO r.   )r6   r<   rA   rE   rJ   rename_axis0rename_axis1rU   rY   r\   r_   re   rj   rm   rr   )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|          } |||          }t          j        t          |d          t          |d                    sJ d S )Nr_   
2012-01-01rb   r   rc   periodsre   rj   rm   Europe/Brusselsrc   ry   tzr   r   r   r   r   r?   r   )	nodecallspecidr   r
   r   r"   r#   r   )requestr3   r   r   r*   df2s         r,   test_methods_copy_keywordr   E   s   T 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Ic3//2s1C1CDDDDDDDr.   c                 4    |                      ddi|          S )Nr   d   rN   r6   serr   s     r,   r:   r:      s    #**As8$*?? r.   c                 0    |                      d |          S r>   r   r   s     r,   r:   r:      s    #**T*55 r.   c                 :    |                      | j        |          S )NrN   )r<   r   r   s     r,   r:   r:      s    #++CID+AA r.   c                 0    |                      | |          S r>   r@   r   s     r,   r:   r:      s    #**3T*:: r.   c                 <    |                      | |          d         S rC   rD   r   s     r,   r:   r:      s    #))Cd)33A6 r.   c                 6    |                      g dd|          S rG   rI   r   s     r,   r:   r:      s    #,,W4,PP r.   c                 0    |                      d|          S rL   rO   r   s     r,   r:   r:      s    #//T/BB r.   c                 0    |                      d|          S )NrS   r?   rT   r   s     r,   r:   r:      s    #**W4*88 r.   c                 2    |                      dd|          S )Nr   r   r?   )	swaplevelr   s     r,   r:   r:      s    #--14-88 r.   c                 2    |                      dd|          S rW   rX   r   s     r,   r:   r:      s    #,,q!$,77 r.   c                 .    |                      |          S r>   r[   r   s     r,   r:   r:      s    #+++66 r.   c                 .    |                      |          S r>   r^   r   s     r,   r:   r:      s    #***55 r.   c                 0    |                      d|          S ra   rd   r   s     r,   r:   r:      s    #--St-<< r.   c                 0    |                      d|          S rg   ri   r   s     r,   r:   r:      s    #//,T/BB r.   c                 0    |                      d|          S rg   rl   r   s     r,   r:   r:      s    #..D.AA r.   c                 0    |                      d|          S ro   rq   r   s     r,   r:   r:      s    #--D-QQ r.   )zrename (dict)r6   r<   rA   rE   rJ   rs   rU   r   rY   r\   r_   re   rj   rm   rr   c                 8   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|          } |||          }t          j	        t          |          t          |                    sJ d S )Nr_   rw   rb   r   rx   re   rj   rm   rz   r{   r   r   r   r}   r?   )r~   r   r   r   r
   r   from_arraysr   r"   r#   r   )r   r3   r   r   r   ser2s         r,    test_methods_series_copy_keywordr   ~   s6   T 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IdOOYs^^<<<<<<<r.   c                     t          g dg dg ddg d          } |                                 }|                                 }|j                                         t          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 t          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*   df_origr   s      r,   test_reset_indexr      sC    
iiiiiooo>>lll
 
 
B ggiiG
..

CH    Ic3//2s1C1CDDDDDIc3//2s1C1CDDDDDCHTN	#s 3 3Yr35G5GHHHHHIc3//2s1C1CDDDDD"g&&&&&r.   r   r   r   c                 X   t          ddg|           }|                                }|                    d          }t          j        t          |          t          |                    sJ |j                            d          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   r   ser_origr   s       r,   test_reset_index_series_dropr      s    
!Qu
%
%
%CxxzzH???%%DIcNNIdOO<<<<<x))!,,,,,DIaL3)))))r.   c                  @   t          g dg dg dd          } |                     d          } | d         }|                     |d                                          }|                     dd                                          }t	          j        ||           d S )	Nr   r   r   dr   r   r   r   )r   r   r   r   ABCr   r   T)observed)r   	set_indexgroupbysumr   r   )r*   keyresultexpecteds       r,   'test_groupby_column_index_in_referencesr      s    	"""<P<P<PQQ
 
B 
c		B
S'CZZdZ++//11Fzz#z--1133H&(+++++r.   c                     t          g d          } t          g dg dd          }|                    |           }d| j        d<   |                                }t          ddgd	d
gdddg          }t          j        ||           d S )N)r   r   r   r   r   r   r   r   r   r   r   r   r   r   r}   )r   r   r   r(   r   r   r   )r   r*   gbr   r   s        r,   test_groupby_modify_seriesr      s     

C	33	4	4B	CBCHQKVVXXF1vRG44QFCCCH&(+++++r.   c                  l   t          g dg dg dd          } |                                 }|                     t          j                  }t          j        t          |d          t          | d                    sJ d|j        d	<   t          j        t          |d          t          | d                    rJ t          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   r   r   )r   r   r6   r7   upperr"   r#   r   r(   r   r   r*   r   r   r   s       r,   test_rename_columnsr      s*    
II	J	JBggiiG
))CI)
&
&CIc3//2s1C1CDDDDDCHTN	#s 3 3Yr35G5GHHHHHIc3//2s1C1CDDDDDyyyyyyOOPPH#x((("g&&&&&r.   c                  l   t          g dg dg dd          } |                     t          j                  }|                                }t          j        t          |d          t          | d                    sJ d| j        d	<   t          j        t          |d          t          | d                    rJ t          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   r6   r7   r   r   r"   r#   r   r(   r   r   )r*   r   df2_origr   s       r,   !test_rename_columns_modify_parentr     s*    
II	J	JB
))CI)
&
&CxxzzHIc3//2s1C1CDDDDDBGDM	#s 3 3Yr35G5GHHHHHIc3//2s1C1CDDDDD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<   t          j        | |           t          j        t          |d          t          | d                    r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   piper"   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"g&&&	#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 t          j        | |           t          j        t          |d          t          | d                    r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   s       r,   test_pipe_modify_dfr   '  s    	--	.	.BggiiG   ''(

CIc3//2s1C1CDDDDD"g&&&	#s 3 3Yr35G5GHHHHHIc3//2s1C1CDDDDDDDr.   c                      t          g dg dg dd          } |                                 }|                     ddg          }t          j        t          |d          t          | d                    sJ d|j        d	<   t          j        t          |d          t          | d                    rJ t          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   8  s     
II	J	JBggiiG
**c3Z*
(
(C Ic3//2s1C1CDDDDDCHTN	#s 3 3Yr35G5GHHHHHIc3//2s1C1CDDDDD"g&&&&&r.   c                     | S r   r   idxs    r,   r:   r:   K  s    C r.   c                 *    |                                  S r   )viewr   s    r,   r:   r:   L      CHHJJ r.   c                 *    |                                  S r   r?   r   s    r,   r:   r:   M  r   r.   c                      t          |           S r   )listr   s    r,   r:   r:   N  s    DII r.   )	identicalr   r   valuesc                    t          g dg dg dd          }|                                }|                     | |j                            }t	          j        t          |d          t          |d                    sJ d|j        d<   t	          j        t          |d          t          |d                    rJ t	          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   r   s       r,   test_reindex_rowsr   H  s     
II	J	JBggiiG
**55??*
+
+C Ic3//2s1C1CDDDDDCHTN	#s 3 3Yr35G5GHHHHHIc3//2s1C1CDDDDD"g&&&&&r.   c                     t          g dg dg dd          } |                                 }|                     d          }|j                                         t          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 t          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   b  s1   	II	J	JBggiiG
''#'

CH   Ic3//2s1C1CDDDDDIc3//2s1C1CDDDDDCHTN	#s 3 3Yr35G5GHHHHHIc3//2s1C1CDDDDD"g&&&&&r.   c                     t          g dg dg dd          } |                                 }|                     d          }|j                                         t          j        t          |d          t          | d                    sJ d|j        d<   t          j        t          |d          t          | d                    rJ t          j
        | |           d S )	Nr   r   r   r   rS   r   r   r   )r   r   select_dtypesr$   r   r"   r#   r   r(   r   r   r   s      r,   test_select_dtypesr   p  s     
II	J	JBggiiG


7
#
#CH   Ic3//2s1C1CDDDDDCHTN	#s 3 3Yr35G5GHHHHH"g&&&&&r.   filter_kwargsitemsr   likeregexc                    t          g dg dg dd          }|                                } |j        di | }t          j        t          |d          t          |d                    s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   filterr"   r#   r   r(   r   r   )r   r*   r   r   s       r,   test_filterr     s     
II	J	JBggiiG
")
$
$m
$
$CIc3//2s1C1CDDDDD CHTN	#s 3 3Yr35G5GHHHHH"g&&&&&r.   c                  `   t          ddgddgddggt          dd          d	d
g          } |                                 }|                     d          }t	          j        t          |d	          t          | d	                    sJ |j        | j        usJ |j        | j        usJ d| j	        d<   t	          j        t          | d	          t          |d	                    rJ t	          j        t          | d
          t          |d
                    sJ t          j        ||           d S )Nr   r   r   r   r   r   
2020-01-01
2020-01-03r   r   r   r    r   )ry   r   )r   r
   r   shiftr"   r#   r   r   r    r(   r   r   r   s      r,   test_shift_no_opr     s7   	
Q!Q!Q |44c

 
 
B
 ggiiG
((1(

CIc3//2s1C1CDDDDD9BH$$$$;bj((((BGDM	"c 2 2Ic34G4GHHHHHIb#..	#s0C0CDDDDD#w'''''r.   c                  0   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 |j        | j        usJ |j        | j        usJ d S )Nr   r   r   r   r   r   r   r   r   r   r   r   ry   rH   )r   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9BH$$$$;bj((((((r.   c                     t          ddgddgddggt          dd          d	d
g          } |                                 }t          dd          |_        |                     dd          }t          j        t          |d	          t          | d	                    s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   
2020-01-02z
2020-01-041Dry   rc   r   r   )r   r
   r   r   r   r"   r#   r   r(   r   r   r   s      r,   test_shift_rows_freqr    s    	
Q!Q!Q |44c

 
 
B
 ggiiG|\::GM
((14(
(
(CIc3//2s1C1CDDDDDBGDM	"c 2 2Ic34G4GHHHHH#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<   t          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(   nanr   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#|$$iL&A&A     
&!rvqkBFA;/<66  H #x(((((r.   c                  d   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 d|j        d<   t          j        |j        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   popr"   r#   r   r   r(   r   r   )r*   r   view_originalr   s       r,   test_popr
    s!   	II	J	JBggiiGqqqEMVVC[[FFM9]C+H+HIIIIIIb#..	-0M0MNNNNNFKNy/L/LMMMMMBGDM	"c 2 2ImS4Q4QRRRRR-11111r.   funcc                 ,    |                      |          S r   rD   xys     r,   r:   r:     s    QWWQZZ r.   c                 :    |                      |j        d          S )Nr   rH   )rE   r   r  s     r,   r:   r:     s    QWWQSqW)) r.   c                 l    |                      |j        j        t          dd                   d          S )Nr   r   r  )rE   r   r(   slicer  s     r,   r:   r:     s'    QWWQSXeAqkk2W;; r.   c                    t          g ddd          }|                                }|ddg                                         } | ||          \  }}t          j        t	          |d          t	          |d                    s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  r*   r   
df_changedr   _s         r,   test_align_framer    s     
--	.	.BggiiGS#J$$&&JT"j!!FCIc3//2s1C1CDDDDDCHTN	#s 3 3Yr35G5GHHHHH"g&&&&&r.   c                  ,   t          ddg          } |                                 }|                                 }|                     |          \  }}t          j        |j        | j                  sJ t          j        |j        |j                  sJ d|j        d<   d|j        d<   t          j        |j        | j                  rJ t          j        |j        |j                  rJ t          j        | |           t          j        ||           d S )Nr   r   r   )	r   r   rE   r"   r#   r   r(   r   r   )r   r   	ser_otherr   ser_other_results        r,   test_align_seriesr    s   
!Q..CxxzzH

I YYy11D
DK44444,3Y5EFFFFFDIaL !SZ88888 0 79IJJJJJ3)))9h/////r.   c                     t          g dg dd          } |                                 }|                     |           \  }}t          j        t          | d          t          |d                    sJ t          j        t          | d          t          |d                    sJ d|j        d<   t          j        | |           d|j        d<   t          j        | |           d S )Nr   r   r   r   r   r   r   r   )	r   r   rE   r"   r#   r   locr   r   )r*   r   r   df3s       r,   test_align_copy_falser!    s    	33	4	4BggiiGxx||HCIb#..	#s0C0CDDDDDIb#..	#s0C0CDDDDDCGFO"g&&&CGFO"g&&&&&r.   c                     t          g dg dd          } t          g 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 d|j        d
<   t          j	        | |           d|j        d<   t          j
        ||           d S )Nr   r   r   r  namer   r  r   r   r  )r   r   r   rE   r"   r#   r   r  r   r   r   )r*   r   r   r   r   r   s         r,   !test_align_with_series_copy_falser%    s1   	33	4	4B

%
%
%CxxzzHggiiG1%%ICIb#..	#s0C0CDDDDDIb#..	#s0C0CDDDDDIc3//41E1EFFFFFCGFO"g&&&DHQK3)))))r.   c                  P   t          g d          } |                                 }| d d                                          }t          j        | j        t          |d                    sJ d|j        d<   t          j        | j        t          |d                    rJ t          j	        | |           | d d                                          }d| j        d<   t          j
        ||                                           |                                 }|j        | j        usJ d S )Nr   r   r   )r   r   to_framer"   r#   r   r   r(   r   r   r   r   )r   r   r*   s      r,   test_to_framer(  &  s   


CxxzzH	QQQ		B CJ	"a(8(899999BGDM 
Ib!,<,<=====3))) 
QQQ		BCHQK"h//11222	B839$$$$$$r.   zmethod, idxc                 V    |                      d                               d          S )NFr0   r?   r   s    r,   r:   r:   C  s%    BGGG'',,%,88 r.   c                 N    |                                                                   S r   )r   r   s    r,   r:   r:   D  s    BNN$$0022 r.   c                 ~    |                      t          j                                       t          j                  S )Nr   )r6   r7   r   r8   r   s    r,   r:   r:   E  s)    BIIciI0077	7JJ r.   c                 V    |                      d                              d          S )NFr0   number)include)r   r   r   s    r,   r:   r:   F  s%    BGGG''55h5GG r.   )zshallow-copyr   r6   r   c                 .   t          g dg dg dd          }|                                } | |          }d|j        d|f<   t          j        ||            | |          }d|j        d<   t          j        |j        d d |d f         |           d S )Nr   r   r   r   r   r   )r   r   r(   r   r   )r3   r   r*   r   r   s        r,   test_chained_methodsr0  @  s     
II	J	JBggiiG &**CCHQV"g&&& &**CBGDM#(111cdd7+W55555r.   objr#  c                    t          t          dd          t          dd          g          | _        |                                 }|                                 }t          j        t          |d          t          | d                    sJ d|j        d<   t          j        t          |d          t          | d                    rJ t          j
        | |           d S )Nz2012-1-1rb   rc   z2012-1-2r   r   )r   r   r   r   r_   r"   r#   r   r(   r   assert_equalr1  obj_origobj2s      r,   test_to_timestampr8  Y  s    vjs333VJS5Q5Q5QRSSCIxxzzHDIdC00)C2E2EFFFFF DIaL	$ 4 4iS6I6IJJJJJOC"""""r.   c                    t          t          d          t          d          g          | _        |                                 }|                     d          }t          j        t          |d          t          | d                    sJ d|j        d<   t          j        t          |d          t          | d                    rJ t          j
        | |           d S )N
2019-12-31
2020-12-31Yr3  r   r   )r   r	   r   r   re   r"   r#   r   r(   r   r4  r5  s      r,   test_to_periodr=  h  s    y..	,0G0GHIICIxxzzH==c=""DIdC00)C2E2EFFFFF DIaL	$ 4 4iS6I6IJJJJJOC"""""r.   c                     t          g dg dg dd          } |                                 }|                     d          }t          j        t          |d          t          | d                    s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#   r   r(   r   r   r   s      r,   test_set_indexr@  w  s    	II	J	JBggiiG
,,s

CIc3//2s1C1CDDDDD 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_indexrC    s]    	++	,	,B\\#F{{}}HBGDM&(+++++r.   c                  V   t          g dg dg dd          } |                                 }|                     d          }t          j        t          |d          t          | d                    sJ d|j        d	<   t          j        t          |d          t          | d                    rJ t          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   )rF  CoW_brG  )	r   r   
add_prefixr"   r#   r   r(   r   r   r   s       r,   test_add_prefixrJ    s+   	II	J	JBggiiG
--

CIc733Yr35G5GHHHHHCHTN	#w 7 72s9K9KLLLLLIc733Yr35G5GHHHHH))iii///JJ H #x((("g&&&&&r.   c                  V   t          g dg dg dd          } |                                 }|                     d          }t          j        t          |d          t          | d                    sJ d|j        d	<   t          j        t          |d          t          | d                    rJ t          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   )rM  b_CoWrN  )	r   r   
add_suffixr"   r#   r   r(   r   r   r   s       r,   test_add_suffixrQ    s+   	II	J	JBggiiG
--

CIc733Yr35G5GHHHHHCHTN	#w 7 72s9K9KLLLLLIc733Yr35G5GHHHHH))iii///JJ H #x((("g&&&&&r.   z	axis, val)r         @c                    t          g dd|dgdd          }|                                }|                    |           }t          j        t          |d          t          |d                    s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   dropnar"   r#   r   r(   r   r   )rH   valr*   r   r   s        r,   test_dropnarV    s    	!S!3??	@	@BggiiG
)))

CIc3//2s1C1CDDDDDCHTN	#s 3 3Yr35G5GHHHHH"g&&&&&r.   rU  r   rR  c                 <   t          d| dg          }|                                }|                                }t          j        |j        |j                  sJ d|j        d<   t          j        |j        |j                  rJ t          j        ||           d S )Nr   r   r   )	r   r   rT  r"   r#   r   r(   r   r   )rU  r   r   r   s       r,   test_dropna_seriesrX    s    
!S!

CxxzzH::<<DDK44444DIaLSZ888883)))))r.   c                 *    |                                  S r   headr   s    r,   r:   r:         27799 r.   c                 ,    |                      d          S )Nr   rZ  r   s    r,   r:   r:         2771:: r.   c                 *    |                                  S r   tailr   s    r,   r:   r:     r\  r.   c                 ,    |                      d          S )Nr   r`  r   s    r,   r:   r:     r^  r.   c                 x   t          g dg dd          }|                                } | |          }|j                                         t	          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 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(   r   r   r3   r*   r   r   s       r,   test_head_tailre    s+    
99	:	:BggiiG
&**CH    	#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	#s 3 3Yr35G5GHHHHH"g&&&&&r.   c                    t          ddgt          ddgt                    ddd          }|                                }|                                }t          j        t          |d          t          |d                    sJ | r;t          r4t          j        t          |d          t          |d                    rJ n3t          j        t          |d          t          |d                    sJ d	|j
        d
<   d|j
        d<   t          j        t          |d          t          |d                    rJ t          j        t          |d          t          |d                    rJ t          j        ||           d S )Nr   r   r  r  dtyper   r   r   r   r   r   r?  )r   r   objectr   r\   r"   r#   r   r   r   r(   r   )using_infer_stringr*   r   r   s       r,   test_infer_objectsrk    sy   	!f63*F;;;!#NN
 
B ggiiG




CIc3//2s1C1CDDDDD Ik I#Ic3$7$72s9K9KLLLLLL	#s 3 3Yr35G5GHHHHHCHTNCHTN	#s 3 3Yr35G5GHHHHH	#s 3 3Yr35G5GHHHHH"g&&&&&r.   c                    t          ddgt          ddgt                    dt          t          d          t          d          gd          t          d	d
gt                    d          }|                                }t          |d          }t          |d          }t          |d          }d|j        d<   d|j        d<   t          d          |j        d<   t          j        |t          |d                    sJ | r-t          r&t          j        |t          |d                    sJ n%t          j        |t          |d                    rJ t          j        |t          |d                    sJ d S )Nr   r   r  r  rg  r:  r;  ri  zw)r   r   r   r   er   r   r   r   r   r?  
2018-12-31r   r   )r   r   ri  r	   r\   r   r(   r"   r#   r   r   )rj  r*   arr_aarr_barr_ds        r,   test_infer_objects_no_referenceru    s   	Qc
&111<(()L*A*AB(   c
&111	
 	


 

B 
				Bb#Eb#Eb#EBGDMBGDMl++BGDME9R#5#566666 ?k ? yS'9'9:::::: #E9R+=+=>>>>>E9R#5#56666666r.   c                     t          ddgt          ddgt                    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<   t          j        |t          | d
                    rJ t          j        |t          | d                    rJ t          j        |t          | d                    sJ d S )Nr   r   r  r  rg  r:  r;  ri  r   r   r   r   r   r   r?  rp  rq  )	r   r   ri  r	   r\   r   r(   r"   r#   )r*   r   rr  rs  rt  s        r,   test_infer_objects_referencerw    sJ   	Qc
&111<(()L*A*AB(  		
 	
	
 	
B aaa5D					Bb#Eb#Eb#EBGDMBGDMl++BGDMyS'9'9:::::yS'9'9:::::E9R#5#56666666r.   kwargsr   )beforeafterrH   c                    t          g dddd          }|                                } |j        di | }|j                                         t          j        t          |d          t          |d                    s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   rY   r$   r   r"   r#   r   r(   r   r   rx  r*   r   r   s       r,   test_truncater}  4  s     
33	4	4BggiiG
"+



CH   Ic3//2s1C1CDDDDDCHTN	#s 3 3Yr35G5GHHHHH"g&&&&&r.   assigndrop_duplicatesc                    t          dg di          }|                                } t          ||                       }|j                                         t          j        t          |d          t          |d                    sJ d|j        d<   t          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   rd  s       r,   test_assign_drop_duplicatesr  H  s    	C#	$	$BggiiG
'"f



CH   Ic3//2s1C1CDDDDDCHTN	#s 3 3Yr35G5GHHHHH"g&&&&&r.   c                    |                                  }|                     ddg          }t          j        |j        | j                  sJ d|j        d<   t          j        |j        | j                  rJ t          j        | |           d S )Nr   r   )r   taker"   r#   r   r(   r   r4  r5  s      r,   	test_taker  V  s     xxzzH88QFDDK44444DIaLSZ88888OC"""""r.   c                 J   t          ddd          | _        |                                 }|                     dd          }t	          j        |j        | j                  sJ d|j        d<   t	          j        |j        | j                  rJ t          j	        | |           d S )Nz
2018-04-09r   1D20minr  z0:00z1:00r   )
r
   r   r   between_timer"   r#   r   r(   r   r4  r5  s      r,   test_between_timer  b  s    <CCCCIxxzzHFF++DDK44444DIaLSZ88888OC"""""r.   c                     t          ddgdd          } t          dddgd          }|                                 }|                     |          }t          j        t          |d          t          | d                    s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   rA   r"   r#   r   r(   r   r   )r*   otherr   r   s       r,   test_reindex_liker  n  s    	!Qc**	+	+BCq!f--..EggiiG
//%
 
 CIc3//2s1C1CDDDDDCHTN	#s 3 3Yr35G5GHHHHH"g&&&&&r.   c                  :   t          g d          } |                                 }|                                 }t          j        | j        |j                  s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   s      r,   test_sort_indexr  {  s    


CxxzzH>>DCJ44444 DIaLSZ888883)))))r.   zobj, kwargsr   byc                 R   |                                  } | j        di |}t          j        t	          |d          t	          | d                    s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   r4  )r1  rx  r6  r7  s       r,   test_sort_valuesr    s    
 xxzzH3?$$V$$DIdC00)C2E2EFFFFF DIaL	$ 4 4iS6I6IJJJJJOC"""""r.   c                 j   |                                  }| d d          } | j        dddi| t          j        t	          | d          t	          |d                    sJ d| j        d<   t          j        t	          | d          t	          |d                    rJ t          j        ||           d S )NinplaceTr   r   r   r  )r1  rx  r6  r   s       r,   test_sort_values_inplacer    s    
 xxzzHqqq6DCO++D+F+++Ic3//41E1EFFFFF CHQK	#s 3 3YtS5I5IJJJJJOD(#####r.   decimals)r   r   c                 t   t          ddgdd          }|                                }|                    |           }t          j        t          |d          t          |d                    sJ | dk    r^t          t          j                  t          d          k     r4t          j        t          |d	          t          |d	                    sJ n3t          j        t          |d	          t          |d	                    rJ |j	        |j	        usJ |j
        |j
        usJ d
|j        d<   d|j        d<   t          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   z
2.4.0.dev0r   r   r?  r   r   )r   r   roundr   r#   r   r   r"   __version__r   r    r(   r   )r  r*   r   r   s       r,   
test_roundr    s   	!Qc**	+	+BggiiG
((H(
%
%CIc3//2s1C1CDDDDD1}}007<3H3HHH 	#s 3 3Yr35G5GHHHHHH#Ic3$7$72s9K9KLLLLL9BH$$$$;bj((((CHTNCHTN	#s 3 3Yr35G5GHHHHH	#s 3 3Yr35G5GHHHHH"g&&&&&r.   c                     t          j        g dddg          } t          dg di|           }|                                }|                    ddg          }t          j        t          |d          t          |d                    sJ d	|j        d
<   t          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}   orderr   r   )r   from_tuplesr   r   reorder_levelsr"   r#   r   r(   r   r   r   s       r,   test_reorder_levelsr    s    "(((  E 
C&e	4	4	4BggiiG


5%.

1
1CIc3//2s1C1CDDDDDCHTN	#s 3 3Yr35G5GHHHHH"g&&&&&r.   c                  z   t          j        g dddg          } t          g d|           }|                                }|                    ddg          }t          j        |j        |j                  sJ d|j        d<   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    s    "(((  E U
+
+
+CxxzzHUEN33DDK44444DIaLSZ888883)))))r.   c                 Z   t          j        g dddg          }|| _        |                                 }|                                 }t          j        |j        | j                  sJ d|j        d<   t          j        |j        | j                  rJ t          j
        | |           d S )Nr  r  r  r  r  r  r   )r   r  r   r   r   r"   r#   r   r(   r   r4  )r1  r   r6  r7  s       r,   test_swaplevelr    s    "#;#;#;E5>RRRECIxxzzH==??DDK44444DIaLSZ88888OC"""""r.   c                     t          g dg dg dd          } |                                 }|                     g dd          }t          j        t          |d          t          | d                    s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   rJ   r"   r#   r   r(   r   r   r   s      r,   test_frame_set_axisr    s    	II	J	JBggiiG
++oooG+
4
4CIc3//2s1C1CDDDDD CHTN	#s 3 3Yr35G5GHHHHH"g&&&&&r.   c                     t          g d          } |                                 }|                     g dd          }t          j        | |          sJ d|j        d<   t          j        ||           rJ t          j        | |           d S )Nr   r   r   r  r   )r   r   rJ   r"   r#   r(   r   r   r  s      r,   test_series_set_axisr    s    


CxxzzH<<g<66DC&&&&& DIaLc*****3)))))r.   c                     t          g d          } |                                 }|                     d          }t          j        | |          sJ d|j        d<   t          j        ||           rJ t          j        | |           d S )Nr   F)rp   r   )r   r   rr   r"   r#   r(   r   r   r  s      r,   test_set_flagsr    s    


CxxzzH===77DC&&&&& DIaLc*****3)))))r.   mapperrM   c                    t          dg dit          g dd                    }|                                } |j        di | }t	          j        t          |d          t          |d                    s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   rP   r"   r#   r   r(   r   r   r|  s       r,   test_rename_axisr    s    	C&eLLLs.K.K.K	L	L	LBggiiG
".
"
"6
"
"CIc3//2s1C1CDDDDDCHTN	#s 3 3Yr35G5GHHHHH"g&&&&&r.   zfunc, tz)rm   zEurope/Berlin)rj   Nc           	      j   t          ddgt          ddd|                    }|                                } t          ||           d          }t	          j        |j        |j                  sJ d|j        d<   t	          j        |j        |j                  rJ t          j	        ||           d S )	Nr   r   z2014-08-01 09:00h)startrc   ry   r|   r}   rh   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CJ44444 DIaLSZ888883)))))r.   c                     t          j        g dddg          } t          g dg dg dd| 	          }|                                }|                    d
          }t          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 t          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   	droplevelr"   r#   r   r(   r   r   r   s       r,   test_droplevelr  /  s@   "#;#;#;E5>RRRE	CC5	Q	Q	QBggiiG
,,q//CIc3//2s1C1CDDDDDIc3//2s1C1CDDDDD CHTN	#s 3 3Yr35G5GHHHHHIc3//2s1C1CDDDDD"g&&&&&r.   c                  b   t          dg di          } |                                 }|                                 }t          j        |j        t          | d                    sJ d|j        d<   t          j        |j        t          | d                    rJ t          j	        | |           d S )Nr   r   r   )
r   r   squeezer"   r#   r   r   r(   r   r   )r*   r   seriess      r,   test_squeezer  B  s    	C#	$	$BggiiGZZ\\F FM9R+=+=>>>>> FKNyS/A/ABBBBB"g&&&&&r.   c            	         t          g dg dg dd          } |                                 }t          d          D ]}|                                 D ]\  }}t	          j        t          ||          t          | |                    sJ d|j        d<   t	          j        t          ||          t          | |                    rJ t          j	        | |           d S )Nr   r   r  r   r   r   )
r   r   ranger   r"   r#   r   r(   r   r   )r*   r   ir$  r   s        r,   
test_itemsr  P  s    	CC	D	DBggiiG 1XX / / 	/ 	/ID##Ic4$8$8)B:M:MNNNNN CHQK'	#t(<(<iD>Q>QRRRRR!"g....	// /r.   rh  rS   Int64c                    t          ddgddd|           }|d d          }|                                }d|||k    <   t          j        t	          |d          t	          |d                    rJ t          j        ||           d S )Nr   r   r   rg  r   r   )r   r   r"   r#   r   r   r   )rh  r*   r   r   s       r,   test_putmaskr  a  s    	!Qaa00	>	>	>Baaa5DggiiGBrRxL	$ 4 4iC6H6HIIIII$(((((r.   c                     t          ddgddd|           }t          |d          }d|||k    <   t          j        |t          |d                    sJ d S )Nr   r   r   rg  r   r   r   r   r"   r#   rh  r*   rr  s      r,   test_putmask_no_referencer  l  sg    	!Qaa00	>	>	>Bb#EBrRxLE9R#5#56666666r.   float64Float64c                     t          ddgdd|           }t          |d          }t          dddgi          |||k    <   t          j        |t          |d                    sJ d S )Nr   r   r   rg  r   rR  r   r  r  s      r,   $test_putmask_aligns_rhs_no_referencer  t  su    	#q,,E	:	:	:Bb#EcC8_--BrRxLE9R#5#56666666r.   zval, exp, raises)rR  TT)r   FFraisesc                    t          ddgddd          }|d d          }|                                }t          g dg dgt          d                    }|r;t          j        t
          d	          5  | ||<   d d d            d S # 1 swxY w Y   d S | ||<   t          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 )Nr   r   r   r   )TFFabcr   zInvalid value)matchr   r   r   )r   r   r   pytestr  	TypeErrorr"   r#   r   r$   r   r   r   )rU  expr  r*   r   r   indexers          r,   "test_putmask_dont_copy_some_blocksr  |  s   	!Qac22	3	3Baaa5DggiiG			3334d5kk  G  -]9O<<< 	 	BwK	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 7#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,,,,,s   5BBBc                 6    |                      | dk    d          S Nr   r   wherer   s    r,   r:   r:         CIIcAgr** r.   c                 6    |                      | dk    d          S r  maskr  s    r,   r:   r:         CHHSAXr** r.   c                    t          g d|           }|                                } ||          }t          j        t	          |          t	          |                    sJ |j        |j        usJ d|j        d<   t          j        t	          |          t	          |                    rJ t          j        ||           d S )Nr   rg  r   r   )	r   r   r"   r#   r   r   r(   r   r   rh  r  r   r   r   s        r,   test_where_mask_noopr    s     %
(
(
(CxxzzHT#YYFIcNNIf,=,=>>>>><sy((((FKN	#	&0A0ABBBBB3)))))r.   c                 6    |                      | dk     d          S r  r  r  s    r,   r:   r:     r  r.   c                 6    |                      | dk    d          S r  r  r  s    r,   r:   r:     r  r.   c                    t          g d|           }|                                } ||          }t          j        t	          |          t	          |                    rJ |j        |j        usJ t          j        ||           d S )Nr   rg  )r   r   r"   r#   r   r   r   r   r  s        r,   test_where_maskr    s     %
(
(
(CxxzzHT#YYF	#	&0A0ABBBBB<sy((((3)))))r.   z
dtype, val)rS   g      %@)r  r   c                 6    |                      | dk     |          S Nr   r  r*   rU  s     r,   r:   r:     s    a-- r.   c                 6    |                      | dk    |          S r  r  r  s     r,   r:   r:     s    a-- r.   c                    t          g dg dd|           }|                                } |||          }t          j        t	          |d          t	          |d                    sJ t          j        t	          |d          t	          |d                    rJ d|j        d<   t          j        t	          |d          t	          |d                    rJ t          j        ||           d S )	Nr   )ir   rg  r   r   r   r?  r  )rh  rU  r  r*   r   r   s         r,   %test_where_mask_noop_on_single_columnr    s     
66e	D	D	DBggiiGT"c]]FIb#..	&#0F0FGGGGG	"c 2 2Ifc4J4JKKKKKFK	"c 2 2Ifc4J4JKKKKK"g&&&&&r.   r  r  c                    t          g ddd          }|                                }t          j                    5   t	          |d         |           |d         dk    dd           d d d            n# 1 swxY w Y   t          j        ||           t          j                    5   t	          |dg         |           |d         dk    dd           d d d            n# 1 swxY w Y   t          j        ||           d S )	N)r   r   r   r   r   r   r   r   Tr  )r   r   r   raises_chained_assignment_errorr  r   )r  r*   r   s      r,   test_chained_where_maskr    sx   	++	,	,BggiiG		+	-	- = =3r#w{At<<<<= = = = = = = = = = = = = = ="g&&&		+	-	- ? ? C5	4  C1a>>>>? ? ? ? ? ? ? ? ? ? ? ? ? ? ?"g&&&&&s#   -A55A9<A9(.C""C&)C&c                     t          dg dit          ddd                    } |                                 }|                     d          }t	          j        t          |d          t          | d                    sJ d	|j        d
<   t	          j        t          |d          t          | d                    rJ t          j	        | |           d S )Nr   )g        N       @      @z1/1/2000r   minr  r}   r3  r   r   )
r   r
   r   asfreqr"   r#   r   r(   r   r   r   s      r,   test_asfreq_noopr    s    		###$QU;;;
 
 
B ggiiG
)))

CIc3//2s1C1CDDDDD CHTN	#s 3 3Yr35G5GHHHHH"g&&&&&r.   c                      t          dddg d          } |                                 }|                                 D ]\  }}d|j        d<   t	          j        | |           d S )Nr   r   r   r   r}   r   )r   r   iterrowsr(   r   r   )r*   r   r  subs       r,   test_iterrowsr    sn    	##999	5	5	5BggiiG++--  3"g&&&&&r.   c                      t          ddt          j        dgi          } | d d          }|                                 }|                     d           d| j        d<   t          j        ||           d S )Nr   r   r   Tr  g      Y@r   )r   r"   r  r   ffillr(   r   r   )r*   r   r   s      r,   test_interpolate_creates_copyr    so    	C#rvq)*	+	+Baaa5DwwyyHHHTHBGDM$)))))r.   c                  P   t          g dg dg dd          } |                                 }|                     d          }|                    dt          j        g d                     t          j        t          | d	          t          |d	                    sJ t          j        t          | d
          t          |d
                    sJ d|j        d<   t          j	        | |           t          j        t          | d	          t          |d	                    sJ d S )Nr   r   r  r   Fr0   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
''u'

CLLBH\\\**+++Ib#..	#s0C0CDDDDDIb#..	#s0C0CDDDDDCGFO"g&&&Ib#..	#s0C0CDDDDDDDr.   zsingle-blockzmixed-blockc                    t          g dt          j        g d|           d          }t          g d          }|                                }|                    d|           t          j        t          |d          t          |                    sJ |j        	                    d          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   rg  r   r  r   r   r  )r   r"   r  r   r   r  r#   r   r$   r   r  r   r   r   )rh  r*   r   r   r   s        r,   test_isetitem_seriesr    sj    
)))5)I)I)IJJ	K	KB


CxxzzHKK3Ib#..	#?????w((+++++ BF6N3))) 
)))5)I)I)IJJ	K	KB


CKK3CGAJyyyrx			/O/O/OPPQQH"h'''''r.   c                     t          g dddd          } t          g ddd          }|                     ddg|           t          j        t	          | d          t	          |d                    sJ t          j        t	          | d	          t	          |d	                    sJ | j                            d          rJ |                                 }d
|j        d<   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$   r   r   r(   r   r   )r*   rhsr   s      r,   test_isetitem_framer  '  s    	33	4	4B
)))!,,
-
-CKKAIb#..	#s0C0CDDDDDIb#..	#s0C0CDDDDDw((+++++wwyyHCHTNCHTN"h'''''r.   r   c                    t          g dg dd          }|                                }|                    |           }t          j        t          |d          t          |d                    sJ d|j        d<   t          j        t          |d          t          |d                    rJ t          j        ||           d S )Nr   r   r   r   r   )	r   r   getr"   r#   r   r(   r   r   )r   r*   r   r   s       r,   test_getr  4  s    	33	4	4BggiiGVVC[[FIfc22Ib#4F4FGGGGGFKN	&# 6 6	"c8J8JKKKKK"g&&&&&r.   z	axis, keyr   )r   r   c                    |dk    }t          g dg dt          j        g d|          d          }|                                }|                    ||           }| dk    s|r3t          j        t          |d	          t          |                    sJ n|j                            d
          sJ | d
k    r|j	        |j
        usJ n|j	        |j	        usJ d
|j        d
<   t          j        ||           d S )NrS   r   r   r  rg  r   r  r   r   r   )r   r"   r  r   xsr#   r   r$   r   r   r    r(   r   r   )rH   r   rh  single_blockr*   r   r   s          r,   test_xsr  A  s   
 G#L	iiiiibhyyy.N.N.NOO
 
B ggiiGUU3TU""FqyyLy	"c 2 2If4E4EFFFFFF{,,Q/////qyy|2:-----|28++++FKN"g&&&&&r.   rH   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 |j        |j        usJ |j        |j        usJ d|j        d<   t          j        ||           d S )N   r   r   r  l2r   lev1lev2r  r  r   r   )levelrH   r   r   )r"   arangereshaper   from_productr   r   	transposer   r  r#   r   r    r   r(   r   r   )r   r$  rH   arrr   r*   r   r   s           r,   test_xs_multiindexr*  [  sG    )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
 
 	
 	
 
 <rx''''>++++FK"g&&&&&r.   c                     t          g dg dd          } t          ddgidg          }|                                 }| d d          }|                     |           t          g dg dd          }t          j        | |           t          j        ||           t          j        t          | d	          t          |d	                    sJ t          j        t          | d          t          |d                    rJ d S )
N      ?r   r  )      @g      @      @r   r         Y@r   r}   )r.  r0  r/  r   )r   r   updater   r   r"   r#   r   )df1r   df1_origr   r   s        r,   test_update_framer4  r  s   
///@@
A
AC
S5'N1#
.
.
.CxxzzHqqq6DJJsOOO5F5F5FGGHHH#x((($)))Ic3//41E1EFFFFF	#s 3 3YtS5I5IJJJJJJJr.   c                  *   t          g d          } t          dgdg          }|                                 }| d d          }|                     |           t          g d          }t          j        | |           t          j        ||           d S )Nr,  r0  r   r}   )r-  r0  r  )r   r   r1  r   r   )ser1r   	ser1_origr   r   s        r,   test_update_seriesr8    s    ///""D5'!%%%D		I7DKK'''((H4***4+++++r.   c                     t          dg di          } t          dgdg          }|                                 }t          j                    5  | d                             |           d d d            n# 1 swxY w Y   t          j        | |           t          j                    5  | dg                             |                                           d d d            n# 1 swxY w Y   t          j        | |           d S )Nr   r   r0  r   r}   )r   r   r   r   r  r1  r   r'  )r*   r   r   s      r,   test_update_chained_assignmentr:    sg   	C#	$	$B5'!%%%DggiiG		+	-	-  
3t              "g&&&		+	-	- * *
C5	)))* * * * * * * * * * * * * * *"g&&&&&s$   A66A:=A:)/C$$C(+C(c                     t          g d          } |                                 }t          |           }| dz  } t          j        t          |           |          rJ t          j        |t          |                     d S Nr   r   )r   r   r   r"   r#   r   assert_numpy_array_equal)r   r   datas      r,   test_inplace_arithmetic_seriesr?    sv    


CxxzzHS>>D1HC 	#55555i&9&9:::::r.   c                      t          g d          } |                                 }| d d          }| dz  } t          j        t	          |           t	          |                    rJ t          j        ||           d S r<  )r   r   r"   r#   r   r   r   )r   r   r   s      r,   -test_inplace_arithmetic_series_with_referencerA    st    


CxxzzHqqq6D1HC	#	$@@@@@8T*****r.   c                  "   t          g ddd          } |                                 }|                                 }t          j        t          | d          t          |d                    sJ 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   r   s      r,   test_transposerD    s    	++	,	,BggiiG\\^^FIb#..	&!0D0DEEEEEFK"g&&&&&r.   c                     t          g ddd          } |                                 }| j        }t          j        t          | d          t          |d                    rJ d|j        d<   t          j        | |           d S )Nr   r   r   r   r   r   r   )	r   r   Tr"   r#   r   r(   r   r   rC  s      r,   test_transpose_different_dtypesrG    s    	--	.	.BggiiGTF	"c 2 2Ifa4H4HIIIIIFK"g&&&&&r.   c                      t          dg did          } | j        }t          j        t	          | d          t	          |d                    rJ d S )Nr   r   r  rg  r   )r   rF  r"   r#   r   r*   r   s     r,   test_transpose_ea_single_columnrJ    sW    	C#7	3	3	3BTF	"c 2 2Ifa4H4HIIIIIIIr.   c                      t          g ddd          } |                                 }d }|                     |           t          j        | |           d S )Nr   r   r   c                     d| j         d<   | S Nr   r   r   r  s    r,   r  z"test_transform_frame.<locals>.func      
r.   )r   r   	transformr   r   )r*   r   r  s      r,   test_transform_framerP    sd    	++	,	,BggiiG   LL"g&&&&&r.   c                      t          g d          } |                                 }d }|                     |           t          j        | |           d S )Nr   c                     d| j         d<   | S rM  r   r  s    r,   r  z#test_transform_series.<locals>.func  rN  r.   )r   r   rO  r   r   )r   r   r  s      r,   test_transform_seriesrS    s\    


CxxzzH   MM$3)))))r.   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   rB  s      r,   test_count_read_only_arrayrV    sf    	!Qa((	)	)BXXZZFFKNsAhsCj111H68,,,,,r.   c                     t          dg di          } t          g d          }|                                }|                     d|d           t	          j        t          |          t          | d                    sJ | j                            d          rJ d| j	        d<   t          j        ||           d S )Nr   r   r   r   )r  valuecolumnr   r?  )r   r   r   insertr"   r#   r   r$   r   r(   r   r   )r*   r   r   s      r,   test_insert_seriesr[    s    	C#	$	$B


CxxzzHII!3sI+++IcNNIb#,>,>?????w((+++++BGDM3)))))r.   c                  $   t          g ddd          } |                                 }|                     d          }t          j        t          | d          t          |d                    sJ d|j        d<   t          j        | |           d S )Nr   r   r   c = a+br   r   r   	r   r   evalr"   r#   r   r(   r   r   rC  s      r,   	test_evalr`    s    	++	,	,BggiiGWWYFIb#..	&#0F0FGGGGGFK"g&&&&&r.   c                  <   t          g ddd          } |                                 }| d d          }|                     dd           t          j        t          | d          t          |d                    sJ d| j        d	<   t          j        ||           d S )
Nr   r   r   r]  Tr  r   r   r   r^  )r*   r   df_views      r,   test_eval_inplacerc    s    	++	,	,BggiiGeGGGItG$$$Ib#..	'30G0GHHHHHBGDM'7+++++r.   c                     t          ddgddgd          } |                                 }d }|                     |d           t          j        | |           t          ddgdd	gd          } |                                 }t          j        d           5  |                     |d           d d d            n# 1 swxY w Y   t          j        | |           d S )
Nr   r   r   r   )r   r   c                     d| d<   | S )Nr   r   r   )rows    r,   rO  z(test_apply_modify_row.<locals>.transform  s    C
r.   r  r   r   )r   r   applyr   r   assert_produces_warning)r*   r   rO  s      r,   test_apply_modify_rowri    s+    
!Qq!f--	.	.BggiiG   HHYQH"g&&& 
!QsCj11	2	2BggiiG		#D	)	) $ $
###$ $ $ $ $ $ $ $ $ $ $ $ $ $ $ "g&&&&&s   B<<C C c                     t          g ddd          } |                                 }|j        | j        usJ |                     g d                                          }|j        | j        usJ |                     d          }|j        | j        usJ |                     g d          }|j        | j        usJ d S )Nr   r   r   )r   r   r         ?)g      ?rk  g      ?)r   r   r   r    r   quantilerI  s     r,   test_reducerm  )  s    	--	.	.BVVXXF<rz))))ZZ			""&&((F>++++[[F<rz))))[[***++F>++++++r.   c                      t          g ddd          } |                                 }|j        | j        usJ |j        | j        usJ t	          g d          }|                                }|j        |j        usJ d S )Nr   r   r   )r   diffr   r    r   )r*   r   r   s      r,   	test_diffrp  8  s    	--	.	.BWWYYF<rx''''>++++


CXXZZF<sy((((((r.   )znumpyr"   r  pandas.compatr   pandaspdr   r   r   r   r   r	   r
   r   pandas._testing_testingr   pandas.tests.copy_view.utilr   pandas.util.versionr   r-   r2   markfilterwarningsparametrizer   r   r   
RangeIndexr   r   r   r   r   r   r   r   r   r   r   r   r   r   r  r  r
  r  r  r!  r%  r(  r0  r8  r=  r@  rC  rJ  rQ  r  rV  rX  re  rk  ru  rw  r}  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  boolr  r  r  r  r  r  r  r  r  r  r  r  r  r*  r4  r8  r:  r?  rA  rD  rG  rJ  rP  rS  rV  r[  r`  rc  ri  rm  rp  r   r.   r,   <module>r~     s        % % % % % %    	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	       1 1 1 1 1 1 ' ' ' ' ' '  .I I I0 899!4!4!455@@BB7733NN@@BB==554433::@@??OO!$	 	 	)  & &NE EO& & 65 :9RE  899!4!4!455??55AA::66PPBB8888776655<<BBAAQQ!$	 	 	)  & &N= =O& & 65 :9R=,' ' '( =2=A#6#6q!f"FGG* * HG*, , ,
, 
, 
,' ' ' ) ) ) E E E$E E E"' ' '  	 	0//  	 	' '	 	' ' ' '' ' ' w&~F ' ' '( ( ($
) 
) 
)( ( ( ) ) )$2 2 2  
));; 	' 	' 	'0 0 0 ' ' '* * *$% % %4 	8	8!<	2	2A6	J	JAN	G	GK	 	CBB  	 	6 6	 	6 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))* * *)* 	 ' ' '$' ' '(7 7 7B7 7 74 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 
$ 
$	 
$ ZZZ00' ' 1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7 7 657 9i"8997 7 :97 +.?AR-STT- - - - UT-* 7G"455
**** 
* 
*  65
* 7G"455
**** * *  65* 'GHH
---- 
' 
'  IH
' &'!233	' 	' 43	'' ' ' ' ' '* * *
E 
E 
E gy!'F   ( ( (.
( 
( 
( se--	' 	' .-	' vx&899gy!'F   ' '  :9', !Q((	6':;;' ' <; )('*K K K, , ,
' 
' 
'; ; ;+ + +' ' '' ' 'J J J	' 	' 	'	* 	* 	*- - -	* 	* 	*' ' '	, 	, 	,' ' '0, , ,	) 	) 	) 	) 	)r.   