Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/frame/indexing/__pycache__/test_where.cpython-37.pyc
Назад
B �5�g}f � @ s� d dl m Z d dlZd dlZd dlmZ d dlZd dlmZm Z m Z m Z m Z m Z mZ d dlmZ ejddddgd �d d � �Zd d � ZG dd� d�Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� ZdS ) � )�datetimeN)� is_scalar)� DataFrame� DatetimeIndex�Series� StringDtype� Timestamp� date_range�isna�default� float_string� mixed_float� mixed_int)�paramsc C sT | j dkr&ttj�dd�dddgd�S | j dkr4|S | j d krB|S | j d krP|S d S ) Nr � � �A�B�C)�columnsr r r )�paramr �np�random�randn)�request�float_string_frameZmixed_float_frameZmixed_int_frame� r �\/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/frame/indexing/test_where.py� where_frame s r c s& dd� � t t� fdd�| �� D ���S )Nc S s t | jjtjtjf�o | jdkS )NZuint8)� issubclass�dtype�typer �integerZfloating)�sr r r �is_ok# s z_safe_add.<locals>.is_okc 3 s. | ]&\}}� |�r ||d fn||fV qdS )� Nr )�.0�cr# )r$ r r � <genexpr>( s z _safe_add.<locals>.<genexpr>)r �dict�items)�dfr )r$ r � _safe_add! s r, c @ s e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Ze j � d e e ejg�dd� �Ze j � ddgdgdggedgdgdgg�eddddgi�dgdgdgged�gejged�ggg�dd � �Ze j � dddgddgddggeddgddgd dgg�ddgddgddggedddgd d d gd ��ejed�ged�ejged!�ed!�ggg�d"d#� �Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Z d.d/� Z d0d1� Z d2d3� Z d4d5� Z d6d7� Z!d8d9� Z"d:d;� Z#d<d=� Z$d>d?� Z%d@dA� Z&e j � dBi dCdDig�dEdF� �Z'dGdH� Z(dIdJ� Z)dDS )K�TestDataFrameIndexingWherec C sV ddd�}|}||kr@d}t jt|d�� |dk W d Q R X d S |dk}|||� d S )NTc S s� t | �}| �||�}| �|j|�}xH|�� D ]<\}}tt�|| | | || �|jd�}tj||dd� q,W t� ||� |r�|j | j k� � s�t �d S )N)�indexF)Z check_names) r, �where�valuesr* r r r. �tm�assert_series_equal�assert_frame_equal�dtypes�all�AssertionError) r+ �cond� check_dtypes�other1�rsZrs2�k�v�expr r r � _check_get- s $ z=TestDataFrameIndexingWhere.test_where_get.<locals>._check_getz6'>' not supported between instances of 'str' and 'int')�matchr )T)�pytest�raises� TypeError)�selfr r r> r+ �msgr7 r r r �test_where_get, s z)TestDataFrameIndexingWhere.test_where_getc C sn t dd� dD ��}d|jdd d �f< |j}tt�d�t�d�t�d�t�d �gdddd gd �}t�||� d S ) Nc S s i | ]}t d gd |d�|�qS )r% r )r )r )r& r' r r r � <dictcomp>G s zCTestDataFrameIndexingWhere.test_where_upcasting.<locals>.<dictcomp>)�float32�float64�int32�int64r r% rG rH rI rJ )r. )r �ilocr4 r r r r1 r2 )rC r+ �result�expectedr r r �test_where_upcastingD s z/TestDataFrameIndexingWhere.test_where_upcastingc C s� d dd�}|}||kr@d}t jt|d�� |dk W d Q R X d S |dkdd � }|||t|�� |dk}|||t|�j� |dk}tdd � |jD ��}|||tj|d � d S ) NTc S s� | � ||�}x�t|j�D ]�\}}|| }| | j}|| �| | j��d�j} t|�r\|} n4t|t j �r�t |d d �|f |jd�j} n || j} | � � r�|n t � | || �} t | |j|d�} t j|| dd� qW |r�t|t j �s�|j| jk� � s�t�d S )NF)r. )r. �name)Z check_dtype)r/ � enumerater r0 �reindexr. �fillnar � isinstancer Zndarrayr r5 r1 r2 r4 r6 ) r+ r7 �otherr8 r: �ir; rL �dr' �o� new_valuesrM r r r � _check_align` s zETestDataFrameIndexingWhere.test_where_alignment.<locals>._check_alignz6'>' not supported between instances of 'str' and 'int')r? r r% c s s | ]}t |jtj� V qd S )N)r r! r r" )r& r# r r r r( � s zBTestDataFrameIndexingWhere.test_where_alignment.<locals>.<genexpr>)r8 )T) r@ rA rB r, r0 r5 r4 r �nan)rC r r rY r+ rD r7 r8 r r r �test_where_alignment^ s z/TestDataFrameIndexingWhere.test_where_alignmentc C s� t tj�dd�dddgd�}|dk}|d jdd �d d �f }d }tjt|d �� |�||� W d Q R X |j d d �d d �f j}t |�}d }tjt|d �� |�||� W d Q R X 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% � z4other must be the same shape as self when an ndarray)r? z,Array conditional must be same shape as selfT) r r r r r0 r@ rA � ValueErrorr/ rK r, �mask)rC r+ r7 Zerr1rD Zerr2r9 r r r �test_where_invalid� s z-TestDataFrameIndexingWhere.test_where_invalidc C s� ddd�}|}||kr@d}t jt|d�� |dk W d Q R X d S |dk}|||� |dk}|||� |dkdd � }|||� d S ) NTc S s� | � � }|�| ��d�}|�| �}|j|tjdd�}|d ksBt�t� ||� |r�xL| j � � D ]>\}}t |j tj�r�|| �� s�t�d�}|| j|ks^t�q^W d S )NT)�inplacerH )�copyZ reindex_likerR r^ r/ r rZ r6 r1 r3 r4 r* r r! r" r5 r ) r+ r7 r8 ZdfiZecondrM � return_valuer; r< r r r � _check_set� s z=TestDataFrameIndexingWhere.test_where_set.<locals>._check_setz6'>' not supported between instances of 'str' and 'int')r? r r% )T)r@ rA rB )rC r r rc r+ rD r7 r r r �test_where_set� s z)TestDataFrameIndexingWhere.test_where_setc C sR t td�tdd�d��}|�|d dk�}||d dk �|j�}t�||� d S )Nr � � )�a�brg r% )r �ranger/ rQ r. r1 r3 )rC r+ rL rM r r r �test_where_series_slicing� s z4TestDataFrameIndexingWhere.test_where_series_slicing�klassc C s� t ddddgi�}dgdgdgg}t dtjddgi�}|�||��}t�||� d|d< dtjdg|d< ddgddgddgg}|�||��}t�||� d S )Nrg r% r\ r FTrh )r r rZ r/ r1 r3 )rC rk r+ r7 rM rL r r r �test_where_array_like� s z0TestDataFrameIndexingWhere.test_where_array_liker7 r% r r\ r rf rg �True�Falsez 2017-01-01z 2017-01-02c C s> t ddddgi�}d}tjt|d�� |�|� W d Q R X d S )Nrg r% r\ r z(Boolean array expected for the condition)r? )r r@ rA r] r/ )rC r7 r+ rD r r r � test_where_invalid_input_single� s z:TestDataFrameIndexingWhere.test_where_invalid_input_singlere � � )rg rh z 2017-01-03c C sF t dddgdddgd��}d}tjt|d�� |�|� W d Q R X d S )Nr% r\ r )rg rh z(Boolean array expected for the condition)r? )r r@ rA r] r/ )rC r7 r+ rD r r r �!test_where_invalid_input_multiple� s z<TestDataFrameIndexingWhere.test_where_invalid_input_multiplec C s� t dddgdddgg�}t dddgdddgg�}|�|�}t d tjdgtjtjdgg�}t�||� d d d g|_|�|�}t tj|j|jd �}t�||� d S )Nr% r\ r re r � TFg �?rg rh r' )r. r )r r/ r rZ r1 r3 r r. )rC r+ r7 rL rM r r r � test_where_dataframe_col_match s z9TestDataFrameIndexingWhere.test_where_dataframe_col_matchc C s� d}t dddgdddgg�}dg}tjt|d �� |�|� W d Q R X t dddgtjtjtjgg�}|�t|��}t� ||� t� d dd dg�}tjt|d �� |�|� W d Q R X t tjtjtjgdddgg�}|�t|��}t� ||� d S ) Nz,Array conditional must be same shape as selfr% r\ r re r rs T)r? F) r r@ rA r] r/ r rZ r r1 r3 �array)rC rD r+ r7 rM �outr r r �test_where_ndframe_align s z3TestDataFrameIndexingWhere.test_where_ndframe_alignc C s� t ddddgddddgd�dd�}t tjtjddgddtjtjgd�dd�}|�|dktj�}t�||� |�� }|j|dktjd d �}|d ks�t�t�||� d S ) Ng �?g @g @g @)rg rh rH )r r\ T)r` )r r rZ r/ r1 r3 ra r6 )rC r+ rM rL rb r r r �test_where_bug. s z)TestDataFrameIndexingWhere.test_where_bugc C s� t tjddddg|d�tjdddd gd d�d ��}t tjtjddgddtjtjgd �d d�}|�|dktj�}t�||� |�� }|j|dktjd d �}|d ks�t�t�||� d S )Nr% r\ r re )r g @g @g @g �?rH )rg rh T)r` ) r r ru rZ r/ r1 r3 ra r6 )rC Z sint_dtyper+ rM rL rb r r r �test_where_bug_mixed? s z/TestDataFrameIndexingWhere.test_where_bug_mixedc C s� t ddgddgddgd��}t tjdgd tjgtjtjgd��}|�� ||kB }|�� }||| < |�||�}t�||� t ddgdd gd ��}t tjdgdtjgd ��}|�� ||kB }|�� }||| < |�||�}t�||� d S ) Nr% r\ r re r rs )r r% r\ rp rq r )r r% )r r rZ r ra r/ r1 r3 )rC rg rh Zdo_not_replacerM rL r r r � test_where_bug_transpositionU s $ z7TestDataFrameIndexingWhere.test_where_bug_transpositionc C s� t tddd�tddd�tj�d�d��}tddd�}d }tjt|d �� ||k W d Q R X ||j d d �d d �f |k }|� � }tj |j d dgd f< tj |j d d �df< t �||� d S )NZ20130102r )ZperiodsZ20130104)r r r i� r% r zF'>' not supported between instances of 'float' and 'datetime.datetime')r? ���r r r )r r r r r r r@ rA rB rK ra rZ �locr1 r3 )rC r+ ZstamprD rL rM r r r �test_where_datetimek s z.TestDataFrameIndexingWhere.test_where_datetimec C s� t dttd��i��t�}d ||dk< t dtdddddd d dtjtjg �i�}t�||� t dtjd d �tjd tjd �g�}d }t j t |d�� |j t |� d dd� W d Q R X d S )NZseries� rf r r% r\ r re r rs ZTest)r r r z boolean setting on mixed-type)r? T)r` )r r ri �astype�floatr rZ r1 r3 r@ rA rB r/ r )rC r+ rM rD r r r �test_where_none� s & z*TestDataFrameIndexingWhere.test_where_nonec C s< t dgd�}|}|jtk�� s"t�|�|�}t�||� d S )Nrg )r )r r4 �objectr5 r6 r/ r1 r3 )rC r+ r7 rL r r r �9test_where_empty_df_and_empty_cond_having_non_bool_dtypes� s zTTestDataFrameIndexingWhere.test_where_empty_df_and_empty_cond_having_non_bool_dtypesc C s dd� }|� }|� |�� �}|jt�|�|�� dd�}t�||� |jt�|�|�� ddd�}|d ksjt�t�||� |� � d�}|jdd � |d d �}|j|dk|d d d�}t�||� |j|dk|d d d�}t�||� |� }|� d �}|�t�|�t d |j |j d��}t�||� d S )Nc S sR t tj�dd��} tj| jdd�df< tj| jdd�df< tj| jdd�d f< | S ) Nr~ r r r re rs r% rp r\ )r r r r rZ rK )r+ r r r �create� s z;TestDataFrameIndexingWhere.test_where_align.<locals>.creater )�axisT)r` r� r c S s | � | dk|�S )Nr )r/ )�x�yr r r �<lambda>� � z=TestDataFrameIndexingWhere.test_where_align.<locals>.<lambda>)r� r. �rowsr% )r. r ) rR Zmeanr/ �pdZnotnar1 r3 r6 �applyr r. r )rC r� r+ rM rL rb r r r �test_where_align� s& z+TestDataFrameIndexingWhere.test_where_alignc C s\ t ddgtjdggddgd�}t ddgddggddgd�}tj||�� dk< t�||� d S ) Ny �? �?r\ y @ �?rg rh )r y @ �?r )r r rZ �absr1 r3 )rC rM r+ r r r �test_where_complex� s z-TestDataFrameIndexingWhere.test_where_complexc C s> t tj�dd��}t ddgddgg�}tddg�}t ddgddggdd�}|j||dd�}t�||� |�� }|j||dd d �}|d ks�t �t�||� t ddgddggdd�}|j||d d�}t�||� |�� }|j||d d d �}|d ks�t �t�||� t ddgd d ggdd�}t ddgddgg�}tdtj g�}t ddgtj tj ggdd�}|j||dd�}t�||� |�� }|j||dd d �}|d k�s�t �t�||� t dtj gdtj gg�}|j||d d�}t�||� t tj ddgdd�tj tj tj gdd�d��}|�� }|j||d d d �}|d k�s.t �t�||� d S )Nr\ Fr r% rH )r r. )r� T)r� r` r r re rJ )r r% ) r r r r r r/ r1 r3 ra r6 rZ ru )rC r+ r^ r# rM rL rb r r r �test_where_axis� sJ z*TestDataFrameIndexingWhere.test_where_axisc C s t jttj�dd��ttjjdddd�dd�gdd d �}td |j|jd �}t d |jd �}t d|jd �}|j ||dd�}td|j|jd �}|d � d�|d< |d � d�|d< t � ||� |�� }|j ||ddd�}|d ks�t�t � ||� |j ||dd�}td|j|jd �}|d � d�|d< |d � d�|d< t � ||� |�� }|j ||ddd�}|d k�sdt�t � ||� |�� jd dd�}|�� }tj|jd d d �f< |� ||�}t � ||� |j ||dd�}t � ||� |�� }|j ||dd�}|d k�s�t�t � ||� |�� }|j ||ddd�}|d k�s,t�t � ||� |�� jd d d�} |�� }tj|jd d �d f< |� || �}t � ||� |j || dd�}t � ||� |�� }|j || dd�}|d k�s�t�t � ||� |�� }|j || ddd�}|d k�s�t�t � ||� d S )Nr~ r\ r )r~ r\ )�sizerJ )r Tr% )Z ignore_indexr� F)r r. )r. r )r� g �?r )r� r` r. g @)r` )r` r� )r� �concatr r r r �randintr r. r r/ r r1 r3 ra r6 ZdroprZ r| ) rC r+ r^ �s1�s2rL rM rb Zd1Zd2r r r � test_where_axis_multiple_dtypes� sp z:TestDataFrameIndexingWhere.test_where_axis_multiple_dtypesc C sH t dddgdddgddd gg�}|�d d � d d � �}t dddgdddgddd gg�}t�||� t�||�|dk|d �� |�d d � dd � �}t dddgdddgdddgg�}t�||� t�||�|d dkd�� |d �dd � dd � �}t dddgdddgd ddgg�}t�||� t�||d �|d dk|d d �� d S )Nr% r\ r re r rs rf rp rq c S s | dkS )Nre r )r� r r r r� H r� z@TestDataFrameIndexingWhere.test_where_callable.<locals>.<lambda>c S s | d S )Nr% r )r� r r r r� H r� c S s | d dkj S )Nr\ r )r0 )r� r r r r� N r� c S s dS )N�c r )r� r r r r� N r� r� r c S s | dkS )Nrp r )r� r r r r� T r� c S s | d S )Nr~ r )r� r r r r� T r� � � � � � � r~ � )r r/ r1 r3 )rC r+ rL r= r r r �test_where_callableE s z.TestDataFrameIndexingWhere.test_where_callablec C s� t tdddg|d�dgd�}t tdddg|d�dgd�}t d d d gdgd�}t tdddg|d�dgd�}|�||�}t�||� d S ) NZ20150101Z20150102Z20150103)�tz�date)r Z20150104Z20150105TF)r r r/ r1 r3 )rC Ztz_naive_fixtureZdf1Zdf2r^ r= rL r r r �test_where_tz_valuesY s z/TestDataFrameIndexingWhere.test_where_tz_valuesc C sz t t�d��dd�td�d�}t�dddgdddgg�}|�|�}t dtjtjgtjtjd ggtd�d�}t� ||� d S ) Nrs r\ r �ABC)r TFr r ) r r �arange�reshape�listru r/ rZ r1 r3 )rC r+ r^ rL rM r r r �test_df_where_change_dtypej s &z5TestDataFrameIndexingWhere.test_df_where_change_dtype�kwargsrT Nc C s t t�d��dd�td�d�}t�dddgdddgg�}|j�d�|_|j�d�|_|j �d�|_ |j |f|�}t j d tj gd dgd �}t j tj tj gd d gd �}t j tj d gdd gd �}t |||d��}t�||� |jj |d d �d f f|�}t|dd�}t�||� d S )Nrs r\ r r� )r TF�categoryr )� categoriesr% re r )r r r r )rO )r r r� r� r� ru r r r r r/ r� Z CategoricalrZ r1 r3 r r2 ) rC r� r+ r^ rL r r r rM r r r �test_df_where_with_categoryv s z6TestDataFrameIndexingWhere.test_df_where_with_categoryc C sl t ddgddggddgd�}|d �d�|d< |�|d dk�}|�� }tj|jdd d �f< t�||� d S )Nr r% rg rh )�datar r� ) r r r/ ra r rZ r| r1 � assert_equal)rC r+ rL rM r r r � test_where_categorical_filtering� s z;TestDataFrameIndexingWhere.test_where_categorical_filteringc C s� t dddgdddgd��}t�dtjd g�}t|�}tj|jtd �}d |dd d �f< |j ||d d �}t dtjdgdtjdgd��� t �}t � ||� t|d d� ddgd�}t dddgdtjdgd��}|d � t �|d< |j ||dd �}t � ||� d S )Nr% r\ r re r rs )r r rf rq )r Fr )r� r r )r. )r r� ru �NAr r Zones�shape�boolr/ r r� r1 r3 )rC r+ ZarrZserr^ rL rM Zser2r r r �test_where_ea_other� s $ z.TestDataFrameIndexingWhere.test_where_ea_other)*�__name__� __module__� __qualname__rE rN r[ r_ rd rj r@ �markZ parametrizer� �tupler ru rl r r r r� �NaTro rr rt rw rx ry rz r} r� r� r� r� r� r� r� r� r� r� r� r� r r r r r- + sR 2$ & !5G r- c C sT t tj�dd��}| t k r"|d }|dk}t�t�� |j|ddd� W d Q R X d S )Nre r r r{ F)Ztry_cast)r r r r r1 �assert_produces_warning� FutureWarningr/ )�frame_or_series�objr^ r r r � test_where_try_cast_deprecated� s r� c C s~ | ddddg�}|� � }| tkr(|d n|}|�|dk �}|d9 }t�||� |�|dkddddg�}|d9 }t�||� d S )Nr% r\ r re r r )ra r r/ r1 r� )r� rL rM �colZ where_resr r r �test_where_copies_with_noop� s r� c C s� | ddddgddddgt � d �}| ddgddgt � d �}td d d d g�}|�||�}| tjddtjgddddgt � d �}t�||� d S ) Nrg rh r' rV Zid1Zid2Zid3Zid4)r. r FT)r r r/ r� r� r1 r� )r� r� Z filtered_objZ filter_serrL rM r r r �test_where_string_dtype� s r� c C sv t dgd dgd ddddgd��} | �| dk�}t tjtjgd td�dgd tjdtjdgd��}t�||� d S )NTre F)ZAAAZBBBZCCC)r )r r/ r ru rZ r� r1 r3 )Zdf_maskrL rM r r r �test_where_bool_comparison� s "r� c C sF t td�tjtd�gddtjgd��} | �| �� d �}t� || � d S )NZ20130101Z20130103r% r\ )r r ) r r r� r� r rZ r/ Znotnullr1 r3 )rM rL r r r �test_where_none_nan_coerce� s r� c C s^ t td��} d}tjt|d�� | �| dkdd�}W d Q R X t ddddd g�}t�||� d S ) Nr zIn a future version of pandas all arguments of DataFrame.where except for the arguments 'cond' and 'other' will be keyword-only)r? r% r~ Fr\ r re )r ri r1 r� r� r/ r3 )r# rD rL rM r r r �"test_where_non_keyword_deprecation� s r� ) r �numpyr r@ Zpandas.core.dtypes.commonr Zpandasr� r r r r r r r Zpandas._testingZ_testingr1 Zfixturer r, r- r� r� r� r� r� r� r r r r �<module> s( $
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка