Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/series/indexing/__pycache__/test_where.cpython-37.pyc
Назад
B �5�gS7 � @ sF d dl Zd dlZd dlm mZ d dlmZ d dl Z d dl m Z m Z m Z mZ d dlmZ dd� Zdd� Zej�dejejfejejfejejfejejfejejfejejfg�d d � �Z d d � Z d d� Z dd� Z dd� Z ej�de!e"ej#e g�dd� �Z$ej�ddd dge dddg�dd dge d �e j%e d �gg�d d � �Z&d!d"� Z'd#d$� Z(ej�d%e)dd&��ej�d'd(d)d)d)d)gd(d)gd)gg�ej�d*d+ej*e�+e,�j-e�+e,�j.g�ej�d,d-d.� d/d.� d0d.� g�d1d2� ����Z/d3d4� Z0d5d6� Z1d7d8� Z2d9d:� Z3d;d<� Z4d=d>� Z5d?d@� Z6dAdB� Z7ej�de e j8g�dCdD� �Z9ej:dEdF� �Z;dS )G� N)� is_integer)�Series� Timestamp� date_range�isnac C s\ t t�d�| d�}|dk }tdd�||< t ttdd��ttdd�� | d�}t�||� d S )N� )�dtype� � � )r �np�arange�range�list�tm�assert_series_equal)Z sint_dtype�s�mask�expected� r �]/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/series/indexing/test_where.py�test_where_unsafe_int s $r c C s` t t�d�| d�}|dk }tdd�||< ttdd��ttdd�� }t || d�}t�||� d S )Nr )r r r r )r r r r r r r )Z float_dtyper r �datar r r r �test_where_unsafe_float s r zdtype,expected_dtypec C sZ t t�d�| d�}dddddg}|dk }t |ttdd�� |d�}|||< t�||� d S ) Nr )r g @g @g @g @g @r )r r r r r r r )r Zexpected_dtyper �valuesr r r r r �test_where_unsafe_upcast' s r c C s: t t�d�dd�} ddddg}| dk}t ttd ��| d d�}|| |< t�| |� t t�d�dd�} | dk }td d �| |< t ttd d ��ttdd�� dd�}t�| |� | j|jks�t�t t�d�dd�} | dk}d gd | |< t d dd dddgd gd dd�}t�| |� t t�d��} | dk}d}t j t |d�� dddd dg| |< W d Q R X t j t |d�� d gd | |< W d Q R X t dd ddg�} | � | d ktj �}t tj tj ddg�}t�||� t td���t�} d | d< | d }t|��s�t�t td���t�} d | | dk< | t| � }t tj dgd�}t�||� d S )Nr �int64)r g @g @g @g @r � �float64r r r � � � z-cannot assign mismatch length to masked array)�match� � )�index)r r r r r r r r �AssertionError�pytest�raises� ValueError�where�nan�astype�floatr )r r r r �msg�resultr r r �test_where_unsafe< sJ $ " r0 c C s t tj�d��} | dk}| �|��� }| | }t�||� | �|| �}t�|| �� � | �|�}| j |j ksrt �|| k s~t �t dddddg| j d�}| �� }|| � |j d d� �� |j �}|�|d d� �}t�||� |�� }|d |j d< |�|d d� | �}t�||� d S )Nr r TF)r% r! )r r �random�randnr* �dropnar r �abs�shaper& r% ZreindexZiloc)r �cond�rsZrs2�s2r r r r � test_wheres s& r9 c C s^ t td��} d}tjt|d�� | �| dkdd�}W d Q R X t ddddd g�}t�||� d S ) Nr z|In a future version of pandas all arguments of Series.where except for the arguments 'cond' and 'other' will be keyword-only)r" r r Fr r! r )r r r Zassert_produces_warning� FutureWarningr* r )r r. r/ r r r r �"test_where_non_keyword_deprecation� s r; c C s t tj�d��} | dk}d}tjt|d�� | �d� W d Q R X tjt|d�� | �|d d� j| � W d Q R X t ddg�} ddg| dd g< t ddg�}t � | |� d }tjt|d�� dddg| dd g< W d Q R X d }tjt|d�� g | dd g< W d Q R X d S ) Nr r z,Array conditional must be same shape as self)r" r r! r TFz-cannot assign mismatch length to masked arrayzrNumPy boolean array indexing assignment cannot assign 0 input values to the 1 output values where the mask is true) r r r1 r2 r' r( r) r* r r r )r r6 r. r r r r �test_where_error� s" " r< �klassc C sF t dddg�}dddg}t tjddg�}|�| |��}t�||� d S )Nr r r! FT)r r r+ r* r r )r= r r6 r r/ r r r �test_where_array_like� s r> r6 r r r r �True�Falsez 2017-01-01z 2017-01-02c C sd t dddg�}d}tjt|d�� |�| � W d Q R X d}tjt|d�� |�dg� W d Q R X d S )Nr r r! z(Boolean array expected for the condition)r" z,Array conditional must be same shape as selfT)r r' r( r) r* )r6 r r. r r r �test_where_invalid_input� s rA c C s� d} t dddg�}dg}tjt| d�� |�|� W d Q R X t dtjtjg�}|�t |��}t�||� t� ddddg�}tjt| d�� |�|� W d Q R X t tjdtjg�}|�t |��}t�||� d S )Nz,Array conditional must be same shape as selfr r r! T)r" F) r r' r( r) r* r r+ r r �array)r. r r6 r �outr r r �test_where_ndframe_align� s rD c C s2 dd� } t td��}tjt| d�d�� ttd��|dd�< W d Q R X ttd��|dd�< t dd d g�}t�|�t j �|� t td ��}tjt| d�d�� ttd��|dd d �< W d Q R X t td ��}ttd ��|dd d �< t dd d dddg�}t�||� t td ��}tjt| d�d�� ttd��|d d�< W d Q R X ttd ��|dd�< t dd ddd dg�}t�||� t td��}tjt| d�d�� ttd��|dd d g< W d Q R X t td��}tjt| d�d�� ttd ��|dd d g< W d Q R X t td��}ttd��|d< t ttd��d dg�}t�||� d S )Nc S s d| � d�S )Nzcannot set using a z/ indexer with a different length than the valuer )�xr r r �<lambda>� � z,test_where_setitem_invalid.<locals>.<lambda>�abc�slice)r" � r r! r r Zabcdefr �b�d�e�f�������a�cz list-liker ) r r r' r( r) r r r r, r r )r. r r r r r �test_where_setitem_invalid� s< rS �sizer r TF�itemg @�boxc C s t �| g�S )N)r rB )rE r r r rF 5 rG rF c C s | gS )Nr )rE r r r rF 5 rG c C s | fS )Nr )rE r r r rF 5 rG c s� t �|| �}t j| td�� t� �fdd�t|�D ��}t� �}|��||< t�||� t� �}|�| |���}t�||� t� �}|� ||���}t�||� d S )N)r c s g | ]\}}|r�n� | �qS r r )�.0�iZuse_item)r rU r r � <listcomp>? s z"test_broadcast.<locals>.<listcomp>) r �resizer r- r � enumerater r r* r )rT r rU rV Z selectionr r r/ r )r rU r �test_broadcast+ s r\ c C s� t tj�d��} | dk}| �� }|j|dd� t�|�� | | � t�|| �|�� | �� }|j|| dd� t�|| �|| �� d S )Nr r T)Zinplace) r r r1 r2 �copyr* r r r3 )r r6 r7 r r r �test_where_inplaceO s r^ c C s� t ttd���} t ttd���}t�| |g�}|�|dk �}t ddtjddtjgddddddgd�}t� ||� d||dk < t ddddddgddddddgd�}t� ||� ||dk d7 < t ddddddgddddddgd�}t� ||� d S ) Nr! r r r )r% r r � ) r r r �pd�concatr* r r+ r r )�s1r8 Zcombr/ r r r r �test_where_dups^ s ( $ $rc c C sN t dddg�} | �| dkd�}t|d �r.t�t|d �s>t�t|d �sNt�t|d t�s`t�|jdksnt�| �| dkdddg�}t|d �r�t�t|d �s�t�t|d �s�t�t|d t�s�t�|jdks�t�| �| dkt�dddg��}t|d ��rt�t|d ��st�t|d ��s&t�t|d t��s:t�|jdk�sJt�d S ) Nr r r! �Xr �object�Y�Z) r r* r r& � isinstance�strr r rB )r �wr r r � test_where_numeric_with_strings s& rk c C s� t ddgdd�} t ddg�}t�ddg�}| �|ddg�}t�||� | �|d�}t�||� | �|d�}t�||� | �|ddg�}t�||� | �|dtjg�}t dd gdd�}t�||� d S ) Nr r ztimedelta64[ns])r r Fg $@re )r r rB r* r r r+ )r r r r7 r r r �test_where_timedelta_coerce� s rl c C s t tddd��} t ddg�}t�ddg�}| �|ddg�}t�||� | �|d�}t�||� | �|d�}t�||� | �|ddg�}t�||� | �|dtjg�}t dd gdd�}t�||� d d g}t d d � |D ��} | �t dd g��}t tj | d g�}t�||� d S )NZ20130102r )�periodsr Fg $@re )r z2016-12-31 12:00:04+00:00z 2016-12-31 12:00:04.010000+00:00c S s g | ] }t |��qS r )r )rW �tr r r rY � s z2test_where_datetime_conversion.<locals>.<listcomp>Tr ) r r r rB r* r r r+ r` �NaT)r r r r7 Z timestampsr r r � test_where_datetime_conversion� s& rp c C sr t tjdddg| d��}t tjdddg| d��}t ddd g�}|�||�}t tjdddg| d��}t�||� d S ) NZ20150101Z20150102Z20150103)�tzZ20160514Z20160515Z20160516TF)r r` � DatetimeIndexr* r r )�tz_naive_fixtureZser1Zser2r r/ �expr r r �test_where_dt_tz_values� s ru c C sH t tj�ddg��} | �| dkd�}t tj�ddg��}t�||� d S )Nr r r )r r` ZarraysZ SparseArrayr* r r )�serr/ r r r r �test_where_sparse� s rw c C s& t g td�} | �g �}t�|| � d S )N)r )r r- r* r r )rv r/ r r r �=test_where_empty_series_and_empty_cond_having_non_bool_dtypes� s rx c C s\ | t jddddtjgdddgd�dd�}| dddddgdd�}|�|dk�}t�||� d S )N�A�B�C)� categories�category)r )r` � Categoricalr r+ r* r Z assert_equal)r= rt Zdf�resr r r �test_where_categorical� s r� c C s� | }t dd|d��d �}t�|d |d tjg�}t�|d tj|d g�}t�dddg�}|�||�}t � ||� |j �||�}t � ||j � t |��||�}t �|t |�� t�|��|d d �d f t�|��}t �|t�|�� d S ) Nz 2001-01-01r! )rm rq r r r TF)r Z _with_freqr` rr ro r~ r rB r* r Zassert_index_equal�_dataZassert_datetime_array_equalr r � DataFrameZassert_frame_equal)rs rq ZdrZlvalsZrvalsr r r r r �#test_where_datetimelike_categorical� s $r� )<�numpyr r' Z pandas.util._test_decorators�utilZ_test_decorators�tdZpandas.core.dtypes.commonr Zpandasr` r r r r Zpandas._testingZ_testingr r r �markZ parametrizeZint8r Zint16Zint32r Zfloat32r r0 r9 r; r< r �tuplerB r> ro rA rD rS r r+ Zfinfor- �max�minr\ r^ rc rk rl rp ru rw rx r� r� Z&skip_array_manager_not_yet_implementedr� r r r r �<module> s\ 7 9 (
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка