Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/indexes/datetimes/__pycache__/test_setops.cpython-37.pyc
Назад
B �5�g�O � @ s� d dl m Z d dlZd dlZd dlm mZ d dlZ d dlm Z m Z m Z m Z mZmZmZ d dlmZ d dlmZmZmZ e ddd�e ddd� ZZG dd � d �ZG d d � d �ZG d d � d �Z dS )� )�datetimeN)� DataFrame� DatetimeIndex�Index� Int64Index�Series� bdate_range� date_range)� BMonthEnd�Minute�MonthEndi� � i� c @ s* e Zd ZddddddgZdd� Zej�d ej e e g�d d � �Z ej�d e�d d� �Z dd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zej�d ddddg�d!d"� �Zej�d#d$d%g�d&d'� �Zd(d)� Zej�d e�d*d+� �Zd,d-� Zd.d/� Z ej�d dddg�d0d1� �Z d2d3� Z dS )4�TestDatetimeIndexSetOpsN�UTCz Asia/Tokyoz US/Easternzdateutil/Asia/Singaporezdateutil/US/Pacificc C s@ t �d�}|d d� }|dd � }|j||d�}t �||� d S )N� � )�sort)�tm� makeDateIndex�union�assert_index_equal)�selfr � everything�first�secondr � r �`/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/indexes/datetimes/test_setops.py� test_union2( s z#TestDatetimeIndexSetOps.test_union2�boxc C sX t �d�}|d d� }|dd � }|j||d�}||j�}|j||d�}t �||� d S )Nr r )r )r r r �valuesr ) r r r r r r �expected�case�resultr r r � test_union3/ s z#TestDatetimeIndexSetOps.test_union3�tzc C sD t ddd|d�}t ddd|d�}t ddd|d�}tt|�t|� �}t ddd|d�}t ddd|d�}t ddd|d�} tt|�t|d d � � �} t ddd|d�} tg |d �} t ddd|d�} | }x~||||f||| | f| | | |fgD ]X\}}}}|j||d �}t�||� |j||d �}|d k�r0t�||� q�t�||� q�W d S ) Nz1/1/2000�Dr )�freq�periodsr$ z1/6/2000r z1/4/2000� � )r$ )r )r r �listr r r )r r$ r �rng1�other1� expected1Zexpected1_notsorted�rng2�other2� expected2Zexpected2_notsorted�rng3�other3� expected3Zexpected3_notsorted�rng�other�expZ exp_notsortedZ result_unionr r r � test_union; s, z"TestDatetimeIndexSetOps.test_unionc C sn t dddg�}t |�� dd�}|j||d�}t�||� |d d� j||d�}t�||� |j|jksjt�d S )Nz 2000-01-03z 2000-01-01z 2000-01-02Zinfer)r& )r r )r � sort_valuesr r r r& �AssertionError)r r �idxZorderedr" r r r �test_union_coverage[ s z+TestDatetimeIndexSetOps.test_union_coveragec C sp t dddd�}t dddd�}|j||d�}t|�t|dd � � }|d krXtt|��}nt|�}t�||� d S )Nz1/1/2012� Z3H)r' r& Z4H)r r )r r r* r �sortedr r )r r Zrng_aZrng_br" r6 r r r �test_union_bug_1730e s z+TestDatetimeIndexSetOps.test_union_bug_1730c C sV t dg�}t dddg�}|j||d�}t ddddg�}|d krF|�� }t�||� d S )Nz2012-05-11 15:19:49.695000z2012-05-29 13:04:21.322000z2012-05-11 15:27:24.873000z2012-05-11 15:31:05.350000)r )r r r8 r r )r r �left�rightr" r6 r r r �test_union_bug_1745q s z+TestDatetimeIndexSetOps.test_union_bug_1745c C sp ddl m} tdd�}||dd� }|j||d�}t|�t|� }|d krXtt|��}nt|�}t�||� d S )Nr )� DateOffsetz 2013-01-01z 2013-02-01� )�minutes)r ) �pandasrB r r r* r r= r r )r r rB r? r@ r" r6 r r r �test_union_bug_4564� s z+TestDatetimeIndexSetOps.test_union_bug_4564c C s@ t ddd�}d |j_|j||d�}t�||� |jd ks<t�d S )NZ20150101r )r' )r )r �_datar& r r r r9 )r r r r" r r r �test_union_freq_both_none� s z1TestDatetimeIndexSetOps.test_union_freq_both_nonec C sn t ddd�}|ddddg }|dddg }|jd ks8t�|jd ksFt�|�|�}t�||� |jd ksjt�d S ) Nz 2016-01-01r )r' r r r) r< � r% )r r& r9 r r r )r �dtir? r@ r" r r r �test_union_freq_infer� s z-TestDatetimeIndexSetOps.test_union_freq_inferc C sv t dddd�}ttj�t|��|�}t dddd�}ttj�t|��|�}t||d��}t dddd�}t�|j |� d S )Nz1/1/1999z1/1/2012ZMS)r& z1/1/1980z 12/1/2001)�s1�s2) r r �np�randomZrandn�lenr r r �index)r r+ rL r. rM Zdfr6 r r r �test_union_dataframe_index� s z2TestDatetimeIndexSetOps.test_union_dataframe_indexc C s@ t t�ddd��}tdddd�}|j||d� |j||d� d S ) Nr � rI z2012-01-03 00:00:00r r% )�startr' r& )r )r rN Zaranger r )r r �i1�i2r r r � test_union_with_DatetimeIndex� s z5TestDatetimeIndexSetOps.test_union_with_DatetimeIndexc s� t �d�}|dd � � |�� �}t �|� �s0t�� fdd�tjttgD �}x&|D ] }|�|�}t �|� �sPt�qPW t dddg�}|�|�}t g t d�}t � ||� d S ) Nr r c s g | ]}|� j ��qS r )r )�.0�klass)r r r � <listcomp>� s z>TestDatetimeIndexSetOps.test_intersection2.<locals>.<listcomp>�a�b�c)Zdtype) r r � intersectionZ equalContentsr9 rN �arrayr r* r �objectr )r r Z intersectZcasesr! r" �thirdr r )r r �test_intersection2� s z*TestDatetimeIndexSetOps.test_intersection2c C s� t ddddd�}t ddddd�}t ddddd�}t ddddd�}t dddd d�}t d d ddd�}tg ddd�} xH||f||f|| fgD ].\} } |�| �} t�| | � | j| jks�t�q�W td d d dg|dd�}td d ddg|dd�}td d g|dd�}td d ddg|dd�}td d g|d d�}t d d d|dd�}tg |dd�} | jd k�sNt�xb||f||f|| fgD ]H\} } |j| |d�} |d k�r�| �� } t�| | � | j| jk�sft��qfW d S )Nz6/1/2000z 6/30/2000r% r: )r& �namez 5/15/2000z 6/20/2000r5 z7/1/2000z 7/31/2000z 2011-01-05z 2011-01-04z 2011-01-02z 2011-01-03)r$ rc z 2011-02-02z 2011-02-03)r& r$ rc )r )r r r^ r r r& r9 r8 ) r r$ r �baser. r0 r1 r3 Zrng4Z expected4r4 r r" r r r �test_intersection� sH z)TestDatetimeIndexSetOps.test_intersectionr& �T�Bc C s |}t dd||d�}|dd� �|�}t|�dks6t�|j|jksFt�|�|dd� �}t|�dksht�|j|jksxt�|dk}|d d� �|dd � �}t�||d d� � |r�|j|jks�t�|dd � �|d d� �}t�||d d� � |�r|j|jk�st�d S )Nz6/1/2000z 6/15/2000)r& r$ r rf r) ���)r r^ rP r9 r& r r )r Ztz_aware_fixturer& r$ r4 r" Z check_freqr r r �test_intersection_empty s" z/TestDatetimeIndexSetOps.test_intersection_emptyc C sZ ddl m} tdddd�}||dd� }t�t�� ||@ }W d Q R X t|�dksVt�d S ) Nr )rB z1/1/2012r< Z12H)r' r& r )�hours)rE rB r r Zassert_produces_warning� FutureWarningrP r9 )r rB Zindex_1Zindex_2r" r r r �test_intersection_bug_1708- s z2TestDatetimeIndexSetOps.test_intersection_bug_1708c C s� dddddg}t ||d�}tddd |d �}t ||d�}t ||d�}tddd |d �}t |d d � |d�} t ||d�} t g |d�} t ||d�} xZ|||f||| f| | | fgD ]:\} }}| �||�}|d kr�t|�r�|�� }t�||� q�W d S ) Nz1/2/2000z1/3/2000z1/1/2000z1/4/2000z1/5/2000)r$ z1/6/2000r% r )r& r' r$ r) )r r � differencerP r8 r r )r r$ r Z rng_datesr+ r, r- r. r/ r0 r1 r2 r3 r4 r5 r Z result_diffr r r �test_difference7 s$ z'TestDatetimeIndexSetOps.test_differencec C s� t dddd�}t dddd�}tddgd d�}|�||�}t�||� t�d||� t dddd�}|�||�}tddgd d�}t�||� t�d||� d S ) NZ20160920Z20160925r% )r& Z20160921Z20160924r& Z20160922)r r rm r r Zassert_attr_equal)r r rQ r5 r Zidx_diffr r r �test_difference_freqR s z,TestDatetimeIndexSetOps.test_difference_freqc C sH t dtddd�dd�}t dtddd�dd�}t|�||��dksDt�d S ) NzQ-JANi� � � �d )r& rT r' �b rI )r r rP rm r9 )r r Zdti1Zdti2r r r �test_datetimeindex_diffb s z/TestDatetimeIndexSetOps.test_datetimeindex_diffc C s� t ddd|d�}|d d� �|dd� �}|j|jks:t�|j|jksJt�|j|jksZt�|d d� �|dd� �}|j|jks�t�|j|jks�t�|j|jks�t�|d d� �|dd� �}|j|jks�t�|jd ks�t�|j|jks�t�|d d� �|d d � �}|j|jk�st�|jd k�s(t�|j|jk�s:t�tt |d d � �d d �}|d d� �|�}|jd k�stt�|j|jk�s�t�|j|jk�s�t�|d d� �|�}|jd k�s�t�|j|jk�s�t�|j|jk�s�t�d S )Nz1/1/2000z1/1/2002r: )rc r$ �2 rr � �< � �K r% r5 )rc ) r r rc r9 r& r$ r^ Zfreqstrr r* )r r$ r4 r" Znofreqr r r �test_setops_preserve_freqg s4 z1TestDatetimeIndexSetOps.test_setops_preserve_freqc C sR t ddddddgdd�}|d d d � �|d d d � �}|d d � }t�||� d S ) Nz 2018-12-31z 2019-03-31z 2019-06-30z 2019-09-30z 2019-12-31z 2020-03-31zQ-DEC)r& rI r r )r r^ r r )r rJ r" r r r r �&test_intersection_non_tick_no_fastpath� s z>TestDatetimeIndexSetOps.test_intersection_non_tick_no_fastpath) �__name__� __module__� __qualname__r$ r �pytest�mark� parametrizerN r_ r r* r# r7 r; r> rA rF rH rK rR rW rb re ri rl rn ro rt rz r{ r r r r r s6 = #r c @ sl e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z e j dd� �Z e j�dddg�dd� �ZdS )�TestBusinessDatetimeIndexc C s t tt�| _d S )N)r �START�ENDr4 )r �methodr r r � setup_method� s z&TestBusinessDatetimeIndex.setup_methodc C s& | j d d� }| j dd� }|j||d�}t|t�s8t�| j d d� }| j dd � }|j||d�}t|t�spt�| j d d� }| j dd� }|j||d�}t|t�s�t�|d kr�t�|j||d�|� n*tt|�t|� �}t�|j||d�|� t t t t � d�}| j j||d�}t|t��s"t�d S )Nr r )r )r& ) r4 r � isinstancer r9 r r r r* r r� r� r )r r r? r@ � the_unionr r4 r r r r7 � s& z$TestBusinessDatetimeIndex.test_unionc C s� t ddt� d�}|dd � }|d d� }|j||d�}|d krLt�||� n0tt|dd � �t|d d� � �}t�||� |dd � }|dd� }|j||d�}|dd � }t�||� d S ) Nz1/1/2000ru )r' r& r rx )r rC �# )r r r r r r r* )r r r4 r+ r. r� r r r r �test_union_not_cacheable� s $ z2TestBusinessDatetimeIndex.test_union_not_cacheablec C s� t ddt� d�}|dd � }|d d� }|�|�}|dd� }t�||� t|t�sXt�|j|jksht�|�|� t��}t�||� |d d� �|dd � �}tg �}t�||� d S )Nz1/1/2000ru )r' r& r rx ) r r r^ r r r� r r9 r& �view)r r4 r+ r. Zthe_intr r r r re � s z+TestBusinessDatetimeIndex.test_intersectionc C s> t dd�}t dd�}|�|�}t�||� |j|jks:t�d S )Nz 11/30/2011z 12/31/2011z 12/10/2011z 12/20/2011)r r^ r r r& r9 )r r[ r\ r" r r r �test_intersection_bug� s z/TestBusinessDatetimeIndex.test_intersection_bugc C s: t �d�t �d�g}t|dd�}|�|�}t�||� d S )Nz 2020-01-01z 2020-02-01r[ )rc )�pd� Timestampr r^ r r )r r r: �resr r r �test_intersection_list� s z0TestBusinessDatetimeIndex.test_intersection_listc C sz ddl m} |d�}tddd�}tddd�}tddd�}tddd�}t|||t� d�}t|||t� d�} |j| |d � d S ) Nr )�timezonez US/Easterni� r r) r )rT �endr$ r& )r )Zpytzr� r r r r ) r r r� r$ � early_start� early_end� late_start�late_end�early_dr�late_drr r r � test_month_range_union_tz_pytz� s z8TestBusinessDatetimeIndex.test_month_range_union_tz_pytzc C sz ddl m} |d�}tddd�}tddd�}tddd�}tddd�}t|||t� d�}t|||t� d�} |j| |d � d S ) Nr )�dateutil_gettzz US/Easterni� r r) r )rT r� r$ r& )r )Z pandas._libs.tslibs.timezonesr� r r r r ) r r r� r$ r� r� r� r� r� r� r r r �"test_month_range_union_tz_dateutil s z<TestBusinessDatetimeIndex.test_month_range_union_tz_dateutilr FNc C sV t t�d�t�d�t�d�g�}|j||d�}t t�d�t�d�g�}t�||� d S )Nz 2019-12-13z 2019-12-12)r )r r� r� r^ r r )r r Zidx1r" r r r r � test_intersection_duplicates s z6TestBusinessDatetimeIndex.test_intersection_duplicates)r| r} r~ r� r7 r� re r� r� r� �tdZskip_if_windows_python_3r� r r� r� r� r r r r r� � s #r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestCustomDatetimeIndexc C s t ttdd�| _d S )N�C)r& )r r� r� r4 )r r� r r r r� ' s z$TestCustomDatetimeIndex.setup_methodc C s� | j d d� }| j dd� }|j||d�}t|t�s8t�| j d d� }| j dd � }|�||�}t|t�snt�| j d d� }| j dd� }|j||d�}t|t�s�t�|d kr�t�|j||d�|� tt t t � d�}| j j||d�}t|t�s�t�d S )Nr r )r )r& ) r4 r r� r r9 r r r r r� r� r )r r r? r@ r� r4 r r r r7 * s" z"TestCustomDatetimeIndex.test_unionc C sF t dddd�}t dddd�}|�|�}t�||� |j|jksBt�d S )Nz 11/30/2011z 12/31/2011r� )r& z 12/10/2011z 12/20/2011)r r^ r r r& r9 )r r[ r\ r" r r r r� J s z-TestCustomDatetimeIndex.test_intersection_bugN)r| r} r~ r� r7 r� r r r r r� & s r� ) r �numpyrN r Z pandas.util._test_decorators�utilZ_test_decoratorsr� rE r� r r r r r r r Zpandas._testingZ_testingr Zpandas.tseries.offsetsr r r r� r� r r� r� r r r r �<module> s $
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка