Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/__pycache__/test_isin.cpython-37.pyc
Назад
B �5�g� � @ sJ d dl Zd dlZd dlZd dlmZmZmZ d dlm Z G dd� d�Z dS )� N)� DataFrame� MultiIndex�Seriesc @ s� e Zd Zdd� Zej�dg eed�e � g �g�dd� �Z dd� Z d d � Z d d � Zd d� Zdd� Zdd� Zdd� Zdd� Zdd� Zej�deddd d gid d�edd d gd d�g�d d � �Zd!d"� Zd#S )$�TestDataFrameIsInc sr t ddddgddddgddd dgd �d d d dgd�� ddd g�� ���}t � �fdd�� jD ��}t�||� d S )N� � � � �a�b�f�n�c)�vals�ids�ids2�foo�bar�baz�qux)�indexc s g | ]}� j | ����qS � )�loc�isin)�.0�s)�df�otherr �Z/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/test_isin.py� <listcomp> s z/TestDataFrameIsIn.test_isin.<locals>.<listcomp>)r r r �tm�assert_frame_equal)�self�result�expectedr )r r r � test_isin s zTestDataFrameIsIn.test_isin�empty)Zdtypec C sD t dddgdddgd��}t d|j|j�}|�|�}t�||� d S )Nr r r �er )�A�BF)r r �columnsr r r! )r" r& r r$ r# r r r �test_isin_empty s z!TestDataFrameIsIn.test_isin_emptyc C s� t dddgdddgd��}ddgi}t d|j|j�}d |jd < |�|�}t�||� t dddgdddgd��}ddg|_t d|j|j�}d |jd < |�|�}t�||� d S ) Nr r r r' r )r( r) r( FT)r r( )r r r* r r r r! )r" r �dr$ r# r r r �test_isin_dict' s z TestDataFrameIsIn.test_isin_dictc C s� t ddddgddddgddd dgd �d d d dgd�}d}tjt|d�� |�d� W d Q R X tjt|d�� |�d� W d Q R X d S )Nr r r r r r r r r )r r r r r r r )r zfonly list-like or dict-like objects are allowed to be passed to DataFrame.isin\(\), you passed a 'str')�matchZaaa)r �pytest�raises� TypeErrorr )r" r �msgr r r � test_isin_with_string_scalar9 s z.TestDataFrameIsIn.test_isin_with_string_scalarc C s� t ddddgdtjddgd��}t ddddgdtjddgd��}t d |j|j�}|�|�}d |d jddg< d |d jddg< t�||� d d g|_|�|�}d |d < t�||� d S )Nr r r r )r( r) r � � FTr( r) �C) r �np�nanr r* r r r r! )r" �df1�df2r$ r# r r r � test_isin_dfM s z TestDataFrameIsIn.test_isin_dfc C sb t dddgdddgd��}tt|d |d ��|d < |d �d g�}t�|td d d gd d�� d S )Nr r r r r r )r( r) r( r) r6 )r r TF)�name)r �list�zipr r Zassert_series_equalr )r" r r# r r r �test_isin_tuples\ s z"TestDataFrameIsIn.test_isin_tuplesc C s� t ddddgdtjddgd��}t ddgddgdtjgddggd d gd �}d }tjt|d �� |�|� W d Q R X t ddgddgdtjgddggd d gddddgd�}tjt|d �� |�|� W d Q R X d d g|_tjt|d �� |�|� W d Q R X d S )Nr r r r )r( r) r r4 r5 r) )r* z+cannot compute isin with a duplicate axis\.)r. r( )r* r )r r7 r8 r/ r0 � ValueErrorr r* )r" r9 r: r2 r r r �test_isin_df_dupe_valuesc s * z*TestDataFrameIsIn.test_isin_df_dupe_valuesc C s� t ddddgddddgd��}t ddgddgddggddgd�}|�|�}t d|j|jd�}d|jd< d|jd < t�||� d S ) Nr r )r( r) r( )r* F)r r* T)r r )r r r r* r Zilocr r! )r" r r r# r$ r r r �test_isin_dupe_selfy s " z%TestDataFrameIsIn.test_isin_dupe_selfc C s� t ddddgdtjddgd�dddd gd �}tddd dgdddd gd �}t d |j|jd �}d|d jd< d|jd < |�|�}t� ||� d S )Nr r r r )r( r) r r r r, )r � F)r r* Tr( ) r r7 r8 r r r* r r r r! )r" r r r$ r# r r r �test_isin_against_series� s * z*TestDataFrameIsIn.test_isin_against_seriesc C s t �ddddddddd d d d g �}tt�d �t�d �d�|d�}tddddddddddddg ddddddddddddg d��}td|j|jd�}|�|�}t � ||� ||_|j � t �}|d d �df |d d �df< t|ddg|d�}|�|�}t � ||� d S )N)r r r )r r r )r r r )r r r )r r r )r r r )r r r )r r r )r r r )r r r )r r r )r r r r4 )r( r) )r r r F)r r* r( r) )r* r )r � from_tuplesr r7 ZonesZzerosr r* r r r! �valuesZastype�bool)r" �idxr9 r: r$ r# r r r �test_isin_multiIndex� s4 " z&TestDataFrameIsIn.test_isin_multiIndexc C s� t dt�ddg�i�}t dt�dd�t�dd�gi�}t dg i�}t � }t dddgi�}|�|�}t�||� |�|�}t�||� |�|�}t�||� |�|�}t�||� d S )N�datez 2014-01-01z 2014-01-02r r r F)r �pd� to_datetimeZ Timedeltar r r! )r" Zdf1_tsZdf1_tdr: Zdf3r$ r# r r r � test_isin_empty_datetimelike� s z.TestDataFrameIsIn.test_isin_empty_datetimelikerF r r r r �categoryc C sN t dddgdddgd��}t dddgd d d gd��}|�|�}t�||� d S ) Nr r r r r5 � )r r TF)r r r r! )r" rF r r$ r# r r r �test_isin_category_frame� s z*TestDataFrameIsIn.test_isin_category_framec C sR t �dddg�}|jdd� tdddg�}|�|�}tdddg�}t�||� d S )Nr r r F)�writeT)r7 �arrayZsetflagsr r r r! )r" Zarrr r# r$ r r r �test_isin_read_only� s z%TestDataFrameIsIn.test_isin_read_onlyN)�__name__� __module__� __qualname__r% r/ �markZ parametrizer �objectr7 rR r+ r- r3 r; r? rA rB rD rI rM r rP rS r r r r r s ( % r ) �numpyr7 r/ ZpandasrK r r r Zpandas._testingZ_testingr r r r r r �<module> s
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка