
    Owg                     8    d dl Z d dlmZ d dlmZ  G d d      Zy)    N)	DataFramec                   *    e Zd Zd Zd Zd Zd Zd Zy)
TestAssignc                    t        g dg dd      }|j                         }|j                  |j                  |j                  z        }|j                         }g d|d<   t        j                  ||       |j                  d       }t        j                  ||       t        j                  ||       |j                  g d      }t        j                  ||       t        j                  ||       |j                  |j                  |j                  z        }|j                  d	d
      j                  dd	i      }t        j                  ||       |j                  |j                  |j                  z         }|j                         }g d|d<   t        j                  ||       |j                  d       }t        j                  ||       y )N                  AB)C)r   g      @r	   r   c                 4    | j                   | j                  z  S N)r   r   xs    ]/var/www/horilla/myenv/lib/python3.12/site-packages/pandas/tests/frame/methods/test_assign.py<lambda>z(TestAssign.test_assign.<locals>.<lambda>       qssQSSy     r   r   r   )axiscolumnsr   )r      	   r   c                 4    | j                   | j                  z   S r   r   r   s    r   r   z(TestAssign.test_assign.<locals>.<lambda>(   r   r   )	r   copyassignr   r   tmassert_frame_equaldroprename)selfdforiginalresultexpecteds        r   test_assignzTestAssign.test_assign   sx   YY78779RTTBDD[)779#
fh/ 01
fh/ 	b(+ [)
fh/
b(+RTTBDD[)==1=-44c3Z4H
fh/ RTTBDD[)779!
fh/ 01
fh/r   c                     t        ddgddgddggddg	      }|j                  g d
|j                  d       }t        g dg dg dgt        d      	      }t	        j
                  ||       y )Nr   r   r	   r   r
   r   r   r   r   )r       r!   c                     | j                   S r   r   r   s    r   r   z1TestAssign.test_assign_multiple.<locals>.<lambda>-   s
    ACC r   )r   DE)r   r   r    r   r   )r	   r   r0   r	   r   )r
   r   r!   r
   r   ABCDEr   r$   r   listr%   r&   r)   r*   r,   r-   s       r   test_assign_multiplezTestAssign.test_assign_multiple+   si    AAA/#sDY"$$-@o?g
 	fh/r   c                    t        ddgddggddg      }|j                  |j                  |j                  z   |j                  |j                  z
        }t        g d	g d
gt	        d            }t        j                  ||       |j                  |j                  |j                  z
  |j                  |j                  z         }t        g dg dgt	        d            }t        j                  ||       y )Nr   r	   r
   r   r   r   r   )r2   r   )r   r	   r
   )r
   r   r    r:   ABDCr   r2   )r   r	   r:   r
   )r
   r   r:   r    ABCD)r   r$   r   r   r6   r%   r&   r7   s       r   test_assign_orderzTestAssign.test_assign_order3   s    AA'#s<RTTBDD[BDD244K8m];T&\R
fh/RTTBDD[BDD244K8m];T&\R
fh/r   c                 |   t        g dg dd      }d}t        j                  t        |      5  |j	                  d        d d d        d}t        j                  t
        |      5  |j	                  |j                  |j                  |j                  z          d d d        y # 1 sw Y   dxY w# 1 sw Y   y xY w)	Nr   r   r   z7assign\(\) takes 1 positional argument but 2 were given)matchc                     | j                   S r   r   r   s    r   r   z,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_badzTestAssign.test_assign_bad@   s    YY78 I]]9C0 	%IIm$	%7]]>5 	-IIrttI,	- 	-	% 	%	- 	-s   B&(5B2&B/2B;c                 V   t        ddgddgd      }|j                  |j                  d       }t        g dg d	gt        d
            }t	        j
                  ||       |j                  d d       }t        g dg d	gt        d
            }t	        j
                  ||       y )Nr   r	   r
   r   r   c                     | d   | d   z   S Nr   r    r   s    r   r   z2TestAssign.test_assign_dependent.<locals>.<lambda>N   s    qv# r   r<   )r   r
   r   r	   )r	   r   r	   r   r=   r   c                     | j                   S r   r   r*   s    r   r   z2TestAssign.test_assign_dependent.<locals>.<lambda>R   s
     r   c                     | d   | d   z   S rJ   rK   rM   s    r   r   z2TestAssign.test_assign_dependent.<locals>.<lambda>R   s    2c7RW;L r   r5   r7   s       r   test_assign_dependentz TestAssign.test_assign_dependentK   s    aV1a&12RTT%>?lL94<P
fh/_0LMlL94<P
fh/r   N)__name__
__module____qualname__r.   r8   r>   rG   rO   rK   r   r   r   r      s    !0F00	-	0r   r   )rB   pandasr   pandas._testing_testingr%   r   rK   r   r   <module>rV      s      M0 M0r   