U
|
ý°dú ã @ s2 d dl Z d dlmZ d dlmZ G dd dZdS )é N)Ú DataFramec @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) Ú
|
TestAssignc C s: t dddgdddgd}| ¡ }|j|j|j d}| ¡ }dd dg|d
|
< t ||¡ |jd d d}t ||¡ t ||¡ |jdd dgd}t ||¡ t ||¡ |j|j|j d }|jdddjd
|
did}t ||¡ |j|j|j d}| ¡ }dddg|d< t ||¡ |jdd d}t ||¡ d S )Né é é é é é ©ÚAÚB)ÚCg @r c S s | j | j S ©N)r r ©Úx© r ú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_assign.pyÚ<lambda> ó z(TestAssign.test_assign.<locals>.<lambda>©r r )Zaxis©Úcolumns©r é é r c S s | j | j S r r
|
r r r r r ( r ) r ÚcopyÚassignr r ÚtmÚassert_frame_equalZdropÚrename)ÚselfÚdfÚoriginalÚresultÚexpectedr r r Ú test_assign s* zTestAssign.test_assignc C s t ddgddgddggddgd }|jd
|
d d g|jd d d}t ddd
|
ddgddd ddgddd ddggtdd }t ||¡ d S )Nr r r r r r r r r r é r c S s | j S r r r r r r r - r z1TestAssign.test_assign_multiple.<locals>.<lambda>)r ÚDÚEZABCDE©r r r Úlistr r ©r r! r# r$ r r r Útest_assign_multiple+ s " & ÿz TestAssign.test_assign_multiplec C s¼ t ddgddggddgd}|j|j|j |j|j d}t dddd gddd
|
d ggtd d}t ||¡ |j|j|j |j|j d }t ddd dgddd d
|
ggtd d}t ||¡ d S )Nr r r r r r r )r' r éÿÿÿÿr ZABDC©r r' ÚABCD)r r r r r* r r r+ r r r Útest_assign_order3 s $ $z TestAssign.test_assign_orderc C s t dddgdddgd}d}tjt|d | d
|
d ¡ W 5 Q R X d }tjt|d |j|j|j|j d W 5 Q R X d S )Nr r r r r r r
|
z7assign\(\) takes 1 positional argument but 2 were given)Úmatchc S s | j S r r r r r r r F r z,TestAssign.test_assign_bad.<locals>.<lambda>z''DataFrame' object has no attribute 'C'r. )r ÚpytestZraisesÚ TypeErrorr ÚAttributeErrorr r )r r! Úmsgr r r Útest_assign_bad@ s zTestAssign.test_assign_badc C s¤ t ddgddgd}|j|jdd d}t ddddgddddggtd d
|
}t ||¡ |jd d d d d}t ddddgddddggtd d
|
}t ||¡ d S ) Nr r r r r
|
c S s | d | d S ©Nr r r r r r r r N r z2TestAssign.test_assign_dependent.<locals>.<lambda>r. r/ r c S s | j S r r ©r! r r r r R r c S s | d | d S r7 r r8 r r r r R r r) r+ r r r Útest_assign_dependentK s $ $z TestAssign.test_assign_dependentN)Ú__name__Ú
|
__module__Ú __qualname__r% r, r0 r6 r9 r r r r r s
|
# r )r2 Zpandasr Zpandas._testingZ_testingr r r r r r Ú<module> s
|
