
    Pi                        d dl Zd dlZd dl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ej                            dd d d d d d d d d d g
g d          d             Zd Zd Zd Zd Zd Zd  ZdS )!    N)	DataFrameIndexSeriesarray)	get_arrayc                 ~    t          | dd          }|d d          }|                    dd          }|j        }||fS )N      ?abr   Tdrop)r   	set_indexindex)
index_datadfviewidxs       {/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pandas/tests/copy_view/index/test_index.py
index_viewr      sJ    	#..	/	/Baaa5D	c	%	%B
(C9    c                      t          ddgdd          } |                     dd          } | j                            d          }d	| j        d
<   t          j        | j        |           d S )N      r
   r   Fr   Tdeepd   r   r   r   r   r   copyiloctmassert_index_equal)r   expecteds     r   test_set_index_update_columnr%      sk    	!Qa((	)	)B	c	&	&Bx}}$}''HBGDM"(H-----r   c                      t          ddgdd          } | d d          }|                     dd          } | j                            d          }d	|j        d
<   t          j        | j        |           d S )Nr   r   r	   r
   r   Tr   r   r   r   r   )r   r   r$   s      r   !test_set_index_drop_update_columnr'      sx    	!Qc**	+	+Baaa5D	c	%	%Bx}}$}''HDIdO"(H-----r   c                      t          ddgdd          } t          ddg          }|                     |          } | j                            d          }d	|j        d
<   t          j        | j        |           d S Nr   r   r	   r
   
      Tr   r   r   )r   r   r   r   r    r!   r"   r#   r   serr$   s      r   test_set_index_seriesr.   (   sx    	!Qc**	+	+B
"b

C	c		Bx}}$}''HCHQK"(H-----r   c                      t          ddgdd          } t          ddg          }|| _        | j                            d          }d	|j        d
<   t          j        | j        |           d S r)   )r   r   r   r    r!   r"   r#   r,   s      r   test_assign_index_as_seriesr0   1   sn    	!Qc**	+	+B
"b

CBHx}}$}''HCHQK"(H-----r   c                      t          ddgdd          } t          ddg          }t          |          }|| _        d }| j                            d          }d	|j        d
<   t          j        | j        |           d S r)   )r   r   r   r   r    r!   r"   r#   )r   r-   	rhs_indexr$   s       r   test_assign_index_as_indexr3   :   s~    	!Qc**	+	+B
"b

Cc

IBHIx}}$}''HCHQK"(H-----r   c                      t          ddg          } t          |           }|                    d          }d| j        d<   t	          j        ||           d S Nr   r   Tr   r   r   r   r   r    r!   r"   r#   r-   r   r$   s      r   test_index_from_seriesr8   E   sS    
!Q..C
**CxxTx""HCHQK#x(((((r   c                      t          ddg          } t          | d          }t          |           }d| j        d<   t	          j        t          |           |          sJ d S )Nr   r   Tr    r   r   )r   r   r   r!   npshares_memory)r-   r   arrs      r   test_index_from_series_copyr>   M   s^    
!Q..C
$


C
C..CCHQKIcNNC0000000r   c                      t          ddg          } t          |           }t          |          }|                    d          }d| j        d<   t	          j        ||           d S r5   r6   r7   s      r   test_index_from_indexr@   U   s^    
!Q..C
**C
**CxxTx""HCHQK#x(((((r   funcc                 6    |                      | j                  S N)_shallow_copy_valuesxs    r   <lambda>rH   a   s    !//!),, r   c                 *    |                                  S rC   )r   rF   s    r   rH   rH   b   s    !&&(( r   c                 0    |                      ddg          S Nr   r   )takerF   s    r   rH   rH   c   s    !&&!Q.. r   c                 0    |                      ddg          S )Nr   )repeatrF   s    r   rH   rH   d   s    !((Aq6"" r   c                 .    | t          dd                   S Nr   r   )slicerF   s    r   rH   rH   e   s    !E!QKK. r   c                     | ddg         S rK    rF   s    r   rH   rH   f   s    !QF) r   c                 H    |                      t          dd                    S rP   )_getitem_slicerQ   rF   s    r   rH   rH   g   s    !""5A;;// r   c                 ,    |                      g           S rC   )deleterF   s    r   rH   rH   h   s    !((2,, r   c                 ,    |                      d          S )Nr   )renamerF   s    r   rH   rH   i   s    !((3-- r   c                 0    |                      dd          S )NInt64Fr:   )astyperF   s    r   rH   rH   j   s    !((7(// r   )
rD   r   rL   rN   getitem_slicegetitem_listrU   rW   rY   r\   )idsc                    t          ddg          \  }}|                    d          }d|j        j        j        v r|                    d          } | |          }d|j        d<   t          j        ||d	
           d S )Nr   r   Tr   r\   r[   r   r   Fcheck_names)	r   r    nodecallspecidr\   r!   r"   r#   )rA   requestr   view_r$   s        r   test_index_opsrh   ^   s    8 QF##JCxxTx""H7<(+++??7++
$s))CEJt#xU;;;;;;r   c                      t          ddg          \  } }|                     d          }|                     d          } d|j        d<   t	          j        | |d	           d S )
Nr   r   Tr   Fr:   aaaar   ra   )r   r    infer_objectsr!   r"   r#   )r   rg   r$   s      r   test_infer_objectsrl      sj    S#J''JCxxTx""H




'
'CEJt#xU;;;;;;r   c                  J   t          g dd          } |                     d          }|                                 }t          j        t          |d          | j                  sJ |j                            d          rJ d|j	        d<   t          j        | |           d S )	Nr   r      r   )nameTr   r   r   r   )r   r    to_framer;   r<   r   rE   _mgr_has_no_referencer!   r"   r#   )r   r$   r   s      r   test_index_to_framert      s    
			
$
$
$CxxTx""H	BIb#..<<<<<w((+++++BGDM#x(((((r   c                  V    t          g d          } | j        }|j        j        du sJ d S )Nrn   F)r   valuesflags	writeable)r   results     r   test_index_valuesrz      s6    
			

CZF<!U******r   c                      t          j        ddg          } t          |           }t          j        | t	          |                    rJ d S rK   )r;   r   r   r<   r   r=   r   s     r   +test_constructor_copy_input_ndarray_defaultr}      sE    
(Aq6

C
**CYs^^4444444r   c                      t          ddgd          } t          |           }t          j        | |j                   rJ d S )Nr   r   r[   dtype)r   r   r"   r<   r|   s     r   &test_constructor_copy_input_ea_defaultr      sD    
Ag
&
&
&C
**CSY///////r   c                  8   t          j        ddgt          j        t           j                            } d| j        _        t          t          |                     }t          j        | t          |                    rJ |j
                            d          sJ t          j        ddgt          j        t           j                            |ddg<   t          ddgt          j        t           j                            }t          j        ||           d S )Nr   r   r   Fr   T)r;   r   r   int8rw   rx   r   r   r<   r   rr   rs   r"   assert_series_equal)r=   r-   r$   s      r   .test_series_from_temporary_index_readonly_datar      s    
(Aq6"'!2!2
3
3
3CCI
s

CYs^^444448%%a(((((1a&0A0ABBBCq!fBHRW$5$5666H3)))))r   )numpyr;   pytestpandasr   r   r   r   pandas._testing_testingr"   pandas.tests.copy_view.utilr   r   r%   r'   r.   r0   r3   r8   r>   r@   markparametrizerh   rl   rt   rz   r}   r   r   rS   r   r   <module>r      s                         1 1 1 1 1 1  . . .. . .. . .. . .. . .) ) )1 1 1) ) ) 
,,  ""  ////	 	 	   6< <7 6<< < <) ) )+ + +5 5 50 0 0	* 	* 	* 	* 	*r   