
    [6g                       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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$ej                            d2d< 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             Z.dT Z/dU Z0dV Z1dW Z2ej                            dXd2dYg          dZ             Z3d[ Z4d\ Z5ej                            d]d^ d fd_ d3fd` d fda d fgg db          dc             Z6ej                            dd ed4d3gdFe           edFd4d3gi          g          df             Z7ej                            dd ed4d3gdFe           edFd4d3gi          g          dg             Z8dh Z9di Z:dj Z;dk Z<ej                            dldmd4ej=        fg          dn             Z>ej                            dodpdqg          dr             Z?ej                            d	ds dt du dv g          dw             Z@dx ZAdy ZBdz ZCej                            d{dFd|d4d}d d4d d}g          d~             ZDej                            d	ddg          d             ZEej                            dd ed4d3g           edFd4d3gi          g          d             ZFej                            dd ed4d3g           edFd4d3gi          g          d             ZGd ZHd ZIej                            d eg ddFe          i f edFg di          ddFifg          d             ZJej                            d eg ddFe          i f edFg di          ddFifg          d             ZKej                            dg d          d             ZLd ZMd ZNej                            dd eg d           edFg di          g          d             ZOd ZPd ZQd ZRej                            d{ddid2dig          d             ZSej                            dddg          d             ZTd ZUd ZVd ZWej                            dddg          d             ZXej                            dddg          d             ZYej                            dddg          d             ZZej                            ddqde[fdg          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 Zad Zbd Zcd Zdej                            dddgddg          d             Zed Zfej                            ddFdFgg          d             Zgej                            dddg          ej                            dddgddg          dƄ                         Zhej                            dd d4g          ej                            dddg          d˄                         Zid̄ Zjd̈́ Zkd΄ Zldτ ZmdЄ Znej                            dddg          d҄             Zodӄ ZpdԄ ZqdՄ Zrdք Zsdׄ Ztd؄ Zudل Zvdڄ Zwdۄ Zxd܄ ZydS )    N)SettingWithCopyWarning)		DataFrameIndex
MultiIndexPeriodSeries	Timestamp
date_rangeoption_contextperiod_range)	get_arrayc                 t   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 | 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indexcolumnsis_npshares_memoryr   _mgrblocksrefshas_referenceiloc)using_copy_on_writedfdf_copys      b/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/pandas/tests/copy_view/test_methods.py	test_copyr,      sF   	II	J	JBggiiG =((((?"*,,,,=RX&&&&&?rz***** 	'3 7 72s9K9KLLLLL ?<&q).<<>>>>><&q).<<>>>>> GL74=A    c                    t          g dg dg dd          }|                    d          }| rc|j        |j        usJ |j        |j        usJ |j                            |j                  sJ |j                            |j                  sJ n |j        |j        u sJ |j        |j        u sJ 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 t          j        |          5  d|j        d
<   d d d            n# 1 swxY w Y   |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   r   r   r    r!   r"   r   r#   r$   r%   r&   r'   tmassert_cow_warning)r(   warn_copy_on_writer)   r*   s       r+   test_copy_shallowr4   +   s   	II	J	JBgg5g!!G  -}BH,,,,bj0000}  *****""2:......}(((("*,,,, 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 "#566 	# 	#!"GL	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	#wt}!!!!	'3 7 72s9K9KLLLLLLLs   G22G69G6r   )TNFmethodc                 D    |                      t          j        |          S )Nr   r   )renamestrlowerr)   r   s     r+   <lambda>r<   T   s    394@@ r-   c                 4    |                      ddg|          S )Nr   r   r7   )reindexr;   s     r+   r<   r<   U   s    S#JTBB r-   c                 0    |                      | |          S Nr   reindex_liker;   s     r+   r<   r<   V   s    $77 r-   c                 <    |                      | |          d         S NrA   r   alignr;   s     r+   r<   r<   W   s    "4003 r-   c                 6    |                      g dd|          S Nr   r   )axisr   set_axisr;   s     r+   r<   r<   X   s    ___7NN r-   c                 0    |                      d|          S Ntestr   r   rename_axisr;   s     r+   r<   r<   Y   s    f4@@ r-   c                 0    |                      d|          S )NrO   r7   rQ   r;   s     r+   r<   r<   Z   s    TBB r-   c                 4    |                      ddi|          S )Nr   int64rA   astyper;   s     r+   r<   r<   [   s    C>== r-   c                 2    |                      dd|          S Nr   rA   swapaxesr;   s     r+   r<   r<   ]       Q55 r-   c                 2    |                      dd|          S Nr   r   rA   truncater;   s     r+   r<   r<   ^   r\   r-   c                 .    |                      |          S r@   infer_objectsr;   s     r+   r<   r<   _   s    ))t)44 r-   c                 .    |                      |          S r@   to_timestampr;   s     r+   r<   r<   `   s    d33 r-   c                 0    |                      d|          S ND)freqr   	to_periodr;   s     r+   r<   r<   a   s    3T:: r-   c                 0    |                      d|          S N
US/CentralrA   tz_localizer;   s     r+   r<   r<   b   s    4@@ r-   c                 0    |                      d|          S rn   
tz_convertr;   s     r+   r<   r<   c   s    |$?? r-   c                 0    |                      d|          S NF)allows_duplicate_labelsr   	set_flagsr;   s     r+   r<   r<   d   s    e$OO r-   )r8   r>   rC   rG   rL   rename_axis0rename_axis1rW   r[   r`   rc   rf   rl   rq   rt   ry   )idsc                 r   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|          }d| j         j        j        v rCd}t          j        t          |          5   |||          }d d d            n# 1 swxY w Y   n |||          }|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 )Nrf   
2012-01-01ri   r   rj   periodsrl   rq   rt   Europe/Brusselsrj   r   tzr   r   r   r   r   r[   "'DataFrame.swapaxes' is deprecatedmatchrA   Fzreindex-r   )nodecallspecidr   r
   r   r1   assert_produces_warningFutureWarning
startswithr!   r"   r   )
requestr5   r   r(   using_array_managerr   r)   msgdf2share_memorys
             r+   test_methods_copy_keywordr   P   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W\*---2'SAAA 	( 	(&$'''C	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( fRd###&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Ls   &D  DDc                 4    |                      ddi|          S )Nr   d   rP   r8   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 )NrP   )r>   r   r   s     r+   r<   r<      s    #++CID+AA r-   c                 0    |                      | |          S r@   rB   r   s     r+   r<   r<      s    #**3T*:: r-   c                 <    |                      | |          d         S rE   rF   r   s     r+   r<   r<      s    #))Cd)33A6 r-   c                 6    |                      g dd|          S rI   rK   r   s     r+   r<   r<      s    #,,W4,PP r-   c                 0    |                      d|          S rN   rQ   r   s     r+   r<   r<      s    #//T/BB r-   c                 0    |                      d|          S )NrU   rA   rV   r   s     r+   r<   r<      s    #**W4*88 r-   c                 2    |                      dd|          S )Nr   r   rA   )	swaplevelr   s     r+   r<   r<      s    #--14-88 r-   c                 2    |                      dd|          S rY   rZ   r   s     r+   r<   r<          #,,q!$,77 r-   c                 2    |                      dd|          S r^   r_   r   s     r+   r<   r<      r   r-   c                 .    |                      |          S r@   rb   r   s     r+   r<   r<      s    #+++66 r-   c                 .    |                      |          S r@   re   r   s     r+   r<   r<      s    #***55 r-   c                 0    |                      d|          S rh   rk   r   s     r+   r<   r<      s    #--St-<< r-   c                 0    |                      d|          S rn   rp   r   s     r+   r<   r<      s    #//,T/BB r-   c                 0    |                      d|          S rn   rs   r   s     r+   r<   r<      s    #..D.AA r-   c                 0    |                      d|          S rv   rx   r   s     r+   r<   r<      s    #--D-QQ r-   )zrename (dict)r8   r>   rC   rG   rL   rz   rW   r   r[   r`   rc   rf   rl   rq   rt   ry   c                 Z   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|          }d| j         j        j        v rCd}t          j	        t          |          5   |||          }d d d            n# 1 swxY w Y   n |||          }|p|du }|r3t          j        t          |          t          |                    sJ d S t          j        t          |          t          |                    rJ d S )Nrf   r~   ri   r   r   rl   rq   rt   r   r   r   r   r   r   r[   z'Series.swapaxes' is deprecatedr   rA   F)r   r   r   r   r
   r   from_arraysr   r1   r   r   r!   r"   r   )	r   r5   r   r(   r   r   r   ser2r   s	            r+    test_methods_series_copy_keywordr      s   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W\*---/'SAAA 	* 	*6#D)))D	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* vc%%%&7$%-L E	$3@@@@@@@#IdOOYs^^DDDDDDDs   D&&D*-D*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   rA   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'   r1   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-   r   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'   r1   assert_series_equalr(   r   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          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sumr1   r   )r)   keyr   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   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   r   r   r   r   r   r   r   r   r   r   r   )r   r   r8   r9   upperr!   r"   r   r'   r1   r   r(   r)   r   r   r   s        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   r8   r9   r   r   r!   r"   r   r'   r1   r   )r(   r)   r   df2_origr   s        r+   !test_rename_columns_modify_parentr   8  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>.testfuncP  s    	r-   r   r   r   r   r   )	r   r   piper!   r"   r   r'   r1   r   r(   r)   r   r   r   r   s         r+   	test_piper   L  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>.testfunch  s    	r-   r   r   r   r   r   )r   r   r   r!   r"   r   r1   r   r   s         r+   test_pipe_modify_dfr   d  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'   r1   r   r   s       r+   test_reindex_columnsr   {  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                     | S r   r   idxs    r+   r<   r<     s    C r-   c                 *    |                                  S r   )viewr   s    r+   r<   r<         CHHJJ r-   c                 *    |                                  S r   rA   r   s    r+   r<   r<     r   r-   c                      t          |           S r   )listr   s    r+   r<   r<     s    DII r-   )	identicalr   r   valuesc                    t          g dg dg dd          }|                                }|                     | |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          j	        ||           d S )
Nr   r   r   r   r   r   r   r   r   )
r   r   r>   r   r!   r"   r   r'   r1   r   )r   r(   r)   r   r   s        r+   test_reindex_rowsr     s>    
II	J	JBggiiG
**55??*
+
+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'   r1   r   r   s       r+   test_drop_on_columnr     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   rU   r   r   r   )r   r   select_dtypesr#   r   r!   r"   r   r'   r1   r   r   s       r+   test_select_dtypesr     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'   r1   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<   | rft	          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   )r   r   )
r   r
   r   shiftr!   r"   r   r'   r1   r   r   s       r+   test_shift_no_opr    sO   	
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 I#Ib#$6$6	#s8K8KLLLLL	"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
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   rJ   )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   rj   r   r   )r   r
   r   r   r  r!   r"   r   r'   r1   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                 z   t          ddgddgddggt          dd          	          }|                    dd
          }t          j        t          |d          t          |d                    sJ t          j        |          5  d|j        d<   d d d            n# 1 swxY w Y   | rt          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 d S )Nr   r   r   r   r   r   r  r  r   r  r   r   )r   r
   r  r!   r"   r   r1   r2   r'   nanr   )r(   r3   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		1	2	2                 -#c<(()B*E*E
 
 	
 	
 	
 fa[261+{3|\::
 
 
 	c8,,,,,- -s   BB"%B"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 t          j	        |          5  d|j        d<   d d d            n# 1 swxY w Y   | 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"   r   r   r'   r1   r2   r   )r(   r3   r)   r   view_originalr   r   s          r+   test_popr  2  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 RA#FM9]C3P3PQQQQQ		1	2	2                 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6s   4DDDfuncc                 ,    |                      |          S r   rF   xys     r+   r<   r<   K  s    QWWQZZ r-   c                 :    |                      |j        d          S )Nr   rJ   )rG   r   r  s     r+   r<   r<   L  s    QWWQSqW)) r-   c                 l    |                      |j        j        t          dd                   d          S )Nr   r   r"  )rG   r   r'   slicer  s     r+   r<   r<   M  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'   r1   r   )r(   r  r)   r   
df_changedr   _s          r+   test_align_framer)  H  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   rG   r!   r"   r   r'   r1   r   )r(   r   r   	ser_otherr   ser_other_results         r+   test_align_seriesr-  a  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   FrA   r   r   r   r   r   )	r   r   rG   r!   r"   r   locr1   r   )r(   r)   r   r   df3s        r+   test_align_copy_falser2  w  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   rJ   r   r   r/  )r   r   r   rG   r!   r"   r   r0  r1   r   r   )r(   r)   r   r   r   r   r   s          r+   !test_align_with_series_copy_falser6    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 t          j        |          5  d|j	        d<   d d d            n# 1 swxY w Y   | r@t          j        |j        t          |d                    rJ t          j
        ||           n3|                                }d|j	        d<   t          j
        ||           |d d                                          }t          j        |          5  d|j	        d<   d d d            n# 1 swxY w Y   | r)t          j        ||                                           d S |                                                                }d|j	        d<   t          j        ||           d S )Nr   r   r   )r   r   to_framer!   r"   r   r   r1   r2   r'   r   r   )r(   r3   r   r   r)   r   s         r+   test_to_framer9    s;   


CxxzzH	QQQ		B CJ	"a(8(899999		1	2	2                  .#CJ	"a0@0@AAAAA
sH---- ==??a
sH--- 
QQQ		B		1	2	2                  ,
b("3"3"5"566666==??++--d
b(+++++s$    BBBEE"%E"axr   c                 j   t          g dg dd          }|                                }d}t          j        t          |          5  |                    ||          }d d d            n# 1 swxY w Y   | 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   r1   r   r   r[   r!   r"   r   r'   r   )r(   r:  r)   r   r   r   s         r+   test_swapaxes_noopr=    ss   	33	4	4BggiiG
.C		#M	=	=	= " "kk"b!!" " " " " " " " " " " " " " "  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&&&&&s   A++A/2A/c                 r   t          g dg ddg d          }|                                }d}t          j        t          |          5  |                    dd	          }d d d            n# 1 swxY w Y   | 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   r   r   r   r   r  r   r   r   r<  )r(   r)   r   r   r   s        r+   test_swapaxes_single_blockr@    s|   	33???	K	K	KBggiiG
.C		#M	=	=	= . .kk'9--. . . . . . . . . . . . . . .  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&&&&&s   A//A36A3c                  8   t          ddgdd          } d}t          j        t          |          5  |                     dd	          } d d d            n# 1 swxY w Y   d
| j        d<   t          d
dgddgdddg          }t          j        | |           d S )Nr   r   r   r   r   r   r   r   )axis1axis2r   r   r   r   r   r   r   )r   r1   r   r   r[   r'   r   )r)   r   r   s      r+   test_swapaxes_read_only_arrayrE    s    	!Qa((	)	)B
.C		#M	=	=	= 9 9[[wi[889 9 9 9 9 9 9 9 9 9 9 9 9 9 9BGDMc1X1a&11#sDDDH"h'''''s   AAAzmethod, idxc                 V    |                      d                               d          S )NFr/   rA   r   s    r+   r<   r<     s%    BGGG'',,%,88 r-   c                 N    |                                                                   S r   )r   r   s    r+   r<   r<     s    BNN$$0022 r-   c                 ~    |                      t          j                                       t          j                  S )Nr   )r8   r9   r   r:   r   s    r+   r<   r<     s)    BIIciI0077	7JJ r-   c                 V    |                      d                              d          S )NFr/   number)include)r   r   r   s    r+   r<   r<     s%    BGGG''55h5GG r-   )shallow-copyr   r8   r   c                 "   t          g dg dg dd          }|                                }| o| j        j        j        dk    } ||          }t          j        |o|          5  d|j        d|f<   d d d            n# 1 swxY w Y   |st          j        ||            ||          }t          j        |o|          5  d|j        d<   d d d            n# 1 swxY w Y   |s(t          j        |j        d d |d f         |           d S d S )Nr   r   r   r   rL  r   r   )	r   r   r   r   r   r1   r2   r'   r   )	r   r5   r   r(   r3   r)   r   df2_is_viewr   s	            r+   test_chained_methodsrO    s    
II	J	JBggiiG *)Xgl.C.F..XK &**C		1Ak	B	B  C               +
b'*** &**C		1Ak	B	B                 :
chqqq#$$w/99999: :s$   'B  BBCC!Cobjr4  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-1ri   rj   z2012-1-2r   r   )r   r   r   r   rf   r!   r"   r   r'   r1   assert_equalr(   rP  obj_origobj2s       r+   test_to_timestamprW    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YrR  r   r   )r   r	   r   r   rl   r!   r"   r   r'   r1   rS  rT  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   rD  r   )	r   r   r   r!   r"   r   r'   r1   r   r   s       r+   test_set_indexr^  3  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'   r1   r   r)   r   r   s      r+   4test_set_index_mutating_parent_does_not_mutate_indexra  D  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   )rd  CoW_bre  )	r   r   
add_prefixr!   r"   r   r'   r1   r   r   s        r+   test_add_prefixrh  M  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   )rk  b_CoWrl  )	r   r   
add_suffixr!   r"   r   r'   r1   r   r   s        r+   test_add_suffixro  b  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   r   r   r"  r   r   r   )	r   r   dropnar!   r"   r   r'   r1   r   )r(   rJ   valr)   r   r   s         r+   test_dropnart  t  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-   rs  r   rp  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   rr  r!   r"   r   r'   r1   r   )r(   rs  r   r   r   s        r+   test_dropna_seriesrv    s    
!S!

CxxzzH::<<D =SZ888888#DK<<<<<DIaL =#DK<<<<<3)))))r-   c                 *    |                                  S r   headr   s    r+   r<   r<         27799 r-   c                 ,    |                      d          S )Nr   rx  r   s    r+   r<   r<         2771:: r-   c                 *    |                                  S r   tailr   s    r+   r<   r<     rz  r-   c                 ,    |                      d          S )Nr   r~  r   s    r+   r<   r<     r|  r-   c                    t          g dg dd          }|                                } | |          }|j                                         |rft	          j        t          |d          t          |d                    rJ t	          j        t          |d          t          |d                    rJ t          j        |          5  d|j	        d<   d d d            n# 1 swxY w Y   |rgt	          j        t          |d          t          |d                    rJ t	          j        t          |d          t          |d                    rJ nit	          j        t          |d          t          |d                    sJ t          j        |          5  d|j	        d<   d d d            n# 1 swxY w Y   t          j
        ||           d S )	Nr   r   r   r   r   r   r   r   )r   r   r#   r   r!   r"   r   r1   r2   r'   r   )r5   r(   r3   r)   r   r   s         r+   test_head_tailr    sG    
99	:	:BggiiG
&**CH    M #Ic3$7$72s9K9KLLLLL#Ic3$7$72s9K9KLLLLL 
	1	2	2                 #Ic3$7$72s9K9KLLLLL#Ic3$7$72s9K9KLLLLLL 	#s 3 3Yr35G5GHHHHH"#566 	 	CHTN	 	 	 	 	 	 	 	 	 	 	 	 	 	 	"g&&&&&s$   C""C&)C&F44F8;F8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   r   r   r   rD  )	r   r   rc   r!   r"   r   r'   r1   r   r   s       r+   test_infer_objectsr    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   rY  rZ  objectdtyper   )r   r   r   r   er   r   r   r   rD  
2018-12-31r   r   r   r   r	   rc   r   r'   r!   r"   )r(   r)   arr_aarr_barr_ds        r+   test_infer_objects_no_referencer    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   rY  rZ  r  r  r   r   r   r   r   r   rD  r  r  r  )r(   r)   r   r  r  r  s         r+   test_infer_objects_referencer    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afterrJ   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'   r1   r   r(   r  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'   r1   r   )r(   r5   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'   r1   rS  rT  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
   r   r   between_timer!   r"   r   r'   r1   rS  rT  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   rD  )	r   r   rC   r!   r"   r   r'   r1   r   )r(   r)   otherr   r   s        r+   test_reindex_liker  Q  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'   r1   r   r(   r   r   r   s       r+   test_sort_indexr  c  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'   r1   rS  )r(   rP  r  rU  rV  s        r+   test_sort_valuesr  t  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                 0   |                                 }|d d          } |j        dddi| t          j        t	          |d          t	          |d                    sJ t          j        |          5  d|j        d<   d d d            n# 1 swxY w Y   | 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  r!   r"   r   r1   r2   r'   rS  )r(   rP  r  r3   rU  r   s         r+   test_sort_values_inplacer    s]   
 xxzzHqqq6DCO++D+F+++Ic3//41E1EFFFFF 
	1	2	2                 K#Ic3$7$749M9MNNNNN
h'''''	#s 3 3YtS5I5IJJJJJJJs   5BBBdecimals)r   r   c                    t          ddgdd          }|                                }|                    |          }| rt          j        t          |d          t          |d                    sJ |dk    r4t          j        t          |d          t          |d                    sJ nt          j        t          |d          t          |d                    r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   r   r   rD  r   r   )	r   r   roundr!   r"   r   r'   r1   r   )r(   r3   r  r)   r   r   s         r+   
test_roundr    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#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   orderr   r   )r   from_tuplesr   r   reorder_levelsr!   r"   r   r'   r1   r   r(   r   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'   r1   r   r   s        r+   test_series_reorder_levelsr    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  r   r   r   r!   r"   r   r'   r1   rS  )r(   rP  r   rU  rV  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   r   r"  r   r   r   )	r   r   rL   r!   r"   r   r'   r1   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   r   r"  r   )r   r   rL   r!   r"   r'   r1   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 t          j        |          5  d|j        d<   d d d            n# 1 swxY w Y   | r.t          j        ||          rJ t          j        ||           d S t          j        ||          sJ t          g d          }t          j        ||           d S )Nr   F)rw   r   r   )	r   r   ry   r!   r"   r1   r2   r'   r   r(   r3   r   r   r   r   s         r+   test_set_flagsr    s=   


CxxzzH===77DC&&&&& 
	1	2	2  	!               .#D#.....
sH-----c*****)))$$
sH-----s   'A>>BBmapperrO   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   r4  r   r   r   r   )
r   r   r   rR   r!   r"   r   r'   r1   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)rt   zEurope/Berlin)rq   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)startrj   r   r   r   ro   r   )
r   r
   r   r  r!   r"   r   r'   r1   r   )r(   r  r   r   r   r   s         r+   test_tz_convert_localizer  4  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                 n   t          j        g dddg          }t          g dg dg 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<   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   	droplevelr!   r"   r   r'   r1   r   r  s        r+   test_droplevelr  J  s   "#;#;#;E5>RRRE	CC5	Q	Q	QBggiiG
,,q//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	#s 3 3Yr35G5GHHHHH I	#s 3 3Yr35G5GHHHHH"g&&&&&r-   c                 <   t          dg di          }|                                }|                                }t          j        |j        t          |d                    sJ t          j        |          5  d|j	        d<   d d d            n# 1 swxY w Y   | 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   r1   r2   r'   r   r0  )r(   r3   r)   r   seriess        r+   test_squeezer  b  sW   	C#	$	$BggiiGZZ\\F FM9R+=+=>>>>> 
	1	2	2  A               ##FM9R3E3EFFFFF
b'***** yS/A/ABBBBBvf~""""""s   :BBBc           	      J   t          g dg dg dd          }|                                }t          d          D ]}|                                D ]\  }}t	          j        t          ||          t          ||                    sJ t          j        |          5  d|j	        d<   d d d            n# 1 swxY w Y   | 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   r1   r2   r'   r   r0  )r(   r3   r)   r   ir5  r   s          r+   
test_itemsr  v  s   	CC	D	DBggiiG 1XX , , 	, 	,ID##Ic4$8$8)B:M:MNNNNN &'9::                                  # ,+Ic4,@,@)BPTBUBUVVVVV%b'2222 vag!+++++	,, ,s   B66B:=B:r  rU   Int64c                    t          ddgddd|          }|d d          }|                                }t          j        |          5  d|||k    <   d d d            n# 1 swxY w Y   | 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   r  r   r   r   )	r   r   r1   r2   r!   r"   r   r   r'   )r(   r  r3   r)   r   r   s         r+   test_putmaskr    sL   	!Qaa00	>	>	>Baaa5DggiiG		1	2	2  28                $#IdC$8$8)B:L:LMMMMM
dG,,,,, 	$ 4 4iC6H6HIIIIIy!######s   

A  A$'A$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   r  r   r   r   r   r!   r"   r(   r  r)   r  s       r+   test_putmask_no_referencer    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   r  r   rp  r   r  r  s       r+   $test_putmask_aligns_rhs_no_referencer    s    	#q,,E	:	:	:Bb#EcC8_--BrRxL ;yS'9'9:::::; ;::r-   zval, exp, warnT)r   FNc                 ^   t          ddgddd          }|d d          }|                                }t          g dg dgt          d                    }|r1t          j                    5  |||<   d d d            n# 1 swxY w Y   n3t          j        |d	          5  |||<   d d d            n# 1 swxY w Y   | 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   zincompatible dtyper   r   r   r   r   r   )r   r   r   r1   r2   r   r!   r"   r   r#   r   r   r'   )	r(   rs  expwarnr3   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  "$$ 	 	BwK	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 '4HIII 	 	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!####	 
 $#s$   -A??BB!B33B7:B7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|          }|                                } ||          }| 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   r  r   r   )r   r   r!   r"   r   r'   r1   r   r(   r  r  r   r   r   s         r+   test_where_mask_noopr    s     %
(
(
(CxxzzHT#YYF 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                 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 t          j        ||           d S )Nr   r  )r   r   r!   r"   r   r1   r   r  s         r+   test_where_maskr
    st     %
(
(
(CxxzzHT#YYF	#	&0A0ABBBBB3)))))r-   z
dtype, val)rU   g      %@)r  r   c                 6    |                      | dk     |          S Nr   r  r)   rs  s     r+   r<   r<     s    a-- r-   c                 6    |                      | dk    |          S r  r  r  s     r+   r<   r<     s    a-- r-   c                 V   t          g dg dd|          }|                                } |||          }| 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   r  r   r   r   rD  r&  )r(   r  rs  r  r)   r   r   s          r+   %test_where_mask_noop_on_single_columnr    s0    
66e	D	D	DBggiiGT"c]]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-   r  r  c                    t          g ddd          }|                                }| rt          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 t          j        t          d	
          5   t	          |d         |          |d         dk    dd           d d d            n# 1 swxY w Y   t          j        d           5  t          dd           5   t	          |dg         |          |d         dk    d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           5  t          dd           5   t	          ||d         dk             |          |d         dk    dd           d d d            n# 1 swxY w Y   d d d            d S # 1 swxY w Y   d S )N)r   r   r   r   r   r   r   r   Tr  inplace methodr   mode.chained_assignment)	r   r   r1   raises_chained_assignment_errorr  r   r   r   r   )r(   r  r)   r   s       r+   test_chained_where_maskr    s   	++	,	,BggiiG M/11 	A 	A"GBsGT""2c7Q;4@@@@	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A
b'***/11 	C 	C$GBuIt$$RWq[!TBBBB	C 	C 	C 	C 	C 	C 	C 	C 	C 	C 	C 	C 	C 	C 	C
b'*****'=MNNN 	A 	A"GBsGT""2c7Q;4@@@@	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A '-- 	G 	G 94@@ G G(C5	4((C1aFFFFG G G G G G G G G G G G G G G	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G '-- 	M 	M 94@@ M M.2c7Q;..r#w{AtLLLLM M M M M M M M M M M M M M M	M 	M 	M 	M 	M 	M 	M 	M 	M 	M 	M 	M 	M 	M 	M 	M 	M 	Ms   -A77A;>A;*.C$$C(+C(!-EE!E9G
.G8GG	GG	GG"G:I&7II&I	I&I	I&&I*-I*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   minr  r   rR  r   r   )
r   r
   r   asfreqr!   r"   r   r'   r1   r   r   s       r+   test_asfreq_noopr  1  s   		###$QU;;;
 
 
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'   r1   r   )r(   r)   r   r(  subs        r+   test_iterrowsr"  E  s    	##999	5	5	5BggiiG++--  3 +
b'*****+ +r-   c                    t          ddt          j        dgi          }|d d          }|                                }t	          j        |          5  |                    d           d d d            n# 1 swxY w Y   t	          j        |          5  d|j        d<   d d d            n# 1 swxY w Y   | rt	          j        ||           d S t          dg di          }t	          j        ||           d S )	Nr   r   r   Tr        Y@r   )r$  r   r   )	r   r!   r  r   r1   r2   ffillr'   r   )r(   r3   r)   r   r   s        r+   test_interpolate_creates_copyr&  O  sv   	C#rvq)*	+	+Baaa5DwwyyH		1	2	2  
              		1	2	2                  .
dH-----c???344
dH-----s$   A44A8;A8B**B.1B.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   r0  r1   r   r   s       r+   test_isetitemr,  a  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|           | rNt          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   r  r   r  r   r   r/  )r   r!   r+  r   r   r*  r"   r   r#   r   r0  r1   r   r   )r(   r  r)   r   r   r   s         r+   test_isetitem_seriesr.  w  sr    
)))5)I)I)IJJ	K	KB


CxxzzHKK3 0	"c 2 2IcNNCCCCC7,,Q///// 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|           | rt          j        t	          |d          t	          |d                    sJ t          j        t	          |d	          t	          |d	                    sJ |j                            d          rJ nft          j        t	          |d          t	          |d                    rJ t          j        t	          |d	          t	          |d	                    rJ |                                }d
|j        d<   d
|j        d<   t          j
        ||           d S )Nr   r   r   r   r   r   r   r   r   r   r   rD  )r   r*  r!   r"   r   r#   r   r   r'   r1   r   )r(   r)   rhsr   s       r+   test_isetitem_framer1    sf   	33	4	4B
)))!,,
-
-CKKA M	"c 2 2Ic34G4GHHHHH	"c 2 2Ic34G4GHHHHH7,,Q//////#Ib#$6$6	#s8K8KLLLLL#Ib#$6$6	#s8K8KLLLLLwwyyHCHTNCHTN"h'''''r-   r   c                 N   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 |rt          |t                    rt          nd }nt          |t                    rt          nd }t          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_assignmentr  r   )r   r   getr!   r"   r   r'   r1   r   
isinstancer9   r   r   r   r   r   )r(   r3   r   r)   r   r   r  s          r+   test_getr6    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'*****  	M$.sC$8$8B==dDD-7T-B-BL))D0&99 	# 	#+D11 # #!"A# # # # # # # # # # # # # # #	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# c4   	&!"g.....74=A%%%%%%s6   E-E8EE	EE	E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|sd
|
j	        d
<   n|r=t          j        |p|dk              5  d
|
j	        d
<   d d d            n# 1 swxY w Y   nct          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 )NrU   r   r   r   r  r  r   r"  r   r   r3  r  r   )r   r!   r+  r   xsr"   r   r#   r   r'   r1   r2   r   r   r   r   )r(   r3   r   rJ   r   r  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///// #w #/A #A	 #"<#<419== 	 	FKN	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 0&99 	# 	#+,BCC # #!"A# # # # # # # # # # # # # # #	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	#  "< "DAII
b'*****wt}!!!!!!sH   2D		DD%E-?E
E-E	E-E	E--E14E1rJ   z
key, level)l1r   r  c                 D   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 |r|dk    rt          nd }n| s
|st          }nd }t          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   )levelrJ   r   r3  r  r   )r!   arangereshaper   from_productr   r   r   r   r8  r"   r   r   r   r   r   r1   r   r'   r   )r(   r3   r   r   rB  rJ   arrr   r)   r   r   r  s               r+   test_xs_multiindexrG    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
 
 	
 	
 	
   %

}}  )< %	,f	5	5 " "'-- 	" 	" !FK	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	"" " " " " " " " " " " " " " " "g&&&&&s6   =F E)F )E-	-F 0E-	1F  FFc                    t          g dg dd          }t          ddgidg          }|                                }|d d          }t          j        |          5  |                    |           d d d            n# 1 swxY w Y   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   )rK  rM  rL  r   )	r   r   r1   r2   updater   r!   r"   r   )r(   r3   df1r   df1_origr   r   s          r+   test_update_framerQ    s   
///@@
A
AC
S5'N1#
.
.
.CxxzzHqqq6D 
	1	2	2  

3               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-----s   B  BBc                    t          g d          }t          dgdg          }|                                }|d d          }|rAt          j                    5  |                    |           d d d            n# 1 swxY w Y   n|                    |           t          g d          }t          j        ||           | rt          j        ||           d S t          j        ||           d S )NrI  rM  r   r   )rJ  rM  r  )r   r   r1   r2   rN  r   )r(   r3   ser1r   	ser1_origr   r   s          r+   test_update_seriesrU    s7   ///""D5'!%%%D		I7D "$$ 	 	KK	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	D'''((H4*** /
tY/////
tX.....s   A::A>A>c                    t          dg di          }t          dgdg          }|                                }| rt          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 t          j        t          d          5  |d                             |           d d d            n# 1 swxY w Y   t          j        d           5  t          dd           5  |dg                             |                                           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           5  t          dd           5  ||d         dk                                 |                                           d d d            n# 1 swxY w Y   d d d            d S # 1 swxY w Y   d S )	Nr   r   rM  r   r   r  r   r  )r   r   r   r1   r  rN  r   r8  r   r   r   )r(   r)   r   r   s       r+   test_update_chained_assignmentrW  /  s   	C#	$	$B5'!%%%DggiiG 8/11 	! 	!sGNN4   	! 	! 	! 	! 	! 	! 	! 	! 	! 	! 	! 	! 	! 	! 	!
b'***/11 	. 	.uIT]]__---	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	.
b'*****'=MNNN 	! 	!sGNN4   	! 	! 	! 	! 	! 	! 	! 	! 	! 	! 	! 	! 	! 	! 	! '-- 	2 	2 94@@ 2 2C5	  1112 2 2 2 2 2 2 2 2 2 2 2 2 2 2	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 '-- 	8 	8 94@@ 8 82c7Q;&&t}}7778 8 8 8 8 8 8 8 8 8 8 8 8 8 8	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8s   A88A<?A<+/C&&C*-C*#EEE*G;/F6*G6F:	:G=F:	>GGG,I=8I5II	II		II Ic                    t          g d          }|                                }t          |          }|dz  }| rHt          j        t          |          |          rJ t          j        |t          |                     d S t          j        t          |          |          sJ t          j        |t          |                     d S Nr   r   )r   r   r   r!   r"   r1   assert_numpy_array_equal)r(   r   r   datas       r+   test_inplace_arithmetic_seriesr\  H  s    


CxxzzHS>>D1HC 
: #IcNND99999
#D)H*=*=>>>>>	#55555
#D)C..99999r-   c                    t          g d          }|                                }|d d          }t          j        |          5  |dz  }d d d            n# 1 swxY w Y   | rHt	          j        t          |          t          |                    rJ t          j        ||           d S t	          j        t          |          t          |                    sJ d S rY  )r   r   r1   r2   r!   r"   r   r   )r(   r3   r   r   r   s        r+   -test_inplace_arithmetic_series_with_referencer^  Z  s    

CxxzzHqqq6D		1	2	2  q               A#IcNNIdOODDDDD
x.....	#	$@@@@@@@s   AAAFc                    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   rA   r   r   r   r   )	r   r   r   r!   r"   r   r'   r1   r   )r(   r   r   r)   r   r   s         r+   test_transposer`  i  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   Tr!   r"   r   r'   r1   r   r(   r)   r   r   s       r+   test_transpose_different_dtypesrd  y  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  r  r   )r   rb  r!   r"   r   )r(   r)   r   s      r+   test_transpose_ea_single_columnrf    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        |          5  |                    |           d d d            n# 1 swxY w Y   | rt          j        ||           d S 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   r1   r2   	transformr   )r(   r3   r)   r   r  s        r+   test_transform_framerl    s    	++	,	,BggiiG   
	1	2	2  
T               +
b'*****+ +s    A""A&)A&c                 
   t          g d          }|                                }d }t          j        |          5  |                    |           d d d            n# 1 swxY w Y   | rt          j        ||           d S d S )Nr   c                     d| j         d<   | S ri  r   r  s    r+   r  z#test_transform_series.<locals>.func  rj  r-   )r   r   r1   r2   rk  r   )r(   r3   r   r   r  s        r+   test_transform_seriesro    s    


CxxzzH   
	1	2	2  d               .
sH-----. .s   AA#&A#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   r1   r   r`  s      r+   test_count_read_only_arrayrr    sf    	!Qa((	)	)BXXZZFFKNsAhsCj111H68,,,,,r-   c                 f   t          g d          }|                                }t          j        t          d          5  |                                }d d d            n# 1 swxY w Y   t          j        t          |          t          |                    sJ | r|j	        
                    d          rJ t          j        |          5  d|j        d<   d d d            n# 1 swxY w Y   | rt          j        ||           d S t          g d          }t          j        ||           d S )Nr   zis deprecatedr   r   r   r   )r   r   r1   r   r   r   r!   r"   r   r#   r   r2   r'   r   r  s         r+   test_series_viewrt    s   


CxxzzH		#M	I	I	I  xxzz              IcNNIdOO<<<<< 29..q11111		1	2	2  	!                .
x-----+++&&
sH-----s$   A""A&)A&C''C+.C+c                    t          dg di          }t          g d          }|                                }|                    d|d           | rOt	          j        t          |          t          |d                    sJ |j                            d          rJ n2t	          j        t          |          t          |d                    rJ d|j	        d<   t          j        ||           d S )Nr   r   r   r   )r0  valuecolumnr   rD  )r   r   r   insertr!   r"   r   r#   r   r'   r1   r   )r(   r)   r   r   s       r+   test_insert_seriesry    s    	C#	$	$B


CxxzzHII!3sI+++ H	#	"c0B0BCCCCC7,,Q//////#IcNNIb#4F4FGGGGGBGDM3)))))r-   c                    t          g 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        ||           d S )Nr   r   r   c = a+br   r   r   )	r   r   evalr!   r"   r   r'   r1   r   rc  s       r+   	test_evalr}    s    	++	,	,BggiiGWWYF P	"c 2 2Ifc4J4JKKKKKK#Ib#$6$6	&#8N8NOOOOOFK"g&&&&&r-   c                    t          g ddd          }|                                }|d d          }|                    dd           t          j        t          |d          t          |d                    sJ t          j        |          5  d|j        d	<   d d d            n# 1 swxY w Y   | rt          j	        ||           d S d S )
Nr   r   r   r{  Tr  r   r   r   )
r   r   r|  r!   r"   r   r1   r2   r'   r   )r(   r3   r)   r   df_views        r+   test_eval_inplacer    s   	++	,	,BggiiGeGGGItG$$$Ib#..	'30G0GHHHHH		1	2	2                 0
gw/////0 0s   B((B,/B,c                 @   t          ddgddgd          }|                                }d }t          j        |          5  |                    |d           d d d            n# 1 swxY w Y   | rt          j        ||           n|j        d         d	k    s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+   rk  z(test_apply_modify_row.<locals>.transform  s    C
r-   r"  )r   r   r   r   r   )r   r   r1   r2   applyr   r0  r   )r(   r3   r)   r   rk  s        r+   test_apply_modify_rowr    s    
!Qq!f--	.	.BggiiG   
	1	2	2 $ $
###$ $ $ $ $ $ $ $ $ $ $ $ $ $ $  %
b'****vf~$$$$ 
!QsCj11	2	2BggiiG		#D	)	) $ $
###$ $ $ $ $ $ $ $ $ $ $ $ $ $ $ "g&&&&&s$   A&&A*-A*C>>DD)znumpyr!   pytestpandas.errorsr   pandaspdr   r   r   r   r   r	   r
   r   r   pandas._testing_testingr1   pandas.tests.copy_view.utilr   r,   r4   markparametrizer   r   r   r   
RangeIndexr   r   r   r   r   r   r   r   r   r   r  r  r  r  r  r  r)  r-  r2  r6  r9  r=  r@  rE  rO  rW  r\  r^  ra  rh  ro  r  rt  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.  r1  r6  r;  rG  rQ  rU  rW  r\  r^  r`  rd  rf  rl  ro  rr  rt  ry  r}  r  r  r   r-   r+   <module>r     s^        0 0 0 0 0 0    
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
       1 1 1 1 1 1  *"M "M "MJ !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F !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< !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.' ' '( 	 	0//  	 	' '	 	'(' ' '&' ' '( w&~F ' ' '$( ( (*I I I( ( (*- - -(7 7 7, 
));; ' ' '"0 0 0,+ + + . . .&!, !, !,H 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))* * *)*  	 ' ' '4' ' '*; ; ;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'2' ' '&* * *& 			!2!2IIsIII>N4O4O PQQ# # RQ#"' ' '"* * *". . .& Xv$6&8I#JKK' ' LK'  02GH * * *&' ' '0# # #(, , ,, 7G"455$ $ 65$  7G"455; ; 65; 9i"899; ; :9; T=13CD $ $ $> 7G"455
**** * *  65*" 7G"455
**** * *  65* 'GHH
---- ' '  IH'$ &'!233M M 43M0' ' '(+ + +. . .$M M M, gy!'F   ( ( (0( ( (" se--& & .-&8 vx&899gy!'F   " "  :9"B !Q((	6':;;' ' <; )('<. . .*/ / /*8 8 82: : :$A A A $//+ + 0/++ + +J J J+ + +. . .- - -. . .(* * *' ' '0 0 0' ' ' ' 'r-   