Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/indexes/__pycache__/common.cpython-37.pyc
Назад
B �5�gLs � @ s� d dl mZ d dlmZ d dlZd dlZd dlZd dlmZ d dl m Z d dl m Z d dl mZ d dlZd dlmZmZmZmZmZmZmZmZmZmZmZm Z m Z d dl m Z d d l!m"Z" G d d � d �Z#G d d � d e#�Z$dS )� )� annotations)�datetimeN)�iNaT)� Timestamp)�is_datetime64tz_dtype)�CategoricalDtype) �CategoricalIndex� DatetimeIndex� Float64Index�Index� Int64Index� IntervalIndex� MultiIndex� PeriodIndex� RangeIndex�Series�TimedeltaIndex� UInt64Index�isna)�DatetimeIndexOpsMixinc @ s� e Zd ZU dZded< ejdd� �Zdd�dd �Zd d � Z ej � d d dg�dd� �Z dd� Z dd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zej � d/e e e j e g�d0d1� �Z!d2d3� Z"d4d5� Z#d6d7� Z$d8d9� Z%d:d;� Z&d<d=� Z'd>d?� Z(d@dA� Z)dBdC� Z*dDdE� Z+dFdG� Z,dHdI� Z-ej � dJdKdL� dMdL� g�dNdO� �Z.dPdQ� Z/ej � dRdSdTg�ej � d d dUg�ej � dVdSdTg�dWdX� ���Z0dYdZ� Z1ej j2d[d\� �Z3d]d^� Z4d_d`� Z5dadb� Z6dcdd� Z7d S )e�Basez/ Base class for index sub-class tests. z type[Index]� _index_clsc C s t d��d S )NzMethod not implemented)�NotImplementedError)�self� r �Q/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/indexes/common.py� simple_index* s zBase.simple_indexr )�returnc C s t d��d S )NzMethod not implemented)r )r r r r � create_index. s zBase.create_indexc C s: d� dddddg�}tjt|d�� | �� W d Q R X d S )N�|zNIndex\(\.\.\.\) must be called with a collection of some kind, None was passedzPDatetimeIndex\(\) must be called with a collection of some kind, None was passedzQTimedeltaIndex\(\) must be called with a collection of some kind, None was passedz:__new__\(\) missing 1 required positional argument: 'data'z2__new__\(\) takes at least 2 arguments \(1 given\))�match)�join�pytest�raises� TypeErrorr )r �msgr r r � test_pickle_compat_construction1 s z$Base.test_pickle_compat_construction�nameN�new_namec C s� |}|r|}n |j pd}|j|d�}|j|ks2t�t|j�dksDt�|jd |ksVt�|| j|jk sjt�|jd|d�}|j|k s�t�d S )Nr )r' � F)�indexr' )r' Zto_framer* �AssertionError�len�columns�values)r r' r �idxZidx_nameZdfr r r � test_to_frameB s zBase.test_to_framec C sb |}dt |�j� �}tjt|d�� |�d� W d Q R X tjt|d�� |�dd� W d Q R X d S )Nz\This method is only implemented for DatetimeIndex, PeriodIndex and TimedeltaIndex; Got type )r r) � )�type�__name__r" r# r �shift)r r r/ r% r r r � test_shiftV s zBase.test_shiftc C s2 |}t jtdd�� t|�|g d� W d Q R X d S )Nz"Index.name must be a hashable type)r )r' )r" r# r$ r2 )r r r/ r r r � test_constructor_name_unhashablec s z%Base.test_constructor_name_unhashablec C s� |}t |t�sHd|_t|�}t�||� t|dd�}d|_t�||� nzddg|_t|�}t�|ttdddddd gd d �ddgd �� t|d dgd �}t�|ttdddddd gd d �d dgd �� d S )N�foo�bar)r' )r7 �one)r7 �two)r8 r9 )Zbazr: )�quxr9 )r; r: �object)�dtype)�names�A�B)� isinstancer r' r �tm�assert_index_equalr> )r r �expected�resultr r r � test_create_index_existing_namej sF z$Base.test_create_index_existing_namec C sf |}t |t�rt�t|�tkr"d S t|j�j}d�d|� d�d|� �g�}tj t |d�� |d W d Q R X d�d|� d�d |� �g�}tj t |d�� d| W d Q R X |� d d �}tj t |d�� |d W d Q R X |� d d �}tj t |d�� d| W d Q R X |� d d �}tj t |d�� |d W d Q R X |� d d �}tj t |d�� d| W d Q R X d S ) Nr z'unsupported operand type\(s\) for \*: 'z ' and 'int'zHcannot perform (__mul__|__truediv__|__floordiv__) with this index type: )r r) z1unsupported operand type\(s\) for \*: 'int' and '�'zKcannot perform (__rmul__|__rtruediv__|__rfloordiv__) with this index type: �*�/z//) rA r r+ r2 r �_datar3 r! r" r# r$ �replace)r r r/ �typZlmsgZrmsgZdiv_errZ floordiv_errr r r �test_numeric_compat� s6 zBase.test_numeric_compatc C sL |}t jtdd�� |�� W d Q R X t jtdd�� |�� W d Q R X d S )Nzcannot perform all)r zcannot perform any)r" r# r$ �all�any)r r r/ r r r �test_logical_compat� s zBase.test_logical_compatc C s |}t �tt|��|� d S )N)rB rC �eval�repr)r r r/ r r r �test_repr_roundtrip� s zBase.test_repr_roundtripc C sB |}|� d�}t�dd �� t|� dt|�ks4t�W d Q R X d S )N�2 zdisplay.max_seq_itemsz...)�repeat�pdZoption_contextrR �strr+ )r r r/ r r r � test_repr_max_seq_item_setting� s z#Base.test_repr_max_seq_item_settingc C s� t |t�rd S t|�|ddd�}t|�|dd�}||k s<t�|�|�sJt�|jdksXt�|jdksft�td|d�}td|d d � d�}t |t�s�|| }|jjdks�t�d S ) NT�mario)�copyr' F)rZ r1 )r* � ���) rA r r2 r+ �equalsr' r r r* )r r* �first�second�s1�s2Zs3r r r �test_copy_name� s zBase.test_copy_namec C s� t |t�rd S |jdd�jdks$t�tjtdd�� |jddgd� W d Q R X t|�j � d�}tjt |d�� |jdggd� W d Q R X d S )NrY )r' z$Length of new names must be 1, got 2)r Zluigiz .name must be a hashable type) rA r rZ r' r+ r" r# � ValueErrorr2 r3 r$ )r r* r% r r r �test_copy_name2� s zBase.test_copy_name2c C s� i }t |t�r|j|d< nt |tttf�r.d S t|�}||jfddi|��}t|j �rh|� d�� |j �}t |t tf�r�|�d �}t�||� t |t�r�|f |jdd�|��}tj|j|jdd� n6t |t�r�n*||jfddi|��}tj|j|jdd� d S ) N�freqrZ T�UTCF)ZordinalrZ Zsame)Z check_same)rA r re r r r r2 r. r r= Z tz_localizeZ tz_convert�tzr r � _with_freqrB rC Zasi8�assert_numpy_array_equalr )r r* Z init_kwargsZ index_typerE r r r �test_ensure_copied_data s& z Base.test_ensure_copied_datac C s~ |j �� |�� }|jr(|dks$t�d S |�|d � |�� }|jdd�}t|ttf�sd||ksdt�|j dkrz||kszt�d S )Nr T)�deepr< ) �_engineZ clear_mappingZ memory_usage�emptyr+ �get_locrA r r Z inferred_type)r r* rE Zresult2Zresult3r r r �test_memory_usage& s zBase.test_memory_usagec C s8 t |t�rd S |�� }t�|��� }tj||dd� d S )NF)Z check_dtype)rA r �argsort�np�arrayrB ri )r �requestr* rE rD r r r � test_argsort; s zBase.test_argsortc C s� t �|�}|�� }t�||� tt|�ttf�r�d}tj t |d�� t j|dd� W d Q R X d}tj t |d�� t j|dd� W d Q R X d}tj t |d�� t j|d d � W d Q R X d S ) Nz%the 'axis' parameter is not supported)r r) )�axisz%the 'kind' parameter is not supportedZ mergesort)�kindz&the 'order' parameter is not supported)�a�b)�order) rq rp rB ri rA r2 r r r" r# rc )r r* rE rD r% r r r �test_numpy_argsortD s zBase.test_numpy_argsortc C sr d}|� � }t|j�|�|jd�}t�|�|�|� |}t�t |��}t|j�|�|jd�}t�|�|�|� d S )Nr1 )r' ) rZ r r. rU r' rB rC rq �aranger, )r r �repr/ rD r r r � test_repeat^ s zBase.test_repeatc C sX d}|}|� |�}t�t� ||�|� d}tjt|d�� tj ||dd� W d Q R X d S )Nr1 z%the 'axis' parameter is not supported)r r )ru )rU rB rC rq r" r# rc )r r r| r/ rD r% r r r �test_numpy_repeati s zBase.test_numpy_repeat�klassc C s� |}t |ttf�r |�d �}dgt|� }|�||��}|}t�||� dgdgt|dd � � }t|j g|dd � � � |j d�}|�||��}t�||� d S )NTFr) )r= ) rA r r rh r, �whererB rC r Z _na_value�tolistr= )r r r r/ ZcondrE rD r r r � test_wheres s "zBase.test_wherec C s> |dd� }t |�sd S |dd� �|�d|d ��s:t�d S )Nr) � r )r, r] �insertr+ )r r* rE r r r �test_insert_base� s zBase.test_insert_basec C s� t |�s d S t|t�rd S |dd � }|�d�}|�|�s>t�|j|jksNt�|d d� }|�d�}|�|�srt�|j|jks�t�t |�}d|� d|� �}tjt |d�� |�|� W d Q R X d S )Nr) r r\ zindex z' is out of bounds for axis 0 with size )r ) r, rA r �deleter] r+ r' r" r# � IndexError)r r* rD rE �lengthr% r r r �test_delete_base� s zBase.test_delete_basec C s� t |t�rd S |�|�s t�|�|�� �s.t�|�|�t��sBt�|�t|��rTt�|�t� |��rht�t |t �s�t |td�}|�|�s�t�|�|�s�t�|j dkr�|�t |��r�t�d S )N)r= r) )rA r r] r+ rZ �astyper< �listrq rr r r Znlevelsr )r r* Z same_valuesr r r � test_equals� s zBase.test_equalsc C s� |}t |�}|dd� }|dd� �|dd� �}|dd� }d}tjt|d�� ||k W d Q R X t�dg| �}t�dg|d dg �} t�||k|� t�||k| � t�|�} t�|dd� �} t�|dd� �|dd� ��} t�|dd� �} tjt|d�� || k W d Q R X t�|| k|� t�|| k| � t | �}t | �}t | �}t | �}tjt|d�� ||k W d Q R X t�||k|� t�||k| � tjtd d�� ||k W d Q R X tjtd d�� ||k W d Q R X tjtd d�� || k W d Q R X d }tjt|d�� ||k W d Q R X tjtd d�� || k W d Q R X t |t ��s�t�dgt |�d ddg �}|d }t�||k|� t� ||kt |�� d S ) Nr r\ ���r) z)Lengths must match|could not be broadcast)r TFzLengths must matchz3Can only compare identically-labeled Series objectsr1 ) r, �appendr" r# rc rq rr rB ri r rA r Zassert_series_equal)r r Zindex_a�nZindex_bZindex_cZindex_dr% Z expected1Z expected2Zarray_aZarray_bZarray_cZarray_dZseries_aZseries_bZseries_cZseries_dZ expected3�itemr r r �test_equals_op� sX zBase.test_equals_opc C s& |}dd� |D �}|� � |ks"t�d S )Nc S s g | ] }t |��qS r )rW )�.0�xr r r � <listcomp>� s z$Base.test_format.<locals>.<listcomp>)�formatr+ )r r r/ rD r r r � test_format� s zBase.test_formatc C s4 | � g �}|�� g kst�|jdd�dgks0t�d S )NT)r' � )r r� r+ )r Z empty_idxr r r �test_format_empty s zBase.test_format_emptyc C s |}|j dd�}tjdgt|� td�}t�|j|� |jdksDt �|j dd�}t� |j �}t|�dkrld S t |t �r�t|d< n t |tttf�r�d S tj|d< t |t�r�t|�||jd�}nJt|�|�}tjdgt|� td�}d|d< t�|j|� |jdk�st �d S )NT)rk F)r= r r) )re )rZ rq rr r, �boolrB ri �_isnan�hasnansr+ �asarrayr. rA r r r r r �nanr r2 re )r Z index_flatr* r/ rD r. r r r �test_hasnans_isnans s* zBase.test_hasnans_isnansc C sn t |�dkr�nZt|t�rV|jdd�}d}tjt|d�� |�|d � W d Q R X �n|jdd�}|�|d �}t� ||� ||k s�t �d}tjt |d�� |�|d g� W d Q R X |jdd�}t � |j�}t|t�r�t|d< n t|tttf�r�d S t j|d< t|t��r t|�||jd�}n t|�|�}t jd gt |� td �}d|d< t�|j|� |j dk�sjt �d S ) Nr T)rk z"isna is not defined for MultiIndex)r z"'value' must be a scalar, passed: r) )re F)r= ) r, rA r rZ r" r# r ZfillnarB rC r+ r$ rq r� r. r r r r r r� r r2 re rr r� ri r� r� )r r* r/ r% rE r. rD r r r � test_fillna& s8 zBase.test_fillnac C s� t |�dkr(t�|�� tjg td�� n�t|t�rb|� � }d}t j t |d�� |�� W d Q R X nn|j s�t�|�� tjt |�td�� t�|�� tjt |�td�� n*t|�}t�|�� |� t�|�� | � d S )Nr )r= z"isna is not defined for MultiIndex)r )r, rB ri r rq rr r� rA r rZ r" r# r r� �zerosZnotnaZones)r r* r/ r% rE r r r � test_nullsM s zBase.test_nullsc C s$ |}|j rt�|d d� j s t�d S )Nr )rm r+ )r r r/ r r r � test_empty_ s zBase.test_emptyc C s, |}|j r(|j||d�}||k�� s(t�d S )N)�how)� is_uniquer! rN r+ )r Z join_typer r/ Zjoinedr r r �test_join_self_uniquee s zBase.test_join_self_uniquec C s< |}t |t�r|�d�}n|}|�dd� �}t�||� d S )N�int64c S s | S )Nr )r� r r r �<lambda>u � z Base.test_map.<locals>.<lambda>)rA r r� �maprB rC )r r r/ rD rE r r r �test_mapk s z Base.test_map�mapperc C s dd� t | |�D �S )Nc S s i | ]\}}||�qS r r )r� �e�ir r r � <dictcomp>| s z!Base.<lambda>.<locals>.<dictcomp>)�zip)r. r* r r r r� | r� z Base.<lambda>c C s t | |�S )N)r )r. r* r r r r� } r� c C s� |}t |t�r"t�dt|�� �� ||j|�}t |t�rNt |t�rN|�d�}n|}|� |�}t � ||� t t jgt|� �}|� |||��}t � ||� d S )Nzskipping tests for r� )rA r r" �skipr2 r. r �dictr� r� rB rC r rq r� r, )r r� r r/ �identityrD rE r r r �test_map_dictlikey s zBase.test_map_dictlikec C s4 |}|� t�}tdd� |D �td�}t�||� d S )Nc S s g | ] }t |��qS r )rW )r� r� r r r r� � s z%Base.test_map_str.<locals>.<listcomp>)r= )r� rW r r< rB rC )r r r/ rE rD r r r � test_map_str� s zBase.test_map_strrZ TFr7 �orderedc C s� |}|r|� |�}t|d�}|j||d�}t|||d�}tj||dd� t|�� �� d d� |�}|j||d�}t|||d�}tj||dd� |dkr�|jd |d�}t||d �}tj||dd� d S ) N)r� )rZ )r' r� T)�exactr\ )r' r= F�category)r' )�renamer r� r rB rC �uniquer� ) r rZ r' r� r r/ r= rE rD r r r �test_astype_category� s zBase.test_astype_categoryc C s� |� � }|jdkst�|d d� }|jdks0t�|�d|d �}|jdksNt�|�dtj�}|jdksjt�|�dtj�}|jdks�t�d S )NTr F)Zdrop_duplicatesr� r+ r� rq r� )r r r* Z index_emptyZ index_dupZindex_naZ index_na_dupr r r �test_is_unique� s zBase.test_is_uniquec C s2 |}t t�|��}|j t t�|��|ks.t�d S )N)r, �gcZ get_referrersrl r+ )r r r* Z nrefs_prer r r �test_engine_reference_cycle� s z Base.test_engine_reference_cyclec C s` |}d}t |ttf� }tjt||d�� |d d �d f }W d Q R X t |tj�s\tt |���d S )Nz&Support for multi-dimensional indexing)r Zcheck_stacklevel) rA r r rB Zassert_produces_warning� FutureWarningrq Zndarrayr+ r2 )r r r/ r% �check�resr r r �test_getitem_2d_deprecated� s z Base.test_getitem_2d_deprecatedc C s. |}|� |d � |�� }|j|jks*t�d S )Nr )rn rZ �_cacher+ )r r r/ rZ r r r �test_copy_shares_cache� s zBase.test_copy_shares_cachec C sX |}|� |d � |�� }|j|jks*t�|�|j�}|j|jk sFt�|ji ksTt�d S )Nr )rn Z_viewr� r+ Z _shallow_copyrJ )r r r/ Z shallow_copyr r r � test_shallow_copy_shares_cache� s z#Base.test_shallow_copy_shares_cachec C s� |d d� }t �ddt jddg�}t�|�|�|ddg |ddg d�� ttdd d�tdd d�tj tdd d�tdd d�gd d �j }t d �t d�g}|d |ddg |d |ddg i}t�|�|�|� d S )N� r) r1 r r� r[ )g �?g @i� � � rf )rg z 2011-11-01z 2011-12-01) rq rr r� rB Zassert_dict_equal�groupbyr r rV �NaTr. r )r r r/ Z to_groupbyZex_keysrD r r r �test_index_groupby� s $ $zBase.test_index_groupby)8r3 � __module__� __qualname__�__doc__�__annotations__r" Zfixturer r r& �markZ parametrizer0 r5 r6 rF rM rP rS rX rb rd rj ro rt rz r} r~ r� �tuplerq rr r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� Zarm_slowr� r� r� r� r� r r r r r # s` 5+ ? ' r c @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dS )� NumericBasezJ Base class for numeric index (incl. RangeIndex) sub-class tests. c C s> t ddg|d�}| �|�}tjddg|d�}t�|j|� d S )Nr) r1 )r= )r r rq rr rB ri rJ )r r= r/ rE rD r r r � test_constructor_unwraps_index s z*NumericBase.test_constructor_unwraps_indexc C s d S )Nr )r r r r r� s zNumericBase.test_wherec C s" |}|d }|� |�dks t�d S )Nr F)Z$_can_hold_identifiers_and_holds_namer+ )r r r/ �keyr r r �test_can_hold_identifiers s z%NumericBase.test_can_hold_identifiersc s< |}t dd� |D ��� � fdd�|D �}|�� |ks8t�d S )Nc s s | ]}t t|��V qd S )N)r, rW )r� r� r r r � <genexpr>$ s z*NumericBase.test_format.<locals>.<genexpr>c s g | ]}t |��� ��qS r )rW �ljust)r� r� )� max_widthr r r� % s z+NumericBase.test_format.<locals>.<listcomp>)�maxr� r+ )r r r/ rD r )r� r r� ! s zNumericBase.test_formatc C s d S )Nr )r r r r rM ( s z NumericBase.test_numeric_compatc C sx |}|}|t jkr:t|d t jgt|dd � � td�}n"t|d tjgt|dd � � �}|�d|�}t � ||� d S )Nr r) )r= ) rV r� r r� r< r rq r� r� rB rC )r Z nulls_fixturer r* Zna_valrD rE r r r �test_insert_na+ s (" zNumericBase.test_insert_nac C s� | j }|tkrtd�}n|tjddd��}tjddd�d }t|�}|d }t�||� d| }t�||� t|�}tjddd�}|| }t�||� t| �}tjddd�}|| }t�||� d S )Nr� r� )r= g������ @Zfloat64)r r rq r{ r rB rC r� )r Z index_clsr/ ZarrrD Zfidxrw rE r r r �$test_arithmetic_explicit_conversions8 s$ z0NumericBase.test_arithmetic_explicit_conversionsc C s@ |}d|� �}t jt|d�� | jdddg|d� W d Q R X d S )Nz:Incorrect `dtype` passed: expected \w+(?: \w+)?, received )r r) r1 r[ )r= )r" r# rc r )r Z invalid_dtyper= r% r r r �test_invalid_dtypeT s z NumericBase.test_invalid_dtypeN) r3 r� r� r� r� r� r� r� rM r� r� r� r r r r r� s r� )%� __future__r r r� �numpyrq r" Z pandas._libsr Zpandas._libs.tslibsr Zpandas.core.dtypes.commonr Zpandas.core.dtypes.dtypesr ZpandasrV r r r r r r r r r r r r r Zpandas._testingZ_testingrB Z pandas.core.indexes.datetimeliker r r� r r r r �<module> s& < o
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка