Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/arrays/sparse/__pycache__/test_array.cpython-37.pyc
Назад
B �5�g� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlm m Z d dl Z d dl mZ d dlmZ d dlmZmZ G dd� d�ZG dd� d�ZG d d � d �Zd d � Zd d� Zej�ddddgd fd ddgdfd ddgdfd ddddgdfddddgdfg dfg�dd� �Z ej�dddej ej gdej dej gddej gg�ej�dej d dg�dd� ��Z dd� Z d d � Z d d � Z!ej�dej dg�d d!� �Z"d"d#� Z#G d$d%� d%�Z$dS )&� N)�IntIndex)�isna)� SparseArray� SparseDtypec @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Ze j � d e e d�e g�dd� �Zdd� Zdd� Ze j � ddeddg�g�dd� �Zdd� Ze j � d e�dd g�dfe�d d g�ejfd d!gd!fe�d"�gejfg�d#d$� �Ze j � d%d&d'd(g�e j � d)dd*g�ej d+d,� ���Z e j � d%d&d'd(g�ej d-d.� ��Z ej d/d0� �Z e j � d1d!e e d!�fd2e d3d�fde d4d�fd5e d6d5�fg�d7d8� �Z!d9d:� Z"d;d<� Z#d=d>� Z$d?d@� Z%dAdB� Z&dCdD� Z'e j � dEddejg�dFdG� �Z(dHdI� Z)dJdK� Z*dLdM� Z+dNdO� Z,dPdQ� Z-dRdS� Z.dTdU� Z/dVdW� Z0dXdY� Z1dZd[� Z2d\d]� Z3d^d_� Z4d`da� Z5dbdc� Z6e j � dde7ddg�dee7d2d ge e8d2�df�fe7ddg�e e7d!d g�fe7ddgddg�e e7d!d ge e d �df�fe j9e7ddg�dhe7ejddgdhdf�e dhe�di��df�e j j:djdk�gdl�e7ddd*g�e;e7dmdndoge e;dm�df�fe7dodpg�e8e7dqdrg�fe7dddg�e<e7dddge e<d�df�fg�dsdt� �Z=dudv� Z>dwdx� Z?dydz� Z@e j � d{dd d|ge�dd g�d}g�d~d� �ZAd�d�� ZBd�d�� ZCe j � d�dddddgd�dfg d�dfdgd�dfd�d�ejd�gd�e<fg�d�d�� �ZDe j � d�ejejejejejgdejejd|ejgdejdd|dgg�e j � dEddg�d�d�� ��ZEd�d�� ZFd�d�� ZGd�d�� ZHd�d�� ZId�d�� ZJe j � d�d�d�d�d�d�d�g�d�d�� �ZKd�d�� ZLd�d�� ZMd�d�� ZNd�d�� ZOd�d�� ZPdS )��TestSparseArrayc C sZ t �t jt jdddt jddt jdg �| _t| j�| _tddddddddddg dd�| _d S ) N� � � � � � r )� fill_value)�np�array�nan�arr_datar �arr�zarr)�self�method� r �[/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/arrays/sparse/test_array.py� setup_method s ( z TestSparseArray.setup_methodc C s� t tjddtjg�}|jttjtj�ks,t�|jjtjks>t�t�|j �sNt�t tjddtjgdd�}|jttjd�ks|t�|j dks�t�t ddddgtjd�}|jttjtj�ks�t�t�|j �s�t�t ddddgtj d�}|jttj d�ks�t�|j dk�st�t ddddgdtj d�}|jttj d�k�s4t�|j dk�sDt�t ddddgd d�}|jttj d�k�spt�|j dk�s�t�t ddddgdd d�}|jttj d�k�s�t�|j dk�s�t�d S )Nr r r )r r )�dtype)r r ) r r r r r �float64�AssertionError�subtype�isnanr �int64)r r r r r �test_constructor_dtype s, z&TestSparseArray.test_constructor_dtypec C s4 t dddgdd�}t dddgtd�}t�||� d S )Nr r r �int)r )r r �tm�assert_sp_array_equal)r �result�expectedr r r �test_constructor_dtype_str8 s z*TestSparseArray.test_constructor_dtype_strc C sX t ddddgtdd�d�}t ddddgdtjd�}t�||� |jjt�d�ksTt�d S )Nr r r ���)r )r r ) r r r r r! r" � sp_valuesr r )r r# r$ r r r � test_constructor_sparse_dtype= s z-TestSparseArray.test_constructor_sparse_dtypec C sP t ddddgdd�}t ddddgtjd�}t�||� |jjt�d�ksLt�d S )Nr r z Sparse[int32])r �int32)r r r) r! r" r' r r )r r# r$ r r r �!test_constructor_sparse_dtype_strC s z1TestSparseArray.test_constructor_sparse_dtype_strc C s� t ddtjdgtd�}|jtt�ks(t�t�|j�s8t�t ddtjdgtdd�}|jttd�ksdt�|jdksrt�ddddg}t |tdd�}|jttd�ks�t�|jdks�t�tj |td�}d d � t ||�D �}tj |tj d�� � s�t�d S ) N�A�B)r )r r Fr g Y@g c s s* | ]"\}}t |�t |�ko ||kV qd S )N)�type)�.0�x�yr r r � <genexpr>Y s z@TestSparseArray.test_constructor_object_dtype.<locals>.<genexpr>)r r r �objectr r r r r r �zipZfromiter�bool_�all)r r �dataZ arr_expected�itr r r � test_constructor_object_dtypeI s z-TestSparseArray.test_constructor_object_dtyper r c C s2 t jtdd�� tddtjg|d� W d Q R X d S )NzCannot convert)�matchr r )r )�pytest�raises� ValueErrorr r r )r r r r r �test_constructor_na_dtype\ s z)TestSparseArray.test_constructor_na_dtypec C s� t jdddd�}ttj|dd��}t�t�� t|�}W d Q R X t�||� t�t�� tt � |��}W d Q R X t�||� d S )Nz 2016-01-01r z US/Pacific)Zperiods�tzzdatetime64[ns])r ) �pdZ date_ranger r �asarrayr! Zassert_produces_warning� UserWarningr" �Series)r Zdtir$ r# r r r �+test_constructor_warns_when_losing_timezonea s z;TestSparseArray.test_constructor_warns_when_losing_timezonec C s� t ddgtdddg�d�}t ddddgdd�}t�||� |jttj�ksNt�|j dks\t�t dddgtddddg�tjdd �}t ddddgtjdd �}t�||� |jttj�ks�t�|j dks�t�t ddgtdddg�dtjd �}t ddddgdtjd �}t�||� |jttj�k�s t�|j dk�s0t�t dddgtddddg�d dd �}t ddddgd d �}t�||� |jttj�k�s�t�|j dk�s�t�d S )Nr r r )r6 � sparse_indexr �integer)�kindr )r6 rD r r )r r )r6 rD r r )r r )r ) r r r! r" r r r r r r )r r r$ �expr r r � test_constructor_spindex_dtypeq s: z.TestSparseArray.test_constructor_spindex_dtyperD Nr c C s� t d|d d�}t dgd d�}t�||� |jttj�ks<t�|jdksJt�t dt ddg�d d�}t dgd d�}t�||� |jttj�ks�t�|jdks�t�d S )Nr )r6 rD r )r r ) r r! r" r r r r r r r )r rD r rG r r r �%test_constructor_spindex_dtype_scalar� s z5TestSparseArray.test_constructor_spindex_dtype_scalarc C sf t ddgtdddg�dd d�}t ddddgdd d�}t�||� |jttj�ksTt�|j dksbt�d S )Nr r r r )r6 rD r r )r r ) r r r! r" r r r r r r )r r rG r r r �0test_constructor_spindex_dtype_scalar_broadcasts� s z@TestSparseArray.test_constructor_spindex_dtype_scalar_broadcastszdata, fill_valuer g �?g @TFz 2017-01-01c C s0 t |�j}t|�r t|�s,t�n ||ks,t�d S )N)r r r r )r r6 r r# r r r �$test_constructor_inferred_fill_value� s z4TestSparseArray.test_constructor_inferred_fill_value�formatZcoo�cscZcsr�size� c C sL dd l }|jj|dd|d�}t�|�}t�|�}|�� �� }t � ||� d S )Nr r g �?)�densityrL ) � scipy.sparse�sparse�randomr � from_spmatrixr r@ �toarray�ravelr! �assert_numpy_array_equal)r rN rL �scipy�matr# r$ r r r �test_from_spmatrix� s z"TestSparseArray.test_from_spmatrixc C sV dd l }|jjddd|d�}d|jd< t�|�}t�|�}|�� � � }t � ||� d S )Nr rO r g �?)rP rL ) rQ rR rS r6 r rT r r@ rU rV r! rW )r rL rX rY r# r$ r r r �*test_from_spmatrix_including_explicit_zero� s z:TestSparseArray.test_from_spmatrix_including_explicit_zeroc C sB dd l }|jjdddd�}tjtdd�� t�|� W d Q R X d S )Nr r r rM )rL znot '4')r9 )rQ rR �eyer: r; r< r rT )r rX rY r r r �test_from_spmatrix_raises� s z)TestSparseArray.test_from_spmatrix_raisesz scalar,dtypeg r r �zr2 c C sR t |dddg|d�}t |||g|d�}t�||� |j|ks@t�|j|ksNt�d S )Nr r r )�indexr )r )r r! r" r r )r Zscalarr r rG r r r �"test_scalar_with_index_infer_dtype� s z2TestSparseArray.test_scalar_with_index_infer_dtypec C s� t �| jd �st�| jd dks&t�| jd dks8t�| jd dksJt�| jd dks\t�| jd dksnt�t�d�}tjt |d�� | jd W d Q R X tjt |d�� | jd W d Q R X | jd | jt | j�d ks�t�d S ) Nr r � r r �bounds)r9 � i����r& ) r r r r r �re�compiler: r; � IndexError�len)r �errmsgr r r � test_get_item� s z TestSparseArray.test_get_itemc C s. d}t jt|d�� | j�d� W d Q R X d S )Nz-'indices' must be an array, not a scalar '2'.)r9 r )r: r; r< r �take)r �msgr r r �test_take_scalar_raises s z'TestSparseArray.test_take_scalar_raisesc C sd t t�| jddg��}t�| j�ddg�|� t t�| jdddg��}t�| j�dddg�|� d S )Nr r r r )r r rj r r! r" r )r rG r r r � test_take s zTestSparseArray.test_takec C s< t jddgtd�d�}|jddgdtjd�}t�||� d S )Nr r )r r T)� allow_fillr )r? r r rj r r r! r" )r �ar# r r r �test_take_all_empty s z#TestSparseArray.test_take_all_emptyc C s� t �dt jdddg�}t|dd�}tt �|dg�dd�}t�|�dg�|� tt �|dddg�dd�}t�|�dddg�|� d S )Nr r r )r r )r r r r rj r! r" )r r6 rR rG r r r �test_take_fill_value s z$TestSparseArray.test_take_fill_valuec C s` t t�| jdg��}t�| j�dg�|� t t�| jdddg��}t�| j�dddg�|� d S )Nr& ���������)r r rj r r! r" r )r rG r r r �test_take_negative& s z"TestSparseArray.test_take_negativer c C sf t t�dddddg�dd�}|jd|d�}t|�r:|jj}t t�|ddddg�dd�}t�||� d S )Nr r r g @)r ) r r r �shiftr r Zna_valuer! r" )r r rR �resrG r r r �test_shift_fill_value- s z%TestSparseArray.test_shift_fill_valuec C s, t jtdd�� | j�dg� W d Q R X d S )Nrb )r9 rc )r: r; rf r rj )r r r r � test_bad_take7 s z TestSparseArray.test_bad_takec C s� t tjtjdtjdg�}|�t�dddg��}t tjtjdg�}t�||� |jt�dddg�dd�}t tjtjtjg�}t�||� |jt�dddg�ddd�}t tjtjdg�}t�||� d }tjt |d �� |jt�ddd g�dd� W d Q R X tjt |d �� |jt�ddd g�dd� W d Q R X d }tjt |d �� |�t�ddg�� W d Q R X tjt |d �� |�t�ddg�� W d Q R X tjt |d �� |jt�ddg�dd� W d Q R X d S )Nr r r r& T)rn F)rn r zInvalid value in 'indices')r9 rt ���z out of bounds value in 'indices'i����r ) r r r rj r r! r" r: r; r< rf )r rR r# r$ rk r r r �test_take_filling; s, $$ z!TestSparseArray.test_take_fillingc C s� t tjddddgdd�}|�t�dddg��}t dtjdgdd�}t�||� |jt�dddg�dd�}t dtjtjgdd�}t�||� |jt�dddg�ddd �}t dtjdgdd�}t�||� d }tjt |d �� |jt�ddd g�dd� W d Q R X tjt |d �� |jt�ddd g�dd� W d Q R X d}tjt |d �� |�t�ddg�� W d Q R X tjt |d �� |�t�ddg�� W d Q R X tjt |d �� |jt�ddg�dd� W d Q R X d S )Nr r r )r r& T)rn F)rn r zInvalid value in 'indices'.)r9 rt rz z out of bounds value in 'indices'i����r ) r r r rj r r! r" r: r; r< rf )r rR r# r$ rk r r r � test_take_filling_fill_value[ s, $$ z,TestSparseArray.test_take_filling_fill_valuec C s* t tjtjtjtjtjg�}|�t�dddg��}t tjtjtjgdd�}t�||� |jt�dddg�dd�}t tjtjtjgdd�}t�||� d}tjt |d �� |�t�dd g�� W d Q R X tjt |d �� |�t�dd g�� W d Q R X tjt |d �� |jt�dd g�dd� W d Q R X d S ) Nr r r& �block)rF T)r z out of bounds value in 'indices')r9 i����r ) r r r rj r r! r" r: r; rf )r rR r# r$ rk r r r �test_take_filling_all_nan} s z)TestSparseArray.test_take_filling_all_nanc s\ � fdd�}� fdd�}t jtdd�� |� W d Q R X t jtdd�� |� W d Q R X d S )Nc s d� j d< d S )Nr r )r r )r r r �setitem� s z.TestSparseArray.test_set_item.<locals>.setitemc s d� j dd�< d S )Nr r r )r r )r r r �setslice� s z/TestSparseArray.test_set_item.<locals>.setslicezassignment via setitem)r9 )r: r; � TypeError)r r r� r )r r � test_set_item� s z TestSparseArray.test_set_itemc C s2 t jtdd�� tt�d��d�� W d Q R X d S )Nz expected dimension <= 1 data)r9 rO )r r )r: r; r� r r �arangeZreshape)r r r r �%test_constructor_from_too_large_array� s z5TestSparseArray.test_constructor_from_too_large_arrayc C s. t | j�}|jdkst�t�|j| jj� d S )Nr )r r r r r! �assert_almost_equalr' )r rw r r r � test_constructor_from_sparse� s z,TestSparseArray.test_constructor_from_sparsec C sp t | jdd�}d|jd d�< | jjd d� dk�� r8t�t | j�}d|jd d�< | jjd d� dk�� slt�d S )NT)�copyr r )r r r' �anyr r5 )r �cpZnot_copyr r r �test_constructor_copy� s z%TestSparseArray.test_constructor_copyc C s� t �ddddddg�}t|dtd�}|jtt�ks6t�t�|j t �ddg�� t�|j j t �ddgt j �� |� � }|jtks�t�t�||� d S )NFT)r r r r )r r r �boolr r r r! rW r' �sp_index�indicesr) �to_dense)r r6 r �denser r r �test_constructor_bool� s z%TestSparseArray.test_constructor_boolc C s� t dddgd d�}|jttj�ks&t�|jr0t�t dddgtjd�}|jttj�ksXt�|jrbt�t dddgtjdd�}|jttjd�ks�t�|js�t�d S )NTF)r )r r )r r r r r4 r r )r r r r r � test_constructor_bool_fill_value� s z0TestSparseArray.test_constructor_bool_fill_valuec C s� t jdt jdgt jd�}t|t jd�}|jtt j�ks:t�t� |j t jddgt jd�� t� |j j t jddgt j d�� |�� }|jt jks�t�t� ||� d S )Ng �?r )r r r r )r r r �float32r r r r r! rW r' r� r� r) r� )r r6 r r� r r r �test_constructor_float32� s z(TestSparseArray.test_constructor_float32c C s t d d ddg�}|�d�}t d d ddgt�d�d�}t�||� tddd�}|�|�}t �tjdd g|j d�t d dd g�|�}t�||� td d�}|�|�}t �tjddgtj d�t d dd g�|�}t�||� t dtj dd gdd�}t jtdd�� |�d� W d Q R X d S )Nr r zSparse[float32]r� )r r )r g g @r r r r ZNA)r9 z Sparse[i8])r �astyper r r! r" r � _simple_newr r r r r r: r; r< )r r r# r$ r r r r � test_astype� s" $ $ zTestSparseArray.test_astypec C s� t ddddgttd�d�}|�t�}t ddddgttd�d�}t�||� |�ttd��}t ddddgttd�d�}t�||� d S )Nr r )r TF)r r r r� r� r! r" )r ro r# r$ r r r �test_astype_bool� s z TestSparseArray.test_astype_boolc C st t �dddg�}t|dd�}t �|�}|�|�}|jt|d�ksDt�|jj|ksTt�t� t � |� � �|�|�� d S )Nr r r )r ) r r r r r� r r r' r! rW r@ r� )r Zany_real_dtype�valsr �typrw r r r �test_astype_all s z TestSparseArray.test_astype_allzarr, dtype, expected�float)r )r zdatetime64[ns]Z1970z NumPy-7619)�reason)Zmarks�0�1�10Z20g $@g 4@c C s |� |�}t�||� d S )N)r� r! r" )r r r r$ r# r r r �test_astype_more s % z TestSparseArray.test_astype_morec C s6 t dtjg�}tjtdd�� |�t� W d Q R X d S )Ng �?zCannot convert non-finite)r9 )r r r r: r; r< r� r )r r r r r �test_astype_nan_raises4 s z&TestSparseArray.test_astype_nan_raisesc C s> t dddg�}|jtdd�}t dddgd d �}t�||� d S ) Nr r r F)r� g �?g @g @g )r )r r� r� r! r" )r r r# r$ r r r �test_astype_copy_false9 s z&TestSparseArray.test_astype_copy_falsec C s� t dtjdgtjd�}d|_|jdks*t�t dddgdtjd�}d|_|jdksTt�d|_|jdksht�tj|_t�|j�s�t�t d d d gd tjd�}d |_|js�t�d|_|jdks�t�tj|_t�|j�s�t�d S ) Ng �?g @)r r r r )r r g������@TF)r r r r r r r r4 )r r r r r �test_set_fill_valueA s" z#TestSparseArray.test_set_fill_value�valr )r r r c C s> t dddgdtjd�}d}tjt|d�� ||_W d Q R X d S )NTF)r r zfill_value must be a scalar)r9 )r r r4 r: r; r< r )r r� r rk r r r � test_set_fill_invalid_non_scalarf s z0TestSparseArray.test_set_fill_invalid_non_scalarc C s2 | j �� }|j| j jk s t�|j| j jks.t�d S )N)r r� r' r r� )r �arr2r r r � test_copyn s zTestSparseArray.test_copyc C s t �| j�� | j� d S )N)r! r� r r� r )r r r r �test_values_asarrays s z#TestSparseArray.test_values_asarrayzdata,shape,dtype)r )r )r r+ r, )r c C s t ||d�}|j|kst�d S )N)r )r �shaper )r r6 r� r �outr r r � test_shapev s zTestSparseArray.test_shaper� c C sB t �|�}t||d�}|�� }t�||� |�� }t�||� d S )N)r )r r r r� r! rW Z_internal_get_values)r r� r r rw �res2r r r �test_dense_repr� s z TestSparseArray.test_dense_reprc s: � fdd�}x(t t� j��D ]}||� || � q W d S )Nc s t �� j| � j�� | � d S )N)r! r� r r� )�i)r r r �_checkit� s z.TestSparseArray.test_getitem.<locals>._checkit)�rangerg r )r r� r� r )r r � test_getitem� s z TestSparseArray.test_getitemc C s8 t dddg�}|dddg }t ddg�}t�||� d S )Nr r r TF)r r! r" )r r r# r$ r r r �test_getitem_arraylike_mask� s z+TestSparseArray.test_getitem_arraylike_maskc C s� | j d d� }t| j �� d d� �}t�||� | j dd � }t| j �� dd � �}t�||� | j dd � }t| j �}t�||� | j d d� }t| j �� d d� �}t�||� d S )Nrs rr i�r )r r r� r! r" )r r# rG r r r � test_getslice� s z TestSparseArray.test_getslicec C s� t �t jdddddt jt jdg �}t|�}|tdd �f }t|dd � �}t�||� t|dd�}|tdd �f }t|dd � dd�}t�||� d}tjt |d�� |dd �d d �f W d Q R X tjt |d�� |dd �d d �f W d Q R X d S )Nr r r r )r ztoo many indices for array)r9 ) r r r r �slicer! r" r: r; rf )r r� rR rw rG rk r r r �test_getslice_tuple� s " z#TestSparseArray.test_getslice_tuplec C s0 t dddg�}|dddg }|j|jks,t�d S )Nr r r F)r r r )r r rw r r r �test_boolean_slice_empty� s z(TestSparseArray.test_boolean_slice_empty�op�add�sub�mul�truediv�floordiv�powc C s� t t|�}tj�d�}tj�d�}tj|d d d�< tj|d d d�< t|�}t|�}d|d d d�< d|d d d�< t|dd�}t|dd�}dd� }tjdd��0 x(||f||fgD ]\} } ||| | � q�W W d Q R X d S ) N� r r )r c S s | ||�}t | |�� |�� �|jd�}t|t �s4t�t�|�� |�� � | ||�� �}t|t �sdt�t�||� | |�� |�}t|t �s�t�t�||� | |d�}t|t �s�t�y | |�� d�}| |jd�}W n tk r� Y n X t�|j|� t�|�� |� d S )N)r r ) r r� r � isinstancer r! r� r" r< ) r� �first�secondrw rG r� Zres3Zres4Zexp_fvr r r � _check_op� s( z8TestSparseArray.test_binary_operators.<locals>._check_op�ignore)r5 )�getattr�operatorr rS Zrandnr r Zerrstate) r r� Zdata1Zdata2Zarr1r� Zfarr1Zfarr2r� Z first_arrZ second_arrr r r �test_binary_operators� s z%TestSparseArray.test_binary_operatorsc C s dd� }|| j � || j� d S )Nc S s t �| �}t �|| � d S )N)r! Zround_trip_pickler" )�objZ unpickledr r r �_check_roundtrip s z5TestSparseArray.test_pickle.<locals>._check_roundtrip)r r )r r� r r r � test_pickle s zTestSparseArray.test_picklec C sd t dddg�}tjdd��@}tjdtd� tjdtd� x |D ]}q>W t|�dksVt�W d Q R X d S ) Nr r r T)�record�always)�action�categoryr )r �warnings�catch_warnings�filterwarnings�DeprecationWarning�PendingDeprecationWarningrg r )r Zsp_arr�w�_r r r �test_generator_warnings s z'TestSparseArray.test_generator_warningsc C s t dtjtjdtjg�}|�d�}t dddddgdtjd�}t�||� t dtjtjdtjgdd�}|�d�}t dddddgdtjd�}t�||� t dtjdddg�}|�d�}t dddddgdtjd�}t�||� t dtjdddgdd�}|�d�}t dddddgdtjd�}t�||� t tjtjtjtjg�}|�d�}t ddddgdtjd�}t�||� t tjtjtjtjgdd�}|�d�}t ddddgdtjd�}t�||� t ddddg�}|�d�}t ddddgdd�}t�||� t ddddg�}|jttj �k�st �|j dk�s t �|�d�}t�||� t ddddgdd�}|jttj �k�s`t �|j dk�spt �|�d�}t ddddgdd�}t�||� t ddddgtjd�}|jttj tjd�k�s�t �t� |j ��s�t �|�d�}t ddddgdd�}t�||� d S )Nr r r& )r r r )r g ) r r r �fillnar r! r" r r r r r r )r �srw rG r r r � test_fillna sZ zTestSparseArray.test_fillnac C s� t dtjtjdtjg�}|�d�}tjdddddgtjd�}t�|�� |� t dtjtjdtjgdd�}|�d�}t dddddgdtjd�}t� ||� d S )Nr r )r r )r )r r ) r r r r� r r r! rW r� r" )r r� rw rG r r r �test_fillna_overlapQ s z#TestSparseArray.test_fillna_overlapc C s� t td�td�ddddddddddg �}tjdddgtjd�}|�� \}t�||� t ddddddddddddg �}|�� \}t�||� d S ) Nr r r r r r � )r )r r� r r r) Znonzeror! rW )r �sar$ r# r r r � test_nonzero^ s ( z TestSparseArray.test_nonzero)Q�__name__� __module__� __qualname__r r r% r( r* r8 r: �mark� parametrizer r r= rC rH r rI rJ r r r r? � TimestampZNaTrK �td�skip_if_no_scipyrZ r[ r] r� r` ri rl rm rp rq ru rx ry r{ r| r~ r� r� r� r� r� r� r� r� r� r� r r� �paramZxfail�strr2 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 s� '" " &%, "2 < r c @ s~ e Zd Zej�ddddgddfdddgddfdddgdd fg�d d � �Zej�ddddgddfdddgddfdddgdd fg�d d � �Zej�ddddgddfdddgddfd dd gdd fg�dd� �Zej�ddddgddfdddgddfd dd gdd fg�dd� �Z dd� Z ej�de � dde j dg�e � dddg�e � dddg�g�ej�ddde j ddg�ej�ddde j fg�dd� ���Zdd � Zej�d e j ddd dd ged �ee � dd!d"d#d$g��fe j dde j dd ged �ee � dd!e j d%d&g��fg�ej�d'ddg�d(d)� ��Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zej�dd e j g�d2d3� �Zd4d5� Zd6d7� Zd8d9� Zd:d;� Zd<d=� Z d>S )?�TestSparseArrayAnalyticsz data,pos,negTFr r r g �?g @g c C sd t |��� }|st�t ||d��� }|s,t�||d< t |��� }|rHt�t ||d��� }|r`t�d S )N)r r )r r5 r )r r6 �pos�negr� r r r �test_allk s z!TestSparseArrayAnalytics.test_allc C s� t �t|��}|st�t �t||d��}|s0t�||d< t �t|��}|rNt�t �t||d��}|rht�d}tjt|d�� t jt|�t �g �d� W d Q R X d S )N)r r z$the 'out' parameter is not supported)r9 )r� )r r5 r r r: r; r< r )r r6 r� r� r� rk r r r �test_numpy_all� s z'TestSparseArrayAnalytics.test_numpy_allc C sd t |��� }|st�t ||d��� }|s,t�||d< t |��� }|rHt�t ||d��� }|r`t�d S )N)r r )r r� r )r r6 r� r� r� r r r �test_any� s z!TestSparseArrayAnalytics.test_anyc C s� t �t|��}|st�t �t||d��}|s0t�||d< t �t|��}|rNt�t �t||d��}|rht�d}tjt|d�� t jt|�|d� W d Q R X d S )N)r r z$the 'out' parameter is not supported)r9 )r� )r r� r r r: r; r< )r r6 r� r� r� rk r r r �test_numpy_any� s z'TestSparseArrayAnalytics.test_numpy_anyc C sp t �d��t�}t|��� }|dks(t�t j|d< t|dd��� }|dksNt�t|t jd��� }|dkslt�d S )NrO g �F@r r )r g D@)r r� r� r� r �sumr r )r r6 r� r r r �test_sum� s z!TestSparseArrayAnalytics.test_sumr r zmin_count, expected)r r r c C sB t ||d�}|j|d�}t�|�r2t�|�s>t�n ||ks>t�d S )N)r )� min_count)r r� r r r )r r r r� r$ Zsparrayr# r r r �test_sum_min_count� s z+TestSparseArrayAnalytics.test_sum_min_countc C s� t �d��t�}t �t|��}|dks*t�t j|d< t �t|dd��}|dksRt�t �t|t jd��}|dksrt�d}tj t |d�� t jt|�t j d � W d Q R X d }tj t |d�� t jt|�|d � W d Q R X d S ) NrO g �F@r r )r g D@z&the 'dtype' parameter is not supported)r9 )r z$the 'out' parameter is not supported)r� ) r r� r� r� r� r r r r: r; r< r )r r6 r� rk r r r �test_numpy_sum� s z'TestSparseArrayAnalytics.test_numpy_sumz data,expectedr r )r g @g @g $@g .@g @g (@�numpyc C s |r t jndd� }|t|��}t�||� |t|t jd��}t�||� |t|dd��}t�||� |r�d}tjt|d�� t jt|�t j d� W d Q R X d}tjt|d�� t jt|�|d � W d Q R X n@d }t � d |� d ��}tjt|d�� t|�j|d � W d Q R X d S )Nc S s | � � S )N)�cumsum)r� r r r �<lambda> � z6TestSparseArrayAnalytics.test_cumsum.<locals>.<lambda>)r r z&the 'dtype' parameter is not supported)r9 )r z$the 'out' parameter is not supported)r� r zaxis(=z) out of bounds)�axis) r r� r r! r" r r: r; r< r rd �escape)r r6 r$ r� r� r� rk r� r r r � test_cumsum s$ z$TestSparseArrayAnalytics.test_cumsumc C sN t �d��t�}t|��� }|dks(t�t j|d< t|��� }|dksJt�d S )NrO g @r gr �q �@)r r� r� r� r �meanr r )r r6 r� r r r � test_mean+ s z"TestSparseArrayAnalytics.test_meanc C s� t �d��t�}t �t|��}|dks*t�t j|d< t �t|��}|dksNt�d}tj t |d�� t jt|�t j d� W d Q R X d}tj t |d�� t jt|�|d � W d Q R X d S ) NrO g @r gr �q �@z&the 'dtype' parameter is not supported)r9 )r z$the 'out' parameter is not supported)r� ) r r� r� r� r� r r r r: r; r< r )r r6 r� rk r r r �test_numpy_mean4 s z(TestSparseArrayAnalytics.test_numpy_meanc C s� t dtjdtjdg�}t dtjdtjdg�}t�t|�|� t�t�|�|� t ddddgdd�}t dddg|jdd�}t�t|�|� t�t�|�|� t ddddgdd�}t dddg|jdd�}t�t|�|� t�t�|�|� t dtjdtjdg�}t t�dtjdtjdg��}t�t�|�|� t ddddgdd�}t t�ddddg�t�d�d�}t�t�|�|� t ddddgdd�}t t�ddddg�t�d�d�}t�t�|�|� d S )Nr r rt r& )r )rD r r )r r r r! r" �absr� �sin)r rR r# r r r � test_ufuncE s* z#TestSparseArrayAnalytics.test_ufuncc C s� t dtjdtjdg�}t dtjdtjdg�}t�t�|d�|� t ddddgdd�}t ddddgdd�}t�t�|d�|� t ddddgdd�}t ddddgdd�}t�t�|d�|� d S )Nr r rt r r& )r r )r r r r! r" r� )r rR r# r r r �test_ufunc_argsb s z(TestSparseArrayAnalytics.test_ufunc_argsc C sh t |gd ddg |d�}t�|�\}}t�t�|��\}}t�|t ||d�� t�|t ||d�� d S )NrO g�������?g������@)r )r r �modfr@ r! r" )r r rR �r1�r2�e1�e2r r r � test_modfp s z"TestSparseArrayAnalytics.test_modfc C s, t dddddgdd�}|j}|dks(t�d S )Nr r r rE )rF � )r �nbytesr )r r r# r r r �test_nbytes_integery s z,TestSparseArrayAnalytics.test_nbytes_integerc C s, t dddddgdd�}|j}|dks(t�d S )Nr r r r} )rF r )r r r )r r r# r r r �test_nbytes_block s z*TestSparseArrayAnalytics.test_nbytes_blockc C s$ t t�dd d dg��}t�|� d S )NZ2012Z2013)r r? � to_datetimer r@ )r r� r r r �test_asarray_datetime64� s z0TestSparseArrayAnalytics.test_asarray_datetime64c C s t ddg�}|jdkst�d S )Nr r g �?)r rP r )r r r r r � test_density� s z%TestSparseArrayAnalytics.test_densityc C s t ddg�}|jdkst�d S )Nr r )r �npointsr )r r r r r � test_npoints� s z%TestSparseArrayAnalytics.test_npointsN) 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 r r r� j s\ $ r� c @ sN e Zd Zej�dddddg�dd� �Zejdd � �Z ejd d � �Z d d � Z dS )� TestAccessor�attrr rP r r' c C s< t ddg�}t�|�}t|j|�}t||�}||ks8t�d S )Nr r )r r? rB r� rR r )r r r �serr# r$ r r r �test_get_attributes� s z TestAccessor.test_get_attributesc C s� dd l }ddddg}ddddg}ddddg}|jj|||ffd d �}tjj�|�}tj�ddddgddddgg�}tjddddg|d d �}t� ||� d S ) Nr r r r r r ra r� r )r z Sparse[int])r_ r ) rQ rR � coo_matrixr? rB Zfrom_coo� MultiIndexZ from_arraysr! Zassert_series_equal) r rX �row�colr6 Zsp_arrayr# r_ r$ r r r � test_from_coo� s zTestAccessor.test_from_cooc C s` dd l }tjdddgtjjdgdddggddgd�dd �}|j�� \}}}t||jj�s\t �d S ) Nr r r r ro �b)�namesz Sparse[int])r_ r ) rQ r? rB r Z from_productrR Zto_coor� r r )r rX r r+ r� r r r � test_to_coo� s zTestAccessor.test_to_cooc C s6 t �dddg�}tjtdd�� |jj W d Q R X d S )Nr r r z.sparse)r9 )r? rB r: r; �AttributeErrorrR rP )r r r r r �test_non_sparse_raises� s z#TestAccessor.test_non_sparse_raisesN) r� r� r� r: r� r� r r� r� r r r r r r r r � s r c C sN t dtjdgdd�} tj| _| �� }t�|�}t�dddg�}t�||� d S )Ng �?g )r FT) r r r r r r@ r r! rW )r r# r$ r r r �*test_setting_fill_value_fillna_still_works� s r c C sT t dtjgdd�} tj| _t jt�tjg�tddg�tttj�d�}t � | |� d S )Ng r )r r r )Z sparse_arrayrD r ) r r r r r� r r r r� r! r" )r r$ r r r � test_setting_fill_value_updates� s r zarr, locr r r r& c C s t | ��� }||kst�d S )N)r Z_first_fill_value_locr )r �locr# r r r �test_first_fill_value_loc� s r! r r c C sF t | |d��� }t�| ��� }t|t �s,t�t�|�}t� ||� d S )N)r ) r �uniquer? rB r� r r r@ r! rW )r r ro r r r r �test_unique_na_fill� s r# c C s. t ddg�} | �� }t dg�}t�||� d S )Nr )r r" r! r" )r r# r$ r r r �test_unique_all_sparse� s r$ c C s� t dddg�} t dddgdd�}| �dddd��}t�||� | �t�dddd���}t�||� | �t�dddd���}t dddgdd�}t�||� d S ) Nr r r rO rc � )r )r r r )r �mapr! r" r? rB )r r$ r# r r r �test_map� s r' c C s@ t dddg�} t ddd gdd�}| �ddd��}t�||� d S )Nr r r rO rc )r )r r )r r& r! r" )r r$ r# r r r �test_map_missing s r( c C sv t tjdg| d�}t dg| d�}t�|�� |� t�ddg|d��}tjdg|d�t�dg�d�}t� |�� |� d S )Nr )r g �?r )ro r )r_ ) r r r r! r" Zdropnar? � DataFrameZ Int64IndexZ assert_equal)r r rG �dfZ expected_dfr r r � test_dropna s r+ c C sJ t �t�d���dd� �} | �� }t �dd� td�D ��}t�||� d S )N)r r c S s t | dd�S )Nr )r )r )r/ r r r r� & r� z1test_drop_duplicates_fill_value.<locals>.<lambda>c S s i | ]}t d gdd�|�qS )g r )r )r )r. r� r r r � <dictcomp>( s z3test_drop_duplicates_fill_value.<locals>.<dictcomp>r ) r? r) r Zzeros�applyZdrop_duplicatesr� r! Zassert_frame_equal)r* r# r$ r r r � test_drop_duplicates_fill_value$ s r. c @ s� e Zd Ze�d��e�Zed Ze e� dddej dg��Z e e� ej ej ej ej ej g��Z e e� ej ej ej ej ej g�dd�Zej�dedgdgfedgd gfe dgdgfe ej gej gfedgdgfg�d d � �Zd S ) � TestMinMaxr r& r r r r )r z"raw_data,max_expected,min_expectedrr c C s4 t |��� }t |��� }||ks$t�||ks0t�d S )N)r �max�minr )r Zraw_dataZ max_expectedZ min_expectedZ max_resultZ min_resultr r r � test_maxmin5 s zTestMinMax.test_maxminN)r� r� r� r r� r� r� Z plain_dataZdata_negr r r Zdata_NaNZ data_all_NaNZdata_NA_filledr: r� r� r2 r r r r r/ , s "$ r/ )%r� rd r� r� r r: Zpandas._libs.sparser Z pandas.util._test_decorators�utilZ_test_decoratorsr� Zpandasr? r Zpandas._testingZ_testingr! Zpandas.core.arrays.sparser r r r� r r r r� r� r! r r# r$ r' r( r+ r. r/ r r r r �<module> sP \ +, ,
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка