Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/indexes/timedeltas/__pycache__/test_indexing.cpython-37.pyc
Назад
B �5�g�/ � @ s� d dl m Z mZ d dlZd dlZd dlZd dlmZmZm Z m Z m Z m Z m Z mZ d dlmZ G dd� d�ZG dd� d�ZG dd � d �ZG d d � d �ZG d d � d �ZG dd� d�ZdS )� )�datetime� timedeltaN)�Index�NaT� Timedelta�TimedeltaIndex� Timestamp�notna�timedelta_range� to_timedeltac @ sb e Zd Zdd� Zdd� Zdd� Zej�de d�e d �e d d d �e d �� � e � d �g�dd� �ZdS )� TestGetItemc C s6 t ddddd�}|d }|�|�s&t�||k s2t�d S )Nz1 dayz31 day�D�idx)�freq�name.)r �equals�AssertionError)�selfr �result� r �c/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/indexes/timedeltas/test_indexing.py� test_ellipsis s zTestGetItem.test_ellipsisc C s, t ddddd�}|dd � j|jks(t�d S )N�1dZ5d�HZ timebucket)r r � )r r r )r �tdir r r � test_getitem_slice_keeps_name s z)TestGetItem.test_getitem_slice_keeps_namec C s. t ddddd�}�x|gD �] }|d }|td�ks8t�|dd� }t ddddd�}t�||� |j|jkspt�|dd d � }t dd d dd�}t�||� |j|jks�t�|d dd� }t ddddd�}t�||� |j|jks�t�|dd d� }tdddddgddd�}t�||� |j|jkst�qW d S )Nz1 dayz31 dayr r )r r r � z5 day� � z9 day�2Di������ z12 dayz24 dayZ3D� ���z4 dayz3 dayz2 dayz-1D)r r r �tm�assert_index_equalr r )r �idx1r r �expectedr r r � test_getitem% s* zTestGetItem.test_getitem�keyz 1970-01-01z 1970-01-02i� r z 1970-01-03r c C s> t ddd�}tjtt�t|��d�� |�|� W d Q R X d S )Nr r )�periods)�match)r �pytest�raises�KeyError�re�escape�repr�get_loc)r r* r r r r �test_timestamp_invalid_keyB s z&TestGetItem.test_timestamp_invalid_keyN)�__name__� __module__� __qualname__r r r) r- �markZ parametrizer r Z to_datetime64�np� datetime64r4 r r r r r s r c @ s e Zd Zdd� Zdd� ZdS )� TestGetLocc C s� t dddg�}x\dD ]T}|�|d |�dks0t�|�|d �� |�dksLt�|�t|d �|�dkst�qW |j|d dtd�d�dks�t�|j|d dt�dd �d�dks�t�|j|d dtd�d�dks�t�t j t d d �� |j|d d d d� W d Q R X t j t dd ��. |j|d d td�� � td�� � gd� W d Q R X x(dD ] \}}|�d|�|k�sFt��qFW |�|d � � �dk�s�t�|�d�dk�s�t�d S )Nz0 daysz1 daysz2 days)N�pad�backfill�nearestr r< r )� tolerance�sz unit abbreviation w/o a number)r, r> �foo)�methodr? ztolerance size must match))r< r )r= r )r> r z 1 day 1 hour) r r3 r Zto_pytimedelta�strr r9 Z timedelta64r r- r. � ValueErrorZto_timedelta64)r r rB �locr r r � test_get_locU s( $ zTestGetLoc.test_get_locc C s` t dddg�}|�t�dks t�|�d �dks2t�|�td��dksHt�|�tj�dks\t�d S )Nz1 days 01:00:00r z2 days 01:00:00r �nan)r r3 r r �floatr9 rG )r Ztidxr r r �test_get_loc_natw s zTestGetLoc.test_get_loc_natN)r5 r6 r7 rF rI r r r r r; T s "r; c @ s e Zd Zdd� ZdS )�TestGetIndexerc C s� t dddg�}t�|�|�tjdddgtjd�� t dd d g�}t�|�|d �tjd ddgtjd�� t�|�|d �tjdddgtjd�� t�|�|d�tjdddgtjd�� |j|dtd�d�}t�|tjdd dgtjd�� d S )Nz0 daysz1 daysz2 daysr r r )�dtypez-1 hourz12 hoursz 1 day 1 hourr< r$ r= r> z1 hour)r? )r r% Zassert_numpy_array_equalZ get_indexerr9 �array�intpr )r r �target�resr r r �test_get_indexer� s """z TestGetIndexer.test_get_indexerN)r5 r6 r7 rP r r r r rJ � s rJ c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� TestWherec C s` t ddddd�}dddg}t|d |d |d gd dd �}|�||d d d � �}t�||� d S ) Nz1 dayr" r r )r+ r r TFr r )r r r$ )r r �wherer% r& )r r �condr( r r r r � test_where_doesnt_retain_freq� s z'TestWhere.test_where_doesnt_retain_freqc C sB t ddddd�}|dd � �� }tttg| �}t|�}ttjtjg| tdd�}t|d t�sdt �|� ||j �}t � ||� |t�� }t|d |d g| tdd�}|� ||�}t � ||� |t�� �d�}t|d |d g| tdd�}|� ||�}t � ||� t�� }t||g| tdd�}|� ||�}t � ||� d S ) Nz1 dayr" r r )r+ r r r )rK r r r )r �tolistr r r �value�object� isinstance�intr rR Zasi8r% r& r �nowZ to_period) r r �tail�i2�maskr( r �tsZperr r r �test_where_invalid_dtypes� s( z#TestWhere.test_where_invalid_dtypesc C sp t ddddd�}t�dddg�}t�dd �}t|d ||gtdd �}|d |ksTt�|�||�}t� ||� d S ) Nz1 dayr" r r )r+ r r TFr �nsr )rK r r ) r r9 rL r: r rW r rR r% r& )r r rS Zdtnatr( r r r r �test_where_mismatched_nat� s z#TestWhere.test_where_mismatched_natN)r5 r6 r7 rT r_ ra r r r r rQ � s rQ c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestTakec C s� t ddddd�}�xx|gD �]l}|�dg�}|td�ks<t�|�dg�}|td�ksXt�|�ddd g�}t dd ddd�}t�||� |j|jks�t�|�dd d g�}t dd d dd�}t�||� |j|jks�t�|�dd dg�}t ddddd�}t�||� |j|jk�st�|�dd dg�}tdd dgdd�}t�||� |jd k�sLt�|�dd dg�}tdd dgdd�}t�||� |jd kst�qW d S )Nz1 dayz31 dayr r )r r r r$ r r z3 dayr# z5 dayr � z8 dayz2 dayz-3Dr" r z4 dayz6 day)r ���z29 day)r �taker r r% r& r r )r r' r r r( r r r � test_take� s4 zTestTake.test_takec C s� t ddddd�}dddd d d d d g}d}tjt|d�� |j|dd� W d Q R X d}tjt|d�� |j||d� W d Q R X d}tjt|d�� |j|dd� W d Q R X d S )Nz1 dayz31 dayr r )r r r � r � r � � r" z1take\(\) got an unexpected keyword argument 'foo')r, r )rA z$the 'out' parameter is not supported)�outz%the 'mode' parameter is not supportedZclip)�mode)r r- r. � TypeErrorre rD )r r �indices�msgr r r �test_take_invalid_kwargs� s z!TestTake.test_take_invalid_kwargsc C s� dddg}t ddddd�}t|d dd �}|�d d d g�}|d d d g }xH||gD ]<}t�||� t|t�snt�|jd ks|t�|j|jksPt�qPW d S ) Nz 1day 02:00:00z1 day 04:00:00z1 day 10:00:00r Z2dr r )�start�endr r )r r r r# r ) r r re r% r& rX r r r )r Ztdsr r( Ztaken1Ztaken2Ztakenr r r � test_take2� s zTestTake.test_take2c C sZ t dddgdd�}|�t�dddg��}t dddgdd�}t�||� |jt�dddg�d d �}t ddd gdd�}t�||� |jt�dddg�d d d �}t dddgdd�}t�||� d}tjt|d�� |jt�dddg�d d � W d Q R X tjt|d�� |jt�dddg�d d � W d Q R X d}tjt |d�� |�t�ddg�� W d Q R X d S )Nz1 daysz2 daysz3 daysZxxx)r r r r$ T)� fill_valuer F)Z allow_fillrt zJWhen allow_fill=True and fill_value is not None, all indices must be >= -1)r, ���r! z3index -5 is out of bounds for (axis 0 with )?size 3) r re r9 rL r% r& r- r. rD � IndexError)r r r r( ro r r r �test_take_fill_value s$ $$z TestTake.test_take_fill_valueN)r5 r6 r7 rf rp rs rw r r r r rb � s $rb c @ sD e Zd Zejdddgd�dd� �Zejdd� �Zd d � Zd d � ZdS ) �TestMaybeCastSliceBoundZ increasing� decreasingN)�paramsc C s |j S )N)�param)r �requestr r r � monotonic) s z!TestMaybeCastSliceBound.monotonicc C sV t ddd�}|dkr$|d d d� }n.|d krRtjdtjd�}tj�|� |�|�}|S )Nz1 Dayr )r+ ry r$ )rK )r r9 ZarangerM �random�shufflere )r r} r Ztakerr r r r - s zTestMaybeCastSliceBound.tdic C s� d}t jt|d�� |jddd� W d Q R X t jt|d�� |jddd� W d Q R X t jt|d�� |�dd d � W d Q R X d S )NzRcannot do slice indexing on TimedeltaIndex with these indexers \[foo\] of type str)r, rA �left)Zside)r- r. rm Z_maybe_cast_slice_boundZget_slice_boundZ slice_locs)r r ro r r r �'test_maybe_cast_slice_bound_invalid_str8 s z?TestMaybeCastSliceBound.test_maybe_cast_slice_bound_invalid_strc C s� |t d�|d�}d}tjt|d�� ||�dd � W d Q R X tjt|d�� ||�dd� W d Q R X tjt|d�� ||�d d� W d Q R X tjt|d�� ||�|d d� W d Q R X d S )Nr )�indexzRcannot do slice indexing on TimedeltaIndex with these indexers \[foo\] of type str)r, rA r$ r )�ranger- r. rm )r r Zframe_or_seriesZ indexer_sl�objro r r r �*test_slice_invalid_str_with_timedeltaindexF s zBTestMaybeCastSliceBound.test_slice_invalid_str_with_timedeltaindex) r5 r6 r7 r- Zfixturer} r r� r� r r r r rx ( s rx )r r r0 �numpyr9 r- Zpandasr r r r r r r r Zpandas._testingZ_testingr% r r; rJ rQ rb rx r r r r �<module> s ( =,/c
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка