Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/indexes/__pycache__/test_common.cpython-37.pyc
Назад
B �5�g8 � @ s� d Z ddlZddlZddlZddlmZ ddlmZ ddl m Z m Z ddl Z ddl mZmZmZmZmZmZ ddlmZ G dd� d�Zej�d dd g�d d � �Zej�d d dg�dd� �Zdd� Z dS )z� Collection of tests asserting things that should be true for any index subclass. Makes use of the `indices` fixture defined in pandas/tests/indexes/conftest.py. � N)�iNaT)�IS64)�is_period_dtype�needs_i8_conversion)�CategoricalIndex� DatetimeIndex� MultiIndex� PeriodIndex� RangeIndex�TimedeltaIndexc @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zej�dd d d d d d!g�d"d#� �Zd$d%� Zd&S )'� TestCommonc C s� t |t�rd S |�g ��|�s"t�xV|j|jgfD ]D}t |jt�rN||jkrNq2d}tjt |d�� |�|� W d Q R X q2W x6ddgfD ](}tjt dd�� |�|� W d Q R X q�W d S )NzTCannot remove 1 levels from an index with 1 levels: at least one level must be left.)�match�wrongz>'Requested level \(wrong\) does not match index name \(None\)') � isinstancer Z droplevel�equals�AssertionError�name�tuple�pytest�raises� ValueError�KeyError)�self�index�level�msg� r �V/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/indexes/test_common.py�test_droplevel s zTestCommon.test_droplevelc C s` |}d}dgg}t jt|d�� |j|d� W d Q R X t jt|d�� |j|d� W d Q R X d S )Nz"Index.name must be a hashable type�1)r )r )�names)r r � TypeError�rename� set_names)r � index_flatr �messageZrenamedr r r �"test_constructor_non_hashable_name: s z-TestCommon.test_constructor_non_hashable_namec C s$ |}t |�|�}t�|j|j� d S )N)�type�tmZ assert_equal�_data)r r$ �a�br r r � test_constructor_unwraps_indexI s z)TestCommon.test_constructor_unwraps_indexc C s |}|� � }t�||� d S )N)Z to_flat_indexr( �assert_index_equal)r r$ r �resultr r r �test_to_flat_indexN s z TestCommon.test_to_flat_indexc C s� |}d}|j }|�|g�}|j |ks(t�|j |ks6t�|j|dd�}|d ksPt�|j |ks^t�|j|gksnt�tjtdd�� |jddd� W d Q R X d }|j|dd� |j |ks�t�|j|gks�t�d S ) Nz#This is the new name for this indexT)�inplacezLevel must be None)r r* r )r )�A�B)r r# r r" r r r r )r r$ r �new_name� original_nameZnew_ind�resr r r r �test_set_name_methodsU s z TestCommon.test_set_name_methodsc C sh ddl m }m} |}x2||fD ]&}||�}||k s6t�|�|�s t�q W |j ddd�}|jdksdt�d S )Nr )�copy�deepcopyT�banana)�deepr )r7 r8 r r r )r r$ r7 r8 r �funcZidx_copyZnew_copyr r r �test_copy_and_deepcopyq s z!TestCommon.test_copy_and_deepcopyc C s� |}|� � }x,d|jd fD ] }|j|d�}t�||� qW d}tjt|d�� |jdd� W d Q R X dt� |j� � �� d�}tjt |d�� |jdd� W d Q R X d S ) Nr )r z.Too many levels: Index has only 1 level, not 4)r � z6Requested level \(wrong\) does not match index name \(z\)r ) �drop_duplicatesr �uniquer( r- r r � IndexError�re�escape�__repr__r )r r$ r �expectedr r. r r r r � test_unique� s zTestCommon.test_uniquec C s� |}t |�st�d� |dgd }|dg }|jdks<t�y|jdksLt�W n tk rb Y nX |�� }t� ||� |j s�t�d� t |j �r�|dgd j }tj|d< n@t|j �r�|j jdgd }t|d< n|jdgd }tj|d< |d d� }|j jdk�r2t|j �||j d �}t|j �j||j d �}|�|�}|�|�} | jdk�sVt�|j |j k�sht�| j |j k�szt�| } x&|| gD ]} | �� }t� || � �q�W d S ) Nz)Skip check for empty Index and MultiIndexr � TFz%Skip na-check if index cannot hold na� )�m�M)�dtype)�lenr �skip� is_uniquer Zhasnans�NotImplementedErrorZ_get_unique_indexr( r- Z _can_hold_nar rJ r) �pdZNaTr Z_ndarrayr �values�np�nan�kindr' Z _simple_newZ _shallow_copy) r r$ r �idxZ idx_uniquer. �valsZ vals_uniqueZidx_nanZidx_unique_nanrD �ir r r �test_get_unique_index� sF z TestCommon.test_get_unique_indexc C s4 |}t |tj�rt�d� |jr*t�d� |d }d||k�� }}|dkrTt|�}|jr�|j |dd�}||kstt �|j |dd�}||ks�t �|j |dd�}||ks�t �|j |dd�} || ks�t �nl|j �r|j |dd�}||ks�t �|j |dd�}||k�s0t �n,d} tj t| d�� |j |dd� W d Q R X d S ) Nz'Skip check for MultiIndex/IntervalIndexzSkip check for empty Indexr �left)Zside�rightz0index must be monotonic increasing or decreasing)r )r rO Z IntervalIndexr rL �emptyZargminrK Zis_monotonic_increasingZ_searchsorted_monotonicr Z searchsortedZis_monotonic_decreasingr r ) r r$ r �valueZ expected_leftZexpected_rightZssm_leftZ ssm_rightZss_leftZss_rightr r r r �test_searchsorted_monotonic� s4 z&TestCommon.test_searchsorted_monotonicc C s� |}t |t�rt�d� t|�dkr.t�d� t|�}tt|��}||�}t|�}tj � |t |d ��}||j | �} t �|�j|d�j } t�| j|d�| � |t �| �j|d��} t�| j|d�| � d S )NzWRangeIndex is tested in test_drop_duplicates_no_duplicates as it cannot hold duplicatesr zXempty index is tested in test_drop_duplicates_no_duplicates as it cannot hold duplicatesg �?)�keep)r r r rL rK r' �list�setrQ �random�choice�intrP rO ZSeries� duplicatedr( �assert_numpy_array_equalr> r- ) r r$ r] r �holder� unique_values� unique_idx�nZduplicated_selectionrT �expected_duplicatedZexpected_droppedr r r �test_drop_duplicates� s" z TestCommon.test_drop_duplicatesc C s| |}t |t�r|}n t|�}tt|��}||�}tjdgt|� dd�}t� |� � |� |� � }t� ||� ||k sxt �d S )NF�bool)rJ )r r r' r^ r_ rQ �arrayrK r( rd rc r> r- r )r r$ r rg re rf ri Zresult_droppedr r r �"test_drop_duplicates_no_duplicates s z-TestCommon.test_drop_duplicates_no_duplicatesc C s. d}t jt|d�� |jdd� W d Q R X d S )Nz6drop_duplicates\(\) got an unexpected keyword argument)r T)r0 )r r r! r> )r r r r r r � test_drop_duplicates_inplace. s z'TestCommon.test_drop_duplicates_inplacec C sZ |}t |�}t|�r t|t�r(t�d� ||d gd �}|jdksHt�|jdksVt�d S )Nz6Skip check for empty Index, MultiIndex, and RangeIndexr rF FT) r' rK r r r rL rM r Zhas_duplicates)r r$ r re rT r r r �test_has_duplicates3 s z TestCommon.test_has_duplicatesrJ �int64�uint64Zfloat64�categoryzdatetime64[ns]ztimedelta64[ns]c C s� t |t�r"dd� t|j�D �|_nd|_d }|dkrDt|j�rdt}n t |t �rd|j d k rd|dkrdt}y$t � |�� |� |�}W d Q R X W n ttttfk r� d S X t |t�r�|j|jks�t�n|j|jks�t�d S )Nc S s g | ]}d t |� �qS )rT )�str)�.0rV r r r � <listcomp>I s z9TestCommon.test_astype_preserves_name.<locals>.<listcomp>rT )rp rq zdatetime64[ns])r r �rangeZnlevelsr r r rJ � FutureWarningr �tzr( �assert_produces_warningZastyper r! rN � SystemErrorr )r r rJ �warnr. r r r �test_astype_preserves_nameB s& z%TestCommon.test_astype_preserves_namec C s: t |tttf�rd }nt}t�|�� |j W d Q R X d S )N)r r r r rw r( ry Zasi8)r r r{ r r r �test_asi8_deprecationd s z TestCommon.test_asi8_deprecationN)�__name__� __module__� __qualname__r r& r, r/ r6 r< rE rW r\ rj rm rn ro r �mark� parametrizer| r} r r r r r s" 3." r � na_position�middlec C s0 t jtd|� �d�� | j|d� W d Q R X d S )Nzinvalid na_position: )r )r� )r r r � sort_values)�index_with_missingr� r r r �$test_sort_values_invalid_na_positiono s r� �first�lastc C s� t | t�rt�d� t�| �� �}| | �� j}t� |�}|dkrXt� d g| |g�}nt� |d g| g�}t | �|�}| j |d�}t �||� d S )Nz,missing value sorting order not well-definedr� )r� )r r r rL rQ �sumZisnaZnotnarP �sortZ concatenater' r� r( r- )r� r� Z missing_countZ not_na_valsZ sorted_valuesrD r. r r r � test_sort_values_with_missingv s r� c C s� t | t�rtst�d� | }|j�|�s,t�|�� �|�s>t�|j }|j |j ksTt�|j |j ksdt�|j |j kstt�t | t tf�s�|j|jks�t�|j |j j d S )N�Overflow)r r r r rL �Tr r Z transposerP �shape�ndim�sizer r �nbytes)r rT rP r r r � test_ndarray_compat_properties� s r� ) �__doc__rA �numpyrQ r Zpandas._libs.tslibsr Z pandas.compatr Zpandas.core.dtypes.commonr r ZpandasrO r r r r r r Zpandas._testingZ_testingr( r r� r� r� r� r� r r r r �<module> s R
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка