Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/indexes/datetimes/__pycache__/test_partial_slicing.cpython-37.pyc
Назад
B �5�g: � @ sv d Z ddlmZ ddlZddlZddlZddlmZmZm Z m Z m Z m Z m Z ddlmZ ddlmZ G dd� d�ZdS )z& test partial slicing on Series/Frame � )�datetimeN)� DataFrame� DatetimeIndex�Index�Series� Timedelta� Timestamp� date_range)� IndexingErrorc @ sd e Zd Zdd� Zdd� Zej�ddddd d d g�d d � �Zdd� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zej�d$ed%�ed&d'd'�e�d(d)�g�ej�d*ejd+d,d,d,gfejd+d+d,d,gfejd,d+d,d,gfej d,d,d,d+gfg�d-d.� ��Z ej�d/d0ed0�ed0�� � g�ej�d1d2ed2�ed2�� � g�d3d4� ��Z d5d6� Z d7S )8� TestSlicingc C s� t ttd���}dddddg}t|�}||d< t ttd��|d��}t�||� t d d d d gitd d d�d d d� d�}t d d itdd d�d d d� d�}t�|jd |� d S )N� z 2018-01-02z 2017-02-10z 2016-03-10z 2015-03-15z 2014-03-16�date)r r �A� � � Z20170101)�periods���)�indexZ20170103z 2017-01-03)r �list�ranger �tm�assert_frame_equalr �loc)�self�dfZ date_listZ date_index�expected� r �i/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/indexes/datetimes/test_partial_slicing.py� test_monotone_DTI_indexing_bug s ("z*TestSlicing.test_monotone_DTI_indexing_bugc C s� t dtddd�dd�}tt�t|��|d�}|d }||jjdk }t� ||� t tj � t|�d�|d�}|j d }||jjdk }t�||� d S ) N�Bi� r i� )�freq�startr )r Z2005r )r r r �np�arange�lenr �yearr �assert_series_equalr �random�randr r )r �dti�s�resultr r r r r �test_slice_year- s zTestSlicing.test_slice_year� partial_dtimeZ2019Z2019Q4zDec 2019z 2019-12-31z 2019-12-31 23z2019-12-31 23:59c C sD t dddd�}ttd�|d�}|| }|jd d� }t�||� d S )Nz 2019-12-31 23:59:55.999999999� r+ )r r! )r r )r r r �ilocr r' )r r. r* �serr, r r r r �#test_slice_end_of_period_resolution: s z/TestSlicing.test_slice_end_of_period_resolutionc C st t dtddd�dd�}tt�t|��|d�}t|d �d ks@t�ttj� t|�d �|d�}t|j d �d kspt�d S ) N�Di� � r i� )r! r" r )r Z2001Q1�Z r Z1Q01) r r r r# r$ r% �AssertionErrorr r( r) r )r r* r+ r r r r �test_slice_quarterN s z TestSlicing.test_slice_quarterc C s� t dtddd�dd�}tt�t|��|d�}t|d �dks@t�ttj� t|�d �|d�}t|j d �dkspt�t � |d |d � d S ) Nr3 i� r i� )r! r" r )r z2005-11� r z11-2005) r r r r# r$ r% r6 r r( r) r r r' )r r* r+ r r r r �test_slice_monthW s z TestSlicing.test_slice_monthc C s� t dtddd�dd�}tt�t|��|d�}|dd� }|d d � }t�||� |dd � }|d d � }t�||� |d d� }|d d � }t�||� |d }||jd ks�t �t j t d d�� |d W d Q R X d S )Nr3 i� r i� )r! r" r )r z2005-05z2006-02Z20050501Z20060228z2005-1-1r z^'2004-12-31'$)�matchz 2004-12-31) r r r r# r$ r% r r' r0 r6 �pytest�raises�KeyError)r �rngr+ r, r r r r �test_partial_slicea s z TestSlicing.test_partial_slicec C sp t dtddd�dd�}tt�t|��|d�}|d }t�||jd d � � t j t d d �� |d W d Q R X d S ) N�Hi� r � i� )r! r" r )r z 2005-1-31� z^'2004-12-31 00'$)r: z 2004-12-31 00) r r r r# r$ r% r r' r0 r; r< r= )r r> r+ r, r r r �test_partial_slice_dailyw s z$TestSlicing.test_partial_slice_dailyc C s� t dtdddddd�dd�}tt�t|��|d�}|d }t�||jd d � � |d }t�||jd d � � |d |jd ks�t �t j t dd�� |d W d Q R X d S )N�Ti� r � r i� )r! r" r )r z2005-1-1�� z 2005-1-1 20�< z2005-1-1 20:00z^'2004-12-31 00:15'$)r: z2004-12-31 00:15) r r r r# r$ r% r r' r0 r6 r; r< r= )r r> r+ r, r r r �test_partial_slice_hourly� s z%TestSlicing.test_partial_slice_hourlyc C s� t dtdddddd�dd�}tt�t|��|d �}|d }t�||jd d � � |d }t�||jd d � � |t d � |jd ks�t �t j t dd�� |d W d Q R X d S )N�Si� r � �; r i� )r! r" r )r z2005-1-1 23:59rG z2005-1-1z2005-1-1 23:59:00z^'2004-12-31 00:00:00'$)r: z2004-12-31 00:00:00)r r r r# r$ r% r r' r0 r r6 r; r< r= )r r> r+ r, r r r �test_partial_slice_minutely� s z'TestSlicing.test_partial_slice_minutelyc C s� t tdddddddd�ddd �}tt�d�|�}t�|d |jd d � � t�|d |jd d � � t�|d |jd d � � t�|d |jd d � � |td� |jd ks�t �t j t dd�� |d W d Q R X d S )Ni� r r rK i6B )� microsecondrE ZUS)r" r r! z2005-1-1 00:00r/ z2005-1-1 00:00:59z2005-1-1 00:01z2005-1-1 00:01:00z2005-1-1 00:00:59.999990z2005-1-1 00:00:00)r: ) r r r r# r$ r r' r0 r r6 r; r< r= )r r> r+ r r r �#test_partial_slice_second_precision� s z/TestSlicing.test_partial_slice_second_precisionc C s� ddddddg}ddd d d d g}�x�t |d d � d �D �]�\}}td| �}tdddddd�}t|| ||| g�}dd dg}td|i|tjd�} | jj|ks�t �xzt ||�D ]l\} } | � || �} | d | } t | tj�s�t �| | ks�t �d| � d�}t jt|d�� | | W d Q R X q�W x�|d |� D ]�}x�dtd d�gdtdd �ggD ]n\}}|| � |�} | d | } | d | } t�| | � t�t�� | | } W d Q R X | | } t�| | � �qNW �q,W x�||d d � D ]n}|d � |�} | d | } t | tj��st �| d k�st �d| � d�}t jt|d�� | | W d Q R X �q�W x�tt ||��|d d � D ]|\}}|d td| � }|� |�} d| � d�}t jt|d�� | d | W d Q R X t jt|d�� | | W d Q R X �qfW q6W d S )Nz%Yz%Y-%mz%Y-%m-%dz %Y-%m-%d %Hz%Y-%m-%d %H:%Mz%Y-%m-%d %H:%M:%Sr& �month�day�hour�minute�secondr z1 i� r r r �a)Zdtypez^'z'$)r: )� enumerater r r r r# Zint64r � resolutionr6 �zip�strftime� isinstancer; r< r= �slicer r' �assert_produces_warning� FutureWarningr r )r �formats� resolutionsZrnumrV �unitZmiddater �valuesr � timestampr Z ts_stringr, �msg�fmt�elementZtheslice�res�tsr r r � test_partial_slicing_dataframe� s` & $ z*TestSlicing.test_partial_slicing_dataframec C st t ddddgddddgdddd gd �td d d d �d�}|jddgdd�}t dggtdgdd�dgd�}|jd }t�||� |jtd d d�ddf }|jd }t�||� d}t j t |d�� |jd W d Q R X t t j �d d �td d d �d��� }|d d � �� }|d }|td � }t�||� |td � }|d }t�||� t |�}|�d �}|jtd � }t�||� d S )!N�ACCT1ZACCT2�ABCZMNPZXYZr r r � )�ACCOUNT�TICKER�valz2013-06-19 09:30:00Z5T)r r! )r rk rl T)�append)�namerm )r �columns)z2013-06-19 09:30:00rh )�tz)z2013-06-19 09:30:00rh ri zToo many indexers)r: )z 2013-06-19rh ri i� z2000-1-1)r r z2000-1-4)r r Z set_indexr r r r r r' r; r< r r# r( r) �stack�copyZxs) r r Zdf_multir r, rb r+ �s2Zdf2r r r �$test_partial_slicing_with_multiindex� s< z0TestSlicing.test_partial_slicing_with_multiindexc C s� t t�d�tddd��}|dddg }|jd d� }td�}t�t�� |dd � }W d Q R X t� ||� t j t d d �� ||d � W d Q R X t�t�� |j dd � }W d Q R X t� ||� t j t d d �� |j |d � W d Q R X d S ) Nr/ z 2014-01-01)r r r rj r z 2014-01-10z"Timestamp\('2014-01-10 00:00:00'\))r: )r r# r$ r r0 r r r[ r\ r' r; r< r= r )r r1 Z nonmonotonicr ra r, r r r �.test_partial_slice_doesnt_require_monotonicity3 s z:TestSlicing.test_partial_slice_doesnt_require_monotonicityc C sT t dgtdd�d�}|jtddd�d � }t�||� |jdd � }t�||� d S )N�1z2016-10-01T00:00:00z2016-10-01T23:59:59)rp r i� r/ r )r r r r r r )r r r, r r r � test_loc_datetime_length_oneH s z(TestSlicing.test_loc_datetime_length_one� datetimelike�20130101i� r z2013-01-01T00:00�nsz op,expectedTFc C sJ t dtd�td�tjtd�gi�}||j|�}t|dd�}t�||� d S )Nr Z20120101rz Z20130103)ro )r r r# �nanr r r r' )r ry �opr r r, r r r � test_selection_by_datetimelikeT s z*TestSlicing.test_selection_by_datetimeliker" z2018-12-02 21:50:00+00:00�endz2018-12-02 21:52:00+00:00c C s� t dddd�}td|dgd�}|||� }|jdd�d d �f }t�||� t|�}t|�}tjtd d �� |||d d � d � W d Q R X tjtd d �� |� d �}|||� W d Q R X d S )Nz2018-12-02 14:50:00-07:00Z1min)r" r r! r r )r rp r r zBoth dates must)r: ���z1:00zThe index must be timezone) r r r0 r r �strr; r< � ValueErrorZ tz_localize)r r" r �idxr r, r r r r �,test_getitem_with_datestring_with_UTC_offsetv s " z8TestSlicing.test_getitem_with_datestring_with_UTC_offsetc C sT t dtd�itdddd�d�}ttd�tdddd�dd�}|jd }t�||� d S ) Nr rB �2000�M)r r! )r � )r ro )r� r )r r r r r r r' )r r r r, r r r �test_slice_reduce_to_series� s z'TestSlicing.test_slice_reduce_to_seriesN)!�__name__� __module__� __qualname__r r- r; �markZ parametrizer2 r7 r9 r? rC rH rL rN rg ru rv rx r r r# Z datetime64�operator�lt�le�eq�gtr~ Z to_pydatetimer� r� r r r r r sV P4 r )�__doc__r r� �numpyr# r; Zpandasr r r r r r r Zpandas._testingZ_testingr Zpandas.core.indexingr r r r r r �<module> s $
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка