Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/arrays/sparse/__pycache__/test_libsparse.cpython-37.pyc
Назад
B �5�gsR � @ s� d dl Z d dlZd dlZd dlm mZ d dlm m Z d dl m Z d dlmZ d dlmZmZmZ dZd ddgdddgd d d gd ddgd d dgd dd gd�Zd dgd d gd gd gd gdgd�Zd gdgd dgddgd dgddgd�Zdgdgd dgddgdgdgd�Zd dgd dgddgd d gg g d�Zdd� ZG dd� d�Z G dd� d�Z G dd� d�Z G dd � d �Z G d d � d �Z G d d � d �Z!dS )!� N)�Series)� BlockIndex�IntIndex�make_sparse_index� � � � � � � � � � )�xloc�xlen�yloc�ylen� intersect_loc� intersect_len� � � � c s` � fdd�}|t � |t� |t� |t� |t� � dgdgg g g g � � g g g g g g � d S )Nc s. � | d | d | d | d | d | d � d S )Nr r r r r r � )�case)� _check_caser �_/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/arrays/sparse/test_libsparse.py�_check_case_dict? s z%check_cases.<locals>._check_case_dictr r )� plain_case� delete_blocks� split_blocks� skip_block� no_intersect)r r r )r r � check_cases> s r$ c @ s e Zd Zdd� Zdd� ZdS )�TestSparseIndexUnionc C s� dd� }dg}dg}dg}dg}dg}dg}|||||||� ddg}ddg}dd g}ddg}ddd g}d ddg}|||||||� d g}dg}d g}dg}d g}d g}|||||||� ddg}ddg}dg}d g}dg}dg}|||||||� ddg}d dg}dg}d g}dg}dg}|||||||� ddg}ddg}ddg}d dg}dg}dg}|||||||� dg}dg}dddg}d ddg}dg}dg}|||||||� ddg}d d g}ddg}ddg}ddddg}d dd dg}|||||||� d S )Nc S s� t t| |�}t t||�}|�|�}t|t �s0t�t�|jtj |tj d�� t�|j tj |tj d�� |� � } |� � } | �| �} t| t �s�t�t�| j|� � j� d S )N)�dtype)r � TEST_LENGTH� make_union� isinstance�AssertionError�tm�assert_numpy_array_equal�blocs�np�array�int32�blengths� to_int_indexr �indices) r r r r �eloc�elen�xindex�yindexZbresultZixindexZiyindexZiresultr r r r V s z?TestSparseIndexUnion.test_index_make_union.<locals>._check_caser r r r r r r r r r � r � r r r )�selfr r r r r r4 r5 r r r �test_index_make_unionU sr z*TestSparseIndexUnion.test_index_make_unionc C s� t dtjdddgtjd��}t dtjddgtjd��}|�|�}t dt�ddddgtj��}|�|�sjt�t dtjg tjd��}t dtjddgtjd��}|�|�}t dt�ddgtj��}|�|�s�t�t dtjg tjd��}t dtjg tjd��}|�|�}t dt�g tj��}|�|��s$t�t dtjdddddgtjd��}t dtjdddddgtjd��}|�|�}t dt�dddddgtj��}|�|��s�t�t dtjddgtjd��}t dtjddgtjd��}d}tjt |d �� |�|� W d Q R X d S ) Nr r r r )r& r r z-Indices must reference same underlying length)�match) r r. r/ r0 r( �equalsr* �pytest�raises� ValueError)r: �a�b�res�exp�msgr r r �test_int_index_make_union� s2 z.TestSparseIndexUnion.test_int_index_make_unionN)�__name__� __module__� __qualname__r; rF r r r r r% T s pr% c @ s* e Zd Zejdd� �Zdd� Zdd� ZdS )�TestSparseIndexIntersectc s* dd� � dd� �� �fdd�}t |� d S )Nc S s | � |�}|�|�st�d S )N)� intersectr= r* )rA rB �expected�resultr r r �_check_correct� s z?TestSparseIndexIntersect.test_intersect.<locals>._check_correctc S s, d}t jt|d�� | �|� W d Q R X d S )Nz-Indices must reference same underlying length)r< )r> r? � ExceptionrK )rA ZlongerrE r r r �_check_length_exc� s zBTestSparseIndexIntersect.test_intersect.<locals>._check_length_excc sx t t| |�}t t||�}t t||�}t td ||�} � |||� � |�� |�� |�� � �|| � �|�� | �� � d S )Nr )r r' r2 ) r r r r r4 r5 r6 r7 rL Z longer_index)rN rP r r r � s z<TestSparseIndexIntersect.test_intersect.<locals>._check_case)r$ )r: r r )rN rP r �test_intersect� s z'TestSparseIndexIntersect.test_intersectc C s� t dtjg tjd��}t dtjddgtjd��}|�|��|�sDt�|�|��|�sXt�|�� }|�� }|�|��|�s|t�|�|��|�s�t�d S )Nr )r& r r )r r. r/ r0 rK r= r* �to_block_index)r: r6 r7 r r r �test_intersect_empty s z-TestSparseIndexIntersect.test_intersect_emptyc C s� t dtjddgtjd��t dtjdddgtjd��t dtjg tjd��t dtjg tjd��g}x<|D ]4}|�|��|�s|t�|�� }|�|��|�sdt�qdW d S )Nr r r )r& r r )r r. r/ r0 rK r= r* rR )r: Zcasesr r r r �test_intersect_identical s z1TestSparseIndexIntersect.test_intersect_identicalN)rG rH rI �tdZskip_if_windowsrQ rS rT r r r r rJ � s rJ c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �TestSparseIndexCommonc C s t dtjddgtjd�dd�}t|t�s,t�|jdks:t�t� |j tjddgtjd�� t dtjg tjd�dd�}t|t�s�t�|jdks�t�t� |j tjg tjd�� t dtjddddgtjd�dd�}t|t�s�t�|jdks�t�t� |j tjddddgtjd�� d S ) Nr r r )r& �integer)�kindr r ) r r. r/ r0 r) r r* �npointsr+ r, r3 )r: �idxr r r �test_int_internal s z'TestSparseIndexCommon.test_int_internalc C s� t dtjddgtjd�dd�}t|t�s,t�|jdks:t�t� |j tjdgtjd�� t� |j tjdgtjd�� t dtjg tjd�dd�}t|t�s�t�|jdks�t�t� |j tjg tjd�� t� |j tjg tjd�� t dtjddddgtjd�dd�}t|t��st�|jdk�s t�t� |j tjdgtjd�� t� |j tjdgtjd�� t dtjdddgtjd�dd�}t|t��s�t�|jdk�s�t�t� |j tjddgtjd�� t� |j tjddgtjd�� d S ) Nr r r )r& �block)rX r r ) r r. r/ r0 r) r r* rY r+ r, r- r1 )r: rZ r r r �test_block_internal. s( " z)TestSparseIndexCommon.test_block_internalc C s �xdD �]�}t dtjddgtjd�|d�}|�d�dks>t�|�d�dksPt�|�d �dksbt�|�d�dkstt�|�d�d ks�t�|�d�dks�t�t dtjg tjd�|d�}x$tdd �D ]}|�|�dks�t�q�W t dtjdd ddgtjd�|d�}|�d�dk�st�|�d�dk�s"t�|�d �d k�s6t�|�d�dk�sJt�|�d�dk�s^t�|�d�dk�srt�t dtjdddgtjd�|d�}|�d�dk�s�t�|�d�dk�s�t�|�d �dk�s�t�|�d�d k�s�t�|�d�dk�s�t�|�d�dkst�qW d S ) N)rW r\ r r r )r& )rX ���r r r )r r. r/ r0 �lookupr* �range)r: rX rZ �ir r r � test_lookupG s4 z!TestSparseIndexCommon.test_lookupc C sJ �xBdD �]8}t dtjddgtjd�|d�}|�tjdddgtjd��}tjdddgtjd�}t�||� |�tjddd dgtjd��}tjdddd gtjd�}t�||� t dtjg tjd�|d�}|�tjddddgtjd��}tjddddgtjd�}t dtjdd ddgtjd�|d�}|�tjdddgtjd��}tjdddgtjd�}t�||� |�tjddd dgtjd��}tjddd dgtjd�}t�||� t dtjdddgtjd�|d�}|�tjdd ddgtjd��}tjd dddgtjd�}t�||� |�tjd ddd gtjd��}tjddd dgtjd�}t�||� qW d S ) N)rW r\ r r r )r& )rX r^ r r r )r r. r/ r0 Z lookup_arrayr+ r, )r: rX rZ rC rD r r r �test_lookup_arrayh s4 z'TestSparseIndexCommon.test_lookup_arrayc C s8 dd� }t dddgddg�}|�� }||� ||� d S )Nc S s� | � d�dkst�| � d�dks$t�| � d�dks6t�| � d�dksHt�| � d�dksZt�| � d�dkslt�| � d �dks~t�| � d �d ks�t�| � d �dks�t�| � d �dks�t�d S )Nr r^ r r r r8 r r � r r r � )r_ r* )�indexr r r �_check� s z8TestSparseIndexCommon.test_lookup_basics.<locals>._checkr r r r r )r r2 )r: rg ZbindexZiindexr r r �test_lookup_basics� s z(TestSparseIndexCommon.test_lookup_basicsN)rG rH rI r[ r] rb rc rh r r r r rV s !$rV c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )�TestBlockIndexc C s� t dtjddgtjd�dd�}t|t�s,t�|jdks:t�t� |j tjdgtjd�� t� |j tjdgtjd�� t dtjg tjd�dd�}t|t�s�t�|jdks�t�t� |j tjg tjd�� t� |j tjg tjd�� t dtjddddgtjd�dd�}t|t��st�|jdk�s t�t� |j tjdgtjd�� t� |j tjdgtjd�� t dtjdddgtjd�dd�}t|t��s�t�|jdk�s�t�t� |j tjddgtjd�� t� |j tjddgtjd�� d S ) Nr r r )r& r\ )rX r r ) r r. r/ r0 r) r r* rY r+ r, r- r1 )r: rZ r r r r] � s( " z"TestBlockIndex.test_block_internalc C sp xjdD ]b}t |tjd|dtjd�dd�}tjd|dtjd�}t�|j|� t�|jtjt |�tjd�� qW d S )N)r r �d �e r r )r& r\ )rX ) r r. �aranger0 r+ r, r- r1 Zones�len)r: ra rZ rD r r r �test_make_block_boundary� s z'TestBlockIndex.test_make_block_boundaryc C sD t dddgddg�}|�|�s"t�|�t dddgddg��r@t�d S )Nr r r r r r )r r= r* )r: rf r r r � test_equals� s zTestBlockIndex.test_equalsc C s� g }g }t d||�}t d||�}d}tjt|d�� t ddgdg� W d Q R X d}tjt|d�� t dddgdd g� W d Q R X d S ) Nr r zBlock 0 extends beyond end)r< r r zBlock 0 overlapsr r )r r> r? r@ )r: �locs�lengthsrf rE r r r �test_check_integrity� s z#TestBlockIndex.test_check_integrityc C sZ ddg}ddg}ddddddd d d d g }t d ||�}|�� }t�|jtj|tjd�� d S )Nr r r r r r r rd r r9 r r r )r& )r r2 r+ r, r3 r. r/ r0 )r: rp rq Zexp_indsr\ Zdenser r r �test_to_int_index� s z TestBlockIndex.test_to_int_indexc C s( t dddgddg�}|�� |ks$t�d S )Nr r r r )r rR r* )r: rf r r r �test_to_block_index� s z"TestBlockIndex.test_to_block_indexN) rG rH rI r] rn ro rr rs rt r r r r ri � s ri c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) � TestIntIndexc C sL d}t jt|d�� tddddgd� W d Q R X d}t jt|d�� tddd dgd� W d Q R X d}t jt|d�� tddd dgd� W d Q R X d }t jt|d�� tddddgd� W d Q R X t jt|d�� tdddd gd� W d Q R X d }t jt|d�� tddddgd� W d Q R X t jt|d�� tddddgd� W d Q R X d S ) NzToo many indices)r< r r r )�lengthr3 z No index can be less than zeror ���z(All indices must be less than the lengthr z#Indices must be strictly increasing)r> r? r@ r )r: rE r r r rr � s& z!TestIntIndex.test_check_integrityc C s t dtjddgtjd�dd�}t|t�s,t�|jdks:t�t� |j tjddgtjd�� t dtjg tjd�dd�}t|t�s�t�|jdks�t�t� |j tjg tjd�� t dtjddddgtjd�dd�}t|t�s�t�|jdks�t�t� |j tjddddgtjd�� d S ) Nr r r )r& rW )rX r r ) r r. r/ r0 r) r r* rY r+ r, r3 )r: rZ r r r r[ s z TestIntIndex.test_int_internalc C sB t ddddddg�}|�|�s"t�|�t dddddg��r>t�d S )Nr r r r r r )r r= r* )r: rf r r r ro % s zTestIntIndex.test_equalsc C s dd� }t |� d S )Nc S s^ t t| |�}t t||�}|�� �� }|�� �� } t|t �s>t�|�|�sLt�| �|�sZt�d S )N)r r' r2 rR r) r* r= ) r r r r r4 r5 r6 r7 ZxbindexZybindexr r r r + s z5TestIntIndex.test_to_block_index.<locals>._check_case)r$ )r: r r r r rt * s z TestIntIndex.test_to_block_indexc C s( t ddddddg�}|�� |ks$t�d S )Nr r r r r r )r r2 r* )r: rf r r r rs 8 s z TestIntIndex.test_to_int_indexN)rG rH rI rr r[ ro rt rs r r r r ru � s &ru c @ s4 e Zd Zdd� Zej�ddddddg�d d � �Zd S ) �TestSparseOperatorsc s � �fdd�}t |� d S )Nc s, t t| |�}t t||�}|�� }|�� } t�|j�d d } t�|j�d d } d} d} �| || | || �\}}}�| || | | | �\}}}|�� �|�s�t�t� ||� ||ks�t�t | |j �}|� t�t��� | �}t | | j �}|� t�t��� | �}� ||�}|� |j �}t� ||j� t� ||j� d S )Ng $@r g Y@r r )r r' r2 r. rl rY r= r* r+ r, r r3 ZreindexZfillna�values)r r r r r4 r5 r6 r7 ZxdindexZydindex�x�yZxfillZyfillZresult_block_valsZrb_indexZbfillZresult_int_valsZri_indexZifillZxseriesZyseriesZ series_result)� python_op� sparse_opr r r ? s. z2TestSparseOperators._op_tests.<locals>._check_case)r$ )r: r} r| r r )r| r} r � _op_tests> s %z TestSparseOperators._op_tests�opname�add�sub�mul�truediv�floordivc C s, t td|� d��}t t|�}| �||� d S )NZsparse_Z_float64)�getattr�splib�operatorr~ )r: r r} r| r r r �test_opf s zTestSparseOperators.test_opN)rG rH rI r~ r> �markZ parametrizer� r r r r rx = s (rx )"r� �numpyr. r> Zpandas._libs.sparseZ_libs�sparser� Z pandas.util._test_decorators�utilZ_test_decoratorsrU Zpandasr Zpandas._testingZ_testingr+ Zpandas.core.arrays.sparser r r r' r r r! r" r# r$ r% rJ rV ri ru rx r r r r �<module> s^ 5 JQ
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка