
    Eg                      2   d dl Zd dlZd dlZd dlmZmZmZ d dlm	Z
  ej        ddg          d             Zej                            d ed           ed          f ej        d d	                              d	           ej        d d	          fgd
dg          d             Zd Zej                            dddej        gej         eg d          fddej        gej         eg d          fddej        gddi eddej        g          fddej        gddd eddej        g          fddej        g eddg           eddej        g          fddej        g edgd	z             eddej        g          ff          d             Zej                            dddej        gej         eddej        g          fddej        gej         eddej        g          fddej        gddi eddej        g          fddej        gddd eddej        g          fddej        g eddg           eddej        g          fddej        g eg d           eddej        g          ff          d             Zd Zd ZdS )    N)CategoricalIndexSeriesignore)paramsc                     | j         S N)param)requests    ]/var/www/sysmax/venv/lib/python3.11/site-packages/pandas/tests/arrays/categorical/test_map.py	na_actionr      s
    =    zdata, categoriesabcbcacab   stringinterval)idsc                     t          | ||          }|                    t          |          }t          t          t          |           t          t          |          |          }t          j        ||           d S )N
categoriesorderedr   )r   mapstrtmassert_categorical_equal)datar   r   r   catresultexpecteds          r   test_map_strr"      su     dz7
C
C
CCWWSIW..FC3sJ#7#7  H 11111r   c                    t          t          d          t          d          d          }|                    d |           }t          t          d          t          d          d          }t          j        ||           t          t          d          t          d	          d
          }|                    d |           }t          t          d          t          d          d
          }t          j        ||           |                    d |           }t          t          j        dgdz  t          j                            }t          j	        ||           t          t          d          t          d	          d
          }d }|                    ||           }t          g dg dd
          }t          j        ||           t          g dg d          }|                    ||           }t          j        ||           |                    dddd|           }t          j        ||           d S )NABABCCBATr   c                 *    |                                  S r	   lowerxs    r   <lambda>ztest_map.<locals>.<lambda>&       qwwyy r   r   ababccbaBACFc                 *    |                                  S r	   r'   r)   s    r   r+   ztest_map.<locals>.<lambda>+   r,   r   bacc                     dS )N    r)   s    r   r+   ztest_map.<locals>.<lambda>0   s    q r   r3      )dtypec                 4    dddd                     |           S )N
         ABC)getr)   s    r   fztest_map.<locals>.f7   s     br**..q111r   )r8   r9   r8   r9   r:   )r9   r8   r:   )r8   r9   r:   r;   indexr8   r9   r:   )r   listr   r   r   r   nparrayint64assert_index_equalr   )r   r   r    expr@   mappers         r   test_maprJ   $   s   
d7mmUT
J
J
JCWW((IW>>F
d7mmUT
J
J
JC,,,
d7mmUU
K
K
KCWW((IW>>F
d7mmUU
K
K
KC,,, WW[[IW66F
!q111
2
2C&#&&& d7mmUU
K
K
KC2 2 2 WWQ)W,,F
***|||U
S
S
SC,,,LLL888FWWVyW11F,,,WW2BR00IWFFF,,,,,r   )r   r@   r!   r3   )FFT   F)r3   rK   c                     t          |           }|                    |d           }t          |t                     rt          j        ||           d S t          j        ||           d S )Nr   )r   r   
isinstancer   r   rG   r   r@   r!   valuesr    s        r   test_map_with_nan_nonerP   F   sj    ( FZZTZ**F(K(( 0
#FH55555
fh/////r   )FFFc                     t          |           }|                    |d          }| d         dk    rt          j        ||           d S t          j        ||           d S )Nr   r   r3   )r   r   r   r   rG   rN   s        r   test_map_with_nan_ignorerR   b   se    ( FZZXZ..FAw!||
#FH55555
fh/////r   c                    g d}g d}t          |          }t          |d d         |d d                   }|                    ||           }t          |g d          }t          j        ||           t          t          |d d         |d d                             }|                    ||           }t          j        ||           d S )N)ar=   r3   rT   )onerK         @rU   rA   r   )rV   rK   rU   )r   )r   r   r   r   r   dictzip)r   orig_values
new_valuesr   rI   r    r!   s          r   test_map_with_dict_or_seriesr\   ~   s    $$$K'''J
k
"
"CJssO;ss+;<<<FWWVyW11F :///BBBH111#k#2#&
3B38899FWWVyW11F11111r   c                      t          g d          } d}t          j        t          |          5  |                     d            d d d            d S # 1 swxY w Y   d S )N)rT   bczThe default value of 'ignore' for the `na_action` parameter in pandas.Categorical.map is deprecated and will be changed to 'None' in a future version. Please set na_action to the desired value to avoid seeing this warning)matchc                     | S r	   r4   r)   s    r   r+   z:test_map_na_action_no_default_deprecated.<locals>.<lambda>   s    ! r   )r   r   assert_produces_warningFutureWarningr   )r   msgs     r   (test_map_na_action_no_default_deprecatedre      s    
ooo
&
&C	5  
	#M	=	=	=                   s   AAA)numpyrD   pytestpandaspdr   r   r   pandas._testing_testingr   fixturer   markparametrizerC   interval_rangerepeatr"   rJ   nanisnarP   rR   r\   re   r4   r   r   <module>rs      s                    
       h'(((  )( 	he%		1a	 	 	'	'	*	*,=B,=a,C,CD 
:   2 2 2- - -D 
Q%%(<(<(<"="=>
Q%%(<(<(<"="=>
Q!U[[%1G%H%HI
QUu--uueUBF5K/L/LM26NFE5>""Krv.//	
 26NFE7Q;E5%())	
 &0 0' &0 
Q++ueRV.D"E"EF
Q%%rv(>"?"?@
Q!U[[%1G%H%HI
QUu--uueUBF5K/L/LM26NFE5>""Krv.//	
 26NF((())E5%())	
 &0 0' &02 2 2$
 
 
 
 
r   