
    Pi                        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                            dej        ej        g          e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dS )    N)	DataFrameSeries)	get_arrayc                  j   t          g dg dd          } g d| d<   | d d          }t          d |j        j        D                       sJ |                                 |j        j        d         j                                        sJ t          j        t          | d          t          |d                    sJ |j        j        d	         j                                        rJ | j        j        d         j                                        rJ | j        j        d	         j                                        sJ | j        j        d
         j                                        rJ d|j
        d<   | j        j        d	         j                                        rJ | j        d         dk    sJ d S )N         )皙?g?g333333?)ab         cc              3   H   K   | ]}|j                                         V  d S )N)refshas_reference).0blks     y/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pandas/tests/copy_view/test_internals.py	<genexpr>z#test_consolidate.<locals>.<genexpr>   s0      FFCsx%%''FFFFFF    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9	:	:BiiBsG UF FF6;3EFFFFFFFF !!! ;a %3355555 Ib#..	&#0F0FGGGGG {!!$)7799999 w~a %33555557>!!//11111w~a %3355555 FKw~a %33555556&>S      r   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          }|                    |          }|                                }|                    d	
          }|j                            | |d           t          j        ||           t          |j
                  D ]<\  }}|| vr3t          j        t          ||          t          ||                    sJ =d S )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66	
 	
	
 	
B **U

CggiiG
''u'

CHMM$TM***"g&&&BJ'' M M3D==#Ib#$6$6	#s8K8KLLLLLM Mr   c                  f   t          dg di          t          d          D ]}                     d           t          j        j        d         j        j                  dk    sJ t          dg di          fdt          d	          D             }t          d
          D ]}                     d           t          j        j        d         j        j                  dk    sJ j        j        d         j        j        dk    sJ t          d          D ]}                     d           j        j        d         j        j        dk    sJ |d d         }t          d          D ]}                     d           j        j        d         j        j        dk    sJ d S )Nr   r   i  Fr9   r   i  c                 <    g | ]}                     d           S )Fr9   )r>   )r   rH   r#   s     r   
<listcomp>z,test_exponential_backoff.<locals>.<listcomp>^   s'    
3
3
312777
3
3
3r   i     i  i  i  i,  )	r   ranger>   lenr   r   r   referenced_blocksclear_counter)rH   dfsr#   s     @r   test_exponential_backoffrT   U   s   	C#	$	$B3ZZ  
Urw~a %788C????	C#	$	$B
3
3
3
3c


3
3
3C2YY  
Urw~a %788C????7>!!/477773ZZ  
U 7>!!/47777
dsd)C3ZZ  
U 7>!!/3666666r   )numpyr   pytestpandasr   r   pandas._testing_testingr@   pandas.tests.copy_view.utilr   r%   markparametrizeintpint8arrayTrJ   rT    r   r   <module>rb      s                     1 1 1 1 1 1! ! !D 27BG"455
hbh|||$$%
hbh|||$$%
hbh|||$$%
Q<<<6779:
Q<<<6779:	HBHlllLLL,,,GHHJK
Q<<<6779:
Q<<<6779:	 M M  65M27 7 7 7 7r   