
    bMh                     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     c/var/www/html/test/jupyter/venv/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          g dg dd          }t          g d          }||d<   | r3t          j        t	          |d          t	          |                    sJ n2t          j        t	          |d          t	          |                    r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   )using_copy_on_writer   sers      r!   test_set_column_with_seriesr(      s     
33	4	4B


CBsG H	"c 2 2IcNNCCCCCC #Ib#$6$6	#GGGGG 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&   r   idxr    s       r!   test_set_column_with_indexr-   3   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<   | r4t          j        t          |d          t          |d                    sJ n3t          j        t          |d          t          |d                    r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&   r   df2s      r!   test_set_columns_with_dataframer7   E   s     
33	4	4B
))),,,77
8
8CBSzN M	"c 2 2Ic34G4GHHHHHH #Ib#$6$6	#s8K8KLLLLL CHTN2c7F9993$?$?$?@@@@@r#   c                 &   t          dg di          }t          g d          }|                                }||d<   | r2t          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&   r   rhsrhs_origs       r!   test_setitem_series_no_copyr@   X   s    	C#	$	$B


CxxzzH BsG D	#	"c0B0BCCCCCBGDM3)))))r#   c                 ,   t          g dg dd          }t          g d          }|                                }||d<   | r2t          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   g   s    	99	:	:B


CxxzzHBsG D	#	"c0B0BCCCCCBGDM3)))))r#   c                 (   t          g ddd          }t          g d          }|                                }||d<   | r2t          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   u   s    	++	,	,B


CxxzzHBsG D	#	"c0B0BCCCCCBGDM3)))))r#   c                 >   t          g dg dgt          j        g dg dg                    }t          ddg          }||d<   t	          j        t          |          |                    d                    rJ | r|j        	                    d          sJ d S 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   r>   s      r!   ,test_setitem_series_column_midx_broadcastingrK      s     
	III&'CDD
 
 
B "b

CBsG	#0D0DQ0G0GHHHHH ,w((+++++, ,++r#   c                 J   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         }t          j        |          5  |dz  }d d d            d S # 1 swxY w Y   d S )Nr
   r   r   r   r   )r   r   assert_produces_warningassert_cow_warning)r&   warn_copy_on_writer   r'   s       r!   %test_set_column_with_inplace_operatorrP      sC   	33	4	4B 
	#D	)	)  
31               
33	4	4B
S'C		1	2	2  q                 s#   AAABBB)numpyr   pandasr   r   r   r   r   pandas._testing_testingr   pandas.tests.copy_view.utilr   r"   r(   r-   r7   r@   rB   rD   rK   rP    r#   r!   <module>rW      s                          1 1 1 1 1 1A A AA A A(9 9 9$A A A&* * ** * ** * *, , ,    r#   