Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/extension/__pycache__/test_sparse.cpython-37.pyc
Назад
B �5�g ? � @ sB d Z ddlZddlZddlmZ ddlmZ ddlZ ddlm Z ddl m Z ddlmZ ddlmZ dd � Zejd d � �Zejdejgd �d d� �Zejdd� �Zejdejgd �dd� �Zejdejgd �dd� �Zejdejgd �dd� �Zejdejgd �dd� �Zejdd� �Z ejdd � �Z ejdejgd �d d � �Z G d d � d �Z G d!d"� d"e ej �Z!G d#d$� d$e ej"�Z#G d%d&� d&e ej$�Z%G d'd(� d(e ej&�Z'G d)d*� d*e ej(�Z)G d+d,� d,e ej*�Z+G d-d.� d.e ej,�Z-G d/d0� d0e ej.�Z/G d1d2� d2e ej0�Z1G d3d4� d4e ej2�Z3G d5d6� d6e ej4�Z5G d7d8� d8e ej6�Z7dS )9aj This file contains a minimal set of tests for compliance with the extension array interface test suite, and should contain no other tests. The test suite for the full functionality of the array is located in `pandas/tests/arrays/`. The tests in this file are inherited from the BaseExtensionTests, and only minimal tweaks should be applied to get the tests passing (by overwriting a parent method). Additional tests should either be added to one of the BaseExtensionTests classes (if they are relevant for the extension interface for all dtypes), or be added to the array-specific tests in `pandas/tests/arrays/`. � N)�PerformanceWarning)�is_object_dtype)� SparseDtype)� SparseArray)�basec C s^ t �| �rt jjdd�}n2t jjdddd�}|d |d krL|d d7 < | |dd d�< |S )N�d )�size� r � � )�np�isnan�random�uniform�randint)� fill_value�data� r �X/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/extension/test_sparse.py� make_data s r c C s t � S )N)r r r r r �dtype+ s r )�paramsc C s t t| j�| jd�}|S )z*Length-100 PeriodArray for semantics test.)r )r r �param)�request�resr r r r 0 s r c C s t t�d�d �S )Nr r )r r Zones)r r r r � data_for_twos7 s r c C s t tjdg| jd�S )z Length 2 array with [NA, Valid]r )r )r r �nanr )r r r r � data_missing< s r c # s � fdd�}|V dS )z1Return different versions of data for count timesc 3 s, x&t | �D ]}tt� j�� jd�V q W d S )N)r )�ranger r r )�count�_)r r r �genF s zdata_repeated.<locals>.genNr )r r! r )r r � data_repeatedB s r" c C s t dddg| jd�S )Nr r r )r )r r )r r r r �data_for_sortingM s r# c C s t dtjdg| jd�S )Nr r )r )r r r r )r r r r �data_missing_for_sortingR s r$ c C s t jS )N)r r r r r r �na_valueW s r% c C s dd� S )Nc S s t �| �ot �|�S )N)�pd�isna)�left�rightr r r �<lambda>^ � zna_cmp.<locals>.<lambda>r r r r r �na_cmp\ s r, c C s" t ddtjtjddddg| jd�S )Nr r r )r )r r r r )r r r r �data_for_groupinga s r- c s2 e Zd Zdd� Zejjdd�� fdd��Z� ZS )�BaseSparseTestsc C s |j ttd�krt�d� d S )Nr z Can't store nan in int array.)r r �int�pytest�skip)�selfr r r r �_check_unsupportedg s z"BaseSparseTests._check_unsupportedz$SparseArray does not support setitem)�reasonc s t � �|� d S )N)�super� test_ravel)r2 r )� __class__r r r6 k s zBaseSparseTests.test_ravel) �__name__� __module__� __qualname__r3 r0 �mark�xfailr6 � __classcell__r r )r7 r r. f s r. c @ s e Zd Zdd� ZdS )� TestDtypec C s |� � tkst�d S )N)Zconstruct_array_typer �AssertionError)r2 r r r r r �test_array_type_with_argq s z"TestDtype.test_array_type_with_argN)r8 r9 r: r@ r r r r r> p s r> c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� TestInterfacec C s t �d� d S )NzWe have values)r0 r1 )r2 r r r r �test_no_values_attributev s z&TestInterface.test_no_values_attributec C s |� � d S )N)�copy)r2 r r r r � test_copyy s zTestInterface.test_copyc C s |� � d S )N)�view)r2 r r r r � test_view} s zTestInterface.test_viewN)r8 r9 r: rB rD rF r r r r rA u s rA c @ s e Zd ZdS )�TestConstructorsN)r8 r9 r: r r r r rG � s rG c sz e Zd Zdd� Z� fdd�Z� fdd�Z� fdd�Z� fd d �Z� fd d �Z� fd d�Z e j j dd�� fdd��Z � ZS )� TestReshapingc C s| t �d|d d� i�}t �ddddgi�}t �ddddgi��d�}|||g}t �|�}t �d d � |D ��}| �||� d S ) N�Ar r r �a�b�c�categoryc S s g | ]}|� d d� ��qS )c S s t �| ��t�S )N)r �asarray�astype�object)�sr r r r* � r+ zCTestReshaping.test_concat_mixed_dtypes.<locals>.<listcomp>.<lambda>)�apply)�.0�xr r r � <listcomp>� s z:TestReshaping.test_concat_mixed_dtypes.<locals>.<listcomp>)r& � DataFramerO �concat�assert_frame_equal)r2 r Zdf1Zdf2Zdf3�dfs�result�expectedr r r �test_concat_mixed_dtypes� s z&TestReshaping.test_concat_mixed_dtypesc s | � |� t� �||� d S )N)r3 r5 �test_concat_columns)r2 r r% )r7 r r r] � s z!TestReshaping.test_concat_columnsc s | � |� t� �||� d S )N)r3 r5 �'test_concat_extension_arrays_copy_false)r2 r r% )r7 r r r^ � s z5TestReshaping.test_concat_extension_arrays_copy_falsec s | � |� t� �||� d S )N)r3 r5 � test_align)r2 r r% )r7 r r r_ � s zTestReshaping.test_alignc s | � |� t� �||� d S )N)r3 r5 �test_align_frame)r2 r r% )r7 r r r` � s z TestReshaping.test_align_framec s | � |� t� �||� d S )N)r3 r5 �test_align_series_frame)r2 r r% )r7 r r ra � s z%TestReshaping.test_align_series_framec s | � |� t� �||� d S )N)r3 r5 � test_merge)r2 r r% )r7 r r rb � s zTestReshaping.test_mergez$SparseArray does not support setitem)r4 c s t � �|� d S )N)r5 �test_transpose)r2 r )r7 r r rc � s z TestReshaping.test_transpose)r8 r9 r: r\ r] r^ r_ r` ra rb r0 r; r< rc r= r r )r7 r rH � s rH c s$ e Zd Zdd� Z� fdd�Z� ZS )� TestGetitemc C s� t j|dd� tt|��D �d�}t�|jj�rTt�|�d��rNt�|j d �slt �n|�d�|j d kslt �|�d�|j d ks�t �d S )Nc S s g | ] }d | �qS )r r )rS �ir r r rU � s z(TestGetitem.test_get.<locals>.<listcomp>)�index� r r ) r& �Seriesr �lenr r �valuesr �get�ilocr? )r2 r rQ r r r �test_get� s &zTestGetitem.test_getc s | � |� t� �||� d S )N)r3 r5 � test_reindex)r2 r r% )r7 r r rn � s zTestGetitem.test_reindex)r8 r9 r: rm rn r= r r )r7 r rd � s rd c sf e Zd Zdd� Z� fdd�Z� fdd�Z� fdd�Z� fd d �Zej j d d �d d� �Z dd� Z � Z S )� TestMissingc C s� t tt�|jj��}tddg|d�}t�|�}| �||� t�|��� }t�|�}| � ||� t�|�� ddg��� }tjg |d�}| � ||� d S )NTF)r r r ) r �boolr& r' r r r Z assert_equalrh �assert_series_equalZdrop)r2 r �expected_dtyper[ rZ r r r � test_isna� s zTestMissing.test_isnac s& t �t�� t� �|� W d Q R X d S )N)�tm�assert_produces_warningr r5 �test_fillna_limit_pad)r2 r )r7 r r rv � s z!TestMissing.test_fillna_limit_padc s& t �t�� t� �|� W d Q R X d S )N)rt ru r r5 �test_fillna_limit_backfill)r2 r )r7 r r rw � s z&TestMissing.test_fillna_limit_backfillc sH t �|j�r"|j�tjjdd�� t� t �� t � � |� W d Q R X d S )Nz'returns array with different fill value)r4 ) r r r �node� add_markerr0 r; r< rt ru r r5 � test_fillna_no_op_returns_copy)r2 r r )r7 r r rz � s z*TestMissing.test_fillna_no_op_returns_copyc s& t �t�� t� �|� W d Q R X d S )N)rt ru r r5 rw )r2 r )r7 r r �test_fillna_series_method� s z%TestMissing.test_fillna_series_methodZ Unsupported)r4 c C s d S )Nr )r2 r r r �test_fillna_series� s z TestMissing.test_fillna_seriesc C st |d }t �|ddgd���|�}t �|j�r<t|j|�}n|j}t �|j||g|d�ddgd��}| �||� d S )Nr r )rI �B)r ) r& rV �fillnar' r r r �_from_sequencerX )r2 r r rZ r r[ r r r �test_fillna_frame� s z TestMissing.test_fillna_frame)r8 r9 r: rs rv rw rz r{ r0 r; r1 r| r� r= r r )r7 r ro � s ro c s� e Zd Zdd� Zdd� Zdd� Zejjdd�d d � �Z d d � Z � fd d�Z � fdd�Z dd� Z ej�dddg�� fdd��Zej�dejejejg�� fdd��Z� ZS )� TestMethodsc s� |d�\}}t �|�}t �|�}|�|dd� �}t �tdd� tt|�t|��D �dd��}| �||� |jd � |�� d d� �}t �t� fd d�t|�D �dd��}| �||� d S ) Nr c S s | |kS )Nr )�x1�x2r r r r* r+ z-TestMethods.test_combine_le.<locals>.<lambda>c S s g | ]\}}||k�qS r r )rS rJ rK r r r rU s z/TestMethods.test_combine_le.<locals>.<listcomp>F)r r c S s | |kS )Nr )r� r� r r r r* r+ c s g | ] }|� k�qS r r )rS rJ )�valr r rU s )r& rh �combiner �zip�listrq rl )r2 r" Z orig_data1Z orig_data2�s1�s2rZ r[ r )r� r �test_combine_le s zTestMethods.test_combine_lec C sp |� ddg�}tjd|idd�}|jd }|�|�}t|jd�rT|jj|jjk sTt �|j j � � |� � kslt �d S )Nr rI F)rC )r r �blocks) �taker& rV rl r~ �hasattr�_mgrrj r r? rI �_values�to_dense)r2 r �arr�df� filled_valrZ r r r �test_fillna_copy_frame s z"TestMethods.test_fillna_copy_framec C sT |� ddg�}t�|�}|d }|�|�}|j|jk s:t�|j�� |�� ksPt�d S )Nr r )r� r& rh r~ r� r? r� )r2 r r� �serr� rZ r r r �test_fillna_copy_series& s z#TestMethods.test_fillna_copy_serieszNot Applicable)r4 c C s d S )Nr )r2 r r r r �test_fillna_length_mismatch0 s z'TestMethods.test_fillna_length_mismatchc C s� |d |d kst �t|�}|d d� \}}t�|j||||g|jd��}t�ddddg�}|�|�}t dd�} t�|j||||g| d��} | � || � |j||||g|jd�} t�ddddg�}|�|| �}t�|j||||g|jd��} | � || � d S ) Nr r r )r TF�floatg ) r? �typer& rh r r r �array�wherer rq ) r2 r r% �clsrJ rK r� ZcondrZ Z new_dtyper[ �otherr r r �test_where_series4 s z TestMethods.test_where_seriesc s& |j jdkrt�d� t� �|� d S )Nr/ z1TODO(SparseArray.__setitem__ will preserve dtype.)r �subtyper0 r1 r5 �test_combine_first)r2 r )r7 r r r� J s z TestMethods.test_combine_firstc s( t �t�� t� �||� W d Q R X d S )N)rt ru r r5 �test_searchsorted)r2 r# � as_series)r7 r r r� Q s z TestMethods.test_searchsortedc C s6 |� d�}|jd |jd< |jd |jd ks2t�d S )Nr r )�shiftZ_sparse_valuesr? )r2 r rZ r r r �test_shift_0_periodsU s z TestMethods.test_shift_0_periods�methodZargmaxZargminc s | � |� t� �|||� d S )N)r3 r5 �test_argmin_argmax_all_na)r2 r� r r% )r7 r r r� \ s z%TestMethods.test_argmin_argmax_all_na�boxc s | � |� t� �||||� d S )N)r3 r5 � test_equals)r2 r r% r� r� )r7 r r r� b s zTestMethods.test_equals)r8 r9 r: r� r� r� r0 r; r1 r� r� r� r� r� � parametrizer� r& r� rh rV r� r= r r )r7 r r� s r� c sD e Zd Zdd� Zdd� Zdd� Zejje dd�� fd d ��Z � Z S ) � TestCastingc C s< t j|dd�}|�t�}t|j�s&t�t|jjj�s8t�d S )NrI )�name) r& rh rO rP r r r? r� r� )r2 �all_datar� rZ r r r �test_astype_object_seriesi s z%TestCasting.test_astype_object_seriesc C s2 t �d|i�}|�t�}t|jjd j�s.t�d S )NrI r ) r& rV rO rP r r� Zarraysr r? )r2 r� r� rZ r r r �test_astype_object_frameq s z$TestCasting.test_astype_object_framec C sX t �|d d� ��t�}ttt|j��}t jdd� |d d� D �|d�}| �||� d S )N� c S s g | ] }t |��qS r )�str)rS rT r r r rU � s z/TestCasting.test_astype_str.<locals>.<listcomp>)r )r& rh rO r� r r rq )r2 r rZ rr r[ r r r �test_astype_str~ s zTestCasting.test_astype_strzno sparse StringDtype)�raisesr4 c s t � �|� d S )N)r5 �test_astype_string)r2 r )r7 r r r� � s z TestCasting.test_astype_string) r8 r9 r: r� r� r� r0 r; r< � TypeErrorr� r= r r )r7 r r� h s r� c sL e Zd ZdZdZdZdZdd� Z� fdd�Z� fdd�Z � fdd �Z � Z S ) �TestArithmeticOpsNc C s |j dkrt�d��d S )Nr z(Incorrected expected from Series.combine)r r0 r1 )r2 r r r r �_skip_if_different_combine� s z,TestArithmeticOps._skip_if_different_combinec s | � |� t� �||� d S )N)r� r5 � test_arith_series_with_scalar)r2 r �all_arithmetic_operators)r7 r r r� � s z/TestArithmeticOps.test_arith_series_with_scalarc s | � |� t� �||� d S )N)r� r5 � test_arith_series_with_array)r2 r r� )r7 r r r� � s z.TestArithmeticOps.test_arith_series_with_arrayc sH |j jdkrn(|�d�dkr6tjjdd�}|j�|� t� � ||� d S )Nr r )�mulZrmul�floordivZ rfloordiv�pow�modZrmodz result dtype.fill_value mismatch)r4 ) r r �stripr0 r; r< rx ry r5 � test_arith_frame_with_scalar)r2 r r� r r; )r7 r r r� � s z.TestArithmeticOps.test_arith_frame_with_scalar) r8 r9 r: Zseries_scalar_excZframe_scalar_excZ divmod_excZseries_array_excr� r� r� r� r= r r )r7 r r� � s r� c @ s e Zd Zdd� ZdS )�TestComparisonOpsc C s� | � |�}t�|||��}t|jt�s*t�|jjt�d�ks@t�tj dd��. t�t |t� |�t� |��|j j d��}W d Q R X t�||� t�|�}|||�}t�||� d S )Nrp �ignore)�all)r )Zget_op_from_namer& rh � isinstancer r r? r� r Zerrstater rN rj r rt rq )r2 rQ r Zop_namer� �oprZ r[ r r r �_compare_other� s z TestComparisonOps._compare_otherN)r8 r9 r: r� r r r r r� � s r� c s* e Zd Zejjdd�� fdd��Z� ZS )� TestPrintingzDifferent repr)r4 c s t � �||� d S )N)r5 �test_array_repr)r2 r r )r7 r r r� � s z TestPrinting.test_array_repr)r8 r9 r: r0 r; r< r� r= r r )r7 r r� � s r� c s. e Zd Zej�dddg�� fdd��Z� ZS )� TestParsing�enginerL �pythonc s0 d}t jt|d�� t� �||� W d Q R X d S )Nz,.*must implement _from_sequence_of_strings.*)�match)r0 r� �NotImplementedErrorr5 � test_EA_types)r2 r� r Z expected_msg)r7 r r r� � s zTestParsing.test_EA_types)r8 r9 r: r0 r; r� r� r= r r )r7 r r� � s r� )8�__doc__�numpyr r0 Z pandas.errorsr Zpandas.core.dtypes.commonr Zpandasr& r Zpandas._testingZ_testingrt Z pandas.arraysr Zpandas.tests.extensionr r Zfixturer r r r r r" r# r$ r% r, r- r. ZBaseDtypeTestsr> ZBaseInterfaceTestsrA ZBaseConstructorsTestsrG ZBaseReshapingTestsrH ZBaseGetitemTestsrd ZBaseMissingTestsro ZBaseMethodsTestsr� ZBaseCastingTestsr� ZBaseArithmeticOpsTestsr� ZBaseComparisonOpsTestsr� ZBasePrintingTestsr� ZBaseParsingTestsr� r r r r �<module> sB -?e!&
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка