Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/arrays/interval/__pycache__/test_interval.cpython-37.pyc
Назад
B �5�g�% � @ s� d dl Zd dlZd dlm mZ d dlZd dlm Z m Z m Z m Z m Z mZmZ d dlmZ d dlmZ eje d ddg�e dddg�fe d d d g�e d d d g�fed dd�eddd�feddd�eddd�fedddd�edddd�fgdd� d�dd� �ZG dd� d�ZG dd � d �ZG d d � d �Zd d � Ze�d!�Z e d"d#� �Z e d$d%� �Z e d&d'� �Z e ej j!d(d d d d ged)dd*d+�gd,d-gd.�d/d0� ��Z"e ej j!d(d d d d ged)dd*d+�gd,d-gd.�d1d2� ��Z#dS )3� N)�Index�Interval� IntervalIndex� Timedelta� Timestamp� date_range�timedelta_range)� IntervalArray� � � � � g g �?g @g @z0 days)�periodsz1 dayZ20170101Z20170102z US/Eastern)r �tzc C s t | d j�S )Nr )�str�dtype)�x� r �`/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/arrays/interval/test_interval.py�<lambda> � r )�params�idsc C s | j S )zC Fixture for building an IntervalArray from various dtypes )�param)�requestr r r �left_right_dtypes s r c @ sh e Zd Zej�dded�ed�fed�ed�feddd�eddd�fg�ej�d ee g�d d � ��Z d S ) �TestAttributesz left, right)r r z0 daysz1 dayz 2018-01-01z 2018-01-02z US/Eastern)r � constructorc C sH ||f||ft jg}t �|dkddg�}|j||d�j}t�||� d S )N�bothF)�closed)�np�nan�array� from_tuplesZis_empty�tmZassert_numpy_array_equal)�selfr �left�rightr Ztuples�expected�resultr r r � test_is_empty) s z TestAttributes.test_is_emptyN) �__name__� __module__� __qualname__�pytest�mark� parametrizer r r r r+ r r r r r ( s r c @ sp e Zd Zej�dddddg�dd� �Zej�ded d dd �ej d d d dgdd �g�dd� �Z dd� Z dd� Z dS )� TestMethods� new_closedr' r( r Zneitherc C s> t jtd�|d�}|�|�}t jtd�|d�}t�||� d S )N� )r )r � from_breaks�range� set_closedr% � assert_extension_array_equal)r& r r3 r# r* r) r r r �test_set_closed? s zTestMethods.test_set_closed�otherr r )r r r r c C sR t �tjddddgdd��}d}tjt|d�� |jd d d g|d � W d Q R X d S ) Nr r r r r' )r z+'value.closed' is 'right', expected 'left'.)�matchTF)r: )�pdZSeriesr r5 r/ �raises� ValueError�where)r& r: Zserr; r r r �test_where_raisesG s z TestMethods.test_where_raisesc C s> t �dddg�}|�� }t �tjtjfdg�}t�||� d S )Nr r r )g �?g @)r r5 �shiftr$ r! r" r% �assert_interval_array_equal)r& �ar* r) r r r � test_shiftT s zTestMethods.test_shiftc C sj t �tddd��}|�d�}|jdddgdd�}t�||� |�d�}|jd ddgdd�}t�||� d S ) N�2000r )r r ���r T)Z allow_fillr )r r5 r rA Ztaker% rB )r& rC r* r) r r r �test_shift_datetime\ s z TestMethods.test_shift_datetimeN) r, r- r. r/ r0 r1 r9 r r r5 r@ rD rG r r r r r2 > s " r2 c @ s e Zd Zdd� Zdd� ZdS )� TestSetitemc C s� |\}}t �||�}|jjjdkrJd}tjt|d�� tj |d< W d Q R X |jjjdkr�d}tjt |d�� t j |d< W d Q R X d S t j |d< t|jgt|dd � � �}t|jgt|dd � � �}t �||�}t�||� d S )N)�m�Mz='value' should be an interval type, got <.*NaTType'> instead.)r; r )�i�uz4Cannot set float NaN to integer-backed IntervalArrayr )r Z from_arraysr �subtype�kindr/ r= � TypeErrorr< ZNaTr> r! �NaNr" r Z _na_value�listr% r8 ) r& r r' r( r* �msgZ expected_leftZexpected_rightr) r r r � test_set_nah s zTestSetitem.test_set_nac C s� t �td��}|�� }|�d�}d}tjt|d�� |d |d< W d Q R X tjt|d�� |d d� |d d�< W d Q R X tjt|d�� |d d� |d d�< W d Q R X tjt|d�� |d d d� |d d �< W d Q R X tjt|d�� t|d d d� �|d d �< W d Q R X tjt|d��" |d d d� � t �|d d �< W d Q R X tjt|d��" |d d d� � d�|d d �< W d Q R X g |d d�< t � ||� d S ) Nr r z*'value.closed' is 'both', expected 'right')r; r r rF �category) r r5 r6 �copyr7 r/ r= r> rQ Zastype�objectr% rB )r& �arr�origr: rR r r r � test_setitem_mismatched_closed~ s( $&& z*TestSetitem.test_setitem_mismatched_closedN)r, r- r. rS rY r r r r rH g s rH c C s* t �ddg�} t| �}d}||ks&t�d S )N)r r )r r zI<IntervalArray> [(0, 1], (1, 2]] Length: 2, dtype: interval[int64, right])r r$ �repr�AssertionError)rW r* r) r r r � test_repr� s r\ �pyarrowc C s� dd l } ddlm} || �� d�}|| �� d�}|| �� d�}|jdksLt�||ksXt�||krdt�t|�t|�ksxt�t|�t|�kr�t�d S )Nr )�ArrowIntervalTyper' r( )r] � pandas.core.arrays._arrow_utilsr^ �int64r r[ �hash)�par^ �p1�p2Zp3r r r �test_arrow_extension_type� s re c C s\ dd l } ddlm} tjdddd�j}| �|�}t|j|�s@t�|jj |j ksRt�|jj | � � ksft�|j � d��| jdddd gd d ��s�t�|j � d ��| jddd dgd d ��s�t�| �d d� tdd�D ��}|j �|�s�t�| j||jd �}|�|�s�t�tjtdd�� | j|dd � W d Q R X tjtdd�� | j||| �� d�d � W d Q R X d S )Nr )r^ r r )�freqr' r r r r` )�typer( c S s g | ]}||d d��qS )r )r' r( r )�.0rK r r r � <listcomp>� s z$test_arrow_array.<locals>.<listcomp>z&Not supported to convert IntervalArray)r; �float64zdifferent 'subtype')r] r_ r^ r< Zinterval_ranger# � isinstancerg r[ r rM r` �storage�field�equalsr6 r/ r= rO rj )rb r^ Z intervalsr* r) r r r �test_arrow_array� s" ((ro c C s dd l } ddlm} t�ddddg�}d |d< | �|�}t|j|�sHt�|jj |j ksZt�|jj | � � ksnt�| jdd dgdd �}| jdd dgdd �}|j � d ��|�s�t�|j � d ��|�s�t�ddd �d d d �ddd �g}| jj|t�d dd g�d�}|j �|��s t�d S )Nr )r^ g g �?g @g @r rj )rg r' r( )r' r( FT)�mask)r] r_ r^ r r5 r# rk rg r[ r rM rj rl rm rn Z StructArrayZ from_pandasr! )rb r^ rW r* r' r( �valsr) r r r �test_arrow_array_missing� s" rr �breaksZ2017�D)r rf �floatzdatetime64[ns])r c C s� dd l }ddlm} t�| �}d |d< t�d|i�}|�|�}t|� d�j |�sTt �|� � }t|d j tj�srt �t�||� |�||g�}|� � }tj||gdd�}t�||� |j|jg |�d�j d�g|jd�}|� � }t�||dd� � d S ) Nr )r^ r rC T)Z ignore_index)rg )�schema)r] r_ r^ r r5 r< � DataFrame�tablerk rm rg r[ � to_pandasr � IntervalDtyper% �assert_frame_equalZ concat_tables�concatZ chunked_array�columnrv ) rs rb r^ rW �dfrx r* Ztable2r) r r r �test_arrow_table_roundtrip� s$ r c C sx dd l }t�| �}d |d< t�d|i�}|�|�}|�� }|jjd ksJt �|� � }t |d j tj �sht �t�||� d S )Nr r rC )r] r r5 r< rw rx Zreplace_schema_metadatarv �metadatar[ ry rk r rz r% r{ )rs rb rW r~ rx r* r r r �+test_arrow_table_roundtrip_without_metadata s r� )$�numpyr! r/ Z pandas.util._test_decorators�utilZ_test_decorators�tdZpandasr< r r r r r r r Zpandas._testingZ_testingr% Zpandas.core.arraysr Zfixturer r r2 rH r\ Z skip_if_noZ pyarrow_skipre ro rr r0 r1 r r� r r r r �<module> s@ $ )1
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка