Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/extension/decimal/__pycache__/test_decimal.cpython-37.pyc
Назад
B �5�g�A � @ sN d dl Z d dlZd dlZd dlZd dlZd dlZd dlm Z d dl m Z d dl mZ d dlmZmZmZmZ ejdd� �Zejdd� �Zejd d � �Zejd d � �Zejd d� �Zejdd� �Zejdd� �Zejdd� �Z ejdd� �Z G dd� d�Z G dd� de ej �Z G dd � 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&�Z+G d'd(� d(e+ej,�Z-G d)d*� d*e+ej.�Z/G d+d,� d,e ej0�Z1G d-d.� d.e ej2�Z3G d/d0� d0e ej4�Z5G d1d2� d2e ej6�Z7G d3d4� d4e ej8�Z9ej:j;d5d6�d7d8� �Z<d9d:� Z=d;d<� Z>ej:�?d=d>d?g�d@dA� �Z@G dBdC� dCe ejA�ZBG dDdE� dEe ejC�ZDG dFdG� dGe�ZEG dHdI� dIeE�ZFeF�G� dJdK� ZHej:�?dLeEeFg�dMdN� �ZIej:�?dOd?d dPdPdQgdPd dPd gfd>dQdPd d gd d dQdQgfg�dRdS� �ZJdTdU� ZKdVdW� ZLdXdY� ZMdZd[� ZNd\d]� ZOd^d_� ZPd`da� ZQdbdc� ZRddde� ZSdS )f� N)� infer_dtype)�base)� DecimalArray� DecimalDtype� make_data� to_decimalc C s t � S )N)r � r r �a/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/extension/decimal/test_decimal.py�dtype s r c C s t t� �S )N)r r r r r r �data s r c C s t dd� td�D ��S )Nc S s g | ]}t �d ��qS )� )�decimal�Decimal)�.0�_r r r � <listcomp> s z!data_for_twos.<locals>.<listcomp>�d )r �ranger r r r � data_for_twos s r c C s t t�d�t�d�g�S )N�NaN� )r r r r r r r � data_missing# s r c C s t t�d�t�d�t�d�g�S )N�1�2�0)r r r r r r r �data_for_sorting( s r c C s t t�d�t�d�t�d�g�S )Nr r r )r r r r r r r �data_missing_for_sorting/ s r c C s dd� S )Nc S s | � � o|� � S )N)�is_nan)�x�yr r r �<lambda>8 � zna_cmp.<locals>.<lambda>r r r r r �na_cmp6 s r" c C s t �d�S )Nr )r r r r r r �na_value; s r# c C s@ t �d�} t �d�}t �d�}t �d�}t| | ||||| |g�S )Nz1.0z0.0z2.0r )r r r )�b�a�c�nar r r �data_for_grouping@ s r( c @ s$ e Zd Zedd� �Zedd� �ZdS )� BaseDecimalc O sp dd� }|j dkr |�|�}n|�� }|j dkr<|�|�}n|�� }t�||� tj|| || f|�|�S )Nc S s$ y t �| �S tk r dS X d S )NF)�math�isnan� TypeError)r r r r �convertL s z0BaseDecimal.assert_series_equal.<locals>.convert�object)r �apply�isna�tm�assert_series_equal)�cls�left�right�args�kwargsr- Zleft_naZright_nar r r r2 J s z BaseDecimal.assert_series_equalc O s� t j|j|j|�dd�|�dd�|�dd�|�dd�|�dd �� d �d � |jd kj}x(|D ] }| j|| || f|�|� q\W |j|d �}|j|d �}t j||f|�|� d S )NZcheck_column_type�equiv� check_namesT� check_exactF�check_categorical�obj� DataFramez.columns)�exactr9 r: r; r< r )�columns) r1 Zassert_index_equalr? �getZdtypes�indexr2 Zdrop�assert_frame_equal)r3 r4 r5 r6 r7 �decimals�colr r r rB a s z BaseDecimal.assert_frame_equalN)�__name__� __module__� __qualname__� classmethodr2 rB r r r r r) I s r) c @ s. e Zd Zdd� Zej�dddg�dd� �ZdS ) � TestDtypec C s d S )Nr )�selfr r r r � test_hashabley s zTestDtype.test_hashable�skipnaTFc C s, t ||d�dkst�t ||d�dks(t�d S )N)rL z unknown-array)r �AssertionError)rJ r r rL r r r �test_infer_dtype| s zTestDtype.test_infer_dtypeN)rE rF rG rK �pytest�mark� parametrizerN r r r r rI x s rI c @ s e Zd ZdS )� TestInterfaceN)rE rF rG r r r r rR � s rR c @ s e Zd ZdS )�TestConstructorsN)rE rF rG r r r r rS � s rS c @ s e Zd ZdS )� TestReshapingN)rE rF rG r r r r rT � s rT c @ s e Zd Zdd� ZdS )� TestGetitemc C sZ t t�d�t�d�g�}|jddgdt�d�d�}t t�d�t�d�g�}| �||� d S )Nz1.0z2.0r ���Tz-1.0)Z allow_fillZ fill_value)r r r Ztake� assert_extension_array_equal)rJ �arr�result�expectedr r r � test_take_na_value_other_decimal� s z,TestGetitem.test_take_na_value_other_decimalN)rE rF rG r[ r r r r rU � s rU c @ s e Zd ZdS )� TestMissingN)rE rF rG r r r r r\ � s r\ c @ s e Zd Zdd� ZdS )�Reducec C sj |dkr8d}t jt|d�� t||�|d� W d Q R X n.t||�|d�}tt�|�|�� }t�||� d S )N)ZmedianZskewZkurtz)decimal does not support the .* operation)�match)rL )rO �raises�NotImplementedError�getattr�npZasarrayr1 Zassert_almost_equal)rJ �s�op_namerL �msgrY rZ r r r � check_reduce� s zReduce.check_reduceN)rE rF rG rf r r r r r] � s r] c @ s e Zd ZdS )�TestNumericReduceN)rE rF rG r r r r rg � s rg c @ s e Zd ZdS )�TestBooleanReduceN)rE rF rG r r r r rh � s rh c s6 e Zd Zej�dddg�dd� �Z� fdd�Z� ZS )� TestMethods�dropnaTFc C s� |d d� }|r&t �||�� �}n|}t�|�j|d�}t�|�j|d�}t�� �"}d|jtj < |� � }|� � } W d Q R X t � || � d S )N� )rj F) rb �arrayr0 �pd�SeriesZ value_countsr � localcontext�traps�InvalidOperationZ sort_indexr1 r2 ) rJ Zall_datarj �request�other�vcsZvcs_ex�ctxrY rZ r r r �test_value_counts� s z TestMethods.test_value_countsc s t � �|�S )N)�super� test_value_counts_with_normalize)rJ r )� __class__r r rx � s z,TestMethods.test_value_counts_with_normalize) rE rF rG rO rP rQ rv rx � __classcell__r r )ry r ri � s ri c @ s e Zd ZdS )� TestCastingN)rE rF rG r r r r r{ � s r{ c s e Zd Z� fdd�Z� ZS )� TestGroupbyc s t � �|� d S )N)rw �test_groupby_agg_extension)rJ r( )ry r r r} � s z&TestGroupby.test_groupby_agg_extension)rE rF rG r} rz r r )ry r r| � s r| c @ s e Zd ZdS )� TestSetitemN)rE rF rG r r r r r~ � s r~ c @ s e Zd Zdd� ZdS )� TestPrintingc C s2 t �|�}|jjt|�ks t�dt|�ks.t�d S )Nz Decimal: )rm rn r �name�reprrM )rJ r �serr r r �test_series_repr� s z TestPrinting.test_series_reprN)rE rF rG r� r r r r r � s r zKraising AssertionError as this is not implemented, though easy enough to do)�reasonc C s8 d} t jt| d�� tjdddgt� d� W d Q R X d S )NzQCannot cast data to extension dtype 'decimal'. Pass the extension array directly.)r^ r r r )r )rO r_ � ValueErrorrm rn r )Zxprr r r �=test_series_constructor_coerce_data_to_extension_dtype_raises� s r� c C s` t t�d�g�} tj| t� d�}t�| �}t�||� tj| dd�}t�dg�}t�||� d S )Nz10.0)r �int64rk )r r r rm rn r r1 r2 )rX rY rZ r r r �"test_series_constructor_with_dtype� s r� c C s� t t�d�g�} tjd| it� d�}t�d| i�}t�||� t t�d�g�} tjd| idd�}t�ddgi�}t�||� d S )Nz10.0�A)r r� rk )r r r rm r= r r1 rB )rX rY rZ r r r �%test_dataframe_constructor_with_dtype� s r� �frameTFc C sf t jtt�d�g�dd�}t�� }d|_| r4|�� }|�t |��}| rN|d }|j j j|jksbt �d S )Nr r% )r� � ) rm rn r r r �Context�precZto_frameZastyper r �contextrM )r� r ru rY r r r �test_astype_dispatches s r� c s6 e Zd Zd� fdd� Zdd� Zef� fdd� Z� ZS ) �TestArithmeticOpsNc s t � j|||d d� d S )N)�exc)rw � check_opname)rJ rc rd rs r� )ry r r r� s z TestArithmeticOps.check_opnamec C s� |}t �|�}t�� }|jtj }|jtj }d|jtj< d|jtj< t �dd� |D ��}| �|||� d|kr�| �|||d � | �||d� | �||d� ||jtj< ||jtj< d S )Nr c S s g | ]}t |d ��qS )r )�int)r �dr r r r , s zBTestArithmeticOps.test_arith_series_with_array.<locals>.<listcomp>�modr r� )rm rn r � getcontextrp �DivisionByZerorq r� ) rJ r Zall_arithmetic_operatorsrd rc r� Z divbyzerotrapZ invalidoptraprs r r r � test_arith_series_with_array! s z.TestArithmeticOps.test_arith_series_with_arrayc s t � j|||d d� d S )N)r� )rw �_check_divmod_op)rJ rc �oprs r� )ry r r r� 7 s z"TestArithmeticOps._check_divmod_op)N)rE rF rG r� r� r` r� rz r r )ry r r� s r� c s6 e Zd Zd � fdd� Zdd� Zdd� Zdd � Z� ZS ) �TestComparisonOpsNc s t � j|||d d� d S )N)r� )rw r� )rJ rc rd rs r� )ry r r r� = s z TestComparisonOps.check_opnamec C s | � |||� d S )N)r� )rJ rc r rd rs r r r �_compare_other@ s z TestComparisonOps._compare_otherc C s" |}t �|�}| �|||d� d S )Ng �?)rm rn r� )rJ r �all_compare_operatorsrd rc r r r �test_compare_scalarC s z%TestComparisonOps.test_compare_scalarc C sR |}t �|�}tj�dddgt|��}t �|�dd� |D � }| �||||� d S )NrV r r c S s g | ]}t �td |���qS )g @)r r �pow)r �ir r r r N s z8TestComparisonOps.test_compare_array.<locals>.<listcomp>)rm rn rb �random�choice�lenr� )rJ r r� rd rc Zalterrs r r r �test_compare_arrayH s z$TestComparisonOps.test_compare_array)N)rE rF rG r� r� r� r� rz r r )ry r r� < s r� c @ s e Zd ZdZddd�ZdS )� DecimalArrayWithoutFromSequencez:Helper class for testing error handling in _from_sequence.NFc C s t d��d S )Nz For the test)�KeyError)r3 Zscalarsr �copyr r r �_from_sequenceU s z.DecimalArrayWithoutFromSequence._from_sequence)NF)rE rF rG �__doc__r� r r r r r� R s r� c @ s e Zd Zedd� �ZdS )�DecimalArrayWithoutCoercionc C s | j |dd�S )NF)Zcoerce_to_dtype)Z_create_method)r3 r� r r r �_create_arithmetic_methodZ s z5DecimalArrayWithoutCoercion._create_arithmetic_methodN)rE rF rG rH r� r r r r r� Y s r� c C s| t }tdd� �}| �td|� |t�d�t�d�g�}t�|�}|�|t j �}tjt�d�t�d�gdd�}t � ||� d S ) Nc S s t S )N)r� )r3 r r r �construct_array_typef s z?test_combine_from_sequence_raises.<locals>.construct_array_typer� z1.0z2.0z4.0r. )r ) r� rH �setattrr r r rm rn �combine�operator�addr1 r2 )� monkeypatchr3 r� rX r� rY rZ r r r �!test_combine_from_sequence_raisesb s r� �class_c C sN | t �d�t �d�g�}|| }tjt �d�t �d�gdd�}t�||� d S )Nz1.0z2.0z4.0r. )r )r r rb rl r1 �assert_numpy_array_equal)r� rX rY rZ r r r �$test_scalar_ops_from_sequence_raisesw s r� z#reverse, expected_div, expected_modr r c C s^ t ddddg�}| r$td|�\}}nt|d�\}}t |�}t |�}t�||� t�||� d S )Nr r � � )r �divmodr1 rW )�reverseZ expected_divZ expected_modrX �divr� r r r �test_divmod_array� s r� c C sT | d d� }t j|tdd�d�}t�|�}t jt�|�tdd�d�}t�||� d S )Nr� r� � )rA )rm rn r rb �absr1 r2 )r r% rc rY rZ r r r �test_ufunc_fallback� s r� c C s8 t dddg�} t�| �}t t�| j��}t�||� d S )Nr r r� )r rb �exp�_datar1 rW )r% rY rZ r r r �test_array_ufunc� s r� c C sH t dddg�} t�| �}t�|�}t�t t�| j���}t�||� d S )Nr r r� )r rm rn rb r� r� r1 r2 )r% rc rY rZ r r r �test_array_ufunc_series� s r� c C sR t dddg�} t�| �}t�|t�d��}t�t�| t�d���}t�||� d S )Nr r r� ) r rm rn rb r� r r r1 r2 )r% rc rY rZ r r r �$test_array_ufunc_series_scalar_other� s r� c C s` t dddg�} t�| �}t�t dddg��}t�|| �}t�| |�}t�||� t�||� d S )Nr r r� r� � )r rm rn rb r� r1 r2 )r% rc rZ �r1�r2r r r � test_array_ufunc_series_defer� s r� c C s� t � d d� } t�dddddgdddddgt| �d��}t�t| d | d g��}|�d�d �dd � �}tj ||d d � |d �|d ��d d � �}tj ||d d � tjt| d | d | d g�tj � d ddg�d�}|�ddg�d �dd � �}tj ||d d � |d �|d |d g��dd � �}tj ||d d � t�ddgt| d | d g�d��}|�d��dd � �}tj ||d d � d S )Nr� r r )�id1�id2rC r� r� rC c S s | j d S )Nr )�iloc)r r r r r � r! z"test_groupby_agg.<locals>.<lambda>F)r9 c S s | j d S )Nr )r� )r r r r r � r! )r r )r r )r r )rA r� c S s | j d S )Nr )r� )r r r r r � r! c S s | j d S )Nr )r� )r r r r r � r! )r� rC c S s | j d S )Nr )r� )r r r r r � r! ) r rm r= r rn r �groupby�aggr1 r2 Z MultiIndex� from_tuplesrB )r �dfrZ rY r r r �test_groupby_agg� s$ & $$r� c C s� dd� }| j td|dd� t� d d� }t�dddddgt|�d ��}t�t|d |d |d |d |d g��}|�d �d �dd� �}t j ||dd� t�t|��}|�t � dddddg���dd� �}t j ||dd� d S )Nc S s t �t �| ��S )N)rb �sumrl )rJ r r r �DecimalArray__my_sum� s z8test_groupby_agg_ea_method.<locals>.DecimalArray__my_sum�my_sumF)�raisingr� r r )�idrC r r� r� r� rC c S s | j �� S )N)�valuesr� )r r r r r � r! z,test_groupby_agg_ea_method.<locals>.<lambda>)r9 c S s | j �� S )N)r� r� )r r r r r � r! ) r� r r rm r= rn r r� r� r1 r2 rb rl )r� r� r r� rZ rY rc r r r �test_groupby_agg_ea_method� s 2$r� c C sx d dd�}| j td|dd� t� }t�t|��}t�|tt|��d��}||dk ||dk |jd |jd d S ) Nc S s t d��d S )Nz0tried to convert a DecimalArray to a numpy array)� Exception)rJ r r r r �DecimalArray__array__� s z;test_indexing_no_materialize.<locals>.DecimalArray__array__Z __array__F)r� )r% r$ g �?r )r r% )N) r� r r rm rn r= r r� �at)r� r� r rc r� r r r � test_indexing_no_materialize� s r� c C sz t �d�t �d�g} tjt �d�t �d�gdd�}tj| dd�}|jdd �}t�||� t�|�jdd �}t�||� d S ) Nz1.1111z2.2222z1.11z2.22r. )r r r )rC ) r r rb rl rm Zto_numpyr1 r� rn )r� rZ r% rY r r r �test_to_numpy_keyword s r� )Tr r* r� �numpyrb rO Zpandasrm Zpandas._testingZ_testingr1 Zpandas.api.typesr Zpandas.tests.extensionr Z$pandas.tests.extension.decimal.arrayr r r r Zfixturer r r r r r r" r# r( r) ZBaseDtypeTestsrI ZBaseInterfaceTestsrR ZBaseConstructorsTestsrS ZBaseReshapingTestsrT ZBaseGetitemTestsrU ZBaseMissingTestsr\ r] ZBaseNumericReduceTestsrg ZBaseBooleanReduceTestsrh ZBaseMethodsTestsri ZBaseCastingTestsr{ ZBaseGroupbyTestsr| ZBaseSetitemTestsr~ ZBasePrintingTestsr rP Zxfailr� r� r� rQ r� ZBaseArithmeticOpsTestsr� ZBaseComparisonOpsTestsr� r� r� Z_add_arithmetic_opsr� r� r� r� r� r� r� r� r� r� r� r� r r r r �<module> st / >
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка