
    bMh                     @    d dl Z d dlmZ d dlmZ  G d d          ZdS )    N)	DataFramec                   ,    e Zd Zd Zd Zd Zd Zd ZdS )
TestAssignc                    t          g dg dd          }|                                }|                    |j        |j        z            }|                                }g d|d<   t          j        ||           |                    d           }t          j        ||           t          j        ||           |                    g d          }t          j        ||           t          j        ||           |                    |j        |j        z            }|                    d	d
                              dd	i          }t          j        ||           |                    |j        |j        z             }|                                }g d|d<   t          j        ||           |                    d           }t          j        ||           d S )N                  AB)C)r   g      @r	   r   c                      | j         | j        z  S N)r   r   xs    f/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pandas/tests/frame/methods/test_assign.py<lambda>z(TestAssign.test_assign.<locals>.<lambda>       qsQSy     r   r   r   )axiscolumnsr   )r      	   r   c                      | 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   s   YYYYYY77887799RTBD[))7799#
fh/// 0011
fh/// 	b(+++ [[[))
fh///
b(+++RTBD[))==1=--44c3Z4HH
fh/// RTBD[))7799!		
fh/// 0011
fh/////r   c                     t          ddgddgddggddg	          }|                    g d
|j        d           }t          g dg dg dgt          d          	          }t	          j        ||           d S )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    AC 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+   s    AAA/#sDDDYYY"$--@@__ooo?g
 
 
 	fh/////r   c                    t          ddgddggddg          }|                    |j        |j        z   |j        |j        z
            }t          g d	g d
gt	          d                    }t          j        ||           |                    |j        |j        z
  |j        |j        z             }t          g dg dgt	          d                    }t          j        ||           d S )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<<<RTBD[BD24K88mmm]]];T&\\RRR
fh///RTBD[BD24K88mmm]]];T&\\RRR
fh/////r   c                    t          g dg dd          }d}t          j        t          |          5  |                    d            d d d            n# 1 swxY w Y   d}t          j        t
          |          5  |                    |j        |j        |j        z              d d d            d S # 1 swxY w Y   d S )	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@   sC   YYYYYY7788 I]9C000 	% 	%IImm$$$	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	%7]>555 	- 	-IIrtI,,,	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	-s#   AAA?*B66B:=B:c                    t          ddgddgd          }|                    |j        d           }t          g dg d	gt          d
                    }t	          j        ||           |                    d d           }t          g dg d	gt          d
                    }t	          j        ||           d S )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122RT%>%>??lllLLL94<<PPP
fh///__0L0LMMlllLLL94<<PPP
fh/////r   N)__name__
__module____qualname__r.   r8   r>   rG   rO   rK   r   r   r   r      s`        !0 !0 !0F0 0 00 0 0	- 	- 	-	0 	0 	0 	0 	0r   r   )rB   pandasr   pandas._testing_testingr%   r   rK   r   r   <module>rV      sn                M0 M0 M0 M0 M0 M0 M0 M0 M0 M0r   