Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/arrays/__pycache__/test_datetimelike.cpython-37.pyc
Назад
B �5�gͻ � @ s� d dl mZ d dlZd dlZd dlZd dlmZmZm Z d dl m Z d dl m mZ d dlZd dlmZmZmZmZ d dlmZ d dlmZmZm Z m Z d dl m Z d dl m!Z! ej"d d d d d dgd�dd� �Z#ej"dd� �Z$ej"dd� �Z%ej"dd� �Z&G dd� d�Z'G dd� de'�Z(G d d � d e'�Z)G d d � d e'�Z*ej+j,d ed!d"d#g�j-ee�.d#d$�ffej/d%d&d d'�j-ee�0d#d$�ffej1d%d&d d'�j-effgd(d)� d*�d+d,� �Z2ej+j,d-ed!d"d#g�j-e�0d#d$�ej3ffej/d%d&d d'�j-e�.d#d$�ej3ffej1d%d&d d'�j-e�0d#d$�e�.d#d$�ej3ffgd.d)� d*�d/d0� �Z4ej+�,d1ej/d2d3d4�j5ej6d2d3d4�j5g�d5d6� �Z7ej+�,d7d8d9g�ej+�,d:e�8d;d<g�ed=d>gd d?�ed;d<gd d@�g�ej+�,dAe9ej5ej5ej:ej;ej<ej=g�dBdC� ���Z>ej+�,d:e�8d;d<g�ed=d>gd d?�ed;d<gd d@�g�ej+�,dDd=d>gdEdFge d;dGdH�gd> g�dIdJ� ��Z?ej+�,dAe9e@ej5ej:g�dKdL� �ZAej+�,dMdNdOg�dPdQ� �ZBej"dRdSejCdTe�DdU�dV�ejCdWe�DdW�dV�gd�dXdY� �ZEej+�,dMdNdOg�dZd[� �ZFdS )\� )� annotationsN)�NaT�OutOfBoundsDatetime� Timestamp)�np_version_under1p18)� DatetimeIndex�Period� PeriodIndex�TimedeltaIndex)� DatetimeArray� PandasArray� PeriodArray�TimedeltaArray)�sequence_to_dt64ns)�sequence_to_td64ns�D�B�W�M�Q�Y)�paramsc C s | j S )N)�param)�request� r �[/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/arrays/test_datetimelike.py�freqstr# s r c C s t jtd�d| d�}|S )z� A fixture to provide PeriodIndex objects with different frequencies. Most PeriodArray behavior is already tested in PeriodIndex tests, so here we just test that the PeriodArray behavior matches the PeriodIndex behavior. z 2000-01-01�d )�start�periods�freq)�pd� period_ranger )r �pir r r � period_index( s r$ c C s t jtd�d| d�}|S )z� A fixture to provide DatetimeIndex objects with different frequencies. Most DatetimeArray behavior is already tested in DatetimeIndex tests, so here we just test that the DatetimeArray behavior matches the DatetimeIndex behavior. z 2000-01-01r )r r r )r! � date_ranger )r �dtir r r �datetime_index6 s r' c C s t dddg�S )a A fixture to provide TimedeltaIndex objects with different frequencies. Most TimedeltaArray behavior is already tested in TimedeltaIndex tests, so here we just test that the TimedeltaArray behavior matches the TimedeltaIndex behavior. z1 Dayz3 Hoursr )r r r r r �timedelta_indexD s r( c @ s� e Zd ZU ded< ejdd� �Zdd� Zej� de j dd d �e j dd d d �e j d d d �e j dd dd�g�dd� �Zej� dddg�ej� dddg�dd� ��Zdd� Zej� dd d e�� jg�d d � �Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zej� d.d/d0g�d1d2� �Z d3d4� Z ej� d5d6d7d8g�d9d:� �Z d;d<� Z d=d>� Z d?d@� Z!dAdB� Z"dCdD� Z#ej� d5e j$e j%e&j'e(e)g�dEdF� �Z*dGdH� Z+ej� dddg�dIdJ� �Z,dKdL� Z-ej� d5e(e&j'e j$e j%g�dMdN� �Z.dOdP� Z/dQdR� Z0dSdT� Z1dUdV� Z2d6S )W� SharedTestsz2type[DatetimeIndex | PeriodIndex | TimedeltaIndex]� index_clsc C s, t jddd�d d d }| j|dd�}|S ) N� �i8)�dtype� i i ʚ;r )r )�np�arange� array_cls)�self�data�arrr r r �arr1dS s zSharedTests.arr1dc C sd |}| � |�}tjtdd�� ||d d� k W d Q R X tjtdd�� ||dg k W d Q R X d S )NzLengths must match)�match� r )r* �pytest�raises� ValueError)r2 r5 r4 �idxr r r �test_compare_len1_raisesY s z$SharedTests.test_compare_len1_raises�resultZ2020� )r �UTC)r �tzz0 days�2020Q1r )r r c C s. t �|�}t||k�st�t||k�r*t�d S )N)r! � Categorical�all�AssertionError�any)r2 r= �expectedr r r � test_compare_with_Categoricalf s z)SharedTests.test_compare_with_Categorical�reverseTF�as_indexc C s� t j||d�}|r t �|�}|| }}|r4|| }}tj|jtd�}| } ||k} t�| |� ||k} t�| | � |s�|s�||k } t�| | � ||k} t�| |� ||k} t�| | � ||k} t�| |� d S )N)�ordered)r- ) r! rB �CategoricalIndexr/ �ones�shape�bool�tm�assert_numpy_array_equal) r2 r5 rI rH rJ �other�left�rightrL Zzerosr= r r r � test_compare_categorical_dtypet s* z*SharedTests.test_compare_categorical_dtypec C s� t jddd�d d d }t j�|� | jtk r4d nd}| j||d�}| j�|�}d d d g}|�|�}|�|�}t � | �|�|� t � d d d g�}|�|�}|�|�}t � | �|�|� d S ) Nr r, )r- r. i i ʚ;r )r r7 � �^ ) r/ r0 �random�shuffler1 r r* � _simple_new�takerO �assert_index_equal�array)r2 r3 r r4 r; Ztakersr= rF r r r � test_take� s zSharedTests.test_take� fill_value� g @c C sj t jddd�d d d }| j|dd�}d |jj� d �}tjt|d �� |jd d gd|d� W d Q R X d S )Nr+ r, )r- r. i i ʚ;r )r zvalue should be a 'z' or 'NaT'. Got)r6 r r7 T)� allow_fillr^ ) r/ r0 r1 � _scalar_type�__name__r8 r9 � TypeErrorrZ )r2 r^ r3 r4 �msgr r r �test_take_fill_raises� s z!SharedTests.test_take_fill_raisesc C s� t jddd�d d d }| j|dd�}|jd d gd d d �}|d tksLt�|jd d gd t jd �}|d tksrt�|jd d gd td �}|d tks�t�d S )Nr+ r, )r- r. i i ʚ;r )r ���r7 T)r` r^ r )r/ r0 r1 rZ r rD �nan)r2 r3 r4 r= r r r �test_take_fill� s zSharedTests.test_take_fillc C sv |j ddgdt|d �d�}|ddg }t�||� d|jj� d�}tjt|d�� |j ddgddd� W d Q R X d S ) Nrf r7 T)r` r^ zvalue should be a 'z' or 'NaT'. Got)r6 �foo) rZ �strrO � assert_equalra rb r8 r9 rc )r2 r5 r= rF rd r r r �test_take_fill_str� s z SharedTests.test_take_fill_strc C s� |}| � |�}|�dt�}| �|�}|�|d d� |dd � |g�}|�t�}| � t�|d d� |dd � |g�d �}t � | � |�|� d S )Nr rf r7 ) r* �insertr r1 �_concat_same_type�astype�objectr/ Z concatenaterO r[ )r2 r5 r4 r; r= �arr2rF r r r �test_concat_same_type� s (z!SharedTests.test_concat_same_typec C s� t jddd�d d d }| j|dd�}|�|d �}|jjj}t||�sNt�|�t �}t||�sft�d | j j � d �}t j t|d �� |�d � W d Q R X d S )Nr+ r, )r- r. i i ʚ;r )r r z'value' should be a �.)r6 ri )r/ r0 r1 Z _unbox_scalar�_datar- �type� isinstancerD r � scalar_typerb r8 r9 r: )r2 r3 r4 r= rF rd r r r �test_unbox_scalar� s z SharedTests.test_unbox_scalarc C sV t jddd�d d d }| j|dd�}|�|d � |�|d d � � |�t� d S ) Nr+ r, )r- r. i i ʚ;r )r r r7 )r/ r0 r1 Z_check_compatible_withr )r2 r3 r4 r r r �test_check_compatible_with� s z&SharedTests.test_check_compatible_withc C sN t jddd�d d d }| j|dd�}|�t|d ��}||d ksJt�d S ) Nr+ r, )r- r. i i ʚ;r )r r )r/ r0 r1 Z_scalar_from_stringrj rD )r2 r3 r4 r= r r r �test_scalar_from_string� s z#SharedTests.test_scalar_from_stringc C sb t jddd�d d d }| j|dd�}d t|�j� d �}tjt|d �� |�d � W d Q R X d S ) Nr+ r, )r- r. i i ʚ;r )r �'z-' does not implement reduction 'not a method')r6 z not a method) r/ r0 r1 ru rb r8 r9 rc Z_reduce)r2 r3 r4 rd r r r �test_reduce_invalid� s z SharedTests.test_reduce_invalid�method�padZbackfillc C sx t jddd�d d d }| j|dd�}t|d < |d kr@|d n|d }|j|d �}|d |ksdt�|d tkstt�d S )Nr+ r, )r- r. i i ʚ;r )r rU r~ r> � )r} )r/ r0 r1 r ZfillnarD )r2 r} r3 r4 r^ r= r r r �%test_fillna_method_doesnt_change_orig� s z1SharedTests.test_fillna_method_doesnt_change_origc C s� t jddd�d d d }| j|dd�}|�|d �}|d ksBt�|j|d d d �}|d ks`t�|�|d d � �}t jd d gt jd�}t�||� |j|d d � d d �}t jd d gt jd�}t�||� |�t �}t r�|dks�t�n |dks�t�d S )Nr+ r, )r- r. i i ʚ;r )r r7 r_ rS )�sider> r ) r/ r0 r1 � searchsortedrD r\ �intprO rP r r )r2 r3 r4 r= rF r r r �test_searchsorted s z SharedTests.test_searchsorted�boxN�indexZseriesc C s� t |t�r\|j}|dd� \}}|d k r\|j�|�|j�|�kr\tjjtdd�}|j� |� |} |d krjn |dkr~| � | �} n t � | �} | � t| d ��} | dks�t�| j t| d �dd�} | dks�t�| � d d � | dd� D ��} tjddgtjd �} t�| | � tjtt�d |jj� d ��d�� | � d� W d Q R X |dk�rHdnd} t �d|��N tjtt�d |jj� d| � d��d�� | � t| d �dg� W d Q R X W d Q R X d S )Nr7 r> z!timezone comparisons inconsistent)r9 �reasonr� r_ rS )r� c S s g | ] }t |��qS r )rj )�.0�xr r r � <listcomp>I s zBSharedTests.test_searchsorted_castable_strings.<locals>.<listcomp>)r- zvalue should be a 'z/', 'NaT', or array of those. Got 'str' instead.)r6 ri �pythonZ StringArrayZArrowStringArray�string_storagez"', 'NaT', or array of those. Got 'z ' instead.Zbaz)rv r r@ �tznamer8 �mark�xfailrc �node� add_markerr* r! �Seriesr� rj rD r/ r\ r� rO rP r9 �re�escapera rb Zoption_context) r2 r5 r� r r� r@ �ts1�ts2r� r4 r= rF Zarr_typer r r �"test_searchsorted_castable_strings* s@ z.SharedTests.test_searchsorted_castable_stringsc C s\ t jdd� tdd�D �dd�}| j|dd�}|d t�|�}|d t�|�}|d d S ) Nc S s g | ]}t j| �qS r )r �value)r� �nr r r r� e s zGSharedTests.test_getitem_near_implementation_bounds.<locals>.<listcomp>r7 r r, )r- �ns)r r )r/ �asarray�ranger1 r! �Indexr� )r2 Zi8valsr4 r� Zserr r r �'test_getitem_near_implementation_boundsb s z3SharedTests.test_getitem_near_implementation_boundsc C s� t |�|jd d �tjf |jd�}|d d �tjf }t�||� |}t |�|jd d�df |jd�}|d d�df }t�||� |d }|d }||ks�t�d S )N)r- r> r )rf r rf )ru rt r/ �newaxisr- rO rk rD )r2 r5 rF r= �arr2dr r r �test_getitem_2do s " zSharedTests.test_getitem_2dc C s� |j d d�tjf }t|�j||jd�}t|�}t|�dks@t�x<|D ]4}t |t|��s\t�|j dksjt�|j|jksFt�qFW d S )Nr> )r- r7 ) rt r/ r� ru rY r- �list�lenrD rv �ndim)r2 r5 �data2dr� r= r� r r r � test_iter_2d� s zSharedTests.test_iter_2dc C s� |j d d�tjf }t|�j||jd�}t|�}t|t�r~dt|�j � d|d � � � d|d � � � d|d � � � d |j� � }n6dt|�j � d|d � d|d � d|d � d |j� � }||ks�t �d S ) Nr> )r- �<z> [ ['r z'], ['r7 r_ z'] ] Shape: (3, 1), dtype: ) rt r/ r� ru rY r- �reprrv r rb Z _repr_baserD )r2 r5 r� r� r= rF r r r � test_repr_2d� s D 6 zSharedTests.test_repr_2dc C s� t jddd�d d d }| j|dd�}|d |d < t jddd�d d d }|d |d < t�|j|� |d d � |d d �< |d d � |d d �< t�|j|� d S ) Nr+ r, )r- r. i i ʚ;r )r r7 r ���r_ )r/ r0 r1 rO rP �asi8)r2 r3 r4 rF r r r � test_setitem� s zSharedTests.test_setitemc C s� |� � d d d� }|jjdkr(|�d �}|}|tkr>t|�}nF|tjkrZt�|�t��}n*|t krv|tj |td��}n||��t�}||d d �< t � ||� d S )Nrf )�mr )r- ) �copyr- �kind� _with_freqr� r/ r\ ro rp r r� rO rk )r2 r� r5 rF �valsr r r �test_setitem_object_dtype� s z%SharedTests.test_setitem_object_dtypec C s� t |t�r\|j}|dd � \}}|d k r\|j�|�|j�|�kr\tjjtdd�}|j� |� |� � }|dd � |ddg< |� � }dd� |dd � D �|d d�< t � ||� |� � }|d |d< |� � }t |d �|d< t � ||� d S ) Nr� z!timezone comparisons inconsistent)r9 r� r r7 c S s g | ] }t |��qS r )rj )r� r� r r r r� � s z1SharedTests.test_setitem_strs.<locals>.<listcomp>r_ rf )rv r r@ r� r8 r� r� rc r� r� r� rO rk rj ) r2 r5 r r@ r� r� r� rF r= r r r �test_setitem_strs� s" z SharedTests.test_setitem_strsc C sd |� � d d d� }t|t�s&|�d �}t�|�}|r>t�|�}|d d d� |d d �< t�||� d S )Nrf ) r� rv r r� r! rB rK rO rk )r2 r5 rI rF �catr r r �test_setitem_categorical� s z$SharedTests.test_setitem_categoricalc C s� t jddd�d d d }| j|dd�}|d }tjtd d �� ||d < W d Q R X tjtd d �� t� |d < W d Q R X d}tjt|d �� |d g|g < W d Q R X d}tjt|d �� |d d� |dd�< W d Q R X d S )Nr+ r, )r- r. i i ʚ;r )r r zindex 12 is out of bounds)r6 � z value should be a.* 'object'z<cannot set using a list-like indexer with a different lengthr7 z=cannot set using a slice indexer with a different length thanr> ) r/ r0 r1 r8 r9 � IndexErrorrc rp r: )r2 r3 r4 �valrd r r r �test_setitem_raises s z SharedTests.test_setitem_raisesc C sp d|j j� d�}tjt|d�� |ddg�|d d�< W d Q R X tjt|d�� |ddg�|d d�< W d Q R X d S ) Nzvalue should be a 'z ', 'NaT', or array of those. Got)r6 r r7 r_ g g �?)ra rb r8 r9 rc )r2 r5 r� rd r r r �test_setitem_numeric_raises s z'SharedTests.test_setitem_numeric_raisesc C s� t jddd�d d d }| j|dd�}|tjd d � }|tjd d �7 }t�||� |tjd d � }|tjd d �8 }t�||� d S ) Nr+ r, )r- r. i i ʚ;r )r r7 )�days)r/ r0 r1 r! � TimedeltarO rk )r2 r3 r4 rF r r r �test_inplace_arithmetic# s z#SharedTests.test_inplace_arithmeticc C s� t jddd�d d d }| j|dd�}d }tjt|d �� |jd d d �}W d Q R X |�� }| jtkrztj j d |j d�}n |� d �}||d < |d d� |d d �< t� ||� d S )Nr+ r, )r- r. i i ʚ;r )r z Passing <class 'int'> to shift)r6 r7 )r^ r rf ) r/ r0 r1 rO �assert_produces_warning� FutureWarning�shiftr� r ra Z _from_ordinalr rk )r2 r3 r4 rd r= rF Zfill_valr r r � test_shift_fill_int_deprecated0 s z*SharedTests.test_shift_fill_int_deprecatedc C s� |}t |�d dkr |d d� }|t |�d }|�� }t|�t|�ksLt�||ksXt�t|t |�d < t|t�s�|t |�d d t |�d d � �� }|jdd�tks�t�|�� }t|�t|�ks�t�||ks�t�|d d� �� tks�t�|d d� jdd�tk�s t�|�dd�}|jd d�}t|�t|�k�s:t�||k�sHt�|jd dd�tk�s`t�|jdd�}t|�j |g|j d �}t � ||� |jddd�}t|�j tg|j d �}t � ||� |jdd�}t � ||� |jddd�}t � ||� d S ) Nr_ r rf r7 F)�skipna)�axis)r� r� )r- ) r� Zmedianru rD r rv r Zmean�reshape�_from_sequencer- rO rk )r2 r5 r4 rF r= rq Z expected2r r r � test_medianB s> ( zSharedTests.test_medianc C sT t jdddgt jd�}| j|| jd�}tj|dd�}| j|| jd�}t�||� d S )Nr7 r_ r> )r- ZInt64)r/ r\ �int64r1 � example_dtyper! rO � assert_extension_array_equal)r2 r4 rF r3 r= r r r �test_from_integer_arrayr s z#SharedTests.test_from_integer_array)3rb � __module__� __qualname__�__annotations__r8 �fixturer5 r< r� � parametrizer! r% �timedelta_ranger" rG rT r] r �now�timere rh rl rr rx ry rz r| r� r� r� r� r� r� r� r� r� r� r/ r\ r� r r� r� r� r� r� r� r� r� r� r r r r r) P sV "" 8 $ 0r) c @ s� e Zd ZeZeZeZdZ e j dd� �Z dd� Z dd� Zdd � Zd d � Zd d � Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Ze j�dej�dd � �Ze j�dej �d d � �Z d d � Z d!d"� Z d#d$� Z d%d&� Z!d'd(� Z"d)S )*�TestDatetimeArrayzM8[ns]c C s |}t jdd||d�}|j}|S )Nz2016-01-01 01:01:00r )r r r@ )r! r% rt )r2 Ztz_naive_fixturer r@ r& �dtar r r r5 � s zTestDatetimeArray.arr1dc C sj | � |�}|jdd�}|tjdd� }|�d �}t�||� |j}|jdd�}|j�d �}t�||� d S )NZ2T)r r7 )�minutes) r* �roundr! r� r� rO r[ rt �assert_datetime_array_equal)r2 r5 r& r= rF r� r r r � test_round� s z TestDatetimeArray.test_roundc C s� t |�}t�|�}|j}||ks$t�t�||� tj|dd�}||ksJt�t�||� tj|dd�}|j}||ksvt�t�||� tj|ddd�}||ks�t�t�||� tj|dd�}||k s�t�t�||� tj|td�}tjt |�td�}t�||� tj|dd�}||j k �st�t� ||��r,t�|j � � }t�||� x<dt gD ]0}tj||d�}t�|��|�}t�||� �qLW d S )NF)r� zdatetime64[ns])r- )r- r� r� �float64)r r/ r� rt rD rO rP r\ rp r� r� �may_share_memoryr� rj ro )r2 r' r4 r= rF r- r r r �test_array_interface� s< z&TestDatetimeArray.test_array_interfacec C sT |}| � |�}t�t|��}tj|td�}t�||� tj|td�}t�||� d S )N)r- )r* r/ r\ r� rp rO rP )r2 r5 r4 r& rF r= r r r �test_array_object_dtype� s z)TestDatetimeArray.test_array_object_dtypec C s� |}| � |�}|j�d�}tj|dd�}t�||� tj|dd�}t�||� tj|ddd�}|j|jksnt�|jd k s|t�tj|ddd�}|j|jks�t�|jd k s�t�d S )NzM8[ns])r- zdatetime64[ns]F)r- r� ) r* r� �viewr/ r\ rO rP �baserD )r2 r5 r4 r& rF r= r r r � test_array_tz� s z TestDatetimeArray.test_array_tzc C s| |}| � |�}|j}tj|dd�}t�||� tj|tjd�}t�||� tj|ddd�}|j|jk sjt�|jd ksxt�d S )Nr, )r- F)r- r� ) r* r� r/ r\ rO rP r� r� rD )r2 r5 r4 r& rF r= r r r �test_array_i8_dtype� s z%TestDatetimeArray.test_array_i8_dtypec C sL t jddgdd�}t|�}|j|ks(t�t|d d� �}|jj|ksHt�d S )Nz 2000-01-01z 2000-01-02zM8[ns])r- r )r/ r\ r rt rD r� )r2 r4 r� r r r �test_from_array_keeps_base� s z,TestDatetimeArray.test_from_array_keeps_basec C sR |}| � |�}t|�t|�ks"t�t�|�}t|t�s:t�t|�t|�ksNt�d S )N)r* r� rD r! r� rv r )r2 r5 r4 r& Zdti2r r r � test_from_dti� s z TestDatetimeArray.test_from_dtic C sN |}| � |�}|�d�}t|tj�s(t�|jdks6t�t|�t|�ksJt�d S )N�O)r* ro rv r/ �ndarrayrD r- r� )r2 r5 r4 r& �asobjr r r �test_astype_object s z$TestDatetimeArray.test_astype_objectc C s� |}t |�}d}tjt|d�� |j|d�}W d Q R X tjt|dd�� |j|d�}W d Q R X t|t�slt�t�t � |�t � |�� d S )Nz0to_perioddelta is deprecated and will be removed)r6 )r F)r6 Zcheck_stacklevel) r rO r� r� Zto_perioddeltarv r rD r[ r! r� )r2 r' r r& r4 rd rF r= r r r �test_to_perioddelta s z%TestDatetimeArray.test_to_perioddeltac C sN |}t |�}|j|d�}|j|d�}t|t�s2t�t�t�|�t�|�� d S )N)r ) r � to_periodrv r rD rO r[ r! r� )r2 r' r r& r4 rF r= r r r �test_to_period$ s z TestDatetimeArray.test_to_periodc C s` |� dd�}|jd krd nt}t�|��" |�d�}|�d�� dd�}W d Q R X t�||� d S )Nr7 rf r )r� r@ � UserWarningrO r� r� Zassert_period_array_equal)r2 r5 r� �warnr= rF r r r �test_to_period_2d0 s z#TestDatetimeArray.test_to_period_2d�propnamec C sN | � |�}|}|j|jks t�t||�}tjt||�|jd�}t�||� d S )N)r- ) r* r rD �getattrr/ r\ r- rO rP )r2 r5 r� r& r4 r= rF r r r �test_bool_properties9 s z&TestDatetimeArray.test_bool_propertiesc C sJ |dkr d S | � |�}|}t||�}tjt||�|jd�}t�||� d S )N)�weekZ weekofyear)r- )r* r� r/ r\ r- rO rP )r2 r5 r� r& r4 r= rF r r r �test_int_propertiesE s z%TestDatetimeArray.test_int_propertiesc C s� |}| � |�}t�� �|j�}|jddgd|d�}|d |ksBt�d|jj� d�}t j t |d�� |jddgd|| d� W d Q R X t j t |d�� |jddgdt d �d� W d Q R X |jd k r�d nd }t�� �|�}d }t j t |d�� |jddgd|d� W d Q R X t j}d|jj� d�}t j t |d�� |jddgd|d� W d Q R X t�d d �}t j t |d�� |jddgd|d� W d Q R X |jd k �r�t�� �d�}d}t j t|d�� |jddgd|d� W d Q R X d S )Nrf r7 T)r` r^ r zvalue should be a 'z' or 'NaT'. Got)r6 Z2014Q1z US/Easternz:Cannot compare tz-naive and tz-aware datetime-like objectsr r� zAustralia/Melbournez2Timezones don't match. .* != 'Australia/Melbourne')r* r r� � tz_localizer@ rZ rD ra rb r8 r9 rc r r r� r/ � timedelta64r: ) r2 r5 r4 r& r� r= rd r@ r� r r r �test_take_fill_validR s6 "" z&TestDatetimeArray.test_take_fill_validc C sP |}|j d kr|�d�}n |�d �}tjtdd�� |�||g� W d Q R X d S )Nr? z to_concat must have the same)r6 )r@ r� r8 r9 r: rn )r2 r5 r4 rQ r r r � test_concat_same_type_invalid~ s z/TestDatetimeArray.test_concat_same_type_invalidc C sf t tjddddd��}t tjddddd��}t �||g�}t t�dddd g��d��}t�||� d S ) N�2000r_ r z US/Central)r r r@ �Hz2000-01-01 00:00:00z2000-01-02 00:00:00z2000-01-01 01:00:00)r r! r% rn � to_datetimer� rO r� )r2 �a�br= rF r r r �$test_concat_same_type_different_freq� s z6TestDatetimeArray.test_concat_same_type_different_freqc C s6 |}|� d�}tjdd� |D �td�}t�||� d S )Nz%Y %bc S s g | ]}|� d ��qS )z%Y %b)�strftime)r� �tsr r r r� � s z3TestDatetimeArray.test_strftime.<locals>.<listcomp>)r- )r� r/ r\ rp rO rP )r2 r5 r4 r= rF r r r � test_strftime� s z TestDatetimeArray.test_strftimec C s> t tdtg��}|�d�}tjdtjgtd�}t� ||� d S )Nz 2019-01-01z%Y-%m-%d)r- ) r r r r� r/ r\ rg rp rO rP )r2 r4 r= rF r r r �test_strftime_nat� s z#TestDatetimeArray.test_strftime_natN)#rb r� r� r r* r r1 r rw r� r8 r� r5 r� r� r� r� r� r� r� r� r� r� r� r� r� � _bool_opsr� � _field_opsr� r� r� r� r� r� r r r r r� | s. * , r� c @ sf e Zd ZeZeZejZ dZ dd� Z dd� Z dd� Z dd � Zej�d ej�d d � �Zd d� Zdd� ZdS )�TestTimedeltaArrayzm8[ns]c C sX t ddg�}t|�}t|�t|�ks(t�t�|�}t|t �s@t�t|�t|�ksTt�d S )Nz1 Dayz3 Hours)r r r� rD r! r� rv )r2 �tdir4 Ztdi2r r r � test_from_tdi� s z TestTimedeltaArray.test_from_tdic C sT t ddg�}t|�}|�d�}t|tj�s.t�|jdks<t�t|�t|�ksPt�d S )Nz1 Dayz3 Hoursr� ) r r ro rv r/ r� rD r- r� )r2 r r4 r� r r r r� � s z%TestTimedeltaArray.test_astype_objectc C s, |}t |�}|�� }|�� }t�||� d S )N)r Zto_pytimedeltarO rP )r2 r( r r4 rF r= r r r �test_to_pytimedelta� s z&TestTimedeltaArray.test_to_pytimedeltac C s. |}t |�}|�� }|�� }t�||j� d S )N)r � total_secondsrO rP �values)r2 r( r r4 rF r= r r r �test_total_seconds� s z%TestTimedeltaArray.test_total_secondsr� c C s< |}t |�}t||�}tjt||�|jd�}t�||� d S )N)r- )r r� r/ r\ r- rO rP )r2 r( r� r r4 r= rF r r r r� � s z&TestTimedeltaArray.test_int_propertiesc C s� t |�}t�|�}|j}||ks$t�t�||� tj|dd�}||ksJt�t�||� tj|dd�}|j}||ksvt�t�||� tj|ddd�}||ks�t�t�||� tj|dd�}||k s�t�t�||� tj|td�}tjt |�td�}t�||� tj|dd�}||j k �st�t� ||��r,t�|j � � }t�||� x<dt gD ]0}tj||d�}t�|��|�}t�||� �qLW d S )NF)r� ztimedelta64[ns])r- )r- r� r� r� )r r/ r� rt rD rO rP r\ rp r� r� r� r� rj ro )r2 r( r4 r= rF r- r r r r� � s< z'TestTimedeltaArray.test_array_interfacec C s� |}t |�}tjdd�}|jddgd|d�}|d |ks<t�t�� }|}d|jj� d�}t j t |d �� |jddgd|d� W d Q R X |� d �}t j t |d �� |jddgd|d� W d Q R X t �d d �}t j t |d �� |jddgd|d� W d Q R X d S ) Nr7 )r� rf T)r` r^ r zvalue should be a 'z' or 'NaT'. Got)r6 r r r� )r r! r� rZ rD r r� ra rb r8 r9 rc r� r/ � datetime64) r2 r( r r4 Ztd1r= r� r� rd r r r r� s z'TestTimedeltaArray.test_take_fill_validN)rb r� r� r r* r r1 r! r� rw r� r r� r r r8 r� r� r r� r� r� r r r r r � s *r c @ s� e Zd ZeZeZeZeg dd�j Z e j dd� �Z dd� Zdd� Zd d � Ze j�d d d g�dd� �Zdd� Ze j�dej�dd� �Ze j�dej�dd� �Zdd� Zdd� Zdd � Zd S ) �TestPeriodArrayr )r c C s |j S )N)rt )r2 r$ r r r r5 * s zTestPeriodArray.arr1dc C sR | � |�}|}t|�t|�ks"t�t�|�}t|t�s:t�t|�t|�ksNt�d S )N)r* r� rD r! r� rv r )r2 r5 r# r4 Zpi2r r r � test_from_pi. s z TestPeriodArray.test_from_pic C sN | � |�}|}|�d�}t|tj�s(t�|jdks6t�t|�t|�ksJt�d S )Nr� )r* ro rv r/ r� rD r- r� )r2 r5 r# r4 r� r r r r� 8 s z"TestPeriodArray.test_astype_objectc C s� |}t j}d|jj� d�}tjt|d�� |jddgd|d� W d Q R X t� dd �}tjt|d�� |jddgd|d� W d Q R X d S ) Nzvalue should be a 'z' or 'NaT'. Got)r6 rf r7 T)r` r^ r r� ) r r� ra rb r8 r9 rc rZ r/ r� )r2 r5 r4 r� rd r r r r� @ s z$TestPeriodArray.test_take_fill_valid�how�S�Ec C sT | � |�}|}t|j|d��}|j|d�}t|t�s8t�t�t�|�t�|�� d S )N)r ) r* r � to_timestamprv rD rO r[ r! r� )r2 r r5 r# r4 rF r= r r r �test_to_timestampN s z!TestPeriodArray.test_to_timestampc C s^ t jdddd�}d}tjt|d�� |�� W d Q R X tjt|d�� |j�� W d Q R X d S )NZ1500r r> )r r z7Out of bounds nanosecond timestamp: 1500-01-01 00:00:00)r6 )r! r" r8 r9 r r rt )r2 r# rd r r r � test_to_timestamp_out_of_bounds[ s z/TestPeriodArray.test_to_timestamp_out_of_boundsr� c C s8 | � |�}|}t||�}t�t||��}t�||� d S )N)r* r� r/ r\ rO rP )r2 r5 r� r# r4 r= rF r r r r� e s z$TestPeriodArray.test_bool_propertiesc C s8 | � |�}|}t||�}t�t||��}t�||� d S )N)r* r� r/ r\ rO rP )r2 r5 r� r# r4 r= rF r r r r� p s z#TestPeriodArray.test_int_propertiesc C s� |}t �|�}t jt|�td�}t�||� t j|td�}t�||� t j|dd�}t�||j� d}tj t |d�� t j|dd� W d Q R X t j|dd�}t �|�� d�}t�||� d S )N)r- r� zEfloat\(\) argument must be a string or a( real)? number, not 'Period')r6 r� ZS20) r/ r� r\ r� rp rO rP r� r8 r9 rc ro )r2 r5 r4 r= rF rd r r r r� z s z$TestPeriodArray.test_array_interfacec C s6 |}|� d�}tjdd� |D �td�}t�||� d S )Nz%Yc S s g | ]}|� d ��qS )z%Y)r� )r� Zperr r r r� � s z1TestPeriodArray.test_strftime.<locals>.<listcomp>)r- )r� r/ r\ rp rO rP )r2 r5 r4 r= rF r r r r� � s z TestPeriodArray.test_strftimec C sB t tdtgdd��}|�d�}tjdtjgtd�}t� ||� d S )Nz 2019-01-01z period[D])r- z%Y-%m-%d) r r r r� r/ r\ rg rp rO rP )r2 r4 r= rF r r r r� � s z!TestPeriodArray.test_strftime_natN) rb r� r� r r* r r1 r rw r- r� r8 r� r5 r r� r� r� r� r r r r� r r� r� r� r� r r r r r $ s r zarr,casting_natsz1 Dayz3 Hoursr r� z 2000-01-01r> )r r c C s t | �jS )N)ru rb )r� r r r �<lambda>� � r )Zidsc C sJ t | ��t| d | d g�}x(|D ] }| �� } || d< t�| |� q"W d S )Nr7 r_ r )ru r� r r� rO rk )r4 Z casting_natsrF �natr r r � test_casting_nat_setitem_array� s r zarr,non_casting_natsc C s t | �jS )N)ru rb )r� r r r r � r c C s8 d}x.|D ]&}t jt|d�� || d< W d Q R X q W d S )Nzwvalue should be a '(Timestamp|Timedelta|Period)', 'NaT', or array of those. Got '(timedelta64|datetime64|int)' instead.)r6 r )r8 r9 rc )r4 Znon_casting_natsrd r r r r � test_invalid_nat_setitem_array� s r r4 r� rU )r c C s� t r tj}ntj}t| d< | �� }| �� }||d �s:t�| jdd�}|d dksVt�| jddd�}|d dkstt�| j| d �� d�}|d |d ks�t�| j| d jdd �d�}|d |d ks�t�t � | |� d S ) Nr r� )r- l���� )r- �na_valuer7 )r F)r� ) r r! Zisnar/ �isnanr r� Zto_numpyrD rO rk )r4 r �originalr= r r r �test_to_numpy_extra� s r rI TFr z 2020-01-01z 2020-02-01r7 r_ )�unit)r �klassc C s< |s | j } | �|| ��}tjddg|jd�}t�||� d S )Nr r7 )r- )rt r� r/ r\ r- rO rP )r r rI r= rF r r r �,test_searchsorted_datetimelike_with_listlike� s r �argr� r� z Europe/London)r@ c C s, d}t jt|d�� | �|� W d Q R X d S )Nz [Unexpected type|Cannot compare])r6 )r8 r9 rc r� )r r rd r r r �:test_searchsorted_datetimelike_with_listlike_invalid_dtype s r c C sB ddgd }| |�}t |dd�}t dd� |D ��}t�||� d S )NrA Z2020Q2r_ r )r c S s g | ] }t |��qS r )r )r� �sr r r r� 2 s z?test_period_index_construction_from_strings.<locals>.<listcomp>)r rO r[ )r �stringsr3 r= rF r r r �+test_period_index_construction_from_strings, s r" r- zM8[ns]zm8[ns]c C s� t jdddg| d�}t|�}ttd�| }||�}||�}t�||� |�|�}|�|�}t�||� tt d�| }||�d }||�d }t� ||� t j t j d�| }||�j}||�j}t� ||� ttd�| }||�}||�}t�||� d S )Nr7 r_ r> )r- )zM8[ns]zm8[ns]r )r/ r\ r r r rO r� r� r r rk r! r� Z to_timedeltar r r[ )r- r3 r4 �clsr= rF �func�idx_clsr r r �test_from_pandas_array6 s* r& � memoryviewr\ �daskz dask.array)Zmarks�xarrayc C s� t jdddgt jd�}| j}|dkr.t|�}nV|dkrLdd l}|�d|�}n8|d krjdd l}|j�|�}n|d kr�dd l}|�|�}||fS ) Nr7 r_ r> )r- r' r\ r �ir( r) )r/ r\ r� r r' Z dask.arrayr) Z DataArray)r r4 �namer3 r\ r( Zxrr r r � array_likesW s r, c C s� |\}}t td�| }||�}|�|�}t�||� ttd�| }||�d }||�d }t�||� tt d�| }||�}||�}t� ||� d S )N)zM8[ns]zm8[ns]r ) r r r� rO r� r r rk r r r[ ) r- r, r4 r3 r# rF r= r$ r% r r r �test_from_obscure_arrayw s r- )G� __future__r r� �numpyr/ r8 Z pandas._libsr r r Z pandas.compatr Z pandas.util._test_decorators�utilZ_test_decorators�tdZpandasr! r r r r Zpandas._testingZ_testingrO Zpandas.core.arraysr r r r Z pandas.core.arrays.datetimesr Z pandas.core.arrays.timedeltasr r� r r$ r' r( r) r� r r r� r� rt r� r% r r" r r� r r\ r� r r� r� r� r� rB rK r r �tupler" r& r Z skip_if_nor, r- r r r r �<module> s� 0 2x~ . " !"
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка