Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/extension/__pycache__/test_categorical.cpython-37.pyc
Назад
B �5�gf% � @ s� d Z ddlZddlZddlZddlZddlmZmZm Z ddl m Z ddl mZ ddlmZ dd� Zejdd � �Zejd d � �Zejd d � �Zejdd� �Zejdd� �Zejdd� �Zejdd� �ZG dd� dej�ZG dd� dej �Z G dd� dej �Z G d d � d ej �Z!G d d � d ej"�Z#G d d!� d!ej$�Z%G d"d#� d#ej&�Z'G d$d%� d%ej(�Z)G d&d'� d'ej*�Z+G d(d)� d)ej,�Z-G d*d+� d+ej.�Z/G d,d-� d-ej0�Z1G d.d/� d/ej2�Z3dS )0aj 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)� Categorical�CategoricalIndex� Timestamp)�CategoricalDtype)�basec C s2 x,t jjttj�dd�} | d | d krP qW | S )N�d )�sizer � )�np�random�choice�list�string� ascii_letters)�values� r �]/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/extension/test_categorical.py� make_data s r c C s t � S )N)r r r r r �dtype+ s r c C s t t� �S )z�Length-100 array for this type. * data[0] and data[1] should both be non missing * data[0] and data[1] should not gbe equal )r r r r r r �data0 s r c C s t tjdg�S )z Length 2 array with [NA, Valid]�A)r r �nanr r r r � data_missing: s r c C s t dddgdddgdd�S )Nr �B�CT)� categories�ordered)r r r r r �data_for_sorting@ s r c C s t dd dgddgdd�S )Nr r T)r r )r r r r r �data_missing_for_sortingE s r c C s t jS )N)r r r r r r �na_valueJ s r c C s t ddd d ddddg�S )N�a�b�c)r r r r r �data_for_groupingO s r# c @ s e Zd ZdS )� TestDtypeN)�__name__� __module__� __qualname__r r r r r$ T s r$ c s2 e Zd Zejjdd�� fdd��Zdd� Z� ZS )� TestInterfacezMemory usage doesn't match)�reasonc s t � �|� d S )N)�super�test_memory_usage)�selfr )� __class__r r r+ Y s z TestInterface.test_memory_usagec C s� |j j}||�� }|d |ks&t�|d |ks6t�||ksBt�||ksNt�x0tjD ]&}||krdqV||kspt�||ksVt�qVW d S )Nr )r r Zisna�AssertionError�tmZ NULL_OBJECTS)r, r r r Z na_value_objr r r � test_contains^ s zTestInterface.test_contains) r% r& r' �pytest�mark�skipr+ r0 � __classcell__r r )r- r r( X s r( c @ s e Zd Zdd� ZdS )�TestConstructorsc C s: |� � }|jd|d�}t||�s$t�|jtg �ks6t�d S )N)� )r )Zconstruct_array_type�_empty� isinstancer. r r )r, r �cls�resultr r r � test_emptyx s zTestConstructors.test_emptyN)r% r& r' r; r r r r r5 w s r5 c @ s e Zd ZdS )� TestReshapingN)r% r& r' r r r r r< � s r< c s* e Zd Zejjdd�� fdd��Z� ZS )� TestGetitemzBackwards compatibility)r) c s t � �|� d S )N)r* �test_getitem_scalar)r, r )r- r r r> � s z TestGetitem.test_getitem_scalar)r% r& r' r1 r2 r3 r> r4 r r )r- r r= � s r= c @ s e Zd ZdS )� TestSetitemN)r% r& r' r r r r r? � s r? c sD e Zd Zejjdd�� fdd��Zejjdd�� fdd��Z� ZS )� TestMissingzNot implemented)r) c s t � �|� d S )N)r* �test_fillna_limit_pad)r, r )r- r r rA � s z!TestMissing.test_fillna_limit_padc s t � �|� d S )N)r* �test_fillna_limit_backfill)r, r )r- r r rB � s z&TestMissing.test_fillna_limit_backfill) r% r& r' r1 r2 r3 rA rB r4 r r )r- r r@ � s r@ c @ s e Zd ZdS )� TestReduceN)r% r& r' r r r r rC � s rC c sL e Zd Zejjdd�� fdd��Zdd� Zejjdd�� fdd ��Z� Z S ) � TestMethodsz Unobserved categories included)r) c s t � �||�S )N)r* �test_value_counts)r, Zall_dataZdropna)r- r r rE � s z TestMethods.test_value_countsc s� |d�\}}t �|�}t �|�}|�|dd� �}t �dd� tt|�t|��D ��}| �||� |jd � |�� dd� �}t �� fdd�t|�D ��}| �||� d S ) N� c S s | | S )Nr )�x1�x2r r r �<lambda>� � z.TestMethods.test_combine_add.<locals>.<lambda>c S s g | ]\}}|| �qS r r )�.0r r! r r r � <listcomp>� s z0TestMethods.test_combine_add.<locals>.<listcomp>r c S s | | S )Nr )rG rH r r r rI � rJ c s g | ] }|� �qS r r )rK r )�valr r rL � s )�pd�Series�combine�zipr �assert_series_equalZiloc)r, Z data_repeatedZ orig_data1Z orig_data2�s1�s2r: �expectedr )rM r �test_combine_add� s z TestMethods.test_combine_addzNot Applicablec s t � �|� d S )N)r* �test_fillna_length_mismatch)r, r )r- r r rW � s z'TestMethods.test_fillna_length_mismatch) r% r& r' r1 r2 r3 rE rV rW r4 r r )r- r rD � s rD c @ s e Zd Zej�deeg�ej�ddej ge d�e j gg�dd� ��Z ej�de jdd gd d �e jd d gd d �e je �d�e �d �gdd �e je �d d�e �dd�gdd �e jdej gdd �g�dd� �Zej�ddejdgdd �fde �e ddd�g�jfg�dd � �Zd S ) � TestCastingr9 r r �2000c C s8 ||�}d}t jttf|d�� |�t� W d Q R X d S )NzCannot (cast|convert))�match)r1 �raises� ValueError� TypeError�astype�int)r, r9 r �s�msgr r r �test_cast_nan_to_int� s z TestCasting.test_cast_nan_to_intrU �2019�2020zdatetime64[ns, UTC])r r ztimedelta64[ns]z period[A-DEC]rF �intervalZInt64c C s" |� d�� |j�}t�||� d S )N�category)r^ r r/ rR )r, rU r: r r r �%test_cast_category_to_extension_dtype� s z1TestCasting.test_cast_category_to_extension_dtypezdtype, expectedzdatetime64[ns]z 2015-01-01T00:00:00.000000000zdatetime64[ns, MET]z2015-01-01 00:00:00+0100ZMET)�tzc C s t dg��|�}||ks t�d S )Nz 2015-01-01)r r^ r. )r, r rU r: r r r �test_consistent_casting� s z#TestCasting.test_consistent_castingN)r% r& r' r1 r2 � parametrizer r r r r rN ZNaTrb rO ZPeriodZIntervalrg �arrayZ DatetimeIndexri r r r r rX � s , rX c sH e Zd Z� fdd�Z� fdd�Zdd� Zdd� Zef� fd d � Z� Z S ) �TestArithmeticOpsc s4 |}|dkr"|j �tjjdd�� t� �||� d S )N�__rmod__z/rmod never called when string is first argument)r) )�node� add_markerr1 r2 �xfailr* � test_arith_frame_with_scalar)r, r �all_arithmetic_operators�request�op_name)r- r r rq � s z.TestArithmeticOps.test_arith_frame_with_scalarc s4 |}|dkr"|j �tjjdd�� t� �||� d S )Nrm z/rmod never called when string is first argument)r) )rn ro r1 r2 rp r* � test_arith_series_with_scalar)r, r rr rs rt )r- r r ru � s z/TestArithmeticOps.test_arith_series_with_scalarc C s0 t �|�}tjtdd�� || W d Q R X d S )Nz"cannot perform|unsupported operand)rZ )rN rO r1 r[ r] )r, r Zserr r r �$test_add_series_with_extension_array s z6TestArithmeticOps.test_add_series_with_extension_arrayc C s d S )Nr )r, r r r �test_divmod_series_array s z*TestArithmeticOps.test_divmod_series_arrayc s t � j|||td�S )N)�exc)r* �_check_divmod_opr] )r, r` �op�otherrx )r- r r ry s z"TestArithmeticOps._check_divmod_op) r% r& r' rq ru rv rw �NotImplementedErrorry r4 r r )r- r rl � s rl c @ sD e Zd Zdd� Zej�dddgddged�ed �gg�d d � �Zd S ) �TestComparisonOpsc C s� | � |�}|dkr>|||�}|�|dd� �}||k�� s�t�n\|dkrr|||�}|�|dd� �}||k�� s�t�n(d}tjt|d�� |||� W d Q R X d S )N�__eq__c S s | |kS )Nr )�x�yr r r rI rJ z2TestComparisonOps._compare_other.<locals>.<lambda>�__ne__c S s | |kS )Nr )r r� r r r rI rJ z7Unordered Categoricals can only compare equality or not)rZ )Zget_op_from_namerP �allr. r1 r[ r] ) r, r` r rt r{ rz r: rU ra r r r �_compare_other s z TestComparisonOps._compare_otherr r r! r r rc rd c C s< t jddg|d�}t jddg|d�}||k}|�� s8t�d S )N���r )r r )r Z from_codesr� r. )r, r �c1�c2r: r r r �test_not_equal_with_na" s z(TestComparisonOps.test_not_equal_with_naN) r% r& r' r� r1 r2 rj r r� r r r r r} s r} c @ s e Zd ZdS )� TestParsingN)r% r& r' r r r r r� 0 s r� )4�__doc__r �numpyr r1 ZpandasrN r r r Zpandas._testingZ_testingr/ Zpandas.api.typesr Zpandas.tests.extensionr r Zfixturer r r r r r r# ZBaseDtypeTestsr$ ZBaseInterfaceTestsr( ZBaseConstructorsTestsr5 ZBaseReshapingTestsr< ZBaseGetitemTestsr= ZBaseSetitemTestsr? ZBaseMissingTestsr@ ZBaseNoReduceTestsrC ZBaseMethodsTestsrD ZBaseCastingTestsrX ZBaseArithmeticOpsTestsrl ZBaseComparisonOpsTestsr} ZBaseParsingTestsr� r r r r �<module> s: 0$
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка