
    bMhW                        d dl Zd dlZd dlZd dlmZ d dlm	Z	 ej
                            dddg ej        ddgej                  f ej        ddgd           ej        ddgej                  f ej        ddgd           ej        ddgej                  f ej        ddg           ej        ddgej                  fg d	 ej        g d	ej                  f ej        g d	           ej        g d	ej                  f ej        g d	d
           ej        g d	ej                  fg  ej        g ej                  fg          d             Zej
                            dg d ej        g dd
          g          d             Zej
                            dddg ej        ddgd
           ej        ddgej                  g          d             Zej
                            dg d ej        d dej        gd          g          d             Zej
                            dddg ej        ddgd           ej        ddge           ej        ddge          g          d             Zd Zej
                            dde ed d          dg          d             ZdS )    N)check_array_indexerzindexer, expected      dtypeint64Int32)TFTbooleanc                 |    t          j        g d          }t          ||           }t          j        ||           d S Nr   r      )nparrayr   tmassert_numpy_array_equal)indexerexpectedarrresults       h/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pandas/tests/indexing/test_check_indexer.pytest_valid_inputr   	   s?    ( (999

C g..F11111    r   )TFNc                     t          j        g d          }t          ||           }t          j        g dt                    }t	          j        ||           d S )Nr   )TFFr   )r   r   r   boolr   r   )r   r   r   r   s       r   test_boolean_na_returns_indexerr   "   sZ    
 (999

C g..Fx,,,D999H11111r   TFc                     t          j        g d          }d}t          j        t          |          5  t          ||            d d d            d S # 1 swxY w Y   d S )Nr   zBoolean index has wrong lengthmatchr   r   pytestraises
IndexErrorr   r   r   msgs      r   test_bool_raise_lengthr&   /   s     (999

C
*C	z	-	-	- * *C)))* * * * * * * * * * * * * * * * * *   AAA)r   r   NInt64c                     t          j        g d          }d}t          j        t          |          5  t          ||            d d d            d S # 1 swxY w Y   d S )Nr   z9Cannot index with an integer indexer containing NA valuesr   )r   r   r!   r"   
ValueErrorr   r$   s      r   test_int_raise_missing_valuesr+   ?   s     (999

C
EC	z	-	-	- * *C)))* * * * * * * * * * * * * * * * * *r'   g        g      ?g       @float64c                     t          j        g d          }d}t          j        t          |          5  t          ||            d d d            d S # 1 swxY w Y   d S )Nr   9arrays used as indices must be of integer or boolean typer   r    r$   s      r   test_raise_invalid_array_dtypesr/   J   s     (999

C
EC	z	-	-	- * *C)))* * * * * * * * * * * * * * * * * *r'   c                     t          j        ddg|           }t          j        g d          }d}t          j        t
          |          5  t          ||           d d d            d S # 1 swxY w Y   d S )Nabr   r   r.   r   )pdr   r   r!   r"   r#   r   )nullable_string_dtyper   r   r%   s       r    test_raise_nullable_string_dtyper5   [   s    hSz)>???G
(999

C
EC	z	-	-	- * *C)))* * * * * * * * * * * * * * * * * *s   A**A.1A.r   )Nc                 b    t          j        g d          }t          ||           }|| k    sJ d S r   )r   r   r   )r   r   r   s      r   !test_pass_through_non_array_likesr7   d   s;    
(999

C g..FWr   )numpyr   r!   pandasr3   pandas._testing_testingr   pandas.api.indexersr   markparametrizer   intpIndexbool_r   r   r&   NAr+   objectr/   r5   Ellipsisslicer7    r   r   <module>rG      s                  3 3 3 3 3 3  Q1a&0001	1a&	(	(	(("(Aq6*I*I*IJ	1a&	(	(	(("(Aq6*I*I*IJ	1a&		828QF"':::;			hbh':':':"(KKKL	%%%	&	&1D1D1DBH(U(U(UVBH(((	:::BH(((999	

 
XRXb((() &2 2' &2 ###XRX.A.A.A%S%S%ST 2 2 2 	u$i000$bh/// * * * hbh1be}GDDDE * * * 	c
#s9---$f---$f---	 * * ** * * T8UU1a[['$JKK  LK  r   