Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/tseries/offsets/__pycache__/test_offsets.cpython-37.pyc
Назад
B �5�g�u � @ s� d Z ddlmZ ddlmZmZ ddlmZmZmZ ddl Z ddl Z ddl m Z mZmZmZ ddlm m mZ ddlmZmZ ddlmZ dd lmZ dd l m Z dd l m Z ddl m!Z" dd l#m$Z$m%Z%m&Z& ddl'm( mZ dd l'm)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z; eee*eeef f Z<G dd� de$�Z=G dd� de$�Z>dd� Z?G dd� d�Z@dd� ZAdd� ZBG dd� d�ZCd d � ZDd d � ZEG d d!� d!�ZFd"d#� ZGd$d%� ZHe jI�Jd&eKejL��d'd(� �ZMd)d*� ZNe jI�Jd&eKejL��d+d,� �ZOe jI�Jd&eKejL��d-d.� �ZPd/d0� ZQd1d2� ZRd3d4� ZSd5d6� ZTd7d8� ZUd9d:� ZVd;d<� ZWe jI�Jd=d>d?d@dAdBg�dCdD� �ZXe jI�JdEdFdGdHgdIdJdKgg�dLdM� �ZYdS )Nz! Tests of pandas.tseries.offsets � )� annotations)�datetime� timedelta)�Dict�List�TupleN)�NaT� Timestamp� conversion� timezones)� _get_offset� _offset_map)�INVALID_FREQ_ERR_MSG)�np_datetime64_compat)�PerformanceWarning)� DatetimeIndex)�Base�WeekDay�assert_offset_equal)�FY5253� BaseOffset�BDay� BMonthEnd� BusinessHour�CustomBusinessDay�CustomBusinessHour�CustomBusinessMonthBegin�CustomBusinessMonthEnd� DateOffset�Day�Easter� FY5253Quarter�LastWeekOfMonth� MonthBegin�Nano�Tick�Week� WeekOfMonthc $ @ st e Zd Zed�ed�ed�ed�ed�ed�ed�ed�ed�ed�ed�ed�ed�ed�ed�ed�ed �ed �ed �ed �ed �ed �ed �ed �ed�ed�ed �ed�ed�ed�ed�ed�ed�eed��d�"Zdd� Zdd� Zdd � Zd d � Z d d � Z d!d"� Z d=d$d%�Z d&d'� Z d(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Zd4d5� Zd6d7� Zd8d9� Zd:d;� Zd<S )>� TestCommonz2011-01-02 09:00:00z2011-01-03 09:00:00z2011-01-31 09:00:00z2011-02-01 09:00:00z2011-01-15 09:00:00z2012-01-01 09:00:00z2011-12-31 09:00:00z2011-12-30 09:00:00z2011-03-01 09:00:00z2011-03-31 09:00:00z2011-01-03 10:00:00z2011-01-08 09:00:00z2011-01-29 09:00:00z2011-01-25 09:00:00z2011-04-24 09:00:00z2011-01-01 10:00:00z2011-01-01 09:01:00z2011-01-01 09:00:01z2011-01-01 09:00:00.001000z2011-01-01 09:00:00.000001z 2011-01-01T09:00:00.000000001Z)"r r � BusinessDayr r r r# �BusinessMonthBegin�MonthEnd� SemiMonthEnd�SemiMonthBegin�BusinessMonthEnd� YearBegin� BYearBegin�YearEnd�BYearEnd� QuarterBegin� BQuarterBegin� QuarterEnd� BQuarterEndr r r' r" r! r r& r �Hour�Minute�Second�Milli�Micror$ c C sR | � |�}d}tjt|d�� d|_W d Q R X tjt|d�� d|_W d Q R X d S )Nz8objects is not writable|DateOffset objects are immutable)�matchT�[ )r �pytest�raises�AttributeError� normalize�n)�self� offset_types�offset�msg� rG �_/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/tseries/offsets/test_offsets.py�test_immutablem s zTestCommon.test_immutablec C sl | � |�}td�| }t|t�s$t�t| tks4t�|t tksDt�t| tksTt�| �t�tksht�d S )NZ20080101)r r � isinstance�AssertionErrorr �apply)rC rD rE �resultrG rG rH �test_return_typev s zTestCommon.test_return_typec C sH | � |�}|jdkst�|d }|jdks.t�|d }|jdksDt�d S )N� ���� )r rB rK )rC rD rE Z neg_offsetZ mul_offsetrG rG rH � test_offset_n� s zTestCommon.test_offset_nc C sL | � |�}t�dd�}tjtdd�� t|�f d|i|j�� W d Q R X d S )Ni� �szargument must be an integer)r< rB )r �npZ timedelta64r> r? � TypeError�type�kwds)rC rD �offZtd64rG rG rH �test_offset_timedelta64_arg� s z&TestCommon.test_offset_timedelta64_argc C s| | � |�}t�||d g|d |d gg�}t�ddgddgg�| }t�||� |t�ddgddgg� }t�||� d S )N� rQ � rO )r rT �array�tmZassert_numpy_array_equal)rC rD rX �expectedrM rG rG rH �test_offset_mul_ndarray� s " z"TestCommon.test_offset_mul_ndarrayc C s2 | � |�}|j}|dkr.t |�}|j|ks.t�d S )N)z<Easter>z<DateOffset: days=1>zLWOM-SAT)r �freqstr� rule_coderK )rC rD rE r` �coderG rG rH �test_offset_freqstr� s z TestCommon.test_offset_freqstrFc C sL |rt |t�rd S | j||d�}t||�}||�}t|t�s@t�||ksLt�|t|��}t|t�sft�||ksrt�d } t|�td� } t|�j dkr�|dks�|r�| j dkr�t } t � | �� || �}W d Q R X t|t�s�t�|dkr�||td� ks�t�n||k�st�t|tj��rd S �x,| jD �] } |�| �} t�| �} t�|| �}||�}t|t��s`t�|| k�snt�|t|| d��}t|t��s�t�|| k�s�t�d } t|| d�td� } t|�j dk�r�|dk�s�|�r�| j dk�r�t } t � | �� || �}W d Q R X t|t��st�|dk�r4|| td� k�sBt�n|| k�s"t��q"W d S )N)rA � r rL r F)�tz)� issubclassr% r �getattrrJ r rK r$ rV �__name__Z nanosecond� UserWarningr] �assert_produces_warningrT Z datetime64r � tz_localizeZ maybe_get_tzr Zlocalize_pydatetime)rC rE �funcname�dtr^ rA �offset_s�funcrM Z exp_warning�tsre �expected_localizeZtz_objZdt_tzrG rG rH �_check_offsetfunc_works� s\ z"TestCommon._check_offsetfunc_worksc C sj t ddddd�}td�}xL||gD ]@}| j|j }| �|d||� t|�� �}| j|d||dd� q"W d S ) Ni� rO � r z2011-01-01 09:00ZrL T)rA )r r � expectedsrh rr r �date)rC rD �sdt�ndtrm r^ rG rG rH � test_apply� s zTestCommon.test_applyc C s: | j �� }ddddddddd d d d g }x|D ]}td �||< q,W td�|d< td�|d< |�� }x |D ]}t|| �� �||< qfW td�td�td�td�td�td�td�td�td�td�td�d� }|�|� tddddd�}td �} xH|| gD ]<} ||j } | �|d | | � ||j } | j|d | | d d � q�W d S ) Nr r# r- r/ r& r7 r8 r9 r: r; r$ r z2011/01/01 09:00z2011-01-03 09:00:00r r z2011-01-02 00:00:00z2011-02-01 00:00:00z2011-01-15 00:00:00z2012-01-01 00:00:00z2011-01-08 00:00:00z2011-01-01 00:00:00) r r r# r- r/ r& r7 r8 r9 r: r; i� rO rs r z2011-01-01 09:00ZZ rollforwardT)rA ) rt �copyr ru �updater r rh rr ) rC rD rt Z no_changesrB � norm_expected�k� normalizedrv rw rm r^ rG rG rH �test_rollforward sP zTestCommon.test_rollforwardc C s� t d�t d�t d�t d�t d�t d�t d�t d�t d�t d�t d�t d�t d�t d�t d�t d�t d�t d�t d�t d�t d�t d �d �}xd D ]}t d �||< q�W |�� }x |D ]}t || �� �||< q�W t d �t d �t d�t d�t d�t d�t d�t d�t d�t d�t d�d� }|�|� tddddd�}td�}xJ||gD ]>} ||j } | �|d| | � ||j } | j|d| | dd� �qBW d S ) Nz2010-12-31 09:00:00z2010-12-01 09:00:00z2010-01-01 09:00:00z2010-12-31 17:00:00z2010-12-11 09:00:00z2010-12-25 09:00:00z2010-10-26 09:00:00z2010-01-26 09:00:00z2010-04-04 09:00:00)r) r r r r* r+ r, r. r0 r1 r2 r3 r4 r5 r6 r r r' r" r! r r ) r r# r- r/ r& r7 r8 r9 r: r; r$ r z2011/01/01 09:00z2010-12-31 00:00:00z2010-12-01 00:00:00z2010-12-15 00:00:00z2010-01-01 00:00:00z2010-12-25 00:00:00z2011-01-01 00:00:00) r r r# r- r/ r& r7 r8 r9 r: r; i� rO rs r z2011-01-01 09:00Z�rollbackT)rA )r ry ru rz r r rh rr ) rC rD rt rB r{ r| r} rv rw rm r^ rG rG rH � test_rollback8 s^ zTestCommon.test_rollbackc C s� | j |j }| �|�}|�|�s$t�t|t�r2d S | j|dd�}|�|�rNt�|ttfkr^d S t |j |j |j �}|�|�s~t�d S )NT)rA ) rt rh r � is_on_offsetrK rf r% r r r �year�month�day)rC rD rm rn Zoffset_nru rG rG rH �test_is_on_offset� s z TestCommon.test_is_on_offsetc C sB |}t ddddd�}| �|�}| j|j }|| }t|�| }x*||gD ] } t| t�sZt�| |ksHt�qHW |�|�} t||d�| } t| t�s�t�| | ks�t�t|t �r�d S | j|dd�}t|� � �}|| }t|�| }x*||gD ] } t| t�s�t�| |ks�t�q�W |�|�} t||d�| } t| t��s0t�| | k�s>t�d S )Ni� rO rs r )re T)rA ) r r rt rh r rJ rK rk rf r% ru ) rC rD �tz_naive_fixturere rm rn r^ Z result_dtZ result_tsrM rq rG rG rH �test_add� s4 zTestCommon.test_addc C s� | � |�}tg |d�}d }t|ttttttt t t t f �r<t }t�|�� || }W d Q R X t�||� t�|�� || }W d Q R X t�||� |j}t�|�� || }W d Q R X t�||� t�|�� || }W d Q R X t�||� d S )N)re )r r rJ r r' r" r r r r r r r! r r] rj Zassert_index_equal�_dataZ assert_equal)rC rD r� rn Zdti�warnrM ZdtarG rG rH � test_add_empty_datetimeindex� s: z'TestCommon.test_add_empty_datetimeindexc C sb | � |�}t�|�}||ks t�t|�tk r^x0|jD ]&}|dkrBq4t||�t||�ks4t�q4W d S )N�calendar)r r] �round_trip_picklerK rV r � _attributesrg )rC rD rX �res�attrrG rG rH �test_pickle_roundtrip� s z TestCommon.test_pickle_roundtripc C sD t dd�}t�|�}||ks t�tddd�}|| || ks@t�d S )N� )�monthsi� rO )r r] r� rK r )rC rX r� Zbase_dtrG rG rH �!test_pickle_dateoffset_odd_inputs� s z,TestCommon.test_pickle_dateoffset_odd_inputsc C sL | � |�}t�� }t�t�� |�|�}W d Q R X |�|�}||ksHt�d S )N) r r �nowr] rj � FutureWarningZonOffsetr� rK )rC rD rX rp rM r^ rG rG rH �test_onOffset_deprecated� s z#TestCommon.test_onOffset_deprecatedc C s@ | � |�}t�t�� |�� }W d Q R X |�� }||ks<t�d S )N)r r] rj r� Z isAnchored� is_anchoredrK )rC rD rX rM r^ rG rG rH �test_isAnchored_deprecated s z%TestCommon.test_isAnchored_deprecatedc C s | � |�}t|�d k st�d S )N)r �hashrK )rC rD rX rG rG rH �test_offsets_hashable s z TestCommon.test_offsets_hashableN)F)rh � __module__� __qualname__r r rt rI rN rR rY r_ rc rr rx r~ r� r� r� r� r� r� r� r� r� rG rG rG rH r( D sh J 7K#( r( c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )�TestDateOffsetc C s t tddd��| _t�� d S )Ni� rO rZ )r r �dr �clear)rC �methodrG rG rH � setup_method s zTestDateOffset.setup_methodc C s: t t� � t td�� t dt� � t dtdd� � d S )NrZ )r� )�reprr )rC rG rG rH � test_repr s zTestDateOffset.test_reprc C s4 t d�dt d� kst�t d�t d�d ks0t�d S )NrZ rO )r rK )rC rG rG rH �test_mul s zTestDateOffset.test_mulc C s� | j tdd� tddd�ks t�| j tdd� tddd�ks@t�| j td� tddd�ks^t�td��� rnt�td��� s~t�tddd �}|tdd� tddd �ks�t�d S ) NrZ )r� i� rQ i� � rO r[ � � )r� r r rK r� )rC r� rG rG rH �test_constructor! s z TestDateOffset.test_constructorc C s t dd��� t dd�ks t�d S )NrZ )r� )r ry rK )rC rG rG rH � test_copy. s zTestDateOffset.test_copyc C s$ t dd�}t dd�}||ks t�d S )NrO )�daysim )r rK )rC Zoffset1Zoffset2rG rG rH �test_eq1 s zTestDateOffset.test_eqN) rh r� r� r� r� r� r� r� r� rG rG rG rH r� s r� c C sB t t� tddd�tddd�� t t� tddd�tddd�� t td�tddd�tddd�� t t� tddd�tddd�� t td�tddd�tddd �� t t� tddd�tddd�� t t� tddd�tddd�� t td� tddd�td dd �� t t� tddd�td dd �� t td� tddd�td d d�� d S )Ni� rO r[ rd i� � rZ i� � i� r� i� rQ � )r r r rG rG rG rH � test_Easter8 s " r� c @ s e Zd Zdd� ZdS )�TestOffsetNamesc C s� t � jdkst�t d�jdks"t�t� jdks2t�tdd�jdksFt�tdd�jd ksZt�tdd�jd ksnt�td d�jd ks�t�td d�jdks�t�ttjd�jdks�t�d S )N�BrZ Z2B�BMr )�weekdayzW-MONrO zW-TUEzW-WEDrQ zW-THUr[ zW-FRIzLWOM-SUN)r r` rK r r& r" r �SUN)rC rG rG rH �test_get_offset_nameI s z$TestOffsetNames.test_get_offset_nameN)rh r� r� r� rG rG rG rH r� H s r� c C s� t jttd�� td� W d Q R X t jttd�� td� W d Q R X dt� fdt� fdt� fdt� fdtd d �fd td d �fd tdd �fdtdd �fdtdd �fg } xH| D ]@\}}t|�}||ks�tdt |�� dt |�� dt |�� d���q�W d S )N)r< Z gibberishzQS-JAN-Br� �bZbmZBmzW-MONr )r� zW-TUErO zW-WEDrZ zW-THUrQ zW-FRIr[ z Expected z to yield z (actual: �)) r> r? � ValueErrorr r r r r& rK r� )�pairs�namer^ rE rG rG rH �test_get_offsetV s" r� c C sH dt dd�fg} x2| D ]*\}}tjttd�� t|� W d Q R X qW d S )Nzw@Satrd )r� )r< )r& r> r? r� r r )r� r� r^ rG rG rH �test_get_offset_legacyp s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestOffsetAliasesc C s t �� d S )N)r r� )rC r� rG rG rH r� x s z TestOffsetAliases.setup_methodc C s4 x.t �� D ]"\}}|d kr q ||�� ks t�q W d S )N)r �itemsry rK )rC r| �vrG rG rH �test_alias_equality{ s z%TestOffsetAliases.test_alias_equalityc C s> ddddddddd d d g }x@|D ]8}|t |�jks6t�|tksBt�|t |�d jks t�q W d ddddddg}d}xB|D ]:}d�||g�}|t |�jks�t�|t |�d jksxt�qxW dddddd d d d d d!d"g }d#d$d%d&d'd(d)d*g}xR|D ]J}xD|D ]<}d�||g�}|t |�jk�s t�|t |�d jks�t�q�W q�W d S )+N�MZMSr� ZBMS�Dr� �H�T�S�L�UrQ �MON�TUE�WED�THU�FRI�SATr� �W�-rd �JAN�FEB�MAR�APR�MAY�JUN�JUL�AUG�SEP�OCT�NOV�DEC�A�AS�BA�BAS�Q�QS�BQ�BQS)r ra rK r �join)rC �lstr| Z suffix_lst�baser� �aliasZbase_lstrG rG rH �test_rule_code� s: z TestOffsetAliases.test_rule_codeN)rh r� r� r� r� r� rG rG rG rH r� w s r� c C s, t jddd�} | j t jdd�dkr(t�d S )NrZ r[ )r� r� )r� )�offsetsr r` rK )ZosetrG rG rH �test_dateoffset_misc� s r� c C sD t dtdd�d�} | jdks t�t dtdd�d�} | jdks@t�d S )NrO r i )rE zB+30Mini����zB-30Min)r r r` rK )rX rG rG rH �test_freq_offsets� s r� c @ s e Zd Zdd� ZdS )� TestReprNamesc s� ddddddddg}d d � |D �}d d d ddddg� |dd � � D �7 }|� fdd �dD �7 }t �� x"|D ]}t|�}|j|ksjt�qjW d S )Nr� r� r� r� r� r� r� r� c S s" g | ]}d D ]}|d | �q qS )) r� r� r� r� r� r� r� r� r� r� r� r� r� rG )�.0�prefixr� rG rG rH � <listcomp>� s z<TestReprNames.test_str_for_named_is_name.<locals>.<listcomp>r� r� r� r� r� r� r� c S s g | ] }d | �qS )zW-rG )r� r� rG rG rH r� � s c s" g | ]}� D ]}d | | �q qS )zWOM-rG )r� �weekr� )r� rG rH r� � s )�1�2�3�4)r r� r r` rK )rC Zmonth_prefixes�namesr� rE rG )r� rH �test_str_for_named_is_name� s z(TestReprNames.test_str_for_named_is_nameN)rh r� r� r� rG rG rG rH r� � s r� c C s | � � }|jd d |j d S )Nr� i g �@)� utcoffsetr� �seconds)rp �orG rG rH �get_utc_offset_hours� s r� c C s | }|� d S )NrG )rD �clsrG rG rH � test_valid_default_arguments� s r� �kwdc C s< |}d| � d�}t jt|d�� |f | di� W d Q R X d S )Nz1__init__\(\) got an unexpected keyword argument '�')r< rQ )r> r? rU )r� � month_classesr� rF rG rG rH �test_valid_month_attributes� s r c C s$ | d�}| d�}|j |j ks t�d S )NrO rZ )r� rK )r� �obj�obj2rG rG rH �test_month_offset_name� s r c C s t f | di� d S )NrO )r )r� rG rG rH � test_valid_relativedelta_kwargs� s r c C s< |}d| � d�}t jt|d�� |f | di� W d Q R X d S )Nz1__init__\(\) got an unexpected keyword argument 'r� )r< rQ )r> r? rU )r� � tick_classesr� rF rG rG rH �test_valid_tick_attributes� s r c C s� t jtdd�� tdd� W d Q R X t jtdd�� ttd�d� W d Q R X t jtdd�� ttjddgtj d�d� W d Q R X d S )Nzargument must be an integer)r< zDoh!)rB rO rZ )Zdtype) r> r? rU r r# r r rT r\ Zint64rG rG rG rH �test_validate_n_error s r c C s, | }t jtdd�� |dd� W d Q R X d S )Nzargument must be an integer)r< g �?)rB )r> r? r� )rD r� rG rG rH �test_require_integers s r c C s2 | }d}t jt|d�� |ddd� W d Q R X d S )Nz2Tick offset with `normalize=True` are not allowed.)r< rQ T)rB rA )r> r? r� )r r� rF rG rG rH �test_tick_normalize_raises s r c C s� t dd d�} tddd�}| �|�}|| | |k}||ks>t�t dd d�} tddd�}| �|�}|| | |k}||ks|t�d S )NrZ )rB r� z"1862-01-13 09:03:34.873477378+0210z Africa/Lusaka)re z"1856-10-24 16:18:36.556360110-0717zPacific/Easter)r& r r� rK )rE rp �fast�slowrG rG rH �test_weeks_onoffset! s r c C s� t dddd�} tddd�}| �|�}|| | |k}||ks@t�t dddd�} td d d�}| �|�}|| | |k}||ks�t�d S ) NrZ r )rB r� r� z"1916-05-15 01:14:49.583410462+0422zAsia/Qyzylorda)re ���rO z"1980-12-08 03:38:52.878321185+0500z Asia/Oral)r' r r� rK )rE rp r r rG rG rH �test_weekofmonth_onoffset2 s r c C s� t ddd�} tddd�}|| | |k}| �|�}||ks>t�t ddd�} td d d�}|| | |k}| �|�}||ks|t�d S ) Nr[ � )rB r� z"1917-05-27 20:55:27.084284178+0200z Europe/Warsaw)re ���rd z"2005-08-27 05:01:42.799392561-0500zAmerica/Rainy_River)r" r r� rK )rE rp r r rG rG rH �!test_last_week_of_month_on_offsetC s r c C s6 t dd�} t� }tjtdd�� | | W d Q R X d S )NrO )r� z Cannot add)r< )r& r r> r? rU )rE �otherrG rG rH �test_week_add_invalidS s r � attribute�hoursr� �weeksr� Zyearsc C s< t f | di�}d}tjt|d�� t|| d� W d Q R X d S )Nr z DateOffset objects are immutable)r< rd )r r> r? r@ �setattr)r rE rF rG rG rH �test_dateoffset_immutable\ s r z weekmask, expected_time, multzMon Tue Wed Thu Fri Satz2018-11-10 09:00:00� zTue Wed Thu Fri Satz2018-11-13 08:00:00� c C s> dg}t dd| |d�}td�|| }t|�}||ks:t�d S )Nz 2018-11-09z08:00z17:00)�start�end�weekmask�holidaysz2018-11-08 08:00)r r rK )r Z expected_timeZmultr ZbhrM r^ rG rG rH �.test_custom_businesshour_weekmask_and_holidaysm s r )Z�__doc__� __future__r r r �typingr r r �numpyrT r> Zpandas._libs.tslibsr r r r Zpandas._libs.tslibs.offsetsZ_libsZtslibsr� Z liboffsetsr r Zpandas._libs.tslibs.periodr Z pandas.compatr Z pandas.errorsr Zpandasr Zpandas._testingZ_testingr] Z#pandas.tests.tseries.offsets.commonr r r Zpandas.tseries.offsetsZtseriesr r r r r r r r r r r r r! r" r# r$ r% r& r' Z _ApplyCasesr( r� r� r� r� r� r� r� r� r� r� r� �markZ parametrize�sortedZ_relativedelta_kwdsr r r r r r r r r r r r r rG rG rG rH �<module> sl T Q&/
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка