Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/extension/base/__pycache__/methods.cpython-37.pyc
Назад
B �5�g>O � @ sl d dl Z d dlZd dlZd dlZd dlmZ d dlZd dl m Z d dl m Z d dlmZ G dd� de�ZdS )� N)� is_bool_dtype)�nargsort)�BaseExtensionTestsc @ s� e Zd ZdZdd� Zej�dddg�dd� �Zd d � Z d d � Z d d� Z dd� Z dd� Z dd� Zdd� Zdd� Zej�dddg�d d � �Zej�dddg�d d � �Zej�d d!d"d#d$d%dejfd&dejfd'd(g�d)d*� �Zd+d,� Zej�d-d.ejd/d0d1ge�d2�d3�fd4ejd1d/d0ge�d2�d3�fg�d5d6� �Zej�d7ddg�d8d9� �Zej�d7ddg�d:d;� �Zej�d7ddg�d<d=� �Z ej�d>e j d?d@� g�ej�ddAd@� e j g�dBdC� ��Z ej�dDdEdFg�dGdH� �Z!ej�dDdEdFg�dIdJ� �Z"dKdL� Z#dMdN� Z$dOdP� Z%dQdR� Z&dSdT� Z'dUdV� Z(dWdX� Z)ej�dYddg�ej�dZdFd/d[d\dEdEgfd0d0d1d/d[d\gfd/dEdEd0d1d/gfg�d]d^� ��Z*d_d`� Z+ej�dad1dFg�dbdc� �Z,ej�dZdddEdEggdEd1dEggd0d0d1ggd1dEd0ggd\dEdEggg�dedf� �Z-ej�dadddEd0d1d\g�dgdh� �Z.didj� Z/dkdl� Z0dmdn� Z1dodp� Z2dqdr� Z3dsdt� Z4ej�dud0d1d/d1d/d[gg�dvdw� �Z5ej�dxd/dyd1ie6dyfdEi e6dzfd1d/gi e6d{fd/d|d}ie7d~fg�dd�� �Z8d�d�� Z9ej�d>e je j e j:g�d�d�� �Z;d�S )��BaseMethodsTestsz%Various Series and DataFrame methods.c C s< t |d�st�d� t�|j�}|jd }|jdks8t�d S )N� value_countsz value_counts is not implemented�dropnaT) �hasattr�pytest�skip�inspect� signaturer � parameters�default�AssertionError)�self�data�sig�kwarg� r �Y/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/extension/base/methods.py� test_value_counts_default_dropna s z1BaseMethodsTests.test_value_counts_default_dropnar TFc C sf |d d� }|r&t �||�� �}n|}t�|�j|d��� }t�|�j|d��� }| �||� d S )N� )r )�np�array�isna�pd�Seriesr � sort_index�assert_series_equal)r Zall_datar �other�result�expectedr r r �test_value_counts s z"BaseMethodsTests.test_value_countsc C s� |d d� � � }t�||�� �}tj||jd�}|jdd��� }t |tj �sttjdt |� gt |� |j d�}n$tjd|j d�}dt |� ||dk< | � ||� d S ) Nr )�dtypeT)� normalize� )�indexg r )�uniquer r r r r r# r r � isinstanceZ Categorical�lenr& r )r r �values�serr r! r r r � test_value_counts_with_normalize' s $z1BaseMethodsTests.test_value_counts_with_normalizec C s8 t �d|i�}|jdd�}t �ddg�}| �||� d S )N�A�columns)�axisr r% )r � DataFrame�countr r )r � data_missing�dfr r! r r r � test_count7 s zBaseMethodsTests.test_countc C s& t �|�}|�� }d}||ks"t�d S )Nr% )r r r1 r )r r2 r+ r r! r r r �test_series_count= s z"BaseMethodsTests.test_series_countc C s$ t �|��t�}t|t j�s t�d S )N)r r �apply�idr( r )r r r r r r �test_apply_simple_seriesD s z)BaseMethodsTests.test_apply_simple_seriesc C s: t �|��� }t �tjdddgtjd��}| �||� d S )N� r r% )r# )r r �argsortr r �int64r )r �data_for_sortingr r! r r r � test_argsortH s z BaseMethodsTests.test_argsortc C sN |� � }tjdddgt�d�d�}|jddd�}|jddd�}t�||� d S ) Nr9 r r% �int)r# r; �safe)Zcasting)r: r r r# Zastype�tm�assert_numpy_array_equal)r �data_missing_for_sortingr r! r r r �test_argsort_missing_arrayM s z+BaseMethodsTests.test_argsort_missing_arrayc C s: t �|��� }t �tjdddgtjd��}| �||� d S )Nr% ���r )r# )r r r: r r r; r )r rB r r! r r r �test_argsort_missingU s z%BaseMethodsTests.test_argsort_missingc C sz |� � dkst�|�� dks t�|�ddddddg�}|� � dksFt�|�� dksVt�|� � dksft�|�� dksvt�d S )Nr% r9 r � )�argmaxr �argmin�take)r r< rB �na_valuer r r r �test_argmin_argmaxZ s z#BaseMethodsTests.test_argmin_argmax�methodrG rH c C s6 d}t jt|d�� t|d d� |�� W d Q R X d S )Nzattempt to get)�matchr )r �raises� ValueError�getattr)r rL r �err_msgr r r � test_argmin_argmax_empty_arrayk s z/BaseMethodsTests.test_argmin_argmax_empty_arrayc C sF d}t |�j||g|jd�}tjt|d�� t||�� W d Q R X d S )Nzattempt to get)r# )rM )�type�_from_sequencer# r rN rO rP )r rL r rJ rQ �data_nar r r �test_argmin_argmax_all_nar s z*BaseMethodsTests.test_argmin_argmax_all_nazop_name, skipna, expected)�idxmaxTr )�idxminTr9 )rG Tr )rH Tr9 rW rX )rG FrD )rH FrD c C s* t �|�}t||�|d�}t�||� d S )N)�skipna)r r rP r@ Zassert_almost_equal)r rB Zop_namerY r! r+ r r r r �test_argreduce_seriesz s z&BaseMethodsTests.test_argreduce_seriesc C sT |}t jtdd�� |jdd� W d Q R X t jtdd�� |jdd� W d Q R X d S )N� )rM F)rY )r rN �NotImplementedErrorrH rG )r rB r r r r �+test_argmax_argmin_no_skipna_notimplemented� s z<BaseMethodsTests.test_argmax_argmin_no_skipna_notimplementedzna_position, expected�lastr9 r r% �intp)r# �firstc C s t ||d�}t�||� d S )N)� na_position)r r@ rA )r rB ra r! r r r r � test_nargsort� s z BaseMethodsTests.test_nargsort� ascendingc C sj t �|�}|j||d�}|jdddg }|sZ|�� dkrJ|jdddg }n|jdddg }| �||� d S )N)rc �keyr9 r r% )r r � sort_values�ilocZnuniquer )r r< rc � sort_by_keyr+ r r! r r r �test_sort_values� s z!BaseMethodsTests.test_sort_valuesc C sN t �|�}|j||d�}|r.|jdddg }n|jdddg }| �||� d S )N)rc rd r9 r r% )r r re rf r )r rB rc rg r+ r r! r r r �test_sort_values_missing� s z)BaseMethodsTests.test_sort_values_missingc C s` t �dddg|d��}|�ddg�}t jdddg|�dddg�d�dddgd�}| �||� d S )Nr% r9 )r- �Br- rj r )r& )r r0 re rI �assert_frame_equal)r r< rc r3 r r! r r r �test_sort_values_frame� s (z'BaseMethodsTests.test_sort_values_frame�boxc C s | S )Nr )�xr r r �<lambda>� � zBaseMethodsTests.<lambda>c C s | � � S )N)r' )rn r r r ro � rp c C s\ ||� |d |d g��}||�}t|�dks2t�t|t|��sDt�|d |d ksXt�d S )Nr r% )rT r) r r( rS )r r rm rL Z duplicatedr r r r � test_unique� s z BaseMethodsTests.test_unique� na_sentinelrD ���c C s^ t j||d�\}}tjdd||ddddgtjd�}|�dddg�}t�||� | �||� d S )N)rr r r% r9 )r# � � ) r � factorizer r r_ rI r@ rA � assert_extension_array_equal)r �data_for_groupingrr �codes�uniques�expected_codes�expected_uniquesr r r �test_factorize� s z BaseMethodsTests.test_factorizec C sh t j||d�\}}|j|d�\}}t�||� | �||� t|�tt �|��ksTt�|j|jksdt�d S )N)rr ) r rv r@ rA rw r) r' r r# )r rx rr Zcodes_1Z uniques_1Zcodes_2Z uniques_2r r r �test_factorize_equivalence� s z+BaseMethodsTests.test_factorize_equivalencec C s^ t �|d d� �\}}tjg tjd�}t|�jg |d d� jd�}t� ||� | � ||� d S )Nr )r# ) r rv r r r_ rS rT r# r@ rA rw )r r ry rz r{ r| r r r �test_factorize_empty� s z%BaseMethodsTests.test_factorize_emptyc C sH |� ddg�}t�d|i�}|jd }|�|�}|jj|jjk sDt�d S )Nr% r- )r r )rI r r0 rf �fillnar- r* r )r r2 �arrr3 � filled_valr r r r �test_fillna_copy_frame� s z'BaseMethodsTests.test_fillna_copy_framec C sL |� ddg�}t�|�}|d }|�|�}|j|jk s:t�|j|ksHt�d S )Nr% r )rI r r r� Z_valuesr )r r2 r� r+ r� r r r r �test_fillna_copy_series� s z(BaseMethodsTests.test_fillna_copy_seriesc C s4 d}t jt|d�� |�|�dg�� W d Q R X d S )Nz!Length of 'value' does not match.)rM r% )r rN rO r� rI )r r2 �msgr r r �test_fillna_length_mismatch s z,BaseMethodsTests.test_fillna_length_mismatchc s� |d�\}}t �|�}t �|�}|�|dd� �}t �dd� tt|�t|��D ��}| �||� |jd � |�� dd� �}t �� fdd�t|�D ��}| �||� d S ) Nr9 c S s | |kS )Nr )�x1�x2r r r ro rp z2BaseMethodsTests.test_combine_le.<locals>.<lambda>c S s g | ]\}}||k�qS r r )�.0�a�br r r � <listcomp> s z4BaseMethodsTests.test_combine_le.<locals>.<listcomp>r c S s | |kS )Nr )r� r� r r r ro rp c s g | ] }|� k�qS r r )r� r� )�valr r r� s )r r �combine�zip�listr rf )r � data_repeated� orig_data1� orig_data2�s1�s2r r! r )r� r �test_combine_le s z BaseMethodsTests.test_combine_lec s� |d�\}}t �|�}t �|�}|�|dd� �}tjdd��. t �|�dd� tt|�t|��D ���}W d Q R X | �||� |j d � |�� d d� �}t �|�� fd d�t|�D ���}| �||� d S ) Nr9 c S s | | S )Nr )r� r� r r r ro rp z3BaseMethodsTests.test_combine_add.<locals>.<lambda>�ignore)Zoverc S s g | ]\}}|| �qS r r )r� r� r� r r r r� " s z5BaseMethodsTests.test_combine_add.<locals>.<listcomp>r c S s | | S )Nr )r� r� r r r ro ( rp c s g | ] }|� �qS r r )r� r� )r� r r r� * s ) r r r� r ZerrstaterT r� r� r rf )r r� r� r� r� r� r r! r )r� r �test_combine_add s * z!BaseMethodsTests.test_combine_addc C sZ t �|d d� �}t j|dd� dddgd�}|�|�}t �|d d� �}| �||� d S )NrF r9 � rt )r& )r r Z combine_firstr )r r r� r� r r! r r r �test_combine_first. s z#BaseMethodsTests.test_combine_first�framezperiods, indicesrF rt c C s� |d d� }t j|dd�}t j|j|dd�dd�}|r�|jdd�jdd��|�}t j|t jdgd dd��|�gdd �}| j}n|�|�}| j}|||� d S ) Nr� r- )�nameT)� allow_fillr% )rj rj )r/ ) r r rI �to_frameZassign�shift�concatrk r ) r r r� �periods�indices�subsetr! r �comparer r r �test_container_shift6 s $ z%BaseMethodsTests.test_container_shiftc C sB |� d�}|d |d ks t�|d |d< |d |d ks>t�d S )Nr r% )r� r )r r r r r r �test_shift_0_periodsM s z%BaseMethodsTests.test_shift_0_periodsr� c C s� |d d� }t |j�r tj}ntj}y|||� W n( tk rZ t�t|�� d�� Y nX t � |�}|� |�}t � |||� |���}| � ||� t �|dgd d��}|� |�}|dkr�tjddddg}ndddtjtjg}t �||d��}| �||� d S )Nr� z does not support diffg �?)r- rj r% r )r r# �operator�xor�sub� Exceptionr r rS r r �diffr� r r0 r �nanrk ) r r r� �op�sr r! r3 r� r r r � test_diffT s& zBaseMethodsTests.test_diff���c C s4 |d d� }|� |�}|j|dd�}| �||� d S )Nr9 T)r� )r� rI rw )r r r� r� r� r r! r r r �test_shift_non_empty_arrayn s z+BaseMethodsTests.test_shift_non_empty_arrayc C s* |d d� }|� |�}|}| �||� d S )Nr )r� rw )r r r� �emptyr r! r r r �test_shift_empty_arrayy s z'BaseMethodsTests.test_shift_empty_arrayc C s8 |� d�}||k st�|d d� � d�}||k s4t�d S )Nr r9 )r� r )r r r r r r �test_shift_zero_copies� s z'BaseMethodsTests.test_shift_zero_copiesc C sp |d d� }|d }|j d|d�}|�ddddg�}| �||� |j d|d�}|�ddddg�}| �||� d S )Nrt r r% )� fill_valuer9 rs rF )r� rI rw )r r r� r� r r! r r r �test_shift_fill_value� s z&BaseMethodsTests.test_shift_fill_valuec C s& t jtdd�� t|� W d Q R X d S )Nzunhashable type)rM )r rN � TypeError�hash)r r r r r �test_not_hashable� s z"BaseMethodsTests.test_not_hashablec C s> t �|�}|r|�� }t j�|�}t j�|�}| �||� d S )N)r r r� �utilZhash_pandas_object� assert_equal)r r �as_framer� r� r r r � test_hash_pandas_object_works� s z.BaseMethodsTests.test_hash_pandas_object_worksc C s� |\}}}|� dddg�}|r(t�|�}|�|�dks:t�|j|dd�dksPt�|�|�dksbt�|j|dd�dksxt�|�|�dks�t�|j|dd�dks�t�|�|� ddg��}tjddgtjd�}t� ||� t�dddg�} |j|| d�dks�t�d S ) Nr9 r r% �right)ZsiderF )r# )�sorter) rI r r Z searchsortedr r r r_ r@ rA ) r r< � as_seriesr� �cr� r� r r! r� r r r �test_searchsorted� s z"BaseMethodsTests.test_searchsortedc C sT |d |d kst �t|�}|d d� \}}t�|j||||g|jd��}t�ddddg�}|rx|jdd�}|� d d�}|� |�} t�|j||||g|jd��} |r�| jdd�} | � | | � t�ddddg�}|j||||g|jd�} |�rt� d| i�} t� d|i�}|� || �} t�|j||||g|jd��} |�rD| jdd�} | � | | � d S ) Nr r% r9 )r# TFr� )r� rD ) r rS r r rT r# r r r� Zreshape�wherer� r0 ) r r rJ r� �clsr� r� r+ Zcondr r! r r r r �test_where_series� s0 z"BaseMethodsTests.test_where_series�repeatsc C s� t |�j|d d� |jd�}|r*t�|�}|r:t�||�n|�|�}t|t�rX|gd n|}dd� t ||�D �}t |�j||jd�}|r�tj||j �|�d�}| � ||� d S )NrF )r# c S s" g | ]\}}t |�D ]}|�qqS r )�range)r� rn �n�_r r r r� � s z0BaseMethodsTests.test_repeat.<locals>.<listcomp>)r& ) rS rT r# r r r �repeatr( r> r� r& r� )r r r� r� � use_numpyr� r r! r r r � test_repeat� s z BaseMethodsTests.test_repeatzrepeats, kwargs, error, msgr/ �negative�shapeZfoo�barz'foo'c C sB t j||d��* |r&tj||f|� n|j|f|� W d Q R X d S )N)rM )r rN r r� )r r r� �kwargs�errorr� r� r r r �test_repeat_raises� s z#BaseMethodsTests.test_repeat_raisesc C sd |� d�}|dd � }| �||� |� ddg�}|�|dg |dg |dd � g�}| �||� d S )Nr r% rF r9 rt )�deleterw Z_concat_same_type)r r r r! r r r � test_delete� s $z BaseMethodsTests.test_deletec C s8 t |�j|d gt|� |jd�}t |�j|gt|� |jd�}tj||dd�}tj||dd�}tj||dd�}|�|�dks�t�|�|�� �dks�t�|�|�dks�t�|�|�dks�t�|d d� �|d d� �dks�t�|d d� �|d d� �dk�st�|�d �dk�st�|dg �|d �dk�s4t�d S )Nr )r# F)Z transposeTr9 rF ) rS rT r) r# r@ Z box_expected�equalsr �copy)r r rJ r� rm Zdata2rU r r r � test_equals s " "$z BaseMethodsTests.test_equalsN)<�__name__� __module__� __qualname__�__doc__r r �markZ parametrizer" r, r4 r5 r8 r= rC rE rK rR rV r r� rZ r] r r# rb rh ri rl r r r' rq r} r~ r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rO r� r� r� r0 r� r r r r r s� * " @@ !$ r )r r� �numpyr r Zpandas.core.dtypes.commonr Zpandasr Zpandas._testingZ_testingr@ Zpandas.core.sortingr Z pandas.tests.extension.base.baser r r r r r �<module> s
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка