
    Owg                        d dl Zd dlZd dlmc mZ d dlZd dlm	Z	 d dl
mZ d dlmZ ej                  d        Zej"                  j$                  ej                  d               Zej                  ej"                  j)                  dej*                  ej,                  g      ej"                  j)                  dd g ej.                  g d      fd	g ej.                  g d      fd
g ej.                  g d      fd d	g ej.                  g dg dg      j0                  fd dg ej.                  g dg dg      j0                  fg d ej.                  g dg dg dg      j0                  fd	dg ej.                  g dg dg      j0                  fd	dg ej.                  g dg dg      j0                  fd	dg ej.                  g dg dg      j0                  fg	      d                      Zd Zy)    N)	DataFrame)	get_arrayc                    t        g dg dd      }g d|d<   |d d  }t        d |j                  j                  D              sJ |j	                          |j                  j                  d   j
                  j                         sJ t        j                  t        |d      t        |d            sJ |j                  j                  d	   j
                  j                         rJ |j                  j                  d   j
                  j                         rJ |j                  j                  d	   j
                  j                         sJ |j                  j                  d
   j
                  j                         rJ | rWd|j                  d<   |j                  j                  d	   j
                  j                         rJ |j                  d   dk(  sJ y y )N         皙?g?g333333?ab         cc              3   P   K   | ]  }|j                   j                            y w)N)refshas_reference).0blks     \/var/www/horilla/myenv/lib/python3.12/site-packages/pandas/tests/copy_view/test_internals.py	<genexpr>z#test_consolidate.<locals>.<genexpr>   s     FCsxx%%'Fs   $&r   r   r   r   g        )r   r   )r   r   r   )r   all_mgrblocks_consolidate_inplacer   r   npshares_memoryr   ilocloc)using_copy_on_writedfsubsets      r   test_consolidater&      s    
9	:BBsG UF F6;;3E3EFFFF ! ;;a %%33555 Ib#.	&#0FGGG {{!!!$))77999 ww~~a %%3355577>>!!!//111ww~~a %%33555 D77>>!$))77999vvf~$$$     c                  4   t        j                  dd      5  t        g dg dd      } | d d  }d|j                  d<   | j                  d   dk(  sJ dt         j                  j
                  _        t        g dg dd      } | d d  }d|j                  d<   | j                  d   d	k(  sJ dt         j                  j
                  _        t        g dg dd      } | d d  }d|j                  d<   | j                  d   dk(  sJ 	 d d d        y # 1 sw Y   y xY w)
Nzmode.copy_on_writeFr   r
   r   r   )r   r   Tr   )pdoption_contextr   r!   optionsmodecopy_on_write)r$   r%   s     r   test_switch_optionsr.   0   s     
		/	7 "Y_=>ADwwt}!!!(,

%Y_=>ADwwt}!!!(-

%Y_=>ADwwt}!!!'" " "s   C-DDdtypez	locs, arr)r   r   )ir   )r   r   r   r	   c           	      *   t        g dg dg dg dg dg dd      }|j                  |      }|j                         }|j                  d       }|j                  j	                  ||d	
       t        j                  ||       | rOt        |j                        D ]6  \  }}||vst        j                  t        ||      t        ||            r6J  y |j                  D ].  }t        j                  t        ||      t        ||            s.J  y )Nr   r   )      	   )
         )         )r   r   r   )r   r   r   defdeepT)inplace)r   astypecopyr   isettmassert_frame_equal	enumeratecolumnsr   r    r   )	r#   locsarrr/   r$   df_origdf2icols	            r   test_iset_splits_blocks_inplacerR   N   s    ( 
 	
	
B **U
CggiG
''t'
CHHMM$TM*"g&

+ 	QFAs}''	"c(:Ic3<OPPP	Q :: 	QC''	"c(:Ic3<OPPP	Qr'   c                     t        dg di      } t        d      D ]  }| j                  d        t        | j                  j
                  d   j                  j                        dk(  sJ t        dg di      } t        d      D cg c]  }| j                  d       }}t        d	      D ]  }| j                  d        t        | j                  j
                  d   j                  j                        d
k(  sJ | j                  j
                  d   j                  j                  dk(  sJ t        d      D ]  }| j                  d        | j                  j
                  d   j                  j                  dk(  sJ |d d }t        d      D ]  }| j                  d        | j                  j
                  d   j                  j                  dk(  sJ y c c}w )Nr   r   i  FrB   r   i  i     i  i  i  i,  )	r   rangerF   lenr   r   r   referenced_blocksclear_counter)r$   rP   dfss      r   test_exponential_backoffrZ   |   s   	C#	$B3Z 
U rww~~a %%778C???	C#	$B(-c

312777
3C
32Y 
Urww~~a %%778C???77>>!!!//47773Z 
U 77>>!!!//4777
ds)C3Z 
U 77>>!!!//3666% 4s   	G!)numpyr   pytestpandas.util._test_decoratorsutil_test_decoratorstdpandasr)   r   pandas._testing_testingrH   pandas.tests.copy_view.utilr   skip_array_manager_invalid_testr&   mark
single_cpur.   parametrizeintpint8arrayTrR   rZ    r'   r   <module>rn      s     ) )    1 ## % $ %F ##" $ "8 ##277BGG"45
hbhh|$%
hbhh|$%
hbhh|$%
Q<6799:
Q<6799:	HBHHlL,GHJJK
Q<6799:
Q<6799:
Q<6799:
Q 6 $ Q<7r'   