Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/indexes/datetimes/__pycache__/test_date_range.cpython-37.pyc
Назад
B �5�gP� � @ s< d Z ddlmZmZmZ ddlZddlZddlZddlmZ ddl m Z ddl m Z m Z mZmZmZ ddlmZ ddlm mZ ddlZddlmZmZmZm Z m Z m Z ddl m Z! dd l"m#Z# ed d d �ed d d � Z$Z%G d d� d�Z&G dd� d�Z'G dd� d�Z(G dd� d�Z)G dd� d�Z*G dd� d�Z+dd� Z,dS )zP test date_range, bdate_range construction from the convenience range functions � )�datetime�time� timedeltaN)�timezone)� timezones)�BDay�CDay� DateOffset�MonthEnd�prefix_mapping)�OutOfBoundsDatetime)� DatetimeIndex� Timedelta� Timestamp� bdate_range� date_range�offsets)�generate_rangei� � i� c @ sB e Zd Zdd� Zdd� Zdd� Zejdd� �Zd d � Z d d � Z d S )�TestTimestampEquivDateRangec C s2 t dddd�}|d }tddd�}||ks.t�d S )N�20090415�20090519z US/Eastern)�tzr )r r �AssertionError)�self�rng�stamp�ts� r �d/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/indexes/datetimes/test_date_range.py� test_date_range_timestamp_equiv0 s z;TestTimestampEquivDateRange.test_date_range_timestamp_equivc C s2 t dddd�}|d }tddd�}||ks.t�d S )Nr r zdateutil/US/Eastern)r r )r r r )r r r r r r r �(test_date_range_timestamp_equiv_dateutil7 s zDTestTimestampEquivDateRange.test_date_range_timestamp_equiv_dateutilc C s> t ddt�d�d�}|d }tdt�d�d�}||ks:t�d S )Nr r z US/Eastern)r r )r �pytzr r r )r r r r r r r �-test_date_range_timestamp_equiv_explicit_pytz> s zITestTimestampEquivDateRange.test_date_range_timestamp_equiv_explicit_pytzc C sF ddl m} tdd|d�d�}|d }td|d�d�}||ksBt�d S )Nr )�dateutil_gettzr r z US/Eastern)r )� pandas._libs.tslibs.timezonesr$ r r r )r Zgettzr r r r r r �1test_date_range_timestamp_equiv_explicit_dateutilE s zMTestTimestampEquivDateRange.test_date_range_timestamp_equiv_explicit_dateutilc C s6 t ddd�}t|ddd�d }t|�}||ks2t�d S )Ni� � � r �D)�periods�freqr )r r r r )r Zdatetime_instance�timestamp_instancer r r r �6test_date_range_timestamp_equiv_from_datetime_instanceO s zRTestTimestampEquivDateRange.test_date_range_timestamp_equiv_from_datetime_instancec C s* t dddd�d }td�}||ks&t�d S )Nz 2014-03-05r r) )r* r+ r )r r r )r r, r r r r �2test_date_range_timestamp_equiv_preserve_frequencyX s zNTestTimestampEquivDateRange.test_date_range_timestamp_equiv_preserve_frequencyN) �__name__� __module__� __qualname__r r! r# �tdZskip_if_windows_python_3r& r- r. r r r r r * s r c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zej j d d � �Z d d� Z dd� Z ej �dddg�dd� �Zej �dddg�dd� �Zej �dddg�d d � �Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zej �d*d+d,d-ged.d/d/�ed.d/d0�d-ged+�ed,�d-ged+d-d1�ed,d-d1�d-ged+d-d1�ed,d-d1�d2gg�d3d4� �Zd5d6� Zd7d8� Z d9d:� Z d;d<� Z ej �d=d>�d?d@� �Z dAdB� Z dCdD� Z!dEdF� Z"dGdH� Z#dIdJ� Z$ej �dKeedLd0dM�d-d1�eedLd0dN�d-d1�geedOdPd/�d-d1�eedOdPdM�d-d1�gg�dQdR� �Z%dSdT� Z&ej �ddUdVdWdXdYdg�dZd[� �Z'd\d]� Z(ej �d^d_d`d2g�dadb� �Z)dcdd� Z*dedf� Z+dgdh� Z,didj� Z-dkdl� Z.dmdn� Z/d2S )o�TestDateRangesc C s6 t d�}tjtdd�� ttjd|d� W d Q R X d S )Nr zCannot generate range with)�match� )�endr* r+ )r �pytest�raisesr r r �min)r r+ r r r �)test_date_range_near_implementation_bound` s z8TestDateRanges.test_date_range_near_implementation_boundc C s\ d}t jt|d�� tdtjdd� W d Q R X t jt|d�� ttjddd� W d Q R X d S )Nz$Neither `start` nor `end` can be NaT)r4 z 2016-01-01r) )�startr6 r+ )r7 r8 � ValueErrorr �pdZNaT)r �msgr r r �test_date_range_natg s z"TestDateRanges.test_date_range_natc C sx t �d �� tdddd�}W d Q R X |d td�ks8t�t|�dksHt�d}tjt|d�� tdd d d � W d Q R X d S ) Nz 1677-09-22i�A r) )r; r* r+ r zCannot generate range with)r4 z 1969-05-04i �� Z30000D)r* r+ ) �tmZassert_produces_warningr r r �lenr7 r8 r )r �dtir> r r r �'test_date_range_multiplication_overflowo s z6TestDateRanges.test_date_range_multiplication_overflowc C sZ t dddd�}t |d t|�dd�}|�|�s2t�t |d t|�dd�}|�|�sVt�d S ) Nz 1677-09-22z 2262-04-11r) )r; r6 r+ r )r; r* r+ ���)r6 r* r+ )r rA �equalsr )r rB �dti2Zdti3r r r �*test_date_range_unsigned_overflow_handling~ s z9TestDateRanges.test_date_range_unsigned_overflow_handlingc C sX d}t jt|d�� tdddd� W d Q R X t jt|d�� tdddd� W d Q R X d S ) NzCannot generate range with)r4 z 1970-02-01i ' �H)r; r* r+ z 1969-11-14)r6 r* r+ )r7 r8 r r )r r> r r r �.test_date_range_int64_overflow_non_recoverable� s z=TestDateRanges.test_date_range_int64_overflow_non_recoverablec C s� t d�}t d�}t||dd�}|d |ks.t�|d |ks>t�t|t|�dd�}t�||� t d�}t d �}t||dd�}|d |ks�t�|d |ks�t�t|t|�dd �}t�||� d S ) Nz 2262-02-23z 1969-11-14z-1H)r; r6 r+ r rD )r6 r* r+ z 1970-02-01z 1677-10-22)r; r* r+ )r r r rA r@ �assert_index_equal) r r; r6 �expectedrB Zstart2Zend2� expected2rF r r r �>test_date_range_int64_overflow_stride_endpoint_different_signs� s zMTestDateRanges.test_date_range_int64_overflow_stride_endpoint_different_signsc C sX d}t jt|d�� tdddd� W d Q R X t jt|d�� tdddd� W d Q R X d S ) NzCannot generate range)r4 z 2016-01-01i�� r) )r* r+ z 1763-10-12)r6 r* r+ )r7 r8 r r )r r> r r r � test_date_range_out_of_bounds� s z,TestDateRanges.test_date_range_out_of_boundsc C s" t dddd�}t|�dks t�d S )Nz1/1/2000 00:00z1/1/2000 00:18Z5min)r+ r( )r rA r )r r r r r �test_date_range_gen_error� s z(TestDateRanges.test_date_range_gen_errorr+ ZASZYSc C s4 t dd|d�}tdddddg|d�}t�||� d S ) Nz1/1/2013z7/1/2017)r+ z 2013-01-01z 2014-01-01z 2015-01-01z 2016-01-01z 2017-01-01)r r r@ rJ )r r+ r �expr r r �test_begin_year_alias� s z$TestDateRanges.test_begin_year_alias�A�Yc C s2 t dd|d�}tddddg|d�}t�||� d S )Nz1/1/2013z7/1/2017)r+ z 2013-12-31z 2014-12-31z 2015-12-31z 2016-12-31)r r r@ rJ )r r+ r rP r r r �test_end_year_alias� s z"TestDateRanges.test_end_year_aliasZBAZBYc C s2 t dd|d�}tddddg|d�}t�||� d S )Nz1/1/2013z7/1/2017)r+ z 2013-12-31z 2014-12-31z 2015-12-31z 2016-12-30)r r r@ rJ )r r+ r rP r r r � test_business_end_year_alias� s z+TestDateRanges.test_business_end_year_aliasc C sx t dddd�}tdddgdd�}t�||� |jdks:t�t dd dd�}tdd d gd d�}t�||� |jd kstt�d S ) Nz 2011-12-31z-2Ar' )r+ r* z 2009-12-31z 2007-12-31)r+ z 2011-01-31z-2Mz 2010-11-30z 2010-09-30)r r r@ rJ r+ r )r r rP r r r � test_date_range_negative_freq� s z,TestDateRanges.test_date_range_negative_freqc C s* t dddd�}td�}|d |ks&t�d S )Nz1/1/2000� ZBMS)r* r+ z 2000-01-03r )r r r )r r Zex_firstr r r �test_date_range_bms_bug� s z&TestDateRanges.test_date_range_bms_bugc s� t �� �d}t�|ddd�}td�� t� �fdd�t|�D �� d�}t�||� td |dd d�}td d �}x |D ]}|�� |kspt �qpW d S ) N�2 FZ2D)r* � normalizer+ r5 c s g | ]}�|� �qS r r )�.0�i)�offset�snapr r � <listcomp>� s z<TestDateRanges.test_date_range_normalize.<locals>.<listcomp>)r+ z1/1/2000 08:15�B� � ) r �todayr r r �ranger@ rJ r r )r �nr �valuesZthe_time�valr )r] r^ r �test_date_range_normalize� s z(TestDateRanges.test_date_range_normalizec C sF t ddtjdddd�d�}|d td �ks.t�|d td �ksBt�d S ) Nz 2013-01-01r5 r r' Znearest)Z startingMonth�weekdayZ variation)r; r* r+ r z 2013-01-31z 2014-01-30)r r ZFY5253r r )r �drr r r �test_date_range_fy5252� s z%TestDateRanges.test_date_range_fy5252c C sV t dddddd�}t dddddd�}d}tjt|d �� t||d d d � W d Q R X d S ) Ni� r � r' �( ra � zVOf the four parameters: start, end, periods, and freq, exactly three must be specified)r4 rW �s)r* r+ )r r7 r8 r< r )r r; r6 r> r r r �#test_date_range_ambiguous_arguments s z2TestDateRanges.test_date_range_ambiguous_argumentsc C sr t dddd�}tdddgd d�}t�||� t d d d dd �}ttd d d�tdd d�tdd d�g�}t�||� d S )Nz 2018-04-24z 2018-04-27r' )r* z2018-04-24 00:00:00z2018-04-25 12:00:00z2018-04-27 00:00:00)r+ z2018-04-01 01:00:00z2018-04-01 04:00:00zAustralia/Sydney)r r* z2018-04-01 01:00:00+1100)r z2018-04-01 02:00:00+1000z2018-04-01 04:00:00+1000)r r r@ rJ r )r �resultrK r r r �#test_date_range_convenience_periods s z2TestDateRanges.test_date_range_convenience_periodszstart,end,result_tz�20180101Z20180103z US/Easterni� r r' )r Nc C s0 t ||d|d�}t ddddd�}t�||� d S )Nr' )r* r rs r) z US/Eastern)r* r+ r )r r@ rJ )r r; r6 Z result_tzrq rK r r r � test_date_range_linspacing_tz' s z,TestDateRanges.test_date_range_linspacing_tzc C s� t ddddddddgd d �}tddd d �}t�||� t dd gd d �}tdd d d �}t�||� t ddddddddd d d dddddddddddddgd d �}tddd d �}t�||� d S )Nz2014-07-04 09:00z2014-07-04 10:00z2014-07-04 11:00z2014-07-04 12:00z2014-07-04 13:00z2014-07-04 14:00z2014-07-04 15:00z2014-07-04 16:00ZBH)r+ z2014-07-07 09:00z2014-07-07 10:00z2014-07-07 11:00z2014-07-07 12:00z2014-07-07 13:00z2014-07-07 14:00z2014-07-07 15:00z2014-07-07 16:00z2014-07-08 09:00z2014-07-08 10:00z2014-07-08 11:00z2014-07-08 12:00z2014-07-08 13:00z2014-07-08 14:00z2014-07-08 15:00z2014-07-08 16:00)r r r@ rJ )r �idxr r r r � test_date_range_businesshour? sV z+TestDateRanges.test_date_range_businesshourc C s d}t jt|d�� tdd� W d Q R X t jt|d�� tdd� W d Q R X t jt|d�� tdd� W d Q R X t jt|d�� tddd � W d Q R X t jt|d�� tddd � W d Q R X t jt|d�� tddd � W d Q R X t jt|d�� t� W d Q R X d S ) NzVOf the four parameters: start, end, periods, and freq, exactly three must be specified)r4 z1/1/2000)r; )r6 rW )r* rH )r; r+ )r6 r+ )r* r+ )r7 r8 r< r )r r> r r r �test_range_misspecifiedt s z&TestDateRanges.test_range_misspecifiedc C s& t td�ddd�}t|�dks"t�d S )Nz1960-04-01 00:00:00�L zQS-JAN)r* r+ )r r rA r )r rq r r r �test_compat_replace� s z"TestDateRanges.test_compat_replacec C sL t jdd�}d}tjt|d��$ ttddd�tddd�|d� W d Q R X d S ) Nrl )�minutez4Offset <DateOffset: minute=5> did not increment date)r4 i� � � )r+ )r r r7 r8 r< r r )r r] r> r r r �test_catch_infinite_loop� s z'TestDateRanges.test_catch_infinite_loopr* )r r5 c C s" t d|dd�}t|�|ks t�d S )NZ20110101zWOM-1MON)r; r* r+ )r rA r )r r* �resr r r � test_wom_len� s zTestDateRanges.test_wom_lenc C sb t d�jddd�}t d�jddd�}t ddd�||g}t|dd �}td d ddd �}t�||� d S ) Nz2010-11-07 01:00:00z US/PacificT)Z ambiguousFz2010-11-07 00:00:00)r rH )r+ z 2010-11-7r' )r; r* r+ r )r Z tz_localizer r r@ rJ )r Zpre_dstZpst_dstZ expect_datarK rq r r r �test_construct_over_dst� s z&TestDateRanges.test_construct_over_dstc C s< t dddd�}ttd�td�td�gdd�}t�||� d S )Nz2013-01-01 00:00:00+09:00z2013/01/01 02:00:00+09:00rH )r+ z2013-01-01 01:00:00+09:00z2013-01-01 02:00:00+09:00)r r r r@ rJ )r rq rK r r r �5test_construct_with_different_start_end_string_format� s zDTestDateRanges.test_construct_with_different_start_end_string_formatc C s4 d}t jt|d�� tddtd�d� W d Q R X d S )Nz.Offset <0 \* MonthEnds> did not increment date)r4 z1/1/2000z1/1/2001r )r+ )r7 r8 r< r r )r r> r r r � test_error_with_zero_monthends� s z-TestDateRanges.test_error_with_zero_monthendsc sT t dd�� tdd� d�}tddd��t� �fdd �td �D �� d�}t�||� d S ) Nr' )�monthsz2011-1-1z 2012-1-31)r+ i� r c s g | ]}�|� �qS r r )r[ r\ )r] r; r r r_ � s z1TestDateRanges.test_range_bug.<locals>.<listcomp>rl )r r r r rd r@ rJ )r rq rK r )r] r; r �test_range_bug� s z TestDateRanges.test_range_bugc C s� t d�}|�tddd��}|�tddd��}t|dd�}|jj|jksJt�|d |ksZt�|d |ksjt�t|dd�}|jj|jks�t�|d |ks�t�|d |ks�t�t||d �}|jj|jks�t�|d |ks�t�|d |ks�t�d S ) Nz US/Easterni� r r' )r; r* r r5 )r6 r* )r; r6 )r Zlocalizer r r �zoner )r r r; r6 rj r r r �test_range_tz_pytz� s z!TestDateRanges.test_range_tz_pytzz start, endi� � r| i� r{ c C s� t ||dd�}|d |ks t�|d |ks.t�t�|jdk�sBt�t ||ddd�}|d |ksbt�|d |ksrt�t�|jdk�s�t�t |jd d�|jd d�ddd�}|d |ks�t�|d |ks�t�t�|jdk�s�t�d S )Nr) )r+ r rD z US/Eastern)r+ r )�tzinfo)r r �np�all�hour�replace)r r; r6 rj r r r � test_range_tz_dst_straddle_pytz� s z.TestDateRanges.test_range_tz_dst_straddle_pytzc s� ddl m� � fdd�}tddd|d�d�}tddd |d�d�}t|d d �}|j|d�ks^t�|d |ksnt�|d |ks~t�t|d d �}|j|d�ks�t�|d |ks�t�|d |ks�t�t||d �}|j|d�ks�t�|d |ks�t�|d |ks�t�d S )Nr )� maybe_get_tzc s � d| �S )Nz dateutil/r )�x)r� r r �<lambda> � z7TestDateRanges.test_range_tz_dateutil.<locals>.<lambda>i� r z US/Eastern)r� r' )r; r* r5 )r6 r* )r; r6 )r% r� r r r r )r r r; r6 rj r )r� r �test_range_tz_dateutil s z%TestDateRanges.test_range_tz_dateutil�1D�3D�2M�7W�3Hc C s� t ddd�}t ddd�}t||d |d�}t||d|d�}t||d|d�}|}|}||d krh|d d� }||d kr�|dd � }t�||� t�||� d S ) Ni� r i� )�closedr+ �left�rightrD r )r r r@ rJ ) r r+ �beginr6 r� r� r� � expected_left�expected_rightr r r �test_range_closed" s z TestDateRanges.test_range_closedc C sj t ddd�}t ddd�}x�dD ]�}t||d |d�}t||d|d�}t||d|d�}|}|}||d krr|d d � }||d kr�|d d � }t�||� t�||� q W t d�}t d�}t ddd�} t ddd�} x�dD ]�}t||d |dd �}t||d|dd �}t||d|dd �}|}|}| |d k�r0|d d � }| |d k�rJ|d d � }t�||� t�||� q�W d S ) Nz2011/1/1z US/Eastern)r z2014/1/1)r� r� r� r� r� rR )r� r+ r� r� rD r r )r� r+ r )r r r@ rJ ) r r� r6 r+ r� r� r� r� r� ZbegintzZendtzr r r �)test_range_closed_with_tz_aware_start_end5 s< z8TestDateRanges.test_range_closed_with_tz_aware_start_endr� r� r� c C s� t ddd|d�}t ddd|d�}t ddd|d�}| } }}|dkrP|dd � }|dkrd|d d � }|d kr�|dd � }|d d � }t�||� t�||� t�||� d S ) Nz 2015-09-12z 2015-12-01zQS-MAR)r+ r� z 2015-09-01r� r r� rD )r r@ rJ )r r� Zright_boundaryZ left_boundaryZ both_boundaryr� r� Z expected_bothr r r �test_range_closed_boundary] s" z)TestDateRanges.test_range_closed_boundaryc C sB t dddd�}|d tddd�ks&t�|d tdd d�ks>t�d S ) NZ2014Z2015�M)r+ r i� r � rD r| )r r r )r rj r r r �test_years_onlyw s z TestDateRanges.test_years_onlyc C s` t dddd�}t dddd�}tddgd d d d �}td d gd d d d �}t�||� t�||� d S )Nz2005-01-12 10:00z2005-01-12 16:00Z345min)r+ z2005-01-13 10:00z2005-01-13 16:00z2005-01-12 10:00:00z2005-01-12 15:45:00zdatetime64[ns]Z345T)�dtyper+ r z2005-01-13 10:00:00z2005-01-13 15:45:00)r r r@ rJ )r Zresult_1Zresult_2Z expected_1Z expected_2r r r � test_freq_divides_end_in_nanos} s z-TestDateRanges.test_freq_divides_end_in_nanosc C sB t ddtdd�d�}t|�dks$t�|d tddd d �ks>t�d S ) Nz2010-09-01 05:00:00rY r� )�hours)r* r+ r i� rn r rl )r r rA r r )r r r r r �test_cached_range_bug� s z$TestDateRanges.test_cached_range_bugc C s. t ddd�}t|ddd�}t|�dks*t�d S )Nz20130220 10:00z US/Eastern)r r5 )r* r )r r rA r )r r; rq r r r � test_timezone_comparaison_bug� s z,TestDateRanges.test_timezone_comparaison_bugc C s< t ddd�}d}tjt|d�� t|ddd� W d Q R X d S ) Nz20130220 10:00z US/Eastern)r z0Inferred time zone not equal to passed time zone)r4 r5 z Europe/Berlin)r* r )r r7 r8 r r )r r; r> r r r � test_timezone_comparaison_assert� s z/TestDateRanges.test_timezone_comparaison_assertc C s> |}t ddd|d�}t ddd|d�d d d� }t�||� d S )Nz 2011-06-01z 2011-01-01z-1MS)r; r6 r+ r Z1MS)r6 r; r+ r rD )r r@ rJ )r Ztz_aware_fixturer rq rK r r r �1test_negative_non_tick_frequency_descending_dates� s z@TestDateRanges.test_negative_non_tick_frequency_descending_dates)0r/ r0 r1 r: r? rC rG rI r7 �markZslowrM rN rO � parametrizerQ rT rU rV rX rh rk rp rr r r rt rv rw ry r} r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r3 _ sl 5 ""( r3 c @ sn e Zd ZdZdd� Zej�dddg�dd� �Zej�dddg�d d � �Z d d � Z ej�dddg�d d� �Z dS )�TestDateRangeTZz#Tests for date_range with timezonesc C s t ddddd�}|j d S )Nz 2012-01-01z 2012-01-10r) ZHongkong)r+ r )r r� )r rj r r r �test_hongkong_tz_convert� s z(TestDateRangeTZ.test_hongkong_tz_convert�tzstrz US/Easternzdateutil/US/Easternc C sV t ddddd�}|jdk�� s"t�t dd|d �}|j}t�dgd �}t�||� d S ) Nz03/06/2012 00:00�� zW-FRIz US/Eastern)r* r+ r r z 2012-11-02rW )r* r )r r� r� r r= �Indexr@ rJ )r r� rj rq rK r r r �#test_date_range_span_dst_transition� s z3TestDateRangeTZ.test_date_range_span_dst_transitionc C s6 t �|�}tdd|d�}tdd|d�}t�||� d S )Nz1/1/2000rW )r* r )r r� r r@ rJ )r r� r rq rK r r r �%test_date_range_timezone_str_argument� s z5TestDateRangeTZ.test_date_range_timezone_str_argumentc C sv ddl m} |}tdddddd|d�}tdddddd|d�}t||d �}||jksVt�t�||g�}||jksrt�d S ) Nr )�fixed_off_no_namei� r' r{ rl )r� r� )r; r6 )Z-pandas.tests.indexes.datetimes.test_timezonesr� r r r r r= r� )r r� �offr; r6 r ru r r r �'test_date_range_with_fixedoffset_noname� s z7TestDateRangeTZ.test_date_range_with_fixedoffset_nonamec C s> t d|d�}|jdkst�tddd|d�}||d ks:t�d S ) Nz3/11/2012 05:00)r rl z3/11/2012 04:00rW rH )r* r+ r r )r r� r r )r r� r r r r r �test_date_range_with_tz� s z'TestDateRangeTZ.test_date_range_with_tzN) r/ r0 r1 �__doc__r� r7 r� r� r� r� r� r� r r r r r� � s r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd \Z Z d\Z Z e j�dee e d�ee �fee �ee e d�fee e d�ee e d�fee e d�ee e d�fg�dd� �ZdS )�TestGenRangeGenerationc C s6 t tttt� d��}t tttdd��}||ks2t�d S )N)r] r` )�listr �START�ENDr r )r �rng1�rng2r r r � test_generate� s z$TestGenRangeGeneration.test_generatec C s6 t tttt� d��}t tttdd��}||ks2t�d S )N)r] �C)r� r r� r� r r )r r� r� r r r �test_generate_cday� s z)TestGenRangeGeneration.test_generate_cdayc C s@ t ttddd�dd��}tddd�tddd�g}||ks<t�d S )Ni� r' � r5 )r; r* � )r� r r r )r r rK r r r �test_1� s z TestGenRangeGeneration.test_1c C sR t ttddd�tddd�d��}tddd�tddd�tddd�g}||ksNt�d S )Ni� r r' )r; r6 r5 )r� r r r )r r rK r r r �test_2� s "z TestGenRangeGeneration.test_2c C s4 t ttddd�tddd�d��}g }||ks0t�d S )Ni� r rl r� )r; r6 )r� r r r )r r rK r r r �test_3� s z TestGenRangeGeneration.test_3c C s| t dddd�}t dddd�}ddd d g}d d d dddddddg }t|ddd d�}t|ddd d�}t�||� t�||� d S )Nz2015-04-15 00:00:03z2016-04-22 00:00:00�Q)r; r6 r+ z2015-06-22 00:00:04�Wz2015-06-30 00:00:03z2015-09-30 00:00:03z2015-12-31 00:00:03z2016-03-31 00:00:03z2015-04-19 00:00:03z2015-04-26 00:00:03z2015-05-03 00:00:03z2015-05-10 00:00:03z2015-05-17 00:00:03z2015-05-24 00:00:03z2015-05-31 00:00:03z2015-06-07 00:00:03z2015-06-14 00:00:03z2015-06-21 00:00:03zdatetime64[ns]zQ-DEC)r� r+ r zW-SUN)r r r@ rJ )r Zresult1Zresult2Zexpected1_listZexpected2_listZ expected1rL r r r � test_precision_finer_than_offset� s0 z7TestGenRangeGeneration.test_precision_finer_than_offset)z 2017-01-01z 2017-01-01)z US/Easternz Europe/Londonz start,end)r c C sV d}t jt|d�� t||� W d Q R X t jt|d�� t||t� d� W d Q R X d S )Nz>Start and end cannot both be tz-aware with different timezones)r4 )r+ )r7 r8 � TypeErrorr r )r r; r6 r> r r r � test_mismatching_tz_raises_err! s z5TestGenRangeGeneration.test_mismatching_tz_raises_errN)r/ r0 r1 r� r� r� r� r� r� Zdt1Zdt2Ztz1Ztz2r7 r� r� r r� r r r r r� � s #r� c @ sV e Zd Zdd� Zdd� Zdd� Zdd� Zej� d d d g�d d � �Z dd� Z dd� Z dS )�TestBusinessDateRangec C s� t ttt� d� t tdt� d� t tdt� d� d}tjt|d�� tddd � W d Q R X tjt|d�� t ddd � W d Q R X d }tjt|d�� t ttd d d� W d Q R X d S ) N)r+ � )r* r+ )r6 r* r+ z periods must be a number, got B)r4 z2011-1-1z2012-1-1r` z>freq must be specified for bdate_range; use date_range insteadrW )r r� r� r r7 r8 r� r )r r> r r r �test_constructor4 s z&TestBusinessDateRange.test_constructorc C sZ t ddd�}t|dd�}|dt� }t|�dks6t�|d |ksFt�|d |ksVt�d S ) Ni� rl � r� )r6 r* � r rD )r r r rA r )r r6 rj � firstDater r r � test_miscD s z TestBusinessDateRange.test_miscc C s� d}d}t jt|d�� t|� W d Q R X t jt|d�� t|dd� W d Q R X t jt|d�� t|dd� W d Q R X t jt|d�� t||� W d Q R X d S )Nz 2007/100/1z%could not convert string to Timestamp)r4 rW )r; r* )r6 r* )r7 r8 r< r r )r Zbadly_formed_dater> r r r �test_date_parse_failureM s z-TestBusinessDateRange.test_date_parse_failurec C sB t dd�}t dd�}|jjt� ks&t�|�|�}t|t�s>t�d S )Nz 12/5/2011z 12/2/2011)r �_datar+ r r �union� isinstancer )r r� r� rq r r r �test_daterange_bug_456] s z,TestBusinessDateRange.test_daterange_bug_456r� r� r� c C s> d}d}t ||d|d�}d}d}t ||dd�}t�||� d S ) Nz 2018-07-21z 2018-07-29r` )r+ r� z 2018-07-23z 2018-07-27r) )r+ )r r@ rJ )r r� r; r6 rq Z bday_startZbday_endrK r r r � test_bdays_and_open_boundariesf s z4TestBusinessDateRange.test_bdays_and_open_boundariesc C s> t j�d��� }t|d ddd�}t|gdd�}t�||� d S )Nr) r r` )r6 r* r+ )r+ )r �max�floor� to_pydatetimer r r@ rJ )r r; r rK r r r �test_bday_near_overflows s z-TestBusinessDateRange.test_bday_near_overflowc C s@ d}t j�d��� }tjt|d�� t|ddd� W d Q R X d S )Nz"Out of bounds nanosecond timestampr) )r4 r5 r` )r* r+ )r r� r� r� r7 r8 r r )r r> r; r r r �test_bday_overflow_errorz s z.TestBusinessDateRange.test_bday_overflow_errorN) r/ r0 r1 r� r� r� r� r7 r� r� r� r� r� r r r r r� 3 s r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z e j � ddd� e D ��dd� �Ze j � ddddg�dd� �ZdS )�TestCustomDateRangec C s� t ttt� d� t tdt� d� t tdt� d� d}tjt|d�� tddd � W d Q R X tjt|d�� t ddd � W d Q R X d S ) N)r+ r� )r* r+ )r6 r* r+ z periods must be a number, got C)r4 z2011-1-1z2012-1-1r� )r r� r� r r7 r8 r� r )r r> r r r r� � s z$TestCustomDateRange.test_constructorc C s\ t ddd�}t|ddd�}|dt� }t|�dks8t�|d |ksHt�|d |ksXt�d S ) Ni� rl r� r� r� )r6 r* r+ r� r rD )r r r rA r )r r6 rj r� r r r r� � s z TestCustomDateRange.test_miscc C sJ t dddd�}t dddd�}|jjt� ks.t�|�|�}t|t�sFt�d S )Nz 12/5/2011r� )r+ z 12/2/2011)r r� r+ r r r� r� r )r r� r� rq r r r r� � s z*TestCustomDateRange.test_daterange_bug_456c C s@ t dddd�}tdddgdd�}t�||� |j|jks<t�d S )Nz 2013-05-01r' r� )r* r+ z 2013-05-02z 2013-05-03)r+ )r r r@ rJ r+ r )r rq rK r r r �test_cdaterange� s z#TestCustomDateRange.test_cdaterangec C sp t ddddd�}tdddg|jd�}t�||� |j|jks@t�d }tjt|d �� t dddd � W d Q R X d S ) Nz 2013-05-01r' r� zSun Mon Tue Wed Thu)r* r+ �weekmaskz 2013-05-02z 2013-05-05)r+ z[a custom frequency string is required when holidays or weekmask are passed, got frequency B)r4 )r* r� ) r r r+ r@ rJ r r7 r8 r< )r rq rK r> r r r �test_cdaterange_weekmask� s z,TestCustomDateRange.test_cdaterange_weekmaskc C st t ddddgd�}tdddg|jd�}t�||� |j|jksBt�d }tjt|d �� t dddgd � W d Q R X d S ) Nz 2013-05-01r' r� )r* r+ �holidaysz 2013-05-02z 2013-05-03z 2013-05-06)r+ z[a custom frequency string is required when holidays or weekmask are passed, got frequency B)r4 )r* r� ) r r r+ r@ rJ r r7 r8 r< )r rq rK r> r r r �test_cdaterange_holidays� s z,TestCustomDateRange.test_cdaterange_holidaysc C sx t dddddgd�}tdddg|jd �}t�||� |j|jksDt�d }tjt|d �� t ddddgd � W d Q R X d S ) Nz 2013-05-01r' r� zSun Mon Tue Wed Thu)r* r+ r� r� z 2013-05-02z 2013-05-05z 2013-05-06)r+ z[a custom frequency string is required when holidays or weekmask are passed, got frequency B)r4 )r* r� r� ) r r r+ r@ rJ r r7 r8 r< )r rq rK r> r r r �%test_cdaterange_weekmask_and_holidays� s" z9TestCustomDateRange.test_cdaterange_weekmask_and_holidaysr+ c C s g | ]}|� d �r|�qS )r� )� startswith)r[ r+ r r r r_ � s z TestCustomDateRange.<listcomp>c C sR t tt|ddgd� |d }d|� �}tjt|d�� t tt|d� W d Q R X d S )Nz Mon Wed Friz 2009-03-14)r+ r� r� ZFOOz!invalid custom frequency string: )r4 )r+ )r r� r� r7 r8 r< )r r+ Zbad_freqr> r r r �test_all_custom_freq� s z(TestCustomDateRange.test_all_custom_freq� start_end)z2018-01-01T00:00:01.000Zz2018-01-03T00:00:01.000Z)z2018-01-01T00:00:00.010Zz2018-01-03T00:00:00.010Z)z2001-01-01T00:00:00.010Zz2001-01-03T00:00:00.010Zc C s2 |\}}t ||ddd�}t|g�}t�||� d S )Nr5 r� )r; r6 r* r� )r r r@ rJ )r r� r; r6 rq rK r r r �&test_range_with_millisecond_resolution� s z:TestCustomDateRange.test_range_with_millisecond_resolutionN)r/ r0 r1 r� r� r� r� r� r� r� r7 r� r� r r� r� r r r r r� � s r� c C sB t jddgd�} tdd| d�}tddd d g| d �}t�||� d S ) Nz15:00z 2020-11-26)r; r� z2020-11-25 15:00r( )r; r* r+ z2020-11-25 15:00:00z2020-11-25 16:00:00z2020-11-27 15:00:00z2020-11-27 16:00:00)r+ )r ZCustomBusinessHourr r r@ rJ )r+ rq rK r r r �$test_date_range_with_custom_holidays s r� )-r� r r r �numpyr� r7 r" r Zpandas._libs.tslibsr Zpandas._libs.tslibs.offsetsr r r r r Z pandas.errorsr Z pandas.util._test_decorators�utilZ_test_decoratorsr2 Zpandasr= r r r r r r Zpandas._testingZ_testingr@ Z pandas.core.arrays.datetimesr r� r� r r3 r� r� r� r� r� r r r r �<module> s4 5 P6RO
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка