Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/arrays/__pycache__/test_datetimes.cpython-37.pyc
Назад
B �5�g 7 � @ sh d Z ddlZddlZddlZddlmZ ddlZddl m Z ddl m Z G dd� d�ZG dd� d�ZdS ) z Tests for DatetimeArray � N)�DatetimeTZDtype)� DatetimeArrayc @ s e Zd Zdd� ZdS )� TestDatetimeArrayComparisonsc C s� |� d�}tt|�}tjdddd d�}t|�}|j|jks>t�|j|jksNt�|}t j t |�t d�}|dkrr| }|||�}t �||� xB|t �|�gD ]0} ||| �}t �||� || |�}t �||� q�W d S )N�_z 2016-01-1ZMS� )�freq�periods�tz)�dtype)�ne�gt�lt)�strip�getattr�operator�pd� date_ranger r �AssertionErrorr �npZones�len�bool�tm�assert_numpy_array_equal�array) �selfZall_compare_operators�opname�op�dti�arr�right�expected�result�other� r# �X/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/arrays/test_datetimes.py� test_cmp_dt64_arraylike_tznaive s" z<TestDatetimeArrayComparisons.test_cmp_dt64_arraylike_tznaiveN)�__name__� __module__� __qualname__r% r# r# r# r$ r s r c @ s� e Zd Zdd� Zej�dddg�ej�ddddg�dd � ��Zej�dee j e j d d g�d d � �Z dd� Z dd� Zdd� Zdd� Zej�dej�� ej�� �� ej�� �� g�dd� �Zdd� Zdd � Zej�d d d g�d d!� �Zd"d#� Zd$d%� Zd&d'� Z ej�d(d)d*g�d+d,� �Z ej�d(d)d*g�d-d.� �Z ej�dd/e � d/�d0e � d1�ej d2d3�d4e j!d5d6d7�d8 d9 d: e �!d5��"d;�d8 d9 d: ej�� �#d<�g �ej�d(d)d*g�d=d>� ��Z$d?d@� Z%dAdB� Z&dCdD� Z'dEdF� Z(dGS )H�TestDatetimeArrayc C s: t jdgtdd�d�}|jtdd�dd�}||ks6t�d S )N�2000z US/Central)r )r F)�copy)r �_from_sequencer �astyper )r r r! r# r# r$ �test_astype_to_same2 s z%TestDatetimeArray.test_astype_to_samer zdatetime64[ns]zdatetime64[ns, UTC]r" zdatetime64[ns, CET]c C sp t jddg|d�}|�� }d }|dk|dkA r2t}t�|�� |�|�}W d Q R X t j|d d �< t�||� d S )N� � )r zdatetime64[ns]) r �Seriesr+ � FutureWarningr �assert_produces_warningr- �NaT�assert_series_equal)r r r" Zser�orig�warn�tr# r# r$ �test_astype_copies9 s z$TestDatetimeArray.test_astype_copiesZuint32�uint64c C s� t �t�d�t�d�g�}t�t�� |�|�}W d Q R X t� |�j dkrVt� d�}n t� d�}t�t�� |�|�}W d Q R X |j |ks�t �t� ||� d S )Nr* Z2001�ur: �int64) r r, r � Timestampr r3 r2 r- r r �kindr r )r r r r! Zexpected_dtyper r# r# r$ �test_astype_intL s z!TestDatetimeArray.test_astype_intc C s: t jdgtdd�d�}tjtdd�� d|_W d Q R X d S )Nr* z US/Central)r )r � tz_localize)�match�UTC)r r, r �pytest�raises�AttributeErrorr )r r r# r# r$ �test_tz_setter_raises_ s z'TestDatetimeArray.test_tz_setter_raisesc C s� |}t jdddgdd�}|d kr&|jnt|d�}t||d�}|�� }t�d��|�}t |�d ��}||d< ||d< t � ||� ||d< ||g|d d�< t � ||� d S ) Nr/ r0 � zM8[ns])r )r z2020-09-08 16:50r ) r r r r r r+ r r= r@ �strr � assert_equal) r Ztz_naive_fixturer �datar r r �ts�setterr# r# r$ �test_setitem_str_impute_tzf s z,TestDatetimeArray.test_setitem_str_impute_tzc C s� t jdddgdd�}t|dtdd�d �}tjtd d �� t�d �|d < W d Q R X tjt dd �� tjd dd�|d < W d Q R X d S )Nr/ r0 rG zM8[ns])r Fz US/Central)r )r+ r z$Cannot compare tz-naive and tz-aware)rA r* r z US/Eastern) r r r r rC rD � TypeErrorr r= � ValueError)r rJ r r# r# r$ � test_setitem_different_tz_raises} s z2TestDatetimeArray.test_setitem_different_tz_raisesc C s: t tjddddd��}tjddd�|d< |jd ks6t�d S )Nr* r0 �Dz US/Central)r r r )r r )r r r r= r r )r �ar# r# r$ �test_setitem_clears_freq� s z*TestDatetimeArray.test_setitem_clears_freq�objc C s2 t jdddd�}|j}||d< |d |ks.t�d S )Nr* r0 rQ )r r r )r r �_datar )r rT r r r# r# r$ �test_setitem_objects� s z&TestDatetimeArray.test_setitem_objectsc C sJ t jddddd�}t|�}|�ddg�}t|jd |jd�}t�||� d S )Nr* r0 rQ z US/Central)r r r r/ )r r )r r r �repeatZasi8r r rI )r r r Zrepeatedr r# r# r$ �test_repeat_preserves_tz� s z*TestDatetimeArray.test_repeat_preserves_tzc C s� t jddddd�}t|��ddg�}|�� }|j�|�s<t�t j|d< |jd d �}t j ddd g|d |d t jgd �}t � ||� d S )Nr* r0 rQ z US/Central)r r r � rG ���F)Zdropnar/ r )�index) r r r rW Z value_countsr[ �equalsr r4 r1 r r5 )r r r r! r r# r# r$ � test_value_counts_preserves_tz� s $z0TestDatetimeArray.test_value_counts_preserves_tz�method�pad�backfillc C s� t jddddd�}t|dd�}t j|d< |d kr8|d n|d }tj|d |d ||d |d gtdd�d�}|j|d�}t�||� |d t jks�t �|d t j ddd�ks�t �d S )Nz 2000-01-01� rQ z US/Central)r r r T)r+ r0 r_ r/ rG r rY )r )r )r^ z 2000-01-03) r r r r4 r, r �fillnar � assert_extension_array_equalr r= )r r^ r r Zfill_valr r! r# r# r$ �test_fillna_preserves_tz� s z*TestDatetimeArray.test_fillna_preserves_tzc C sd t jdddd�}|j�dd��� }t j|d< t j|d< |jd d �}|�� }|d |d< t�||� |jd d �}|�� }|�� }|d |d< |d |d< t�||� |� |j jdd��}|j j d s�t �|j j d r�t �t�||� |jd d �}t�||� |jd d �} t�| |� t � |�} | jd d �} t � |�} t�| | � | jd d �} t � |�} t�| | � d S )Nz 2016-01-01� z US/Pacific)r r rG r0 )r r/ )r/ r r_ )r^ )r r r` )r0 r )r/ r/ �F)�orderZ F_CONTIGUOUSZ C_CONTIGUOUS)r r rU �reshaper+ r4 rb r rc Z_from_backing_dataZ_ndarray�flagsr Z DataFrameZassert_frame_equal) r r �dtaZres1Z expected1Zres2Z expected2Zdta2Zres3Zres4Zdf�resr r# r# r$ �test_fillna_2d� s: z TestDatetimeArray.test_fillna_2dc C s� d}t tjdd|d��}t�|�}tjtjd|d�tjd|d�gtd�}t� ||� tj|td�}t� ||� tj|d d�}tjd d gd d�}t� ||� d S ) Nz US/Central�2017r0 )r r z2017-01-01T00:00:00)r z2017-01-02T00:00:00)r zM8[ns]z2017-01-01T06:00:00z2017-01-02T06:00:00) r r r r �asarrayr r= �objectr r )r r rJ r! r r# r# r$ �test_array_interface_tz� s z)TestDatetimeArray.test_array_interface_tzc C sv t tjddd��}tjddgdd�}t�|�}t�||� tj|td�}tjt� d�t� d�gtd�}t�||� d S )Nrm r0 )r z2017-01-01T00:00:00z2017-01-02T00:00:00zdatetime64[ns])r ) r r r r r rn r r ro r= )r rJ r r! r# r# r$ �test_array_interface s z&TestDatetimeArray.test_array_interfacer[ TFc C s� t jddd�d d d }t|dd��d �}|r:t�|�}|�|d �}|�|d �d ��}||ksht�|�|d d � �}|�|d d � �d ��}t � ||� d S ) N� �i8)r � i i ʚ;rQ )r z Asia/Tokyor0 rB re ) r �aranger r@ r �Index� searchsortedZ tz_convertr r rI )r r[ rJ r r r! r# r# r$ � test_searchsorted_different_tz s z0TestDatetimeArray.test_searchsorted_different_tzc C s� t jddd�d d d }t|dd�}|r4t�|�}|�d �}d }tjt|d �� |� |d � W d Q R X tjt|d �� |� |� W d Q R X tjt|d �� |� |d � W d Q R X tjt|d �� |� |� W d Q R X d S ) Nrr rs )r rt i i ʚ;rQ )r z Asia/Tokyoz:Cannot compare tz-naive and tz-aware datetime-like objects)rA r ) r ru r r rv r@ rC rD rN rw )r r[ rJ r Zmismatch�msgr# r# r$ �$test_searchsorted_tzawareness_compat& s z6TestDatetimeArray.test_searchsorted_tzawareness_compatr/ g �?r4 r0 )�days�invalidrr rs )r rt i i ʚ;ztimedelta64[ns]rQ c C sj t jddd�d d d }t|dd�}|r4t�|�}d �d d g�}tjt|d �� |� |� W d Q R X d S ) Nrr rs )r rt i i ʚ;rQ )r �|z0searchsorted requires compatible dtype or scalarz<value should be a 'Timestamp', 'NaT', or array of those. Got)rA ) r ru r r rv �joinrC rD rN rw )r r" r[ rJ r ry r# r# r$ � test_searchsorted_invalid_types: s z1TestDatetimeArray.test_searchsorted_invalid_typesc C s� t jddd�}|j}tt�|jd��}|d }x4||�� |�� gD ] }|jd|d�}t � ||� qBW |� d�}|� d�}|d }x.||�� gD ] }|jd|d�}t � ||� q�W d S )Nz 2016-01-01rG )r r/ ���)� fill_valuerB ) r r rU r r Zroll� to_pydatetime� to_datetime64�shiftr �assert_datetime_array_equalr@ )r r rj r �fvr� r! r# r# r$ �test_shift_fill_valueX s z'TestDatetimeArray.test_shift_fill_valuec C s� t jddd�}|j}|d �d�}x<||�� gD ],}tjtdd�� |jd|d � W d Q R X q0W |�d�}|d �d �}xB||�� |� � gD ],}tjtdd�� |jd|d � W d Q R X q�W d S ) Nz 2016-01-01rG )r r� rB zCannot compare)rA r/ )r� ) r r rU r@ r� rC rD rN r� r� )r r rj r� r| r# r# r$ �%test_shift_value_tzawareness_mismatchj s z7TestDatetimeArray.test_shift_value_tzawareness_mismatchc C sT t jdddd�}|j}t jddd�}d}tjt|d �� |jd |d � W d Q R X d S ) Nz 2016-01-01rG rB )r r z2020-10-18 18:44z US/Pacific)r z,Timezones don't match. 'UTC' != 'US/Pacific')rA r/ )r� )r r rU r= rC rD rO r� )r r rj r� ry r# r# r$ �test_shift_requires_tzmatchz s z-TestDatetimeArray.test_shift_requires_tzmatchc C sd t jdddd�}|j�dd�}|�d �}|�� �d ��|j�}t�||� |�d�}t�||� d S )Nz 1994-05-12� z US/Pacific)r r rG rY ) r r rU rh r@ Zravel�shaper r� )r r rj r! r Z roundtripr# r# r$ �test_tz_localize_t2d� s z&TestDatetimeArray.test_tz_localize_t2dN))r& r' r( r. rC �markZ parametrizer9 �intr Zint32r< r? rF rM rP rS r r= �nowr� r� rV rX r] rd rl rp rq rx rz Z timedelta64Z Timedeltaru �viewZ to_periodr r� r� r� r� r# r# r# r$ r) 1 sL $ ) r) )�__doc__r �numpyr rC Zpandas.core.dtypes.dtypesr Zpandasr Zpandas._testingZ_testingr Zpandas.core.arraysr r r) r# r# r# r$ �<module> s !
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка