Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/indexes/interval/__pycache__/test_indexing.cpython-37.pyc
Назад
B �5�giI � @ s� d dl Z d dlZd dlZd dlmZ d dlmZmZm Z m Z m Z m Z m Z mZ d dlmZ G dd� d�ZG dd� d�ZG dd � d �ZdS ) � N)�InvalidIndexError)�NA�CategoricalIndex�Interval� IntervalIndex�NaT� Timedelta� date_range�timedelta_rangec @ s� e Zd Zej�dddddg�dd� �Zej�dd d d d d ddddg �dd� �Zej�ddd d ddddg�dd� �Zej�dddddg�ej�dd d d d g�d d!� ��Z ejjd"e d#d$d%�e d#d$d&d'�e d(d$d%�gd)d*� d+�d,d-� �Z ejjd.e d#d$d%�e d/d$d%�fe d#d$d&d'�e d/d$d&d'�fe d(d$d%�e d0d$d%�fgd1d*� d+�d2d3� �Z ejjd4e d5d$d6d7�e d5d$d6d&d8�e d9d$d6d7�e�d:d;d;�e�ddd�gd<d*� d+�d=d>� �Zej�d?dgd@g�dAdB� �ZdCdD� ZdES )F� TestGetLoc�side�right�left�both�neitherc C s0 t jddg|d�}�xddgddgddgddgddgd dgd dggD ]�}t�d |d � d |d � d |� d��}||kr�|ddgkr�|�tdd|d��dks�t�nX|ddgkr�|�tdd|d��dks�t�n.tjt |d�� |�t|d|i�� W d Q R X qDtjt |d�� |�t|d|i�� W d Q R X qDW d S )N)r � )� � )�closedr r r r � g @���z Interval(z, z , closed='z'))�matchr ) r � from_tuples�re�escape�get_locr �AssertionError�pytest�raises�KeyError)�selfr r �idx�bound�msg� r$ �a/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/indexes/interval/test_indexing.py�test_get_loc_interval s 6& z TestGetLoc.test_get_loc_interval�scalarg �r g �?r g �?r g @r g @c C s� ddddd�ddddd�ddddddd�ddd�d�}t jdd g|d �}||| �� krt|�|�|| | ks�t�n(tjtt|�d �� |�|� W d Q R X d S ) Nr r )g �?r g @r )r g �?r g @)r g �?r r g @r )g �?g @)r r r r )r r )r r )r )r ) r r �keysr r r r r �str)r r r' Zcorrectr! r$ r$ r% �test_get_loc_scalar+ s z TestGetLoc.test_get_loc_scalarr g @� � c C s` t jdg|d�}||d kr4|�|�}|dks\t�n(tjtt|�d�� |�|� W d Q R X d S )N)r r+ )r r )r )r r r r r r r r) )r r' r �index�resultr$ r$ r% � test_get_loc_length_one_scalarA s z)TestGetLoc.test_get_loc_length_one_scalar� other_closedz left, right)r r+ )r r )r r, )r, � c C s� t jdg|d�}t|||d�}||d krB|�|�}|dks�t�n>tjtt� d|� d|� d|� d��d�� |�|� W d Q R X d S ) N)r r+ )r r z Interval(z, z , closed='z'))r ) r r r r r r r r r r )r r r r r0 r- �intervalr. r$ r$ r% � test_get_loc_length_one_intervalL s $z+TestGetLoc.test_get_loc_length_one_interval�breaksZ20180101r )�periodsz US/Eastern)r5 �tzz0 daysc C s t | j�S )N)r) �dtype)�xr$ r$ r% �<lambda>d � zTestGetLoc.<lambda>)Zidsc C sb t �|�}|d j}|�|�}d}||ks.t�t|d j|d j�}|�|�}d}||ks^t�d S )Nr )r � from_breaks�midr r r r r )r r4 r- �valuer. �expectedr2 r$ r$ r% �(test_get_loc_datetimelike_nonoverlapping] s z3TestGetLoc.test_get_loc_datetimelike_nonoverlapping�arraysZ20180103z2 daysc C s t | d j�S )Nr )r) r7 )r8 r$ r$ r% r9 � r: c C sr t j|� }|d jtd� }|�|�}tddd �}||ks>t�t|d j|d j �}|�|�}d}||ksnt�d S )Nr z12 hoursr ) r � from_arraysr< r r �slicer r r r )r r@ r- r= r. r> r2 r$ r$ r% �%test_get_loc_datetimelike_overlappingu s z0TestGetLoc.test_get_loc_datetimelike_overlapping�valuesz 2018-01-04z-1D)r5 �freq)r5 rE r6 z3 daysg @g �c C s t | j�S )N)r) r7 )r8 r$ r$ r% r9 � r: c C s> t �|dd � |d d� �}|�|d �}d}||ks:t�d S )Nr r r )r rA r r )r rD r- r. r> r$ r$ r% �test_get_loc_decreasing� s z"TestGetLoc.test_get_loc_decreasing�key)r r c C sD t �dddddg�}t|�}tjt|d�� |�|� W d Q R X d S )N)r r )r r )r r+ )r1 � )r rH )r )r r r) r r r r )r rG r! r# r$ r$ r% � test_get_loc_non_scalar_errors� s z)TestGetLoc.test_get_loc_non_scalar_errorsc C s� t tjtdd�tjg�}t�dddg�}x6d tjtgD ]&}||ksFt�|�|�}t� ||� q6W xJt t� dd�t� dd�gD ],}t jtt|�d�� |�|� W d Q R X q|W d S )Nr r TFr �ns)r )r �np�nanr �arrayr r r �tm�assert_numpy_array_equalr Z timedelta64Z datetime64r r r r) )r r- r> rG r. r$ r$ r% �test_get_indexer_with_nans� s z%TestGetLoc.test_get_indexer_with_nansN)�__name__� __module__� __qualname__r �mark� parametrizer& r* r/ r3 r r r? rC rK �arangerF rI rP r$ r$ r$ r% r s: ($ r c @ sN e Zd Zej�dedddd�gdgfedddd�gdgfeddd d�gdgfeddd d�gdgfedddd�gdgfed ddd�gdgfed d dd�gdgfedddd�ed ddd�gddgfedddd�edddd�gddgfedddd�edddd�gddgfedddd�edddd�gddgfg �dd� �Zej�ddgdgfd gdgfd gd gfdgd gfd gdgfdgdgfdgdgfdgdgfdgdgfdgdgfdgdgfddgd dgfdddgd ddgfddddgd dddgfdddddgd ddddgfg�dd� �Zej�ddge � d dd �g�dd� �Z ej�d ddg�d d � �Z ej�d e �d d!d"d#g�e �d#d!d"e jg�e jd#d!d"gd d�dd d ddde jgd$d$d%d&gg�d'd(� �Zd)d*� Zej�d+d,d-d"gd fd.d/d0gdfd#d#d!gdfd#d1d"gd fg�d2d3� �Zej�ddgdgd gffd gd gg ffd gd gg ffdgd dgg ffd gd dgg ffdgd ddgg ffdgddgg ffdgdgg ffdgdgg ffdgdgd gffdgdgd gffddgd dd ddgg ffdddgd dd dddgg ffddddgd dd ddddgdgffdddddgd dd ddddd ddg dgffg�d4d5� �Zd6d7� Zd8d9� Zd:S );�TestGetIndexerzquery, expectedr r r )r r r r r r r g �?g �?r+ r1 c C s@ dddg}t j|dd�}|�|�}tj|dd�}t�||� d S )N)r r )r r )r+ r1 r )r �intp)r7 )r r � get_indexerrK rM rN rO )r �queryr> �tuplesr- r. r$ r$ r% � test_get_indexer_with_interval� s z-TestGetIndexer.test_get_indexer_with_intervalg �g @r g @g @c C s@ dddg}t j|dd�}|�|�}tj|dd�}t�||� d S )N)r r )r r )r r r )r rX )r7 )r r rY rK rM rN rO )r rZ r> r[ r- r. r$ r$ r% �#test_get_indexer_with_int_and_float� s z2TestGetIndexer.test_get_indexer_with_int_and_float�itemc C sB t jdg|d�}|�|�}tjdgt|� dd�}t�||� d S )N)r r+ )r r rX )r7 )r r rY rK rM �lenrN rO )r r^ r r- r. r> r$ r$ r% �test_get_indexer_length_one� s z*TestGetIndexer.test_get_indexer_length_one�sizec C sL t jdg|d�}|�tdd|�g| �}tjdg| dd�}t�||� d S )N)r r+ )r r r+ rX )r7 )r r rY r rK rM rN rO )r ra r r- r. r> r$ r$ r% �$test_get_indexer_length_one_interval� s z3TestGetIndexer.test_get_indexer_length_one_interval�target)r1 � )r r )r r )r r Zfoo�barZbazc C s@ t �dddg�}t||d�}|�|�}|�|�}t�||� d S )N)r r )r r )r r )�ordered)r r r rY rN rO )r rc rf r- Zcategorical_targetr. r> r$ r$ r% � test_get_indexer_categorical s z+TestGetIndexer.test_get_indexer_categoricalc C s� t �td��}|�t tjg��}t|�}|�|�}tjdtj d�}t � ||� |dd � �|d d d� �}tj dddddgtj d�}t � ||� |�|�|��}tj ddddddddddg tj d�}t � ||� d S ) Nr+ )r7 r r r r r r ) r r; �range�appendrK rL r rY rV rX rN rO rM )r �iiZii2Zci2r. r> r$ r$ r% �&test_get_indexer_categorical_with_nans s $z5TestGetIndexer.test_get_indexer_categorical_with_nansztuples, closed)r r )r r )r r+ )r r )r, r1 )r r c C s> t j||d�}d}tjt|d�� |�ddg� W d Q R X d S )N)r zKcannot handle overlapping indices; use IntervalIndex.get_indexer_non_unique)r r r )r r r r r rY )r r[ r r- r# r$ r$ r% �test_get_indexer_errors* s z&TestGetIndexer.test_get_indexer_errorsc C sf dddg}t j|dd�}|�|�\}}tj|d dd�}tj|d dd�}t�||� t�||� d S ) N)r g @)r r )r r r )r r rX )r7 r )r r Zget_indexer_non_uniquerK rM rN rO ) r rZ r> r[ r- Zresult_indexerZresult_missingZexpected_indexerZexpected_missingr$ r$ r% �.test_get_indexer_non_unique_with_int_and_float> s z=TestGetIndexer.test_get_indexer_non_unique_with_int_and_floatc C s� t �dddg�}t �ddddg�}|�|�}tjddddgtjd �}t�||� |�|d d � �}tjd dgtjd �}t�||� d S ) N)r r )r r+ )r r )r, r1 )rd � r r r )r7 r )r r rY rK rM rX rN rO )r Zidx1Zidx2r. r> r$ r$ r% � test_get_indexer_non_monotonica s z-TestGetIndexer.test_get_indexer_non_monotonicc C sT t tjtjg�}t tjg�}|jr&t�|�|�}tjddgtjd�}t� ||� d S )Nr r )r7 ) r rK rL Z_index_as_uniquer Zget_indexer_forrM rX rN rO )r r- �otherr. r> r$ r$ r% rP m s z)TestGetIndexer.test_get_indexer_with_nansN)rQ rR rS r rT rU r r\ r] rK rV r` rb r r rL rg rk rl rm ro rP r$ r$ r$ r% rW � s� """0 ( & "8 rW c @ s� e Zd Zdd� Zdd� Zej�dddgddgdd gdd gg�ej�d d d dgdd d gd d dgd dd gd d dd gg�dd� ��ZdS )� TestSliceLocsc C s� t �dddg�}|jtdd�tdd�d�dks2t�|jtdd�d �dksLt�|jtdd�d �dksft�|jtdd�d �d ks�t�|jtdd�tdd�d�d ks�t�t �dddg�}|jtdd�tdd�d�d ks�t�|jtdd�d �d ks�t�|jtdd�d �d k�s t�|jtdd�d �dk�s&t�|jtdd�tdd�d�dk�sJt�t �dddg�}|jtdd�tdd�d�dk�s~t�|jtdd�d �dk�s�t�|jtdd�d �dk�s�t�|jtdd�d �dk�s�t�|jtdd�tdd�d�dk�s�t�t �dddg�}tjtt� d�d�� |jtdd�tdd�d� W d Q R X tjtt� d�d�� |jtdd�d � W d Q R X |jtdd�d �dk�s�t�tjtt� d�d�� |jtdd�d � W d Q R X tjtt� d�d�� |jtdd�tdd�d� W d Q R X t �ddddg�}|jtdd�tdd�d�dk�s0t�|jtdd�d �dk�sLt�|jtdd�d �dk�sht�|jtdd�d �dk�s�t�|jtdd�tdd�d�dk�s�t�d S )N)r r )r r )r r r r r )�start�end)r r )rr )rs )r r )r r )r r )r r zR"Cannot get left slice bound for non-unique label: Interval(0, 2, closed='right')")r zS"Cannot get right slice bound for non-unique label: Interval(0, 2, closed='right')")r r ) r r � slice_locsr r r r r r r )r r- r$ r$ r% � test_slice_locs_with_intervalz s\ """ $$ $ $ $$ z+TestSliceLocs.test_slice_locs_with_intervalc C s t �dddg�}|�dd�dks$t�|�dd�dks8t�|�dd�dksLt�|�dd�d ks`t�|�dd �d kstt�|�dd �d ks�t�t �dddg�}|�dd�d ks�t�|�dd�dks�t�|�dd�dks�t�|�dd�dks�t�|�dd �dks�t�|�dd �dk�st�d S )N)r r )r r )r r r r r )r r r )r r r )r r )r r )r r )r r )r r )r r )r r )r r rt r )r r- r$ r$ r% �-test_slice_locs_with_ints_and_floats_succeeds� s z;TestSliceLocs.test_slice_locs_with_ints_and_floats_succeedsrZ r r r r r r[ )r r )r r )r r c C s< |\}}t �|�}tjtdd�� |�||� W d Q R X d S )Nzt'can only get slices from an IntervalIndex if bounds are non-overlapping and all monotonic increasing or decreasing')r )r r r r r rt )r r[ rZ rr �stopr- r$ r$ r% �+test_slice_locs_with_ints_and_floats_errors� s z9TestSliceLocs.test_slice_locs_with_ints_and_floats_errorsN) rQ rR rS ru rv r rT rU rx r$ r$ r$ r% rq y s O$rq )r �numpyrK r Z pandas.errorsr Zpandasr r r r r r r r Zpandas._testingZ_testingrN r rW rq r$ r$ r$ r% �<module> s ( ( >
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка