
    i                        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,                  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y)!    N)	DataFrameIndexSeriesarray)	get_arrayc                 n    t        | dd      }|d d  }|j                  dd      }|j                  }||fS )N      ?abr   Tdrop)r   	set_indexindex)
index_datadfviewidxs       d/app/cer_product_mecsu/.venv/lib/python3.12/site-packages/pandas/tests/copy_view/index/test_index.py
index_viewr      s?    	#.	/Ba5D	c	%B
((C9    c                      t        ddgdd      } | j                  dd      } | j                  j                  d      }d	| j                  d
<   t        j                  | j                  |       y )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%      s\    	!Qa(	)B	c	&Bxx}}$}'HBGGDM"((H-r   c                      t        ddgdd      } | d d  }| j                  dd      } | j                  j                  d      }d	|j                  d
<   t        j                  | j                  |       y )Nr   r   r	   r
   r   Tr   r   r   r   r   )r   r   r$   s      r   !test_set_index_drop_update_columnr'      se    	!Qc*	+Ba5D	c	%Bxx}}$}'HDIIdO"((H-r   c                      t        ddgdd      } t        ddg      }| j                  |      } | j                  j	                  d      }d	|j
                  d
<   t        j                  | j                  |       y Nr   r   r	   r
   
      Tr   r   r   )r   r   r   r   r    r!   r"   r#   r   serr$   s      r   test_set_index_seriesr.   (   sf    	!Qc*	+B
"b
C	c	Bxx}}$}'HCHHQK"((H-r   c                      t        ddgdd      } t        ddg      }|| _        | j                  j                  d      }d	|j                  d
<   t        j                  | j                  |       y r)   )r   r   r   r    r!   r"   r#   r,   s      r   test_assign_index_as_seriesr0   1   s_    	!Qc*	+B
"b
CBHxx}}$}'HCHHQK"((H-r   c                      t        ddgdd      } t        ddg      }t        |      }|| _        d }| j                  j	                  d      }d	|j
                  d
<   t        j                  | j                  |       y r)   )r   r   r   r   r    r!   r"   r#   )r   r-   	rhs_indexr$   s       r   test_assign_index_as_indexr3   :   sm    	!Qc*	+B
"b
Cc
IBHIxx}}$}'HCHHQK"((H-r   c                      t        ddg      } t        |       }|j                  d      }d| j                  d<   t	        j
                  ||       y Nr   r   Tr   r   r   r   r   r    r!   r"   r#   r-   r   r$   s      r   test_index_from_seriesr8   E   sE    
!Q.C
*CxxTx"HCHHQK#x(r   c                      t        ddg      } t        | d      }t        |       }d| j                  d<   t	        j
                  t        |       |      sJ y )Nr   r   Tr    r   r   )r   r   r   r!   npshares_memory)r-   r   arrs      r   test_index_from_series_copyr>   M   sK    
!Q.C
$
C
C.CCHHQKIcNC000r   c                      t        ddg      } t        |       }t        |      }|j                  d      }d| j                  d<   t	        j
                  ||       y r5   r6   r7   s      r   test_index_from_indexr@   U   sN    
!Q.C
*C
*CxxTx"HCHHQK#x(r   funcc                 8    | j                  | j                        S N)_shallow_copy_valuesxs    r   <lambda>rH   a   s    !//!)), r   c                 "    | j                         S rC   )r   rF   s    r   rH   rH   b   s    !&&( r   c                 (    | j                  ddg      S Nr   r   )takerF   s    r   rH   rH   c   s    !&&!Q. r   c                 (    | j                  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!QK. r   c                     | ddg   S rK    rF   s    r   rH   rH   f   s    !QF) r   c                 8    | j                  t        dd            S rP   )_getitem_slicerQ   rF   s    r   rH   rH   g   s    !""5A;/ r   c                 $    | j                  g       S rC   )deleterF   s    r   rH   rH   h   s    !((2, r   c                 $    | j                  d      S )Nr   )renamerF   s    r   rH   rH   i   s    !((3- r   c                 (    | j                  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      \  }}|j                  d      }d|j                  j                  j                  v r|j                  d      } | |      }d|j                  d<   t        j                  ||d	
       y )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   ^   su    8 QF#JCxxTx"H7<<((+++??7+
s)CEJJt#xU;r   c                      t        ddg      \  } }| j                  d      }| j                  d      } d|j                  d<   t	        j
                  | |d	       y )
Nr   r   Tr   Fr:   aaaar   ra   )r   r    infer_objectsr!   r"   r#   )r   rg   r$   s      r   test_infer_objectsrl      sW    S#J'JCxxTx"H




'CEJJt#xU;r   c                  B   t        g dd      } | j                  d      }| j                         }t        j                  t        |d      | j                        sJ |j                  j                  d      rJ d|j                  d<   t        j                  | |       y )	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#.<<<ww((+++BGGDM#x(r   c                  j    t        g d      } | j                  }|j                  j                  du sJ y )Nrn   F)r   valuesflags	writeable)r   results     r   test_index_valuesrz      s-    
	
CZZF<<!!U***r   c                      t        j                  ddg      } t        |       }t        j                  | t	        |            rJ y rK   )r;   r   r   r<   r   r=   r   s     r   +test_constructor_copy_input_ndarray_defaultr}      s:    
((Aq6
C
*CYs^4444r   c                  |    t        ddgd      } t        |       }t        j                  | |j                         rJ y )Nr   r   r[   dtype)r   r   r"   r<   r|   s     r   &test_constructor_copy_input_ea_defaultr      s8    
Ag
&C
*CSYY////r   c                  H   t        j                  ddgt        j                  t         j                              } d| j                  _        t        t        |             }t        j                  | t        |            rJ |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                  ||       y )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
3CCII
s
CYs^44488%%a(((1a&0ABCq!fBHHRWW$5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) 
, " //	  6<76<<)+50	*r   