
    Mh                     p    S SK rS SKrS SKJr  S SKJr  \R                  " SS/S9S 5       r	 " S S5      r
g)	    N)CategoricalTF)paramsc                     U R                   $ )z3Boolean 'allow_fill' parameter for Categorical.take)param)requests    [/var/www/html/env/lib/python3.13/site-packages/pandas/tests/arrays/categorical/test_take.py
allow_fillr	      s     ==    c                   P    \ rS rSrS rS rS rS rS rS r	S r
S	 rS
 rS rSrg)TestTake   c                     [        SS/5      n[        R                  " S 5         UR                  SS/5      nS S S 5        WR	                  U5      (       d   eg ! , (       d  f       N'= f)Nabr   )r   tmassert_produces_warningtakeequals)selfcatresults      r   test_take_default_allow_fill%TestTake.test_take_default_allow_fill   sU    3*%''-XXq"g&F . }}S!!!! .-s   A
A'c                     [        SS/5      n[        R                  " S 5         UR                  SS/5        S S S 5        g ! , (       d  f       g = f)Nr   r   r   )r   r   r   r   )r   r   s     r   test_take_positive_no_warning&TestTake.test_take_positive_no_warning   s:    3*%''-HHaV .--s   A
Ac                     [        / SQ5      nU(       a  SnOSn[        R                  " [        US9   UR	                  SS/US9  S S S 5        g ! , (       d  f       g = f)Nr   r   r   indices are out-of-boundsz2index 4 is out of bounds for( axis 0 with)? size 3match      r	   r   pytestraises
IndexErrorr   r   r	   r   msgs       r   test_take_boundsTestTake.test_take_bounds   sG    /*-CFC]]:S1HHaV
H3 211s   A
Ac                     [        / SS/S9nU(       a  SnOSn[        R                  " [        US9   UR	                  S/US9  S S S 5        g ! , (       d  f       g = f)	Nr   r   
categoriesr    z-cannot do a non-empty take from an empty axesr!   r   r%   r&   r*   s       r   test_take_emptyTestTake.test_take_empty'   sK    "#s4-CAC]]:S1HHaSZH0 211s   A
Ac                     [        / SQSS/US9nUR                  / SQSS9n[        / SQUR                  US9n[        R                  " X45        g )	N)r   r   r   r   r   r   r0   ordered)r         Fr%   r   r   r   r   r   r0   r   assert_categorical_equalr   r5   r   r   expecteds        r   test_positional_takeTestTake.test_positional_take1   sL    .C:wW)6
 	##F5r
   c                     [        SS// SQUS9nUR                  SS/SS9n[        SS/UR                  US9n[        R                  " X45        g )	Nr   r   r   r   cr4   r6   r   Fr%   r9   r;   s        r   test_positional_take_unobserved(TestTake.test_positional_take_unobserved9   sN    3*'R1a&U3SzcnngV
##F5r
   c                     [        / SQ5      nUR                  / SQSS9n[        S[        R                  [        R                  /SS/S9n[        R
                  " X#5        g )Nr8   )r   r   r   Tr%   r   r   r/   )r   r   npnanr   r:   r   r   r   r<   s       r   test_take_allow_fillTestTake.test_take_allow_fill?   sK    /*+$7RVVRVV4#sL
##F5r
   c                     [        / SQ5      nUR                  / SQSSS9n[        / SQ/ SQS9n[        R                  " X#5        g )N)r   r   r6   )r   r   r6   Tr   )r	   
fill_value)r   r   r   r/   r   r   r   r:   rG   s       r    test_take_fill_with_negative_one)TestTake.test_take_fill_with_negative_oneF   s:    *%*"E{zB
##F5r
   c                     [        / SQ5      nUR                  / SQSSS9n[        / SQ/ SQS9n[        R                  " X#5        g )Nr@   r   r6   r   r   TrK   r	   r   r/   rL   rG   s       r   test_take_fill_valueTestTake.test_take_fill_valueM   s:    /**F?K
##F5r
   c                     [        / SQ5      nSn[        R                  " [        US9   UR	                  / SQSSS9  S S S 5        g ! , (       d  f       g = f)Nr@   z9Cannot setitem on a Categorical with a new category \(d\)r!   rP   dTrQ   )r   r'   r(   	TypeErrorr   )r   r   xprs      r   test_take_fill_value_new_raises(TestTake.test_take_fill_value_new_raisesT   s=    /*J]]9C0HHZCDHA 100s   A
A N)__name__
__module____qualname____firstlineno__r   r   r,   r1   r=   rB   rH   rM   rR   rX   __static_attributes__rZ   r
   r   r   r      s5    "
4166666Br
   r   )numpyrE   r'   pandasr   pandas._testing_testingr   fixturer	   r   rZ   r
   r   <module>re      s@        e}% &
KB KBr
   