
    .iG                     v    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y)    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             y )	N                  abint64)dtypecr   name)r   nparrayshares_memoryr   tmassert_series_equalr   )dfarrs     a/var/www/app/trading-bot/venv/lib/python3.12/site-packages/pandas/tests/copy_view/test_setitem.pytest_set_column_with_arrayr"      sm    	3	4B
((9G
,CBsG 	"c 2C888CF2c7F93$?@    c                  4   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             y )Nr
   r   r   r   r   r   )r   r   r   r   r   ilocr   r   r   sers     r!   test_set_column_with_seriesr(      s     
3	4B

CBsGIb#.	#??? CHHQK88A;!2c7F93$?@r#   c                  "   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 y )Nr
   r   r   r   r   r   d)r   r   r   r   r   valuesr   )r   idxr    s      r!   test_set_column_with_indexr-   /   s    	3	4B
	
CBsG 	"c 2CJJ???
Q
C
**CBsG	"c 2C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             y )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	4B
),7
8CBSzNIb#.	#s0CDDDCHHTN2c7F93$?@r#   c                     t        dg di      } t        g d      }|j                         }|| d<   t        j                  t        |      t        | d            sJ d| j                  d<   t        j                  ||       y )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   so    	C#	$B

CxxzH BsGIcNIb#,>???BGGDM3)r#   c                     t        g dg dd      } t        g d      }|j                         }|| d<   t        j                  t        |      t        | d            sJ d| j                  d<   t        j                  ||       y )Nr
   )g?g?g333333?r   r   r   r9   r5   r;   r=   s      r!   (test_setitem_series_no_copy_single_blockrB   ]   sm    	9	:B

CxxzHBsGIcNIb#,>???BGGDM3)r#   c                     t        g ddd      } t        g d      }|j                         }|| d<   t        j                  t        |      t        | d            sJ d| j                  d<   t        j                  ||       y )Nr
   r   r   r   r   r9   r:   r;   r=   s      r!   'test_setitem_series_no_copy_split_blockrD   j   sm    	+	,B

CxxzHBsGIcNIb#,>???BGGDM3)r#   c                     t        g dg dgt        j                  g dg dg            } t        ddg      }|| d<   t	        j
                  t        |      | j                  d            rJ | j                  j                  d      sJ y )	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     
	I&&'CD
B "b
CBsG	#0D0DQ0GHHH77$$Q'''r#   c                      t        g dg dd      } t        j                  d       5  | dxx   dz  cc<   d d d        t        g dg dd      } | d   }|dz  }y # 1 sw Y   &xY w)Nr
   r   r   r   r   )r   r   assert_produces_warningr&   s     r!   %test_set_column_with_inplace_operatorrN      sf    	3	4B 
	#	#D	) 
31 
3	4B
S'C1HC s   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      sJ       1AA 9$A*
*
*
(
r#   