Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/series/indexing/__pycache__/test_datetime.cpython-37.pyc
Назад
B �5�g6 � @ s d Z ddlmZmZ ddlZddlmZmZ ddlZddl Z ddl Z ddl m Z ddlZddlmZmZmZmZmZ ddlmZ dd� Zd d � Ze j�d d d g�dd� �Z dd� Z dd� Z dd� Z dd� Z dd� Z!dd� Z"d d � Z#d d � Z$d d!� Z%d"d#� Z&d$d%� Z'd&d'� Z(dS )(z< Also test support for datetime64[ns] in Series / DataFrame � )�datetime� timedeltaN)�gettz�tzutc)�index)� DataFrame�Series� Timestamp� date_range� period_rangec C s� t dtddd�tddd�d�} tt�t| ��| d�}|d dksDt�|d dksTt�|d dksdt�|td dd � dks|t�|ttd dd �� dks�t�tj t d d �� |d W d Q R X t � |dd� |td dd�td dd�� � d S )NzWOM-1FRIi� � i� )�freq�start�end)r �0 z1/2/2009z2009-1-2i� � z ^'2009-1-3'$)�matchz2009-1-3z3/6/2009z 2009-06-05� � � ) r r r �np�arange�len�AssertionErrorr �pytest�raises�KeyError�tm�assert_series_equal)�dti�s� r! �`/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/series/indexing/test_datetime.py�test_fancy_getitem s r# c C s� t dtddd�tddd�d�} tt�t| ��| d�}d|d< |d dksLt�d |d < |d d ksdt�d |d d �< |dd � d k�� s�t�d S )NzWOM-1FRIi� r i� )r r r )r ���r ���z1/2/2009���z 2009-06-05�6 )r r r r r r r �all)r r r! r! r" �test_fancy_setitem2 s r) � tz_source�pytzZdateutilc C s, | dkrt j}ndd� }d}td|d|d�d�}ttj�|�|d �}|�� }d |d < |d |d < t� ||� |�� }d |d < |d |d < t� ||� |�� }d |t dddd|d�d�< |d |t dddd|d�d�< t� ||� |�� }t dddd�� |d��}|� � }d ||< |d ||< t� ||� d S )Nr+ c S s | dkrt � S t| �S )N�UTC)r r )�xr! r! r" �<lambda>F � z2test_getitem_setitem_datetime_tz.<locals>.<lambda>�2 z1/1/1990�Hz US/Eastern)�periodsr �tz)r r z1990-01-01 09:00:00+00:00� z1990-01-01 03:00:00-06:00i� r � r, )�tzinfor z US/Central)r+ �timezoner r r �random�randn�copyr r r r � tz_localizeZ to_pydatetime)r* Ztzget�N�rng�ts�result�dtr! r! r" � test_getitem_setitem_datetime_tz@ s0 rA c C sJ d} t d| ddd�}ttj�| �|d�}|d }|d }||ksDt�|�� }d |d< |d |d< t�||� |dd � }|dd � }t�||� |�� }d |dd �< |dd � |dd �< t�||� d}d }||j |k|j |k@ }|dd � }t�||� d }d }||j |k|j |k@ }|dd � }t�||� d}t dddd�}t� t �� || }W d Q R X |d }||k�szt�|�� }t� t �� d |t dddd�< W d Q R X t� t �� |d |t dddd�< W d Q R X t�||� t� t ��& |t dddd�t dddd�� }W d Q R X |dd � }t�||� |�� }t� t ��& d |t dddd�t dddd��< W d Q R X t� t ��. |dd � |t dddd�t dddd��< W d Q R X t�||� t dddd�}t dddd�}d}t jt|d�� ||j |k|j |k@ W d Q R X tt dddd���|j�}tt dddd���|j�}||j |k|j |k@ }|dd � }t�||� ||j d }|d }||k�s�t�||j dd � }|dd � }t�||� |�� }d ||j dd � < |jdd � |jdd �< t�||� |d }|dd� }t�||� |�� }d |d< |dd� |d< t�||� d S )Nr0 z1/1/1990r1 z US/Eastern)r2 r r3 )r z1990-01-01 04:00:00r4 r z1990-01-01 07:00:00� z1990-01-01 04:00:00-0500z1990-01-01 07:00:00-0500z:Cannot compare tz-naive and tz-aware datetime-like objectsi� r � zJInvalid comparison between dtype=datetime64\[ns, US/Eastern\] and datetime)r z 1990-01-02� r )r r r r8 r9 r r: r r r r �assert_produces_warning� FutureWarningr r � TypeErrorr r; r6 �iloc) r< r= r> r? �expected�lb�rb�msgZnaiver! r! r" �"test_getitem_setitem_datetimeindexf s� * * 2 " rM c C s� d} t d| dd�}ttj�| �|d�}|d }|d }||ksBt�|�� }d|d< |d |d< t�||� |dd � }|dd � }t�||� |�� }d|dd �< |dd � |dd �< t�||� d}d }||j |k|j |k@ }|dd � }t�||� ||j d }|d }||k�s t�||j dd � }|dd � }t�||� |�� }d||j dd � < |j dd � |j dd �< t�||� d S ) Nr0 z1/1/1990r1 )r2 r )r z 1990-01-01 04r4 r z 1990-01-01 07rB ) r r r r8 r9 r r: r r r rH )r< r= r> r? rI rJ rK r! r! r" � test_getitem_setitem_periodindex� s@ rN c C s� t dd�} | �d�} tt| �| d�}td�}tjtt� t |��d�� || W d Q R X d||< || dkspt �tt| �| d�}|d d d� }tjtt� t |��d�� || W d Q R X d||< || dks�t �d S ) Nz1/1/2000z1/7/2000r )r z1/8/2000)r r r$ ) r �repeatr r r r r r �re�escape�reprr )r r Zstampr! r! r" �test_datetime_indexing s rS c C s | }|j �� }x�|D ]�}|| }|j |k}|j |k�� }|| }|dkrVt�||� nt�||d � |�� }d||< tt� |d|�|j d�}t�||� qW t ddd�} t j t t�t| ��d�� || W d Q R X d|t ddd�< |t ddd� dks�t�d S )Nr r )r i� r )r )r �unique�sumr r Zassert_almost_equalr: r r �wherer r r r rP rQ rR r ) Z(rand_series_with_duplicate_datetimeindexr> Zuniques�dater? �mask�totalrI �cp�keyr! r! r" �*test_indexing_with_duplicate_datetimeindex s& r\ c C s8 | � tdd� g }tdd�}tdd�}tddd d d �}d }x\t|�D ]P}|�|� |�|| � |�|| | � |�|| | | � |d | 7 }qDW tj�dt |�d d �}x |D ]}|| ||d < q�W t tj� t |�d�|t d�d�} |d } | j | } | | j k�st�| j| t | j| g �dk�s4t�d S )N� _SIZE_CUTOFFi� r )�secondsi � )� microsecondsi� � r � � iL r r r4 ZABCD)r �columns)�setattr�libindexr r �range�appendr r8 �randintr r r9 �listr r �loc) � monkeypatch�dates�secZhalf_sec�d�n�iZduplicate_positions�p�df�pos� timestampr! r! r" �!test_loc_getitem_over_size_cutoff; s* ru c C s� | � tdd� d}tdd|d�}|jjs,t�ttj� t |��|d�}|d }|| }||j ksbt�|| t |j |g �d ks�t�d S ) Nr] i� iL z1/1/2000�T)r r2 )r r r ) rd re r Z_engineZover_size_thresholdr r r r8 r9 r r rj )rk ro �idxr rs rt r! r! r" �+test_indexing_over_size_cutoff_period_index] s rx c s" t ddd�} ttj�t| ��| d�� t�� dd� � dd � � dd� g��x*� jD ] }� | }�| }||ksTt �qTW � �fdd �}|t dd�� t � t �� |t d d�� W d Q R X |t dd �� |t dd �� |t d d�� |t d d�� �d �� }� d }|j�d �|_t �||� d S )Nz 2011-01-01z 2011-01-15)r r )r r r4 ���c s: �| � � }|�� }� | }|j�d �|_t�||� d S )N)r: � sort_indexr � _with_freqr r )Zslobjr? rI )r> �ts2r! r" �compare~ s z(test_indexing_unordered.<locals>.comparez 2010-12-30z 2011-01-16z 2011-01-6z 2011-01-06z 2011-01-8z 2011-01-12Z2011)r r r r8 �randr �pd�concatr r �slicer rE rF rz r{ r )r= �trI r? r} r! )r> r| r" �test_indexing_unorderedq s& ( r� c C sh t tddd�ddd�} tt�t| ��| d�}|�tj�d��}|d }x |j D ]}|j dksNt �qNW d S )Ni� r ra �M)r2 r )r Z2005) r r r r r r Ztaker8 Z permutationr �yearr )r= r> r? r� r! r! r" �test_indexing_unordered2� s r� c C s� t dddd�} ttj�t| ��| d�}|d }d|_td|i�}t� t �� |d d }W d Q R X t� ||� d|d< |d }d|_d|j d < t� t �� |d d }W d Q R X t� ||� d S ) Nz2001-1-1ra r� )r2 r )r �2001�Ar )r� r� ) r r r r8 r~ r �namer r rE rF r rj )rw r> rI rr r? r! r! r" � test_indexing� s r� c C sp t dddd�} ttt| ��| d�}|d }t�||� t dddd�} ttt| ��| d�}|d }t�||� d S ) Nz2013-05-31 00:00z2013-05-31 23:00r1 )r r r )r z2013-05z2013-05-31 23:59�S)r r rf r r r )rw r> rI r! r! r" �)test_getitem_str_month_with_datetimeindex� s r� c C sL t d�t tddddddd��g} ttt| ��| d�}|d }t�||� d S ) Nz2013-05-31 00:00i� r � � �; i?B )r Z2013)r r r rf r r r )rw r> rI r! r! r" �(test_getitem_str_year_with_datetimeindex� s r� c C s� t tj�dd�dddddgtdddd d �d �} tjtd d �� | d W d Q R X d}tjt|d �� | | jd W d Q R X d S )Nr �open�high�low�close�volumez2012-01-02 18:01:00z US/Centralr )r2 r3 r )rc r z^'2012-01-02 18:01:02'$)r z2012-01-02 18:01:02zBTimestamp\('2012-01-02 18:01:02-0600', tz='US/Central', freq='S'\)r ) r r r8 r~ r r r r r )rr rL r! r! r" �*test_getitem_str_second_with_datetimeindex� s r� ))�__doc__r r rP Z dateutil.tzr r �numpyr r r+ Z pandas._libsr re Zpandasr r r r r r Zpandas._testingZ_testingr r# r) �markZ parametrizerA rM rN rS r\ ru rx r� r� r� r� r� r� r! r! r! r" �<module> s2 &m- "%
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка