Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/indexes/interval/__pycache__/test_interval_tree.cpython-37.pyc
Назад
B �5�g� � @ s� d dl mZ d dlZd dlZd dlmZ d dlmZ d dl m Z dd� Z ej ddd d gd �d d � �Zej e d�e d�dgd�dd� �Zej ejddd�ejdd d�ejdd d�ejd ddddejgd d�gd�dd� �ZG dd� d�ZdS ) � )� permutationsN)� IntervalTree)�IS64c C s t jjt dd�}t j| |d�S )z� Skip parameters in a parametrize on 32bit systems. Specifically used here to skip leaf_size parameters related to GH 23440. z$GH 23440: int type mismatch on 32bit)�reason)�marks)�pytest�mark�skipifr �param)r r � r �f/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/indexes/interval/test_interval_tree.py� skipif_32bit s r �class�int64�float64�uint64)�scope�paramsc C s | j S )N)r )�requestr r r �dtype s r � � � )r c C s | j S )zd Fixture to specify IntervalTree leaf_size parameter; to be used with the tree fixture. )r )r r r r � leaf_size s r � )r � � c C s | j }t||d |d�S )Nr )r )r r )r r �leftr r r �tree# s r c @ s8 e Zd Zdd� Zej�dddg�dd� �Zdd � Zej�dddg�d d � �Z d d � Z ej�de d�e d�e d�dg�dd� �Z ej�de jdddgdd�e �ddd g�d fe jdddgdd�e �d ddg�d fe �dde jg�e �d de jg�d fe jdddgdd�e �ddd g�d fe �dde jg�e �dde jg�d fg�ej�d d d!� eed��D ��d"d#� ��Zej�d d$d!� eed��D ��d%d&� �Zej�d'e jg dd�e jg dd�fe jdgdd�e jdgdd�fe �e jg�e �e jg�fe �e jgd �e �e jgd �fg�d(d)� �Zejje d*d+�d,d-� �Zd.S )/�TestIntervalTreec C sf |� t�dddg��}tjdddgdd�}t�||� tjtd d �� |� t�d g�� W d Q R X d S ) Ng �?g @g @r r ����intp)r z6'indexer does not intersect a unique set of intervals')�matchg @)� get_indexer�np�array�tm�assert_numpy_array_equalr �raises�KeyError)�selfr �result�expectedr r r �test_get_indexer1 s z!TestIntervalTree.test_get_indexerz!dtype, target_value, target_dtype)r l r )r r r c C sf t jddg|d�t jddg|d� }}t||�}|�t j|g|d��}t jdgdd�}t�||� d S )Nr r )r r r r! )r$ r% r r# r& r' ) r* r � target_value� target_dtyper �rightr r+ r, r r r �test_get_indexer_overflow; s & z*TestIntervalTree.test_get_indexer_overflowc C s� |� t�dddg��\}}|d d� }tjdgdd�}t�||� t�|dd� �}tjddgdd�}t�||� t�|dd � �}tjd gdd�}t�||� |}tjd gdd�}t�||� d S ) Ng �?g @g @r r r! )r r r r )�get_indexer_non_uniquer$ r% r& r' �sort)r* r �indexer�missingr+ r, r r r �test_get_indexer_non_uniqueG s z,TestIntervalTree.test_get_indexer_non_uniquec C s� t jddg|d�t jddg|d� }}t||�}t j|g|d�}|�|�\}} t jdgdd�} t�|| � t jdgdd�} t�| | � d S )Nr r )r r r r r! )r$ r% r r2 r& r' ) r* r r. r/ r r0 r �targetZresult_indexerZresult_missingZexpected_indexerZexpected_missingr r r �$test_get_indexer_non_unique_overflowZ s & z5TestIntervalTree.test_get_indexer_non_unique_overflowc C s� t jdddg|d�}t||d �}tjtdd�� |�t �dg�� W d Q R X |�t �dg��\}}t �|�}t jdddgdd�}t � ||� |}t jg dd�}t � ||� d S ) Nr )r r z6'indexer does not intersect a unique set of intervals')r" g �?r r! ) r$ r% r r r( r) r# r2 r3 r& r' )r* r r r r4 r5 r+ r, r r r �test_duplicatesj s z TestIntervalTree.test_duplicatesr r r �d i' c C s� t jddd�}|�d�}dt �d� �d�}t||d ||d�}t�||�|d �� |jr`|n|}t�||�|d �� |j r�|n|}t�||�|d �� d S ) Ni� r )r r! r g �?)�closedr g �?g ) r$ �arangeZastypeZonesr r& r' r# Z closed_leftZ closed_right)r* r; r �x�found� not_foundr r, r r r �test_get_indexer_closed| s z(TestIntervalTree.test_get_indexer_closedzleft, right, expectedr r r )r r r r TF�orderc c s | ]}t |�V qd S )N)�list)�.0r= r r r � <genexpr>� s zTestIntervalTree.<genexpr>c C s, t || || |d�}|j}||ks(t�d S )N)r; )r �is_overlapping�AssertionError)r* r; rA r r0 r, r r+ r r r �test_is_overlapping� s z$TestIntervalTree.test_is_overlappingc c s | ]}t |�V qd S )N)rB )rC r= r r r rD � s c C sP t jddd�t �dd� }}t|| || |d�}|j}|dk}||ksLt�dS ) z*shared endpoints are marked as overlappingr r )r r r )r; ZbothN)r$ r< r rE rF )r* r; rA r r0 r r+ r, r r r � test_is_overlapping_endpoints� s z.TestIntervalTree.test_is_overlapping_endpointsz left, rightc C s t |||d�}|jdks t�d S )N)r; F)r rE rF )r* r; r r0 r r r r �test_is_overlapping_trivial� s z,TestIntervalTree.test_is_overlapping_trivialzGH 23440)r c C s\ t jddd�t �t j�jgd }}t||�}|jj}dt �t j�j d }||ksXt�d S )N�e r )r �2 r ) r$ r< Ziinfor �maxr �rootZpivotrF )r* r r0 r r+ r, r r r �test_construction_overflow� s $ z+TestIntervalTree.test_construction_overflowN)�__name__� __module__� __qualname__r- r r Z parametrizer1 r6 r8 r9 r r@ r$ r% �nanr �rangerG rH rI r r rN r r r r r 0 s6 $$$$$(*( 0r )� itertoolsr �numpyr$ r Zpandas._libs.intervalr Z pandas.compatr Zpandas._testingZ_testingr& r Zfixturer r r< r% rR r r r r r r �<module> s " *
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка