
    .i                        d dl Zd dlZd dlmZmZ d dlmZ d dl	m
Z
 d Z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      j"                  fd dg ej                   g dg d
g      j"                  fg d ej                   g dg d
g d
g      j"                  fddg ej                   g dg d
g      j"                  fddg ej                   g dg d
g      j"                  fg      d               Zd Zy)    N)	DataFrameSeries)	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 d|j                  d<   | j                  j                  d	   j
                  j                         rJ | j                  d   dk(  sJ y )N         )皙?g?g333333?)ab         cc              3   P   K   | ]  }|j                   j                            y w)N)refshas_reference).0blks     c/var/www/app/trading-bot/venv/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)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 FKKww~~a %%3355566&>S       dtypez	locs, arr)r   r   )ir	   )r   r   r	   r
   c                    t        g dg dg dg dg dt        g dt              d      }|j                  |      }|j	                         }|j	                  d	
      }|j
                  j                  | |d       t        j                  ||       t        |j                        D ]6  \  }}|| vst        j                  t        ||      t        ||            r6J  y )Nr   r   )      	   )
         )         )r   r   r   )r&   )r   r   r   defFdeepT)inplace)r   r   objectastypecopyr   isettmassert_frame_equal	enumeratecolumnsr   r   r   )locsarrr&   r"   df_origdf2icols           r   test_iset_splits_blocks_inplacerJ   .   s    $ 
v6	
	
B **U
CggiG
''u'
CHHMM$TM*"g&BJJ' M3D=##Ib#$6	#s8KLLLM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  Fr9   r   i  i     i  i  i  i,  )	r   ranger>   lenr   r   r   referenced_blocksclear_counter)r"   rH   dfss      r   test_exponential_backoffrR   U   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pandasr   r   pandas._testing_testingr@   pandas.tests.copy_view.utilr   r$   markparametrizeintpint8arrayTrJ   rR    r%   r   <module>r`      s]      1!D 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:	M 6M27r%   