Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/arrays/sparse/__pycache__/test_arithmetics.cpython-37.pyc
Назад
B �5�g#P � @ s0 d dl Z d dlZd dlZd dlmZ d dlZd dlm Z d dl m Z d dl mZmZ ejddgd�dd � �Zejd d gd�d d � �ZG dd� d�Zej�de je jg�dd� �Zdd� Zdd� Zej�dejej g�ej�ded d ddg�eddddg�g�dd � ��Z ej�d ed d d g�e� d dd g�fed d d gdd �e� d dd g�fed d d gdd �e� d dd g�fed d d gdd �e� d dd g�fed d d gdd �e� d dd g�fg�ej�dejej g�d d!� ��Z d"d#� Z!d$d%� Z"ej�d&d d g�d'd(� �Z#ej�d&d ej$g�ej�de j%e j&g�d)d*� ��Z'dS )+� N)�np_version_under1p20)�ops)� SparseArray� SparseDtype�integer�block)�paramsc C s | j S )z.kind kwarg to pass to SparseArray/SparseSeries)�param)�request� r �a/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/arrays/sparse/test_arithmetics.py�kind s r TFc C s | j S )N)r )r r r r �mix s r c @ s e Zd ZejZeZdd� Zdd� Z dd� Z dd� Z d d � Z e j�d d d dg�e j�ddd dg�dd� ��Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Ze j�dd&d'ejg�d(d)� �Ze j�dd&d'ejg�d*d+� �Z d,d-� Z d.d/� Z d0d1� Z dS )2�TestSparseArrayArithmeticsc C s t �||� d S )N)�tm�assert_numpy_array_equal)�self�a�br r r �_assert" s z"TestSparseArrayArithmetics._assertc C s� t jddd��� |r$|||��� }n|||��� }|tjtjgkrR||d |�}n |||�}|tjtjgkr�t � |�} | � � r�t j || < | � ||� W d Q R X d S )N�ignore)�invalid�divideg �?) �np�errstate�to_dense�operator�truedivr Zrtruediv�floordiv� rfloordiv�isinf�any�nanr ) r r r �a_dense�b_denser �op�result�expected�maskr r r �_check_numeric_ops% s z-TestSparseArrayArithmetics._check_numeric_opsc C sF t || j�st�t |jt�s t�|jjtjks2t�t |jt �sBt�d S )N) � isinstance�_klass�AssertionError�dtyper �subtyper �bool_� fill_value�bool)r �resr r r �_check_bool_result: s z-TestSparseArrayArithmetics._check_bool_resultc C s� t jdd���� | �||k� | �||k�� ||k� | �||k� | �||k�� ||k� | �||k� | �||k�� ||k� | �||k� | �||k�� ||k� | �||k� | �||k�� ||k� | �||k � | �||k �� ||k � | �||k� | �||k�� ||k� | �||k� | �||k�� ||k� | �||k� | �||k�� ||k� | �||k� | �||k�� ||k� | �||k� | �||k�� ||k� | �||k � | �||k �� ||k � W d Q R X d S )Nr )r )r r r3 r r )r r r r# r$ r r r �_check_comparison_ops@ s2 z0TestSparseArrayArithmetics._check_comparison_opsc C s� | � ||@ � | �||@ �� ||@ � | � ||B � | �||B �� ||B � | � ||@ � | �||@ �� ||@ � | � ||B � | �||B �� ||B � d S )N)r3 r r )r r r r# r$ r r r �_check_logical_opsk s z-TestSparseArrayArithmetics._check_logical_ops�scalarr � � r0 N� c C s� |}t s\|tjtjgkr\|tjkr,|dkr,n0|tjkr@|dkr@n tjjtdd�}|j � |� | � t j dddt j ddddt j g �} | j| ||d�} | �| || |||� d S )Nr zGH#38172)�raises�reasonr7 r9 )r r0 )r r r r r �pytest�mark�xfailr, �node� add_marker�_baser r" r+ r) ) r r r �all_arithmetic_functionsr0 r6 r r% r= �valuesr r r r �test_float_scalary s $z,TestSparseArrayArithmetics.test_float_scalarc C s� | � tjdddtjddddtjg �}| j||d�}| �|d|d� | �|d|d� | �|d|d� | j||dd�}| �|d|d� | �|d|d� | �|d|d� | j||dd�}| �|d|d� | �|d|d� | �|d|d� d S )Nr7 r9 r )r r8 )r r0 )rA r r" r+ r4 )r r rC r r r r � test_float_scalar_comparison� s $z7TestSparseArrayArithmetics.test_float_scalar_comparisonc C sx |}| � ddddddddddg �}| � ddddddddddg �}| j||dd�}| j||dd�} | �|| ||||� d S ) Ng g �?g @g @g @g @r )r r0 )rA r+ r) ) r r r rB r r% rC �rvaluesr r r r r �"test_float_same_index_without_nans� s z=TestSparseArrayArithmetics.test_float_same_index_without_nansc C s� |}t s:|tjkr:|r |dks:tjjtdd�}|j�|� | � t j dddt j ddddt j g �}| � t j dddt j ddddt j g �}| j ||d �} | j ||d �} | � | | ||||� d S ) Nr zGH#38172)r: r; r7 r9 r r8 � )r )r r r r< r= r> r, r? r@ rA r r" r+ r) ) r r r rB r r% r= rC rF r r r r r � test_float_same_index_with_nans� s $$z:TestSparseArrayArithmetics.test_float_same_index_with_nansc C s� | � tjdddtjddddtjg �}| � tjdddtjddddtjg �}| j||d�}| j||d�}| �||||� | � ddd d dddd ddg �}| � dd d d dddd d dg �}| j||dd �}| j||dd �}| �||||� d S )Nr7 r9 r r8 rH )r g g �?g @g @g @g @)r r0 )rA r r" r+ r4 )r r rC rF r r r r r � test_float_same_index_comparison� s $$ z;TestSparseArrayArithmetics.test_float_same_index_comparisonc C s6 |}| � tjdddtjddddtjg �}| � dtjddtjddddtjg �}| j||d�}| j||d�}| �||||||� | �||d ||d ||� | j||dd�}| j||d�}| �||||||� | j||dd�}| j||dd�}| �||||||� | j||dd�}| j||dd�}| �||||||� d S )Nr7 r9 r r8 � )r )r r0 )rA r r" r+ r) ) r r r rB r% rC rF r r r r r �test_float_array� s $$ z+TestSparseArrayArithmetics.test_float_arrayc C s6 |}| � tjdddtjddddtjg �}| � dtjddtjddddtjg �}| j|dd�}| j|dd�}| �||||||� | �||d ||d ||� | j|ddd �}| j|dd�}| �||||||� | j|ddd �}| j|ddd �}| �||||||� | j|ddd �}| j|ddd �}| �||||||� d S ) Nr7 r9 r r8 rK r )r r )r r0 )rA r r" r+ r) )r r rB r% rC rF r r r r r � test_float_array_different_kind� s $$ z:TestSparseArrayArithmetics.test_float_array_different_kindc C s | � tjdddtjddddtjg �}| � dtjddtjddddtjg �}| j||d�}| j||d�}| �||||� | �||d ||d � | j||dd�}| j||d�}| �||||� | j||dd�}| j||dd�}| �||||� | j||dd�}| j||dd�}| �||||� d S )Nr7 r9 r r8 rK )r )r r0 )rA r r" r+ r4 )r r rC rF r r r r r �test_float_array_comparison s $$z6TestSparseArrayArithmetics.test_float_array_comparisonc C s� |}t j}| jddddddddddg |d�}| jddddddddddg |d�}| j|||d�}|jt|�kspt�| j|||d�} | jt|�ks�t�| �|| ||||� | �|| d ||d ||� | j|d||d�}|jt|�ks�t�| j|||d�} | jt|�k�s t�| �|| ||||� | j|d||d�}|jt|�k�sDt�| j|d||d�} | jt|�k�sjt�| �|| ||||� | j|d||d�}|jt|dd �k�s�t�| j|d||d�} | jt|dd �k�s�t�| �|| ||||� d S ) Nr r7 r9 )r- r8 rK )r- r )r0 r- r )r0 )r �int64rA r+ r- r r, r) ) r r r rB r% r- rC rF r r r r r �test_int_array s2 "" z)TestSparseArrayArithmetics.test_int_arrayc C s. d}| j ddddddddddg |d�}| j ddddddddddg |d�}| j|||d�}| j|||d�}| �||||� | �||d ||d � | j|||dd �}| j|||d�}| �||||� | j|||dd �}| j|||dd �}| �||||� | j|||dd �}| j|||dd �}| �||||� d S ) NrO r r7 r9 )r- r8 rK )r- r )r- r r0 )rA r+ r4 )r r r- rC rF r r r r r �test_int_array_comparison: s ""z4TestSparseArrayArithmetics.test_int_array_comparisonTFc C sl | j ddddgtjd�}| j ddddgtjd�}| j||tj|d�}| j||tj|d�}| �||||� d S )NTF)r- )r r- r0 )rA r r/ r+ r5 )r r r0 rC rF r r r r r �test_bool_same_indexR s z/TestSparseArrayArithmetics.test_bool_same_indexc C st | j ddddddgtjd�}| j ddddddgtjd�}| j||tj|d�}| j||tj|d�}| �||||� d S )NTF)r- )r r- r0 )rA r r/ r+ r5 )r r r0 rC rF r r r r r �test_bool_array_logical] s z2TestSparseArrayArithmetics.test_bool_array_logicalc C s� |}t s8|tjtjgkr8|r8tjjtdd�}|j � |� d}| � t j dddt j ddddt j g �}| j ddddddddddg |d �} | j||d �} | j| |d �} | jt|�ks�t�| �| | || ||� | �| | d || d ||� | j||dd �} | j| |d �} | jt|�k�st�| �| | || ||� | j||dd �} | j| |dd �} | jt|�k�sZt�| �| | || ||� | j||dd �} | j| |dd �} | jt|dd �k�s�t�| �| | || ||� d S ) NzGH#38172)r: r; rO r7 r9 r r8 rK )r- )r )r r0 )r0 )r r r r r r< r= r> r, r? r@ rA r r" r+ r- r r) ) r r r rB r r% r= �rdtyperC rF r r r r r �test_mixed_array_float_inth s4 $" z5TestSparseArrayArithmetics.test_mixed_array_float_intc C sp d}| � tjdddtjddddtjg �}| j ddddddddddg |d�}| j||d�}| j||d�}|jt|�ksxt�| �||||� | �||d ||d � | j||dd �}| j||d�}|jt|�ks�t�| �||||� | j||dd �}| j||dd �}|jt|�k�st�| �||||� | j||dd �}| j||dd �}|jt|dd �k�s\t�| �||||� d S ) NrO r7 r9 r r8 rK )r- )r )r r0 )r0 )rA r r" r+ r- r r, r4 )r r rT rC rF r r r r r �test_mixed_array_comparison� s( $"z6TestSparseArrayArithmetics.test_mixed_array_comparisonc C sl t ddddg�}t ddddg�}||A }tjjj�dtjdddgdd��}t dddg|d �}t� ||� d S ) NTFrH r r7 r9 Zint32)r- )Z sparse_index) r �pd�coreZarrays�sparseZIntIndexr �arrayr �assert_sp_array_equal)r �s�tr& Zsp_indexr' r r r �test_xor� s "z#TestSparseArrayArithmetics.test_xor) �__name__� __module__� __qualname__r rZ rA r r+ r r) r3 r4 r5 r< r= � parametrizerD rE rG rI rJ rL rM rN rP rQ r" rR rS rU rV r^ r r r r r s. + $ & r r% c C s@ t ddgdd�}| |ddg�}| |t ddg��}t�||� d S )Nr r7 )r0 )r r r[ )r% �arrr& r' r r r �test_with_list� s rd c C s@ t ddgdd�} t�ddgddgg�}| �|�}|tks<t�d S )Nr r7 )r0 r9 r8 rH )r rW � DataFrame�__add__�NotImplementedr, )rc Zdfr& r r r �test_with_dataframe� s rh c C s6 t ddgdd�} | t�d� }| d }t�||� d S )Nr r7 )r0 r9 )r r rZ r r[ )rc r& r' r r r �test_with_zerodim_ndarray� s ri �ufuncrc ���r7 c C s8 | |�}| |j �}t| t�|��|d�}t�||� d S )N)r0 )r0 r r �asarrayr r[ )rj rc r& r0 r' r r r � test_ufuncs� s rm za, br9 )r0 c C sD | ||�}| t �|�t �|��}t|t�s.t�t�t �|�|� d S )N)r rl r* r r, r r )rj r r r& r' r r r �test_binary_ufuncs� s rn c C sL t ddddg�} t�ddddg�}|| 7 }t�ddddg�}t�||� d S )Nr r9 r7 r8 )r r rZ r r )�sparray�ndarrayr' r r r �test_ndarray_inplace� s rq c C sN t ddddg�} t�ddddg�}| |7 } t ddddgdd�}t�| |� d S )Nr r9 r7 r8 )r0 )r r rZ r r[ )ro rp r' r r r �test_sparray_inplace� s rr r0 c C s� t �ddddg�}t|| d�}| }t| | d�}t�||� t�|� }t�|�}t�||� t�d|i� }t�d|i�}t� ||� d S )NTF)r0 �A) r rZ r r r[ rW ZSeriesZassert_series_equalre Zassert_frame_equal)r0 rc ro r& r' r r r � test_invert� s rt c C sL t �ddt jdg�}t||d�}| |�}t| |�| |�d�}t�||� d S )Nr r7 r9 )r0 )r rZ r" r r r[ )r% r0 rc ro r& r' r r r � test_unary_op s ru )(r �numpyr r< Z pandas.compatr ZpandasrW Zpandas._testingZ_testingr Z pandas.corer Zpandas.core.arrays.sparser r Zfixturer r r r= rb �eq�addrd rh ri �abs�exprm rZ Zgreaterrn rq rr rt r" �pos�negru r r r r �<module> s@ . $
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка