Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/indexes/interval/__pycache__/test_constructors.cpython-37.pyc
Назад
B �5�g�D � @ s d dl mZ d dlZd dlZd dlmZ d dlmZ d dl m Z m Z m Z m Z mZmZmZmZmZmZmZ d dlmZ d dlmZ d dlm m Z ej ddgd�d d � �Z G d d � d �Z G d d� de �Z!G dd� de �Z"G dd� de �Z#G dd� de �Z$dd� Z%dS )� )�partialN)�is_categorical_dtype)� IntervalDtype) � Categorical�CategoricalIndex� Float64Index�Index� Int64Index�Interval� IntervalIndex� date_range�notna� period_range�timedelta_range)� IntervalArrayZfoo)�paramsc C s | j S )N)�param)�request� r �e/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/indexes/interval/test_constructors.py�name s r c @ sj e Zd ZdZej�d�ej�ddddddgej d d d �e e d d ��e e� ddd��e dd d�e dd dd�edd d�g�dd� ��Zej�de ddddd g�d fe ddddd g�d fe ddddd g�d fe ddddd g�d fe d d!d�d fedd!d�d fg�d"d#� �Zej�de ddddd g�e ddddd g�e ddddd g�e ddddd g�e d d!d�edd!d�g�d$d%� �Zej�d�ej�dejgd ejgd ejgd& g�d'd(� ��Zej�d�ej�dg ejg d d �ejg d d �ejg d d �ejg d d �g�d)d*� ��Zej�ded+�ed,�ejed,�ed �ejed,�d-d �g�d.d/� �Zej�d0eeg�d1d2� �Z d3d4� Z d5S )6�ConstructorTestsz� Common tests for all variations of IntervalIndex construction. Input data to be supplied in breaks format, then converted by the subclass method get_kwargs_from_breaks to the expected format. z+ignore:Passing keywords other:FutureWarning�breaks� � � �\ i� � �int64)�dtypei����� � � g �?Z20180101)�periodsz US/Eastern)r# �tzz1 dayc C s� | � ||�}|f ||d�|��}|j|ks.t�|j|ks<t�|jjt|dd�ksTt�t�|j t |d d� �� t�|j t |dd � �� d S )N)�closedr r r ���� ) �get_kwargs_from_breaksr% �AssertionErrorr r �subtype�getattr�tm�assert_index_equal�leftr �right)�self� constructorr r% r � result_kwargs�resultr r r �test_constructor' s z!ConstructorTests.test_constructorzbreaks, subtyper r' � � �float64zdatetime64[ns]ztimedelta64[ns]z 2017-01-01� c C s� d }|dkr |j jdkr t}t�|�� | �|�|��}W d Q R X |f |�}| �|�}t|d�}xH|t|�fD ]8} t�|�� |f d| i|��} W d Q R X t� | |� qnW d S )Nr )�M�mr/ r ) r �kind� FutureWarningr, �assert_produces_warningr( Zastyper �strr- ) r0 r1 r r* �warnZexpected_kwargs�expectedr2 �iv_dtyper r3 r r r �test_constructor_dtype> s z'ConstructorTests.test_constructor_dtypec C s� d }t |t�r |jtkr t}t|j�}| �|�}xN|t|�fD ]>}t � |�� |f |dd�|��}W d Q R X |jj dks>t �q>W d S )Nr. )r r% ) � isinstancer �funcr r< r r r( r> r, r= r% r) )r0 r1 r r? rA r2 r r3 r r r � test_constructor_pass_closed\ s z-ConstructorTests.test_constructor_pass_closed�2 c C sl | � |�}|f d|i|��}tj}tj|d d� td�}|j|ksFt�|jj|ksVt�t � t�|�|� d S )Nr% r& )r ) r( �npr7 �array�objectr% r) r r* r, �assert_numpy_array_equal)r0 r1 r r% r2 r3 �expected_subtype�expected_valuesr r r �test_constructor_nanx s z%ConstructorTests.test_constructor_nanc C sv | � |�}|f d|i|��}tjg td�}t|dtj�}|jsBt�|j|ksPt�|j j |ks`t�t � t�|�|� d S )Nr% )r r ) r( rG rH rI r+ r �emptyr) r% r r* r, rJ )r0 r1 r r% r2 r3 rL rK r r r �test_constructor_empty� s z'ConstructorTests.test_constructor_empty� 0123456789Z abcdefghijz<U1c C s2 d}t jt|d�� |f | �|�� W d Q R X d S )NzIcategory, object, and string subtypes are not supported for IntervalIndex)�match)�pytest�raises� TypeErrorr( )r0 r1 r �msgr r r �test_constructor_string� s z(ConstructorTests.test_constructor_string�cat_constructorc C s` t |t�r |jtkr t�� tjddd�}t� |�}||�}| � |�}|f |�}t � ||� d S )Nr r )r ) rC r rD r rR �skiprG �aranger � from_breaksr( r, r- )r0 r1 rW r r@ Z cat_breaksr2 r3 r r r �"test_constructor_categorical_valid� s z3ConstructorTests.test_constructor_categorical_validc C s | � td��}d}tjt|d�� |f ddi|�� W d Q R X d}tjt|d�� |f ddi|�� W d Q R X d }tjt|d�� |f ddi|�� W d Q R X td dd �}| � |�}d }tjt|d�� |f |� W d Q R X | � tdd d ��}d}tjt|d�� |f |� W d Q R X d S )Nr z8closed must be one of 'right', 'left', 'both', 'neither')rQ r% �invalidz)dtype must be an IntervalDtype, got int64r r z(data type ["']invalid["'] not understoodz 2000-01-01)r# z:Period dtypes are not supported, use a PeriodIndex insteadr& z+left side of interval must be <= right side)r( �rangerR rS � ValueErrorrT r )r0 r1 ZfillerrU r# Zperiods_kwargsZdecreasing_kwargsr r r �test_generic_errors� s& z$ConstructorTests.test_generic_errorsN) �__name__� __module__� __qualname__�__doc__rR �mark�filterwarnings� parametrizerG rY r r] r r r r4 rB rE �nanrM rH rO �tuple�listrI rV r r r[ r_ r r r r r sZ 6 r c @ sZ e Zd ZdZejdd� �Zd dd�Zdd� Zej � d e j e j fe j e j fg�d d � �Zd S )�TestFromArraysz+Tests specific to IntervalIndex.from_arraysc C s t jS )N)r � from_arrays)r0 r r r r1 � s zTestFromArrays.constructorr/ c C s |dd� |dd� d�S )z� converts intervals in breaks format to a dictionary of kwargs to specific to the format expected by IntervalIndex.from_arrays Nr& r' )r. r/ r )r0 r r% r r r r( � s z%TestFromArrays.get_kwargs_from_breaksc C s� t td�dd�}d}tjt|d��" t�|d d� |dd � � W d Q R X ddd g}d d g}d }tjt|d�� t�||� W d Q R X d S ) N� 01234abcdeT)�orderedzIcategory, object, and string subtypes are not supported for IntervalIndex)rQ r& r' r r5 r z(left and right must have the same length)r ri rR rS rT r rk r^ )r0 �datarU r. r/ r r r �test_constructor_errors� s & z&TestFromArrays.test_constructor_errorszleft_subtype, right_subtypec C sp t jd|d�}t jdd|d�}t�||�}t|�}t|�}t j}t�|j|� t�|j |� |j j |kslt �dS )z:mixed int/float left/right results in float for both sides� )r r' r N) rG rY r rk r r7 r, r- r. r/ r r* r) ) r0 Z left_subtypeZ right_subtyper. r/ r3 Z expected_leftZexpected_rightrK r r r �test_mixed_float_int� s z#TestFromArrays.test_mixed_float_intN)r/ )r` ra rb rc rR �fixturer1 r( ro rd rf rG r r7 rq r r r r rj � s rj c @ s@ e Zd ZdZejdd� �Zddd�Zdd� Zd d � Z d d � Z d S )�TestFromBreaksz+Tests specific to IntervalIndex.from_breaksc C s t jS )N)r rZ )r0 r r r r1 s zTestFromBreaks.constructorr/ c C s d|iS )z� converts intervals in breaks format to a dictionary of kwargs to specific to the format expected by IntervalIndex.from_breaks r r )r0 r r% r r r r( s z%TestFromBreaks.get_kwargs_from_breaksc C s< t td�dd�}d}tjt|d�� t�|� W d Q R X d S )Nrl T)rm zIcategory, object, and string subtypes are not supported for IntervalIndex)rQ )r ri rR rS rT r rZ )r0 rn rU r r r ro s z&TestFromBreaks.test_constructor_errorsc C s* dg}t �|�}t �g �}t�||� dS )z3breaks of length one produce an empty IntervalIndexr N)r rZ r, r- )r0 r r3 r@ r r r �test_length_one) s z TestFromBreaks.test_length_onec C s: t �d�}t�|�j}|jjd ks6|jj|jjk s6t�d S )Nr8 ) rG rY r rZ �_data�_left�base�_rightr) )r0 r r3 r r r � test_left_right_dont_share_data0 s z.TestFromBreaks.test_left_right_dont_share_dataN)r/ ) r` ra rb rc rR rr r1 r( ro rt ry r r r r rs s rs c @ s8 e Zd ZdZejdd� �Zd dd�Zdd� Zd d � Z d S ) �TestFromTuplesz+Tests specific to IntervalIndex.from_tuplesc C s t jS )N)r � from_tuples)r0 r r r r1 : s zTestFromTuples.constructorr/ c C sl t |�dkrd|iS tt|dd� |dd� ��}t|ttf�rHd|iS t|�r^d|�|�iS dt�|�iS )z� converts intervals in breaks format to a dictionary of kwargs to specific to the format expected by IntervalIndex.from_tuples r rn Nr& r' ) �lenri �ziprC rh r � _constructor�comZasarray_tuplesafe)r0 r r% �tuplesr r r r( > s z%TestFromTuples.get_kwargs_from_breaksc C s� dddg}d}t jt|j|d�d�� t�|� W d Q R X dddg}d}t jt|j|d�d�� t�|� W d Q R X dd d g}t jt|j|d�d�� t�|� W d Q R X d S ) N)r r' r5 )r r6 z5IntervalIndex.from_tuples received an invalid item, 2)�t)rQ )r5 z>IntervalIndex.from_tuples requires tuples of length 2, got {t})r5 r r6 )r8 � )rR rS rT �formatr r{ r^ )r0 r� rU r r r ro M s z&TestFromTuples.test_constructor_errorsc C s> dt jt jfdg}t�|�}t�dt jdg�}t�||� d S )N)r r' )r5 r )rG rg r r{ r, r- )r0 Zna_tupleZ idx_na_tupleZidx_na_elementr r r �test_na_tuples^ s z TestFromTuples.test_na_tuplesN)r/ ) r` ra rb rc rR rr r1 r( ro r� r r r r rz 7 s rz c @ s e Zd ZdZejeeedd�gddgd�dd� �Z d$d d �Z d d � Z dd� Z dd� Z ej�d�ej�dg dfejejgdfedddd�edddd�gdfedddd�eddd d�gdfejed�dd�d fg�d d � ��Zej�d eejeeg�d d � �Zd!d"� Zd#S )%�TestClassConstructorsz6Tests specific to the IntervalIndex/Index constructors�interval)r r r )r Zidsc C s |j S )N)r )r0 r r r r r1 i s z!TestClassConstructors.constructorr/ c sv t |�dkrd|iS � fdd�t|dd� |dd� �D �}t|t�rNd|iS t|�rdd|�|�iS dtj|td�iS ) z� converts intervals in breaks format to a dictionary of kwargs to specific to the format expected by the IntervalIndex/Index constructors r rn c s( g | ] \}}t |�r t||� �n|�qS r )r r )�.0r. r/ )r% r r � <listcomp>y s z@TestClassConstructors.get_kwargs_from_breaks.<locals>.<listcomp>Nr& r' )r ) r| r} rC ri r r~ rG rH rI )r0 r r% �ivsr )r% r r( p s z,TestClassConstructors.get_kwargs_from_breaksc C s dS )z� override the base class implementation since errors are handled differently; checks unnecessary since caught at the Interval level Nr )r0 r1 r r r r_ � s z)TestClassConstructors.test_generic_errorsc C s d S )Nr )r0 r r r rV � s z-TestClassConstructors.test_constructor_stringc C s� t dddd�t dddd�g}d}tjt|d �� ||� W d Q R X d }tjt|d �� |d � W d Q R X d }tjt|d �� |ddg� W d Q R X d S ) Nr r' r/ )r% r5 r r. z-intervals must all be closed on the same side)rQ zPIntervalIndex\(...\) must be called with a collection of some kind, 5 was passedr8 z?type <class 'numpy.int(32|64)'> with value 0 is not an interval)r rR rS r^ rT )r0 r1 r� rU r r r ro � s z-TestClassConstructors.test_constructor_errorsz+ignore:Passing keywords other:FutureWarningz data, closed�both�neitherr r )r% r5 r8 r. c C sL t |t�r|�� }ndd� |D �}tj||d�}|||d�}t�||� d S )Nc S s$ g | ] }t |�r |j|jfn|�qS r )r r. r/ )r� Zivr r r r� � s zGTestClassConstructors.test_override_inferred_closed.<locals>.<listcomp>)r% )rC r Z to_tuplesr{ r, r- )r0 r1 rn r% r� r@ r3 r r r � test_override_inferred_closed� s z3TestClassConstructors.test_override_inferred_closed�values_constructorc C sX t dd�t dd�t dd�g}||�}t|td�}t|�tks@t�t�|jt� |�� d S )Nr r' r5 r )r ) r r rI �typer) r, rJ �valuesrG rH )r0 r� � intervalsr� r3 r r r �test_index_object_dtype� s z-TestClassConstructors.test_index_object_dtypec C sX t dddd�t dddd�t dddd�t dd d d�g}t|�}t|td �}t�||� d S ) Nr r' r. )r% r5 r/ r r� r6 r� )r )r r rI r, r- )r0 r� r3 r@ r r r �test_index_mixed_closed� s z-TestClassConstructors.test_index_mixed_closedN)r/ ) r` ra rb rc rR rr r r r r1 r( r_ rV ro rd re rf rG rg r rZ r] r� ri rH r r� r� r r r r r� f s* $ r� c C sd t tjd�} d}tjt|d�� tg | dd� W d Q R X tjt|d�� tg | dd� W d Q R X d S )Nr. z*closed keyword does not match dtype.closed)rQ r� )r r% )r rG r rR rS r^ r r )r rU r r r �test_dtype_closed_mismatch� s r� )&� functoolsr �numpyrG rR Zpandas.core.dtypes.commonr Zpandas.core.dtypes.dtypesr Zpandasr r r r r r r r r r r Zpandas._testingZ_testingr, Zpandas.core.arraysr Zpandas.core.common�core�commonr rr r r rj rs rz r� r� r r r r �<module> s 4 A1&/r
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка