Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/indexes/datetimes/__pycache__/test_indexing.cpython-37.pyc
Назад
B �5�g�l � @ s" d dl mZm Z mZmZ d dlZd dlZd dlmZ d dl Z d dl m Z m Z m Z mZmZmZ d dlmZ d dlmZmZ e ddd�e ddd� ZZG d d � d �ZG d d � d �ZG d d� d�ZG dd� d�Z G dd� d�Z G dd� d�Z G dd� d�Z G dd� d�Z G dd� d�Z!G dd � d �Z"dS ) � )�date�datetime�time� timedeltaN)�InvalidIndexError)� DatetimeIndex�Index� Timestamp� bdate_range� date_range�notna)�BDay�CDayi� � i� c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )� TestGetItemc C s8 t dddddd�}|d }|�|�s(t�||k s4t�d S )Nz 2011-01-01z 2011-01-31�Dz Asia/Tokyo�idx)�freq�tz�name.)r �equals�AssertionError)�selfr �result� r �b/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/indexes/datetimes/test_indexing.py� test_ellipsis! s zTestGetItem.test_ellipsisc C sD t ddd�}t ddd�}t||ddd�}|dd � j|jks@t�d S ) Nz2013-07-01 00:00:00zAmerica/Los_Angeles)r z2013-07-02 00:00:00�HZ timebucket)r r r )r r r r )r �st�etZdrr r r � test_getitem_slice_keeps_name+ s z)TestGetItem.test_getitem_slice_keeps_namec C sZ t ddddd�}t dddddd�}�x0||gD �]"}|d }|td|jd �ksRt�|dd � }t dd d|jdd�}t�||� |j|jks�t�|dd d � }t ddd|jdd�}t�||� |j|jks�t�|ddd� }t ddd|jdd�}t�||� |j|jk�s t�|dd d� }td ddddgd|jdd�}t�||� |j|jks.t�q.W d S ) Nz 2011-01-01z 2011-01-31r r )r r z Asia/Tokyo)r r r r )r � z 2011-01-05� � z 2011-01-09�2Di������ z 2011-01-12z 2011-01-24Z3D� ���z 2011-01-04z 2011-01-03z 2011-01-02z-1D)r r r r �tm�assert_index_equalr r )r �idx1�idx2r r �expectedr r r � test_getitem2 s: zTestGetItem.test_getitemc C s� t tt�}|d d� }t|�tj�d d� dd�}t�||� |j |j ksNt �|j |j ks^t �|d d d� }|j t � d ks�t �|dddddg }t |�dks�t �t |t�s�t �|j d ks�t �|d |t�d� ks�t �d S ) Nr! �B)r r' r& r# r r )r �START�ENDr �view�np�ndarrayr) r* r r r �len� isinstance�int_)r �rng�smaller�exp�sliced� fancy_indexedr r r �test_dti_business_getitem[ s z%TestGetItem.test_dti_business_getitemc C sR t tt�}t�t�� |d d �d f }W d Q R X |jd d �d f }t�||� d S )N)r r0 r1 r) �assert_produces_warning� FutureWarning�values�assert_numpy_array_equal)r r8 r@ r- r r r �/test_dti_business_getitem_matplotlib_hackaroundo s z;TestGetItem.test_dti_business_getitem_matplotlib_hackaroundc C s� t ttdd�}|d d� }t|�tj�d d� dd�}t�||� |j |j ksRt �|j |j ksbt �|d d d� }|j t � d ks�t �|dddddg }t |�dks�t �t |t�s�t �|j d ks�t �|d |t�d� ks�t �d S ) N�C)r r! r' r& r# r r )r r0 r1 r r2 r3 r4 r) r* r r r r5 r6 r7 )r r8 r9 r: r; r< r r r �test_dti_custom_getitemw s z#TestGetItem.test_dti_custom_getitemc C sV t ttdd�}t�t�� |d d �d f }W d Q R X |jd d �d f }t�||� d S )NrC )r )r r0 r1 r) r> r? r@ rA )r r8 r@ r- r r r �-test_dti_custom_getitem_matplotlib_hackaround� s z9TestGetItem.test_dti_custom_getitem_matplotlib_hackaroundc C sv t dddd�}|dddg }|d }|d }|d }|td �ksDt�|td �ksTt�|td �ksdt�|jd ksrt�d S ) Nz1/1/2005z 12/1/2005�M)�start�endr r r& r! r r# z 2/28/2005z 4/30/2005z 6/30/2005)r r r r )r �dtiZdti2Zv1Zv2Zv3r r r �test_getitem_int_list� s z!TestGetItem.test_getitem_int_listN) �__name__� __module__� __qualname__r r r. r= rB rD rE rJ r r r r r s )r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) � TestWherec C s` t ddddd�}dddg}t|d |d |d gd dd �}|�||d d d � �}t�||� d S ) N�20130101r& r r )�periodsr r TFr r )r r r( )r r �wherer) r* )r rI �condr- r r r r � test_where_doesnt_retain_freq� s z'TestWhere.test_where_doesnt_retain_freqc C s� t dddd�}x6tjtjgD ]&}|jt|�|d�}|}t�||� q W |� � }t tjtjg|dd � � � �}|�t|�|�}t�||� |� � }t tjtjg|dd � � � �}|�t|�|j �}t�||� d S )NrO r& z US/Eastern)rP r )�otherr# ) r r3 �nan�pd�NaTrQ r r) r* �copyr �tolistZ_values)r �i�arrr r- �i2r r r �test_where_other� s zTestWhere.test_where_otherc C s� t dddd�}|dd � �� }ttjtjg| �}t|�}|�||j�}ttjjtjjg| t d�}t � ||� |� d �}|�||�}t|d |d g|dd � �� t d�}t � ||� |� d �� d �}|�||�}t|d |d g| t d�}t � ||� |j�d �} |�|| �}t| d | d g| t d�}t|d tj��sHt�t � ||� |�||j�}ttjjtjjg| t d�}t|d t��s�t�t � ||� tjd d �} |�|| �}t| | g| t d�}|d | k�s�t�t � ||� d S ) NrO r& z US/Eastern)rP r r# )�dtyper r r ztimedelta64[ns]r' )�days)r rY r rV rW r rQ r@ Zasm8�objectr) r* � tz_localizeZ to_periodZasi8r2 r6 r3 � timedelta64r �value�int� Timedelta) r rI �tailr\ �maskr r- Znaive�piZtda�tdr r r �test_where_invalid_dtypes� s: ( z#TestWhere.test_where_invalid_dtypesc C st |}t dd|d�}t�dddg�}t�dd�}t|d ||d gtd �}|d |ksXt�|�||�}t� ||� d S ) Nz 2013-01-01r& )rP r TFrW �nsr r# )r^ r ) r r3 �arrayrb r r` r rQ r) r* )r �tz_aware_fixturer rI rR Ztdnatr- r r r r �test_where_mismatched_nat� s z#TestWhere.test_where_mismatched_natc C sv t dddd�}|�t|��}|}t�||� |�� }ttjtjg|dd � � � �}|�t|��}|}t�||� d S )NrO r& z US/Eastern)rP r r# ) r rQ r r) r* rX r rV rW rY )r rZ r r- r\ r r r � test_where_tz� s zTestWhere.test_where_tzN)rK rL rM rS r] rj rn ro r r r r rN � s * rN c @ sH e Zd Zdd� Zdd� Zej�ddddg�d d � �Zd d � Z d d� Z dS )�TestTakec C s� t ddddd�}t dddddd�}�x|||gD �]n}|�dg�}|td|jd �ksVt�|�dd d g�}t dd d|jdd�}t�||� |j|jks�t�|�dd d g�}t ddd|jdd�}t�||� |j|jks�t�|�dd d g�}t ddd|jdd�}t�||� |j|jk�st�|�dd dg�}tdd dgd |jdd�}t�||� |jd k�s\t�|�dd dg�}tdd dgd |jdd�}t�||� |jd ks.t�q.W d S )Nz 2011-01-01z 2011-01-31r r )r r z Asia/Tokyo)r r r r )r r r# z 2011-01-03r' z 2011-01-05r$ � z 2011-01-08z 2011-01-02z-3Dr& r! z 2011-01-04z 2011-01-06���z 2011-01-29) r �taker r r r) r* r r )r r+ r, r r r- r r r � test_take sJ 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 )Nz 2011-01-01z 2011-01-31r r )r r r � r! � r" � � r& z1take\(\) got an unexpected keyword argument 'foo')�matchr# )�fooz$the 'out' parameter is not supported)�outz%the 'mode' parameter is not supportedZclip)�mode)r �pytest�raises� TypeErrorrs � ValueError)r r �indices�msgr r r �test_take_invalid_kwargs6 s z!TestTake.test_take_invalid_kwargsr Nz US/Easternz Asia/Tokyoc C s� t dddd�t dddd�t dddd�t dddd�g}tddd |d d �}t|d d |d �}|�d dddg�}|d dddg }xX||gD ]L}t�||� t|t�s�t�|jd ks�t�|j |j ks�t�|j |j ks�t�q�W d S )Ni� r � rx � � z2010-01-01 09:00z2010-02-01 09:00r r )rG rH r r r )r r r r! ru � � ) r r r rs r) r* r6 r r r r )r r �datesr r- Ztaken1Ztaken2Ztakenr r r � test_take2G 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 )Nz 2011-01-01z 2011-02-01z 2011-03-01�xxx)r r r r( T)� fill_valuerW F)� allow_fillr� zJWhen allow_fill=True and fill_value is not None, all indices must be >= -1)ry ���r% z out of bounds) r rs r3 rl r) r* r} r~ r� � IndexError)r r r r- r� r r r �test_take_fill_valuec s$ $$z TestTake.test_take_fill_valuec C sb t dddgddd�}|�t�ddd g��}t dddgddd�}t�||� |jt�ddd g�d d �}t ddd gddd�}t�||� |jt�ddd g�d d d�}t dddgddd�}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 )Nz 2011-01-01z 2011-02-01z 2011-03-01r� z US/Eastern)r r r r r( T)r� rW F)r� r� zJWhen allow_fill=True and fill_value is not None, all indices must be >= -1)ry r� r% z out of bounds) r rs r3 rl r) r* r} r~ r� r� )r r r r- r� r r r �"test_take_fill_value_with_timezone� s, $$z+TestTake.test_take_fill_value_with_timezone) rK rL rM rt r� r} �mark� parametrizer� r� r� r r r r rp s 4 rp c @ s� e Zd Zej�dddddg�dd� �Zdd � Zd d � Zd d � Z dd� Z ej�de � d�e � d�e d�g�dd� �Zdd� Zdd� ZdS )� TestGetLoc�methodN�pad�backfill�nearestc C s� t ddd�}|�|d |�dks$t�|�|d �� |�dks@t�|�t|d �|�dks\t�|d k r�|j|d |t�d�d�dks�t�d S )Nz 2000-01-01r& )rP r z0 days)� tolerance)r �get_locr � to_pydatetime�strrV re )r r� r r r r � test_get_loc_method_exact_match� s z*TestGetLoc.test_get_loc_method_exact_matchc C s, t ddd�}|jddd�dks"t�|jddd�dks8t�|jddd d �dksPt�|jddt�d �d �dksnt�|jddt�dd �d �dks�t�|jddtd�d �dks�t�tj t d d�� |jdddd � W d Q R X tj t dd�� |jdddd � W d Q R X tj t dd��. |jddt�d�� � t�d�� � gd � W d Q R X |jddd�t dd�k�s^t�|jddd�t dd�k�s|t�|jddd�dk�s�t�|jddd�dk�s�t�tj t dd�� |jddd� W d Q R X tj t d d�� |jdd d� W d Q R X tj t d d�� |�d � W d Q R X tj td d�� |�t d�� W d Q R X t�dd!g�}|jd"dd�dk�snt�|jd#dd�dk�s�t�|jddd�t dd�k�s�t�t dd$d%d&�}tj|�td'��t�d'g�d(d)� tj|�td'd*��t�g �d(d)� d+}tj t|d�� |jtd'd*�dd� W d Q R X d S ),Nz 2000-01-01r& )rP r� )r� r z 2000-01-01T12r z1 day)r� r� Z1Dr z unit abbreviation w/o a number)ry rz z'2000-01-01T03'z 2000-01-01T03z2 hoursz+tolerance size must match target index sizeZ1day�2000z2000-01Z1999Z2001r# z'1999'r� z'2001'r� z'foobar'Zfoobarzslice\(None, 2, None\)z 2000-01-04z 2000-01-02z 2000-01-03� r )rP r r� F)Z check_dtype� z:cannot yet lookup inexact labels when key is a time object)r r� r rV re r3 rb r r} r~ r� �KeyError�to_timedelta64�slicer � to_datetimer) rA r rl �NotImplementedError)r r r� r r r � test_get_loc� sZ zTestGetLoc.test_get_locc C sD t dddd�}ttjg�}|�|�}tjg tjd�}t� ||� d S )Nr� �+ iH7 )�minute�second� microsecond)r^ ) r r rV rW r� r3 rl �intpr) rA )r ZticrI �locr- r r r �test_get_loc_time_nat� s z TestGetLoc.test_get_loc_time_natc C sH t tddd�tddd�dd�}tddd�}|j|dd �}|d ksDt�d S ) Nz2019-12-12 00:00:00z US/Eastern)r z2019-12-13 00:00:00Z5s)r z2019-12-12 10:19:25r� )r� i )r r r� r )r rI �keyr r r r �test_get_loc_tz_aware s z TestGetLoc.test_get_loc_tz_awarec C s� t ddg�}|�tj�dks t�|�d �dks2t�|�tj�dksFt�|�tj�dksZt�|�t�d��dksrt�t j t dd�� |�t� d�� W d Q R X d S )Nz1/3/2000rW r )ry ) r r� rV rW r r3 rU ZNAZ datetime64r} r~ r� rb )r �indexr r r �test_get_loc_nat s zTestGetLoc.test_get_loc_natr� r r c C s8 t ddd�}d}tjt|d�� |�|� W d Q R X d S )Nz 1970-01-01r" )rP z,Cannot index DatetimeIndex with [Tt]imedelta)ry )r r} r~ r r� )r r� rI r� r r r �"test_get_loc_timedelta_invalid_key s z-TestGetLoc.test_get_loc_timedelta_invalid_keyc C s2 t dg�}tjtdd�� |�d� W d Q R X d S )Nz1/3/2000r� )ry z1/1/2000)r r} r~ r� r� )r r� r r r �!test_get_loc_reasonable_key_error$ s z,TestGetLoc.test_get_loc_reasonable_key_errorc C s. t dd�}|�d�}tdd�}||ks*t�d S )Nz1/1/2000z1/1/2010Z2009i� iE )r r� r� r )r r8 r r- r r r �test_get_loc_year_str* s z TestGetLoc.test_get_loc_year_str)rK rL rM r} r� r� r� r� r� r� r� rV re r r� r� r� r r r r r� � s E ,r� c @ sR e Zd Zdd� Zej�ddddgdddgdddgdddgddd gg�d d � �Zd S ) � TestContainsc C s, t ddddd�}t||g�}||ks(t�d S )Ni� r� r! � r� )r r r )r �dZixr r r �!test_dti_contains_with_duplicates3 s z.TestContains.test_dti_contains_with_duplicates�valsr r r( �2015Z2016Z2014c C s t |�}|d |kst�d S )Nr )r r )r r� r r r r �test_contains_nonunique8 s z$TestContains.test_contains_nonuniqueN)rK rL rM r� r} r� r� r� r r r r r� 2 s r� c @ s� e Zd Zdd� Zdd� Zej�deddd�e d�ge d �eddd �gg�d d � �Z ej�d eddd�e d �gddgfe d �eddd�gddgfeddd�eddd�gddgfg�dd� �Z dd� Z dS )�TestGetIndexerc C s: t ddd�}|�|�dd� ��}|�|�}t�||� d S )Nz1/1/2000r� )rP c S s | � � S )N)r )�xr r r �<lambda>L � z;TestGetIndexer.test_get_indexer_date_objs.<locals>.<lambda>)r � get_indexer�mapr) rA )r r8 r r- r r r �test_get_indexer_date_objsI s z)TestGetIndexer.test_get_indexer_date_objsc C s� t ddd�}tjdddgtjd�}t�|�|�|� |d t�dd d g� }t�|�|d �tjd ddgtjd�� t�|�|d �tjdddgtjd�� t�|�|d�tjdddgtjd�� t�|j|dt� d�d�tjdd dgtjd�� t� d�t� d�t� d�� � g}t�|j|ddd� |D �d�tjdd dgtjd�� t� d�� � t� d�� � dg}d}t j t |d�� |j|d|d� W d Q R X t j t dd�� |j|dg ddd� W d Q R X d S )Nz 2000-01-01r& )rP r r r# )r^ z-1 hourz12 hoursz 1 day 1 hourr� r( r� r� z1 hour)r� c S s g | ]}t �|��qS r )r3 rb )�.0r� r r r � <listcomp>j s z3TestGetIndexer.test_get_indexer.<locals>.<listcomp>z2 hourrz z*Could not convert 'foo' to NumPy timedelta)ry zabbreviation w/o a number)r� r� )r r3 rl r� r) rA r� rV Z to_timedeltare r� r} r~ r� )r r r: �targetZtol_rawZtol_badr� r r r �test_get_indexerP s8 """ z TestGetIndexer.test_get_indexerr� i� r z 2020-01-02z 2020-01-01r# c C sB t td�td�g�}|�|�}tjddgtjd�}t�||� d S )Nz 2020-01-01z 2020-01-02r r )r^ )r r r� r3 rl r� r) rA )r r� r@ r r- r r r � test_get_indexer_mixed_dtypesy s z,TestGetIndexer.test_get_indexer_mixed_dtypesztarget, positionsi' r( r c C s> t td�td�g�}|�|�}tj|tjd�}t�||� d S )Nz 2020-01-01z 2020-01-02)r^ )r r r� r3 rl r� r) rA )r r� Z positionsr@ r r- r r r �#test_get_indexer_out_of_bounds_date� s z2TestGetIndexer.test_get_indexer_out_of_bounds_datec C sL t dddd�}|dddg }d}tjt|d �� |j|d d � W d Q R X d S ) Nz1/1/2000z3/1/2000r/ )r r r r# z0index must be monotonic increasing or decreasing)ry r� )r� )r r} r~ r� r� )r r8 Zrng2r� r r r �*test_get_indexer_pad_requires_monotonicity� s z9TestGetIndexer.test_get_indexer_pad_requires_monotonicityN) rK rL rM r� r� r} r� r� r r r� r� r� r r r r r� H s ) , r� c @ s e Zd Zdd� Zdd� ZdS )�TestMaybeCastSliceBoundc C sR t dddd�}|�dd�}td�}||ks.t�|�dd�}td �}||ksNt�d S ) NZ1Hr r� )r rP rH z 2015-01-02�rightz 2015-01-02 23:59:59.999999999�leftz2015-01-02 00:00:00)r �_maybe_cast_slice_boundr r )r Z empty_idxr� r: r� r r r �"test_maybe_cast_slice_bounds_empty� s z:TestMaybeCastSliceBound.test_maybe_cast_slice_bounds_emptyc C s0 t ddg�}|�dd�}td�}||ks,t�d S )NZ2017z 2017-01-01r� )r r� r r )r r r r- r r r �)test_maybe_cast_slice_duplicate_monotonic� s zATestMaybeCastSliceBound.test_maybe_cast_slice_duplicate_monotonicN)rK rL rM r� r� r r r r r� � s r� c @ s e Zd Zdd� ZdS )�TestDatetimeIndexc C s t ddd�}t�dd�}tj||d�}|d }tjtdd ��( t� t �� |� ||� W d Q R X W d Q R X t� t �� |� ||�}W d Q R X |d ks�t �t� t �� |� ||� � �}W d Q R X |d ks�t �t� t �� |� ||�� �}W d Q R X |d ks�t �d S ) Nz 2016-01-01r& )rP ru rv )r� r zhas no attribute '_values')ry rq )r r3 ZarangerV ZSeriesr} r~ �AttributeErrorr) r> r? � get_valuer r� Z to_datetime64)r rI r[ Zserr� r r r r �test_get_value� s z TestDatetimeIndex.test_get_valueN)rK rL rM r� r r r r r� � s r� c @ s� e Zd Zej�deeeg�ej�ddddg�ej�dddg�d d � ���Z ej�deeeg�ej�ddddg�ej�d d d g�ej�dddg�dd� ����Z ej�deeeg�ej�ddddg�dd� ��Z dS )�TestGetSliceBounds�box�kind�getitemr� Nzside, expected)r� r' )r� r! c C sh |}t dd��|�}|dddd�}|d kr.d nt} t�| �� |j|||d�} W d Q R X | |ksdt�d S )Nz 2000-01-03z 2000-02-11i� r rq )�year�month�day)r� �side)r ra r? r) r> �get_slice_boundr ) r r� r� r� r- rm r r� r� �warnr r r r �%test_get_slice_bounds_datetime_within� s z8TestGetSliceBounds.test_get_slice_bounds_datetime_withinr� r� r� zyear, expected)i� r )i� r� c C sh |}t dd��|�}||ddd�} |d kr.d nt} t�| �� |j| ||d�} W d Q R X | |ksdt�d S )Nz 2000-01-03z 2000-02-11r rq )r� r� r� )r� r� )r ra r? r) r> r� r ) r r� r� r� r� r- rm r r� r� r� r r r r �&test_get_slice_bounds_datetime_outside� s z9TestGetSliceBounds.test_get_slice_bounds_datetime_outsidec C sp |}t ddg��|�}|ddd�}|d kr.d nt}t�|�� |�||ddd��}W d Q R X d} || kslt�d S )Nz 2010-01-01z 2010-01-03i� r r# )r r )r ra r? r) r> Z slice_locsr ) r r� r� rm r r� r� r� r r- r r r �test_slice_datetime_locs� s z+TestGetSliceBounds.test_slice_datetime_locs) rK rL rM r} r� r� r r r r� r� r� r r r r r� � s r� c @ s e Zd Zdd� ZdS )�TestIndexerBetweenTimec C sP t dddd�}d}tjt|d��& |�tddd d�tddd d �� W d Q R X d S ) Nz1/1/2000z1/5/2000Z5min)r zGCannot convert arg \[datetime\.datetime\(2010, 1, 2, 1, 0\)\] to a time)ry i� r r# r! )r r} r~ r� Zindexer_between_timer )r r8 r� r r r �test_indexer_between_time s z0TestIndexerBetweenTime.test_indexer_between_timeN)rK rL rM r� r r r r r� s r� )#r r r r �numpyr3 r} Z pandas.errorsr ZpandasrV r r r r r r Zpandas._testingZ_testingr) Zpandas.tseries.offsetsr r r0 r1 r rN rp r� r� r� r� r� r� r� r r r r �<module> s* _ ' Y4
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка