
    Mh                     8    S SK r S SKJr  S SKJr   " S S5      rg)    N)	DataFramec                   2    \ rS rSrS rS rS rS rS rSr	g)	
TestAssign   c                    [        / SQ/ SQS.5      nUR                  5       nUR                  UR                  UR                  -  S9nUR                  5       n/ SQUS'   [
        R                  " X45        UR                  S S9n[
        R                  " X45        [
        R                  " X5        UR                  / SQS9n[
        R                  " X45        [
        R                  " X5        UR                  UR                  UR                  -  S9nUR                  S	S
S9R                  SS	0S9n[
        R                  " X45        UR                  UR                  UR                  -   S9nUR                  5       n/ SQUS'   [
        R                  " X45        UR                  S S9n[
        R                  " X45        g )N                  AB)C)r   g      @r
   r   c                 4    U R                   U R                  -  $ N)r   r   xs    X/var/www/html/env/lib/python3.13/site-packages/pandas/tests/frame/methods/test_assign.py<lambda>(TestAssign.test_assign.<locals>.<lambda>       qssQSSy    r   r   r	   )axiscolumnsr   )r   r   	   r   c                 4    U R                   U R                  -   $ r   r   r   s    r   r   r   (   r   r   )	r   copyassignr   r   tmassert_frame_equaldroprename)selfdforiginalresultexpecteds        r   test_assignTestAssign.test_assign   sh   YY78779RTTBDD[)779#
f/ 01
f/ 	b+ [)
f/
b+RTTBDD[)==1=-44c3Z4H
f/ RTTBDD[)779!
f/ 01
f/r   c                     [        SS/SS/SS//SS/S	9nUR                  / S
QUR                  S S9n[        / SQ/ SQ/ SQ/[        S5      S	9n[        R
                  " X#5        g )Nr	   r   r
   r   r   r   r   r   r   )r      r"   c                     U R                   $ r   r   r   s    r   r   1TestAssign.test_assign_multiple.<locals>.<lambda>-   s    ACCr   )r   DE)r	   r   r   r	   r   )r
   r   r2   r
   r   )r   r   r"   r   r   ABCDEr   r%   r   listr&   r'   r*   r+   r-   r.   s       r   test_assign_multipleTestAssign.test_assign_multiple+   sg    AAA/#sDY"$$-@o?g
 	f/r   c                    [        SS/SS//SS/S9nUR                  UR                  UR                  -   UR                  UR                  -
  S9n[        / S	Q/ S
Q/[	        S5      S9n[
        R                  " X#5        UR                  UR                  UR                  -
  UR                  UR                  -   S9n[        / SQ/ SQ/[	        S5      S9n[
        R                  " X#5        g )Nr	   r
   r   r   r   r   r   )r5   r   )r	   r
   r   )r   r   r   r>   ABDCr   r5   )r	   r
   r>   r   )r   r   r>   r   ABCD)r   r%   r   r   r9   r&   r'   r:   s       r   test_assign_orderTestAssign.test_assign_order3   s    AA'#s<RTTBDD[BDD244K8m];T&\R
f/RTTBDD[BDD244K8m];T&\R
f/r   c                    [        / SQ/ SQS.5      nSn[        R                  " [        US9   UR	                  S 5        S S S 5        Sn[        R                  " [
        US9   UR	                  UR                  UR                  UR                  -   S9  S S S 5        g ! , (       d  f       Nd= f! , (       d  f       g = f)	Nr   r   r   z7assign\(\) takes 1 positional argument but 2 were given)matchc                     U R                   $ r   r!   r   s    r   r   ,TestAssign.test_assign_bad.<locals>.<lambda>F   s    r   z''DataFrame' object has no attribute 'C'r@   )r   pytestraises	TypeErrorr%   AttributeErrorr   r   )r*   r+   msgs      r   test_assign_badTestAssign.test_assign_bad@   s    YY78 I]]9C0IIm$ 17]]>5IIrttI, 65 10 65s   B$2B0
B-0
B>c                 >   [        SS/SS/S.5      nUR                  UR                  S S9n[        / SQ/ S	Q/[        S
5      S9n[        R
                  " X#5        UR                  S S S9n[        / SQ/ S	Q/[        S
5      S9n[        R
                  " X#5        g )Nr	   r
   r   r   r   c                     U S   U S   -   $ Nr   r    r   s    r   r   2TestAssign.test_assign_dependent.<locals>.<lambda>N   s    qv#r   r@   )r	   r   r	   r
   )r
   r   r
   r   rA   r   c                     U R                   $ r   r!   r+   s    r   r   rS   R   s    r   c                     U S   U S   -   $ rQ   rR   rU   s    r   r   rS   R   s    2c7RW;Lr   r8   r:   s       r   test_assign_dependent TestAssign.test_assign_dependentK   s    aV1a&12RTT%>?lL94<P
f/_0LMlL94<P
f/r   rR   N)
__name__
__module____qualname____firstlineno__r/   r;   rB   rM   rW   __static_attributes__rR   r   r   r   r      s    !0F00	-	0r   r   )rH   pandasr   pandas._testing_testingr&   r   rR   r   r   <module>ra      s      M0 M0r   