
    PiG                     x    d dl Zd dlmZmZmZmZmZ d dlm	Z
 d dlmZ d Zd Zd Zd Zd Zd	 Zd
 Zd Zd ZdS )    N)	DataFrameIndex
MultiIndex
RangeIndexSeries)	get_arrayc                     t          g dg dd          } t          j        g dd          }|| d<   t          j        t	          | d          |          rJ d|d<   t          j        | d         t          g dd                     d S )	N                  abint64)dtypecr   name)r   nparrayshares_memoryr   tmassert_series_equalr   )dfarrs     w/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pandas/tests/copy_view/test_setitem.pytest_set_column_with_arrayr"      s    	33	4	4B
(999G
,
,
,CBsG 	"c 2 2C88888CF2c7F9993$?$?$?@@@@@    c                  T   t          g dg dd          } t          g d          }|| d<   t          j        t	          | d          t	          |                    sJ d|j        d<   |j        d         dk    sJ t          j        | d         t          g dd                     d S )Nr
   r   r   r   r   r   )r   r   r   r   r   ilocr   r   r   sers     r!   test_set_column_with_seriesr(      s     
33	4	4B


CBsGIb#..	#????? CHQK8A;!2c7F9993$?$?$?@@@@@r#   c                  4   t          g dg dd          } t          g d          }|| d<   t          j        t	          | d          |j                  rJ t          dd          }|j        }|| d<   t          j        t	          | d          |          rJ d S )Nr
   r   r   r   r   r   d)r   r   r   r   r   valuesr   )r   idxr    s      r!   test_set_column_with_indexr-   /   s    	33	4	4B
			

CBsG 	"c 2 2CJ?????
Q

C
*CBsG	"c 2 2C8888888r#   c                  >   t          g dg dd          } t          g dg dd          }|| ddg<   t          j        t          | d          t          |d                    sJ d	|j        d
<   t          j        | d         t          g dd                     d S )Nr
   r   r   )      	   )
         )r   r*   r   r*   r   r   r   r   )r   r   r   r   r%   r   r   r   )r   df2s     r!   test_set_columns_with_dataframer7   A   s     
33	4	4B
))),,,77
8
8CBSzNIb#..	#s0C0CDDDDDCHTN2c7F9993$?$?$?@@@@@r#   c                  "   t          dg di          } t          g d          }|                                }|| d<   t          j        t          |          t          | d                    sJ d| j        d<   t          j        ||           d S )Nr   r
   r   r   d   r   r   	r   r   copyr   r   r   r%   r   r   r   rhsrhs_origs      r!   test_setitem_series_no_copyr@   O   s    	C#	$	$B


CxxzzH BsGIcNNIb#,>,>?????BGDM3)))))r#   c                  (   t          g dg dd          } t          g d          }|                                }|| d<   t          j        t          |          t          | d                    sJ d| j        d<   t          j        ||           d S )Nr
   )g?g?g333333?r   r   r   r9   r5   r;   r=   s      r!   (test_setitem_series_no_copy_single_blockrB   ]   s    	99	:	:B


CxxzzHBsGIcNNIb#,>,>?????BGDM3)))))r#   c                  $   t          g ddd          } t          g d          }|                                }|| d<   t          j        t          |          t          | d                    sJ d| j        d<   t          j        ||           d S )Nr
   r   r   r   r   r9   r:   r;   r=   s      r!   'test_setitem_series_no_copy_split_blockrD   j   s    	++	,	,B


CxxzzHBsGIcNNIb#,>,>?????BGDM3)))))r#   c                  6   t          g dg dgt          j        g dg dg                    } t          ddg          }|| d<   t	          j        t          |          |                     d                    rJ | j        	                    d          sJ d S )	Nr
   )r   r   r   )r   r   r   )columnsr2   r3   r   r   )
r   r   from_arraysr   r   r   r   _get_column_array_mgr_has_no_reference)r   r>   s     r!   ,test_setitem_series_column_midx_broadcastingrK   w   s     
	III&'CDD
 
 
B "b

CBsG	#0D0DQ0G0GHHHHH7$$Q'''''''r#   c                      t          g dg dd          } t          j        d           5  | dxx         dz  cc<   d d d            n# 1 swxY w Y   t          g dg dd          } | d         }|dz  }d S )Nr
   r   r   r   r   )r   r   assert_produces_warningr&   s     r!   %test_set_column_with_inplace_operatorrN      s    	33	4	4B 
	#D	)	)  
31               
33	4	4B
S'C1HCCCs   AAA)numpyr   pandasr   r   r   r   r   pandas._testing_testingr   pandas.tests.copy_view.utilr   r"   r(   r-   r7   r@   rB   rD   rK   rN    r#   r!   <module>rU      s                          1 1 1 1 1 1A A AA A A 9 9 9$A A A* * *
* 
* 
*
* 
* 
*
( 
( 
(
 
 
 
 
r#   