Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/indexes/period/__pycache__/test_constructors.cpython-37.pyc
Назад
B �5�g�O � @ s� d dl Zd dlZd dlmZ d dlmZ d dlmZm Z m Z m Z m Z m Z mZmZ d dlmZ d dlmZ G dd� d�ZG dd � d �ZG d d � d �ZdS ) � N)�IncompatibleFrequency)� PeriodDtype)�Index�NaT�Period� PeriodIndex�Series� date_range�offsets� period_range)� PeriodArrayc @ s� e Zd Zdd� Zdd� Zej�dee j e e j g�dd� �Zdd � Zd d � Zd d � Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zej�ddd d g�d d � �Zd d!� Zd"d#� Zd$d%� Zd&d'� Z d(d)� Z d*d+� Z d,d-� Z ej�d.d/d0ge � d/d0g�g�d1d2� �Z d3d4� Z!d5d6� Z"d7d8� Z#ej�d9d:d;d<d=d>g�ej�d?d@dAdBdCdDg�dEdF� ��Z$dGdH� Z%dIdJ� Z&ej�d9d;dKd:d<dLd=d>dMdNdOdPg �dQdR� �Z'dSdT� Z(dS )U�TestPeriodIndexc C s� t ddd�tt ddd�g}t�t|�t|�� t�tt�|��tt�|��� tjtt ddd�g}t�t|�t|�� t�tt�|��tt�|��� t ddd�tt ddd�g}t�t|�t|t d�� t�tt�|��tt�|�t d�� d S )Nz2011-01�M)�freqz2011-03�D)�dtype) r r �tm�assert_index_equalr r �np�array�nan�object)�self�arr� r �c/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/indexes/period/test_constructors.py�"test_construction_base_constructor s z2TestPeriodIndex.test_construction_base_constructorc C s: t d�}dddg}t||d�}t||d�}t�||� d S )Nr z 2011-01-01z 2012-03-04z 2014-05-01)r )r r r r r )r r �values�result�expectedr r r �'test_base_constructor_with_period_dtype& s z7TestPeriodIndex.test_base_constructor_with_period_dtype�values_constructorc C sV t ddd�tt ddd�g}||�}t|td�}t|�tks>t�t�|jt � |�� d S )Nz2011-01r )r z2011-03)r ) r r r r �type�AssertionErrorr �assert_numpy_array_equalr r r )r r! �periodsr r r r r �test_index_object_dtype. s z'TestPeriodIndex.test_index_object_dtypec C s6 t ddd�}tdddd�}t|dd�}t�||� d S )Nz4/2/2012�B)r � )�startr% r )r) r% )r r r r )r �pr �indexr r r � test_constructor_use_start_freq> s z/TestPeriodIndex.test_constructor_use_start_freqc C s: t �dd��d�dd� }t �t �dd�d�dd� }t||d d �}td d d d �}t�||� t||dd �}t�|j |j � t||d�}t�||� dddg}ddg}d}t j t |d�� t||dd� W d Q R X t j t |d�� t||dd� W d Q R X dddg}dddg}t||dd�}tdddd�} t�|| � d S )Ni� i� � � ���� � � zQ-DEC)�year�quarterr Z1990Q3Z2009Q2)r z2Q-DEC)r3 r4 i� z Mismatched Period array lengths)�matchr )r3 �monthr �2M� z2007-01)r% r ) r �arange�repeat�tiler r r r r$ �asi8�pytest�raises� ValueError) r �years�quartersr+ r �index2�months�msg�idx�expr r r � test_constructor_field_arraysF s* z-TestPeriodIndex.test_constructor_field_arraysc C s, t jtdd�� tdddd� W d Q R X d S )NzInvalid frequency: X)r5 z2007-1-1i� �X)r% r )r= r> r? r )r r r r �test_constructor_Ue s z"TestPeriodIndex.test_constructor_Uc C sb t tddd�tddd�dd�}ttddd�tddd�tddd�tddd�gdd�}t�||� d S ) Nr0 �N)Zordinalr r- )r) �endr r. r8 )r )r r r r r )r rE rF r r r �test_constructor_nanoj s z%TestPeriodIndex.test_constructor_nanoc C sl t jddt jd��d�}t �t jddddgt jd�d�}t||d �}t�|j t |�� t�|j t |�� d S ) Ni� i� )r r- r0 r. r8 �( )r3 r4 ) r r9 �int64r: r; r r r r r3 r r4 )r r@ rA Zpindexr r r �%test_constructor_arrays_negative_yeary s z5TestPeriodIndex.test_constructor_arrays_negative_yearc C s> d}t jt|d��" ttdd�ttd��dd� W d Q R X d S )NzQuarter must be 1 <= q <= 4)r5 i� i� r- zQ-DEC)r3 r4 r )r= r> r? r �range�list)r rD r r r �!test_constructor_invalid_quarters� s z1TestPeriodIndex.test_constructor_invalid_quartersc C s, t dddd�}t dddd�}t�||� d S )Nz2007-01g %@r )r% r r( )r r r )r r rF r r r �test_constructor_corner� s z'TestPeriodIndex.test_constructor_cornerc C s� t dddd�}t�t|j�|� t�tt|j��|� d}tjt|d�� t|j � W d Q R X tjt|d�� tt|j �� W d Q R X d}tjt |d�� tt dd d �d � W d Q R X tt |��}t�||� t|�}t�||� t|dd �}t�||� t|t �� d �}t�||� |jdk�s&t�t|d d �}t�||�d �� |jd k�sTt�t|t �d �d �}t�||�d �� |jd k�s�t�t|dd �}|�dd�}t�||� d S )Nz2007-01r2 r )r% r z)freq not specified and cannot be inferred)r5 z 'Period' object is not iterableZ2007�A)r )�datar7 r. r �e)r r r r r rQ r= r> r? r< � TypeErrorr �iterr ZMonthEndr r# Zasfreq)r rE rD r rF r r r � test_constructor_fromarraylike� s: z.TestPeriodIndex.test_constructor_fromarraylikec C sR t jdddt jd�}|�t �d��}d}tjt|d�� t|dd � W d Q R X d S ) Ni�� i�� �d )r zM8[us]z Wrong dtype: datetime64\[us\])r5 r )r ) r r9 rN �viewr r= r> r? r )r �valsrD r r r � test_constructor_datetime64arr� s z.TestPeriodIndex.test_constructor_datetime64arr�boxN�seriesr+ c C s^ t dddd�}|d kr |j}n|dkr.t|�}t|dd�}tdd d d gdd�}t�||� d S ) NZ2017r- r )r% r r_ r )r z 2017-01-31z 2017-02-28z 2017-03-31z 2017-04-30)r Z_valuesr r r r )r r^ rU r r r r r �!test_constructor_datetime64arr_ok� s z1TestPeriodIndex.test_constructor_datetime64arr_okc C s t ddgdd�}t ddgdd�}t�||� |jdks:t�t ddgd d�}t ddgd d�}t�||� |jd kstt�t d d gd d�}t |dd�}t ddgdd�}t�||� |jdks�t�t |dd�}t�||� |jdks�t�d}tjt|d�� t dgddd� W d Q R X d S )Nz2013-01z2013-03z period[M])r r )r z 2013-01-05z 2013-03-05z period[3D]�3Dz 2013-01-01z 2013-01-02r z&specified freq and dtype are different)r5 z2011-01z period[D])r r )r r r r r# r= r> r )r rE rF �resrD r r r �test_constructor_dtype� s& z&TestPeriodIndex.test_constructor_dtypec C s^ t g dd�}t|t �st�t|�dks*t�|jdks8t�tjtdd�� t g � W d Q R X d S )Nr )r r zfreq not specified)r5 )r � isinstancer# �lenr r= r> r? )r rE r r r �test_constructor_empty� s z&TestPeriodIndex.test_constructor_emptyc C s� t tddd�ttddd�g�}t dddgdd�}t�||� t t�tddd�ttddd�g��}t�||� t tttddd�tddd�g�}t ddddgdd�}t�||� t t�tttddd�tddd�g��}t�||� t ttddgdd�}t�||� tjt dd�� t ttg� W d Q R X tjt dd�� t t�ttg�� W d Q R X tjt dd�� t ddg� W d Q R X tjt dd�� t t�ddg�� W d Q R X d S )Nz2011-01r )r r zfreq not specified)r5 ) r r r r r r r r= r> r? )r rE rF r r r �test_constructor_pi_nat� s2 " z'TestPeriodIndex.test_constructor_pi_natc C s� d}t jt|d��$ ttddd�ttddd�g� W d Q R X t jt|d��* tt�tddd�ttddd�g�� W d Q R X t jt|d��$ tttddd�tddd�g� W d Q R X t jt|d��* tt�ttddd�tddd�g�� W d Q R X d S )Nz5Input has different freq=D from PeriodIndex\(freq=M\))r5 z2011-01r )r r )r= r> r r r r r r )r rD r r r � test_constructor_incompat_freq s (((z.TestPeriodIndex.test_constructor_incompat_freqc C s� t dttddd�g�}t dddgdd�}t�||� t dttddd�g�}t dddgdd�}t�||� t tddd�tdg�}t dddgdd�}t�||� d S )Nz2011-01r )r r z 2011-01-01r z 2012-01-01)r r r r r )r rE rF r r r �test_constructor_mixed3 s z&TestPeriodIndex.test_constructor_mixedc C s� t ddddd�}tjtdd�� |j|dd� W d Q R X |j|jdd�}t�||� d }tjt|d��" t|j�jt |j �|j d � W d Q R X t|j�j|j |j d �}|j|dd�}t�||� d S ) Nz2007-01r* r. r )�namer% r z<class .*PeriodIndex'>)r5 )rj z Should be numpy array of type i8)r ) r r= r> r# � _simple_new�_datar r r" r r< r )r rE r rD r r r r �test_constructor_simple_new@ s &z+TestPeriodIndex.test_constructor_simple_newc C sV t g ddd�}tjtdd�� |j|dd� W d Q R X |j|jdd�}t�||� d S )Nr r* )r rj z<class .*PeriodIndex'>)r5 )rj )r r= r> r# rk rl r r )r rE r r r r �!test_constructor_simple_new_emptyR s z1TestPeriodIndex.test_constructor_simple_new_empty�floatsg�������?g������ @c C sN t jtdd�� t�|� W d Q R X d}t jt|d�� t|� W d Q R X d S )Nz<class )r5 z9PeriodIndex does not allow floating point in construction)r= r> r# r rk rW )r ro rD r r r �test_constructor_floats[ s z'TestPeriodIndex.test_constructor_floatsc C sX d}t jt|d�� tdddd� W d Q R X t jt|d�� tdddd� W d Q R X d S )Nz start and end must not be NaT)r5 r z 2011-01-01r )r) rK r )r= r> r? r )r rD r r r �test_constructor_natd s z$TestPeriodIndex.test_constructor_natc C s\ t dddg�}|d }t||d�}dd� t||�D �}ttt|��}t|�}t�||� d S )Ni� i� i� i� )r3 r4 c S s& g | ] }|d d�d|d d���qS )r �d�Qr0 r )�.0�tr r r � <listcomp>o s zETestPeriodIndex.test_constructor_year_and_quarter.<locals>.<listcomp>)r r �ziprQ �mapr r r )r r3 r4 rE �strsZlopsr* r r r �!test_constructor_year_and_quarterk s z1TestPeriodIndex.test_constructor_year_and_quarterc C s t dddd�}tddddgdd�}t�||� t d d d d �}td d dddgd d�}t�||� t dddd�}tdddgdd�}t�||� d}tjt|d�� tdgdd� W d Q R X d}tjt|d�� tdgd d� W d Q R X d}tjt|d�� t ddd d � W d Q R X d S ) Nz2014-01r7 r- )r) r r% z2014-03z2014-05z2014-07)r z 2014-01-02z 2014-01-15ra )r) rK r z 2014-01-05z 2014-01-08z 2014-01-11z 2014-01-14z2014-01-01 17:00Z4Hr8 )rK r r% z2014-01-01 09:00z2014-01-01 13:00z;Frequency must be positive, because it represents span: -1M)r5 z2011-01z-1Mz:Frequency must be positive, because it represents span: 0MZ0M)r% r )r r r r r= r> r? )r �pidxr rD r r r �test_constructor_freq_multt s* z*TestPeriodIndex.test_constructor_freq_multr rT r r �T�S�multr0 r. r8 r- r1 c C s> t |�| }td|dd�}td|dd��|�}t�||� d S )Nz 2014-04-01r( )r) r r% )�strr r Z to_periodr r )r r r Zfreqstrr{ r r r r �%test_constructor_freq_mult_dti_compat� s z5TestPeriodIndex.test_constructor_freq_mult_dti_compatc C sj x,dD ]$}t ddg|d�}t ddgdd�}qW x6dD ].}tdd|d �}t dd gdd�}t�||� q4W d S ) N)Z1D1HZ1H1Dz 2016-01-01z 2016-01-02)r z2016-01-01 00:00z2016-01-02 00:00Z25Hr. )r) r% r z2016-01-02 01:00)r r r r )r r r{ r r r r � test_constructor_freq_combined� s z.TestPeriodIndex.test_constructor_freq_combinedc C s� t dddd�}t|�dks t�t dddd�}t|�dks<t�t dddd�}t|�d ksZt�t d dd d�}t|�d ksxt�t d dd d�}t|�dks�t�t dddd�}t|�dks�t�t dddd�}t|�dks�t�t dddd�}t|�dks�t�tdd �}t |dd�}t|�dk�st�|j|jk�s*t�|d |k�s<t�td d �}t |d d �}t|�d k�sdt�|j|jk�svt�|d |k�s�t�td d!�}t |d d �}t|�t|�k�s�t�||k�� �s�t�|j|jk�s�t�td"d �}t ||d#�}t|td$d �g�}t|�d%k�st�|d |k�s$t�tt�|td$d �g��}t|�d%k�sNt�|d |k�s`t�|td d&�g}d'}t j t |d(�� t|� W d Q R X t�|�}t j t |d(�� t|� W d Q R X t j t d)d(�� td d*� W d Q R X d S )+NrT z1/1/2001z 12/1/2009)r r) rK � rs �$ r �l r z 12/31/2009i� r' i- �Hz12/31/2001 23:00i8" ZMinz1/1/2001 23:59i� r~ z1/1/2001 23:59:59i�Q z 02-Apr-2005r2 )r) r% r z 2006-12-31�Wr( )rK r% ���Z1wz 2005-05-01)r) rK z 2005-05-05r. �wz9Input has different freq=W-SUN from PeriodIndex\(freq=B\))r5 zpass as a string instead)r� r0 ) r re r# r r �allr r r r= r> r rW )r �pir) �i1Zend_intv�i2r\ rD r r r �test_constructor� s` z TestPeriodIndex.test_constructorrs r' �L�UrJ r� c C s, t d|dd�}t|j|d�}t�||� d S )Nz 2001/04/01r0 )r) r r% )r )r r r r r )r r �orgrE r r r �test_recreate_from_data� s z'TestPeriodIndex.test_recreate_from_datac C sf dddg}t |dd�}tdd� |D ��}|�t�}t|t�s@t�tdd � |D ��sVt�t�||� d S ) Ni� i� i� rT )r c S s g | ] }t |��qS r )r� )rt �numr r r rv � s zDTestPeriodIndex.test_map_with_string_constructor.<locals>.<listcomp>c s s | ]}t |t�V qd S )N)rd r� )rt Zresir r r � <genexpr>� s zCTestPeriodIndex.test_map_with_string_constructor.<locals>.<genexpr>) r r rx r� rd r# r� r r )r �rawr+ r rb r r r � test_map_with_string_constructor� s z0TestPeriodIndex.test_map_with_string_constructor))�__name__� __module__� __qualname__r r r= �markZ parametrizerQ r r r r Z_from_sequencer& r, rG rI rL rO rR rS rY r] r` rc rf rg rh ri rm rn rp rq rz r| r� r� r� r� r� r r r r r s@ * ( ( " D&r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestShallowCopyc C s( t g dd�}|�� }|}t�||� d S )Nr )r )r Z_viewr r )r rE r r r r r �test_shallow_copy_empty s z'TestShallowCopy.test_shallow_copy_emptyc C s8 t dddd�}tjtdd�� |�|j� W d Q R X d S )Nz 2018-01-01r8 �2D)r% r Zndarray)r5 )r r= r> r# � _shallow_copyr< )r r� r r r � test_shallow_copy_disallow_i8 s z-TestShallowCopy.test_shallow_copy_disallow_i8c C s6 t dddd�}tjtdd�� |�|� W d Q R X d S )Nz 2018-01-01r8 r� )r% r r )r5 )r r= r> r# r� )r r� r r r �0test_shallow_copy_requires_disallow_period_index s z@TestShallowCopy.test_shallow_copy_requires_disallow_period_indexN)r� r� r� r� r� r� r r r r r� s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestSeriesPeriodc C s t tdddd��| _d S )Nz 2000-01-01r( r )r% r )r r r_ )r �methodr r r � setup_method s z TestSeriesPeriod.setup_methodc C s8 d}t jt|d�� ttdddd�td� W d Q R X d S )Nz(Cannot cast PeriodIndex to dtype float64)r5 z 2000-01-01r( r )r% r )r )r= r> rW r r �float)r rD r r r �!test_constructor_cant_cast_period s z2TestSeriesPeriod.test_constructor_cant_cast_periodc C s8 t tddd�td�d�}t tddd��}t�||� d S )Nz1/1/2000r( )r% r )r )r r r r Zassert_series_equal)r �srF r r r � test_constructor_cast_object! s z-TestSeriesPeriod.test_constructor_cast_objectN)r� r� r� r� r� r� r r r r r� s r� )�numpyr r= Zpandas._libs.tslibs.periodr Zpandas.core.dtypes.dtypesr Zpandasr r r r r r r r Zpandas._testingZ_testingr Zpandas.core.arraysr r r� r� r r r r �<module> s ( p
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка