Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/arithmetic/__pycache__/test_numeric.cpython-37.pyc
Назад
B �5�g�� � @ s� U d dl mZ d dlmZ d dlmZ d dlmZ d dlZd dl m Z d dl Z d dl Z d dlZd dlmZmZmZmZmZmZmZmZmZ d dlmZ d dl m Z d d l m Z e j!eeej"gd �d d � �Z#d d� Z$eegZ%de&d< eddd�gZ'de&d< e'�(dd� dD �� G dd� d�Z)G dd� d�Z*G dd � d �Z+G d d � d �Z,G d d � d �Z-G d!d"� d"�Z.G d#d$� d$�Z/G d%d&� d&�Z0d'd(� Z1d)d*� Z2e j3�4d+d d,d-gd d-d.gfg�d/d0� �Z5d1d2� Z6dS )3� )� annotations)�abc)�Decimal)� combinationsN)�Any) � Float64Index�Index� Int64Index� RangeIndex�Series� Timedelta�TimedeltaIndex� UInt64Index�array)�ops)� expressions)�paramsc C s | j S )zL Fixture to test behavior for Index, Series and tm.to_array classes )�param)�request� r �Z/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/arithmetic/test_numeric.py�box_pandas_1d_array s r c C s8 t �t �| ���� r4t �t �| ���� s,t�|d9 }|S )z_ Helper to adjust the expected result if we are dividing by -0.0 as opposed to 0.0 ���)�npZsignbitr �any�all�AssertionError)�zero�expectedr r r �adjust_negative_zero( s r z list[Any]� ser_or_index� �( �leftsc C s( g | ] }t D ]}|d ddg|d��q qS )r! � � )�dtype)r )�.0r& �clsr r r � <listcomp>> s r) ) �i1�i2�i4�i8�u1�u2Zu4�u8�f2Zf4�f8c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestNumericComparisonsc C s� t �d�tdddg�k}dtdddg�k}t�||� tdddg�t �d�k }tdddg�dk }t�||� t �dddg�d tdddg�k}dtdddg�k}t�||� d S )Nr � � � g )r �float64r �tm�assert_series_equalr )�self�resultr r r r �(test_operator_series_comparison_zerorankI s "z?TestNumericComparisons.test_operator_series_comparison_zerorankc C s� t j�� }t �dtd�i�}d}tjt|d�� ||k W d Q R X tjt|d�� ||k W d Q R X tjt|d�� ||k W d Q R X tjt|d�� ||k W d Q R X ||k�� �� r�t �||k� � � � s�t �d S )N�x� zI'[<>]' not supported between instances of 'numpy.ndarray' and 'Timestamp')�match) �pd� Timestamp�now� DataFrame�range�pytest�raises� TypeErrorr r r )r: �ts�df�msgr r r � test_df_numeric_cmp_dt64_raisesU s z6TestNumericComparisons.test_df_numeric_cmp_dt64_raisesc C sL t tj�d�dd�}t tj�d��}t�d�|_t�|| d|| � d S )Nr> r )�namez 2000-01-01r4 ) r r �random�randnr@ rA rL r8 r9 )r: �a�br r r �test_compare_invalidi s z+TestNumericComparisons.test_compare_invalidc C s� |}|t k r|ntj}ttj�d��}tj||dd�}|dk}ttjdt d��}tj||dd�}t� ||� |dk}t� || � d}t j t |d�� |dk W d Q R X d S )Ni�� F)� transposerO )r& z0Invalid comparison between dtype=float64 and str)r? )r r Zndarrayr rM rN r8 � box_expectedZzeros�bool� assert_equalrE rF rG )r: �box_with_array�boxZxbox�objr; r rJ r r r �$test_numeric_cmp_string_numexpr_pathq s z;TestNumericComparisons.test_numeric_cmp_string_numexpr_pathN)�__name__� __module__� __qualname__r< rK rQ rY r r r r r3 H s r3 c @ s� e Zd Zej�dejee g�ejjde dd� d�dd� ��Z ej�dejee g�ejjde dd� d�d d � ��Z ejjd e d d �e d d ��� e d d ��� e d d ��� �d�e d d ��� �d�gdd� d�dd� �Zejjd e d d �e d d ��� e d d ��� gdd� d�dd� �Zdd� Zej�de dd�e dd��� e dd��� e dd��� �d�e�d �e�d d �ej�d �ej�d �g�d d!� �Zej�dej�� � � ejjd"d#�� � ej�� � � ej g�ej� d$�d%d&� ��Z d'S )(�.TestNumericArraylikeArithmeticWithDatetimeLike�box_cls�leftc C s t | �jt| j� S )N)�typerZ �strr& )r= r r r �<lambda>� � z7TestNumericArraylikeArithmeticWithDatetimeLike.<lambda>)Zidsc C sp t jdddgdd�}||�}tdddg�}t|t�s<|tkrDt|�}|| }t�||� || }t�||� d S ) Nr4 r5 r6 zm8[s])r& Z10sZ40sZ90s)r r r � isinstancer r8 rU )r: r_ r^ �rightr r; r r r �test_mul_td64arr� s z?TestNumericArraylikeArithmeticWithDatetimeLike.test_mul_td64arrc C s t | �jt| j� S )N)r` rZ ra r& )r= r r r rb � rc c C s� t jdddgdd�}||�}tdddg�}t|t�s<|tkrDt|�}|| }t�||� || }t�||� d }tjt |d �� || W d Q R X tjt |d �� || W d Q R X d S ) Nr! r" �Z zm8[s])r& Z1sZ2sZ3sz Cannot divide)r? ) r r r rd r r8 rU rE rF rG )r: r_ r^ re r r; rJ r r r �test_div_td64arr� s z?TestNumericArraylikeArithmeticWithDatetimeLike.test_div_td64arr� scalar_tdr4 )�daysztimedelta64[s]ztimedelta64[ms]c C s t | �jS )N)r` rZ )r= r r r rb � rc c C sf |}|}t dd� tt|��D ��}t�||�}t�||�}|| }t�||� || }t�||� d S )Nc S s g | ]}t |d ��qS ))rj )r )r' �nr r r r) � s z`TestNumericArraylikeArithmeticWithDatetimeLike.test_numeric_arr_mul_tdscalar.<locals>.<listcomp>)r rD �lenr8 rS rU ) r: ri � numeric_idxrV rW �indexr r; Zcommuter r r � test_numeric_arr_mul_tdscalar� s zLTestNumericArraylikeArithmeticWithDatetimeLike.test_numeric_arr_mul_tdscalarc C s t | �jS )N)r` rZ )r= r r r rb � rc c C sr |}t �d��t j�}tj||dd�}|�d��d�}tj||dd�}|| }t�||� || }t�||� d S )Ni N F)rR ztimedelta64[D]ztimedelta64[ns])r �arange�astype�int64r8 rS �viewrU )r: ri rV rW �arrrX r r; r r r �*test_numeric_arr_mul_tdscalar_numexpr_path� s zYTestNumericArraylikeArithmeticWithDatetimeLike.test_numeric_arr_mul_tdscalar_numexpr_pathc C sr |}|dd� }t ddg�}t�||�}t�||�}|| }t�||� d}tjt|d�� || W d Q R X d S )Nr4 r6 z3 Daysz36 Hoursz$cannot use operands with types dtype)r? )r r8 rS rU rE rF rG ) r: Z three_daysrm rV rW rn r r; rJ r r r � test_numeric_arr_rdiv_tdscalar� s zMTestNumericArraylikeArithmeticWithDatetimeLike.test_numeric_arr_rdiv_tdscalar�other� )�hourszm8[h]�NaT�Dr6 r c C s� |}t �||�}d}tjt|d�� || W d Q R X tjt|d�� || W d Q R X tjt|d�� || W d Q R X tjt|d�� || W d Q R X d S )Nz�unsupported operand type|Addition/subtraction of integers and integer-arrays|Instead of adding/subtracting|cannot use operands with types dtype|Concatenation operation is not implemented for NumPy arrays)r? )r8 rS rE rF rG )r: rm rw rV rW r_ rJ r r r �"test_add_sub_timedeltalike_invalid s zQTestNumericArraylikeArithmeticWithDatetimeLike.test_add_sub_timedeltalike_invalid�UTC)�tzz*ignore:elementwise comp:DeprecationWarningc C s� |}t �||�}d�ddddddg�}tjt|d�� || W d Q R X tjt|d�� || W d Q R X tjt|d�� || W d Q R X tjt|d�� || W d Q R X d S ) N�|zunsupported operand typez+Cannot (add|subtract) NaT (to|from) ndarrayz3Addition/subtraction of integers and integer-arraysz;Concatenation operation is not implemented for NumPy arrayszBoperand type\(s\) all returned NotImplemented from __array_ufunc__z(can only perform ops with numeric values)r? )r8 rS �joinrE rF rG )r: rm rw rV rW r_ rJ r r r �!test_add_sub_datetimelike_invalid" s" zPTestNumericArraylikeArithmeticWithDatetimeLike.test_add_sub_datetimelike_invalidN)!rZ r[ r\ rE �mark� parametrizer r r r r# rf rh r Zto_timedelta64Zto_pytimedeltarq ro ru rv Z timedelta64r@ �offsetsZMinuteZSecondr| rA rB � to_pydatetimeZ to_datetime64rz �filterwarningsr� r r r r r] � sL r] c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zej� d e j e j g�d d � �Z ej� d ejejejg�d d� �Zej� d ejejejg�dd� �Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Z d#d$� Z d%d&� Z d'd(� Z d)d*� Z d+S ),�TestDivisionByZeroc C sv |}t tjtjtjtjtjgtjd�}t||�}|| }t�||� t|�� d�t� |�� d� }t� |t|�� d S )N)r& r- ) r r �nan�infr7 r r8 �assert_index_equalr rq r r9 )r: r rm �idxr � expected2r; � ser_compatr r r � test_div_zeroL s " z TestDivisionByZero.test_div_zeroc C sv |}t tjtjtjtjtjgtjd�}t||�}|| }t�||� t|�� d�t� |�� d� }t� |t|�� d S )N)r& r- ) r r r� r� r7 r r8 r� r rq r r9 )r: r rm r� r r� r; r� r r r �test_floordiv_zeroY s " z%TestDivisionByZero.test_floordiv_zeroc C sl |}t tjtjtjtjtjgtjd�}|| }t�||� t|��d�t�|��d� }t� |t|�� d S )N)r& r- ) r r r� r7 r8 r� r rq r r9 )r: r rm r� r r; r� r r r � test_mod_zerof s " z TestDivisionByZero.test_mod_zeroc C s� |}t tjtjtjtjtjgtjd�}t tjtjtjtjtjgtjd�}t||�}t||�}t�|d |� t�|d |� d S )N)r& r r4 ) r r r� r� r7 r �divmodr8 r� )r: r rm r� ZexleftZexrightr; r r r �test_divmod_zeroo s "" z#TestDivisionByZero.test_divmod_zero�opc C sb t |t�rd S |d }ttj tj tj tjtjgtjd�}t||�}|||�}t� ||� d S )Nr6 )r& ) rd r r r r� r� r7 r r8 r� )r: r rm r� r� r r; r r r �test_div_negative_zeroz s ( z)TestDivisionByZero.test_div_negative_zero�dtype1c C s� |}t ddddgdd��|�}t ddddgdd��|�}tjd d ��$ t |j�tj�|j d d d �}W d Q R X tj|jdd�< |jd kr�|jdkr�t j r�|dkr�|�d�}|| }t � ||� |� || �r�t�d S )Nr6 � r> � �first)rL r �second�ignore)r r7 )r& rL rr Zfloat32)r rq r �errstate�valuesr7 r� �ilocr& �exprZ USE_NUMEXPRr8 r9 �equalsr ) r: Zswitch_numexpr_min_elementsr� �any_real_dtype�dtype2r� r� r r; r r r �test_ser_div_ser� s z#TestDivisionByZero.test_ser_div_serc C s� |}t ddg��|�}t ddg��|�}|| || f}t|�}|d �tj�|d< tj|d d< t||�}t�|d |d � t�|d |d � t|j |�}t�|d |d � t�|d |d � d S )Nr4 r r5 ) r rq �listr r7 r� r� r8 r9 r� )r: r� r� r� r_ re r r; r r r �test_ser_divmod_zero� s z'TestDivisionByZero.test_ser_divmod_zeroc C s� t tjdg�}t tjdg�}|| || f}t||�}t�|d |d � t�|d |d � t|j|�}t�|d |d � t�|d |d � d S )Ng �?g @r r4 )r r r� r� r8 r9 r� )r: r_ re r r; r r r �test_ser_divmod_inf� s z&TestDivisionByZero.test_ser_divmod_infc C sz t �dgd �}t j�d�}tdgd �}|t|� }t�||� t|�| }t�||� t|�t|� }t�||� d S )Nr r> g )r r rM rN r r8 r9 )r: Z zero_array�datar r; r r r �test_rdiv_zero_compat� s z(TestDivisionByZero.test_rdiv_zero_compatc C sD t dddgdd�}t tj tjtjgdd�}|d }t�||� d S )Nr r r4 r� )rL )r r r� r� r8 r9 )r: �serr r; r r r �test_div_zero_inf_signs� s z*TestDivisionByZero.test_div_zero_inf_signsc C s> t dddgdd�}t dtjdgdd�}d| }t�||� d S )Nr r r4 r� )rL g )r r r� r8 r9 )r: r� r r; r r r �test_rdiv_zero� s z!TestDivisionByZero.test_rdiv_zeroc C sD t dddgdd�}|d }t tj tjtjgdd�}t�||� d S )Nr r r4 r� )rL )r r r� r� r8 r9 )r: r� r; r r r r �test_floordiv_div� s z$TestDivisionByZero.test_floordiv_divc C sn t �ddddgddddgd��}|| }tddddg�}ttjtjtjdg�}t �||d��}t�||� d S ) Nr6 r� r> r� r )r� r� g �?r4 )r@ rC r r r� r8 �assert_frame_equal)r: rI r; r� r� r r r r �test_df_div_zero_df� s z&TestDivisionByZero.test_df_div_zero_dfc C s� t �ddddgddddgd��}tddddg�}ttjtjtjdg�}t �||d��}tjd d �� |j�d �|j }W d Q R X t j||j|j d �}t � ||� d S ) Nr6 r� r> r� r )r� r� g �?r4 r� )r �float)rn �columns) r@ rC r r r� r� r� rq rn r� r8 r� )r: rI r� r� r rt r; r r r �test_df_div_zero_array� s z)TestDivisionByZero.test_df_div_zero_arrayc C s� t �ddddgddddgd��}|d }t jtj|j|jd�}tj|jdd�df< t� ||� tj d d �� |j � d �d }W d Q R X t j||j|jd�}t� ||� d S ) Nr6 r� r> r� r )r� r� )rn r� r4 r� )r r7 ) r@ rC r r� rn r� r� r� r8 r� r� r� rq )r: rI r; r rt �result2r r r �test_df_div_zero_int s z'TestDivisionByZero.test_df_div_zero_intc C sJ t �tj�dd��}|d }|| }|| }|�d��|�d��rFt�d S )Nr! r> r )r@ rC r rM rN �fillnar� r )r: rI r� �res�res2r r r �(test_df_div_zero_series_does_not_commute s z;TestDivisionByZero.test_df_div_zero_series_does_not_commutec C s� t �ddddgddddgd��}tddddg�}|s>|�d�}ttjtjtjdg�}t �||d��}|| }t�||� t jddddgddddgd�dd �}tddddgd d �}ttjtjtjdg�}t �||d��}|| }t�||� d S ) Nr6 r� r> r� r )r� r� r7 F)�copyrr )r& )r@ rC r rq r r� r8 r� )r: Zusing_array_managerrI r� r� r r; r r r �test_df_mod_zero_df s $z&TestDivisionByZero.test_df_mod_zero_dfc C s� t �ddddgddddgd��}tddddgdd�}ttjtjtjdg�}t �||d��}tjd d �� |j|j }W d Q R X t j||j|jdd �}tj|j dd�d f< t � ||� d S ) Nr6 r� r> r� r )r� r� r7 )r& r� )r )rn r� r& r4 ) r@ rC r r r� r� r� rn r� r� r8 r� )r: rI r� r� r rt r� r r r �test_df_mod_zero_array8 s z)TestDivisionByZero.test_df_mod_zero_arrayc C s� t �ddddgddddgd��}|d }t jtj|j|jd�}t�||� tjdd �� |j � d �d }W d Q R X t j||j|jd�}t�||� d S ) Nr6 r� r> r� r )r� r� )rn r� r� )r r7 ) r@ rC r r� rn r� r8 r� r� r� rq )r: rI r; r rt r� r r r �test_df_mod_zero_intI s z'TestDivisionByZero.test_df_mod_zero_intc C sJ t �tj�dd��}|d }|| }|| }|�d��|�d��rFt�d S )Nr! r> r )r@ rC r rM rN r� r� r )r: rI r� r� r� r r r �(test_df_mod_zero_series_does_not_commuteW s z;TestDivisionByZero.test_df_mod_zero_series_does_not_commuteN)!rZ r[ r\ r� r� r� r� rE r� r� �operator�truediv�floordivr� r rr r7 �uint64r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� K s( "" r� c @ s� e Zd Zdd� Zdd� Zdd� Zej�de j e j e j g�dd � �Zd d � Zd d � Zdd� Zdd� Zdd� Zdd� Zej�de je jg�dd� �Zdd� Zdd� Zd d � Zd d � Zej�d e j d!d"d#d$e j g�d%d&� �Z d'd(� Z d)d*� Z!d+d,� Z"d-S ).�TestMultiplicationDivisionc C sl |}t td�g�}t td�g�}t�||�}t�||�}|td� }t�||� |td� }t�||� d S )Nr! r> r5 )r r r8 rS rU )r: rV rW r� r r; r r r �test_divide_decimale s z.TestMultiplicationDivision.test_divide_decimalc C sb t ddgdd�}t ddgdd�}t dtj g�}|�|�}tj||dd � || }t�||� d S ) Nr4 r r� )rL g{�G�z��g{�G�z��r� F)� check_names)r r r� �divr8 r9 )r: r� r� r r; r r r �test_div_equiv_binopu s z/TestMultiplicationDivision.test_div_equiv_binopc C sH |}|d }|� d�}t�||� |d }t|jd �}t�||� d S )Nr4 r7 r5 )rq r8 r� r r� )r: rm r� r; r r r r � test_div_int� s z'TestMultiplicationDivision.test_div_intr� c C s* |}t �||�}||d�}t �||� d S )Nr4 )r8 rS rU )r: r� rm rV r� r; r r r �test_mul_int_identity� s z0TestMultiplicationDivision.test_mul_int_identityc C sb |}|| }|t jddd� }t�||d � t|t�r<dnd}|t jd|d� }t�||� d S )Nr> rr )r& r� )r r r8 r� rd r rp )r: rm r� �didxr; � arr_dtyper r r �test_mul_int_array� s z-TestMultiplicationDivision.test_mul_int_arrayc C sH |}|| }t |t�rdnd}|ttjd|d�� }t�|t|�� d S )Nr� rr r> )r& )rd r r r rp r8 r9 )r: rm r� r� r� r; r r r �test_mul_int_series� s z.TestMultiplicationDivision.test_mul_int_seriesc C sB |}t jddd�}|t|d � }t||d �}t�||� d S )Nr> r7 )r& g�������?)r rp r r8 r9 )r: rm r� Zrng5r; r r r r �test_mul_float_series� s z0TestMultiplicationDivision.test_mul_float_seriesc C s* |}t |t�s&|| }t�||d � d S )Nr5 )rd r r8 r� )r: rm r� r; r r r �test_mul_index� s z)TestMultiplicationDivision.test_mul_indexc C s8 |}d}t jt|d�� |tjddd� W d Q R X d S )Nz,cannot perform __rmul__ with this index type)r? �20130101r> )�periods)rE rF rG r@ � date_range)r: rm r� rJ r r r �test_mul_datelike_raises� s z3TestMultiplicationDivision.test_mul_datelike_raisesc C sb |}d}t jt|d�� ||dd� W d Q R X t jt|d�� |t�ddg� W d Q R X d S )Nz(operands could not be broadcast together)r? r r6 r4 r5 )rE rF � ValueErrorr r )r: rm r� rJ r r r � test_mul_size_mismatch_raises� s z8TestMultiplicationDivision.test_mul_size_mismatch_raisesc C sJ |}|}t ||jd��}t�||�}t�||�}||d�}t�||� d S )Ng @)r r� r8 rS rU )r: r� rm rV rW r� r r; r r r �test_pow_float� s z)TestMultiplicationDivision.test_pow_floatc C sF |}|}t |jd �}t�||�}t�||�}|d }t�||� d S )Nr5 )r r� r8 rS rU )r: rm rV rW r� r r; r r r � test_modulo� s z&TestMultiplicationDivision.test_moduloc C sn |}t |d�}tjdd�� t |jd�\}}W d Q R X t|�t|�f}x"t||�D ]\}}t�||� qRW d S )Nr5 r� )r )r� r r� r� r �zipr8 r� ) r: rm r� r; r� �modr �r�er r r �test_divmod_scalar� s z-TestMultiplicationDivision.test_divmod_scalarc C s� |}t j|jj|jjd�d }t||�}t jdd�� t|j|�\}}W d Q R X t|�t|�f}x"t||�D ]\}} t � || � qlW d S )N)r& r5 r� )r ) r �onesr� �shaper& r� r� r r� r8 r� ) r: rm r� rw r; r� r� r r� r� r r r �test_divmod_ndarray� s z.TestMultiplicationDivision.test_divmod_ndarrayc C s� |}t j|jj|jjd�d }t|t|��}t jdd�� t|j|�\}}W d Q R X t|�t|�f}x"t||�D ]\}} t � || � qpW d S )N)r& r5 r� )r ) r r� r� r� r& r� r r� r� r8 r9 ) r: rm r� rw r; r� r� r r� r� r r r �test_divmod_series� s z-TestMultiplicationDivision.test_divmod_seriesrw � i��gX9��v�@g ��Q� �c C s� t j�dd�}dd� }||�}t�|t �|� ||| �� t�t �|�| ||| �� t�|t �|� ||| �� t�t �|�| ||| �� d S )Nr> r6 c S s t j| td�dddgd�S )NZABCDEZjimZjoeZjolie)rn r� )r@ rC r� )r= r r r rb s z?TestMultiplicationDivision.test_ops_np_scalar.<locals>.<lambda>)r rM rN r8 r� r )r: rw �vals�frI r r r �test_ops_np_scalar s z-TestMultiplicationDivision.test_ops_np_scalarc C s� t �� }d|_t�d|i�}t j|| ||d dd� t j|| ||d dd� t j||k ||d k dd� t j|| ||d dd� d S )NrH �AF)r� )r8 �makeTimeSeriesrL r@ rC r9 )r: rH rI r r r �test_operators_frame s z/TestMultiplicationDivision.test_operators_framec C s( t jdd��� t�ddddgddddgd��}|d |d }t|d j|d j d d �}t j|jdd�< t� ||� |d d }tt j|j d d �}t� ||� |� d �}|d |d }t|d j|d j �}t� ||� |� d �}|d |d }|d |d }|� |��rt �W d Q R X d S )Nr� )r r6 r� r> r� r )r� r� r� r� r7 )r& )rn rL )r r� r@ rC r r� r� r� r8 r9 rn rq r� r )r: �pr; r r� r r r � test_modulo2 s" z'TestMultiplicationDivision.test_modulo2c C sn t jdd��X tddg�}|d }tt jt jg�}t�||� d| }tt jdg�}t�||� W d Q R X d S )Nr� )r r r4 g )r r� r r� r8 r9 )r: �sr; r r r r �test_modulo_zero_int6 s z/TestMultiplicationDivision.test_modulo_zero_intN)#rZ r[ r\ r� r� r� rE r� r� r� �mulr �rmulr� r� r� r� r� r� r� r� �pow�rpowr� r� r� r� r� r r� r� r� r� r� r� r r r r r� a s& " &r� c @ s� e Zd Zej�dedddged�dd�edddged�dd�ed d ej ej ged �dd�fedddged�dd�eddddged �dd�edd d ej ged �dd�fg�dd� �Z ej�de j ddddgied�d�e j ddddgied�d�e j dd d ej ej gied �d�fe j ddddgied�d�e j dddddgied �d�e j ddd d ej gied �d�fg�dd� �Z dd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd S ) �TestAdditionSubtractionzfirst, second, expectedr4 r5 r6 �ABCr= )rn rL ZABDg @g @�ABCDr� r> c C s$ t �|| |� t �|| |� d S )N)r8 r9 )r: r� r� r r r r �test_add_seriesH s z'TestAdditionSubtraction.test_add_series)rn c C s$ t �|| |� t �|| |� d S )N)r8 r� )r: r� r� r r r r �test_add_frames\ s z'TestAdditionSubtraction.test_add_framesc C s� t t�dd��}d| }|�dd� �}t�||� t�d|i�}d| }t�d|�dd� �i�}t�||� t�� }d|_ tj � � � � }d � d d g�}tjt|d �� || W d Q R X tjt|d �� || W d Q R X d S ) Nr> r! �foo_c S s d| S )Nr� r )r= r r r rb u rc zDTestAdditionSubtraction.test_series_frame_radd_bug.<locals>.<lambda>r� c S s d| S )Nr� r )r= r r r rb z rc rH r zunsupported operand typezConcatenation operation)r? )r r8 Z rands_array�mapr9 r@ rC r� r� rL rA rB r� r� rE rF rG )r: r� r; r �framerH rB rJ r r r �test_series_frame_radd_bugq s$ z2TestAdditionSubtraction.test_series_frame_radd_bugc C s t tj�d��}||j�� }||j }t�||� t tj ddd�tj ddd�d�}||j�� }||j }t�||� d}t j t |d�� ||j� � }W d Q R X tjtj�dd�tj ddd�d�}t�d�|d < |d |j�� |d < |d |j |d < tj|d |d d d � d S )Nr> r� )r� )rn zcannot subtract period)r? r5 Z20130102�dater r; F)r� )r r rM rN rn Z to_seriesr8 r9 r@ r� rE rF rG Z to_periodrC rA )r: r� r r; rJ rI r r r �test_datetime64_with_index� s& z2TestAdditionSubtraction.test_datetime64_with_indexc C s� |}t j|ddddgd�}tj�d�}t|t�|j�d�}|| }|| }x`|�� D ]T\}} xJ| �� D ]>\} } || | d } t�| �s�| | ks�t �qft�| �sft �qfW qTW xd|�� D ]X\}} xN| �� D ]B\} } || | || } t�| �s�| | ks�t �q�t�| �s�t �q�W q�W || } |d }t � | |� t �d d d d gi�}t � || t �d d tj d gi�� xld D ]d}t j|d�t jdg|d�t jdg|d�g}x2|D ]*}|| � |��s�t �t � || |� �q�W �qjW d S )Nr{ �C�Br� )r� r� )rn r5 rO rP ZaaZbb)r� rr )r& )r� r& r )rn r& )r@ rC r rM r r r� �items�isnanr r8 r� r� r� )r: Z float_framer� Zframe2�garbageZ colSeriesZidSumZ seriesSum�col�seriesr� �valZorigVal�addedr rI r& �framesr r r �test_frame_operators� s< " z,TestAdditionSubtraction.test_frame_operatorsc sx d dd�� � fdd�}t �� �d�}|||d � |||d d d� � ||d� � fd d �}||d� |||d � d S ) NFc S sp |rt �| �n| }|r t �|�n|}|||�}|�||�}t|t�r`|j�| j�s`|j�d �|_t� ||� d S )N) r �abs�combinerd r rn r� � _with_freqr8 r9 )r� rw r� �pos_onlyr_ re Zcython_or_numpy�pythonr r r � _check_op� s z@TestAdditionSubtraction.test_series_operators.<locals>._check_opc s� ddddddg}x |D ]}� | |t t|�� qW � | |tjdd� � | |tj� � | |tj� � | |tj� � | |tj� � | |tj� � | |tj dd� � | |tj � d S ) N�add�subr� r� r� r� T)r ) �getattrr� r� r ZraddZrsubZrtruediv� rfloordivr� r� Zrmod)r� rw Z simple_ops�opname)r r r �check� s z<TestAdditionSubtraction.test_series_operators.<locals>.checkrH r5 r> c sJ � | |t j� � | |t j� � | |t j� � | |t j� � | |t j� d S )N)r� �gt�ge�eq�lt�le)r� rw )r r r �check_comparators� s zHTestAdditionSubtraction.test_series_operators.<locals>.check_comparatorsr4 )F)r8 r� �rename)r: r �tserr r )r r �test_series_operators� s z-TestAdditionSubtraction.test_series_operatorsc C sF dd� }t �� �d�}|||d � |||d d d� � ||d� d S )Nc S s� t | |�}t|tj�rPt| �t|�krPg }x(|D ]}|�|� |�tj� q0W n|}t�|�}tj dd�� t | j t�|��}W d Q R X xNt ||�D ]@\}}t � t�|�|� |j| jks�t�t �|j| j�d �� q�W d S )Nr� )r )r� rd r �Iterablerl �appendr r� Zasarrayr� r� r� r8 Zassert_almost_equalrL r r� rn r )r� rw �resultsZother_nprk � expectedsr; r r r r r s z2TestAdditionSubtraction.test_divmod.<locals>.checkrH r5 r> )r8 r� r )r: r r r r r � test_divmod s z#TestAdditionSubtraction.test_divmodc C s| t �� �d�}|d }t||�}ttjgt|� |jdd�}ttj gt|� |jdd�}t � |d |� t � |d |� d S )NrH r )rn rL r4 ) r8 r� r r� r r r� rl rn r� r9 )r: r rw r; �exp1�exp2r r r �test_series_divmod_zero# s z/TestAdditionSubtraction.test_series_divmod_zeroN)rZ r[ r\ rE r� r� r r� r r� r� r@ rC r� r� r� r r r r r r r r r� D s, * " . -+ r� c @ sp e Zd Zej�deeee e g�dd� �Z ej�deeee g�dd� �Z ej�deeee g�dd� �Z dd � Zd S ) �TestUFuncCompat�holderc C sh |t kr t nt}|tkr$tdd�}n|tjddd��}t�|�}|t�tjddd���}t�||� d S )Nr r> rr )r& )r r r r rp �sinr8 rU )r: r! rW r� r; r r r r �test_ufunc_compat6 s z!TestUFuncCompat.test_ufunc_compatc C s |dddddgdd�}|t kr"t nt}t�|�}|jdkrDt||�sHt�tt�t�dddddg��dd�}t � ||�}t � ||� t� |d �}|jdkr�t||�s�t�td d d d d gdd�}t � ||�}t � ||� |d }|jdkr�t||�s�t�tdddddgdd�}t � ||�}t � ||� |d }|jdk�rBt||��sFt�tddd d dgdd�}t � ||�}t � ||� |d }|jdk�r�t||��s�t�td d dddgdd�}t � ||�}t � ||� |d }|jdk�r�t||��s�t�td d d d d gdd�}t � ||�}t � ||� d S )Nr4 r5 r6 r� r> r= )rL r2 g @g �?g �?g �?g @g @g @g @g @g @g �g ) r r r �sqrtr& rd r r r r8 rS rU �divide)r: r! r� rW r; �expr r r �test_ufunc_coercionsE s@ " z$TestUFuncCompat.test_ufunc_coercionsc C s� |dddgdd�}|t kr t nt}t�|�}t|t�s:t�tdddgdd�}tddd gdd�}t� |d t� ||�� t� |d t� ||�� d S ) Nr4 r5 r6 r= )rL g g �?g @g @r ) r r r �modfrd �tupler r r8 rU rS )r: r! rX rW r; r r r r r �!test_ufunc_multiple_return_valueso s z1TestUFuncCompat.test_ufunc_multiple_return_valuesc C sX t dddgdddgdd�}tj�|ddgd� t dddgdddgdd�}t�||� d S ) Nr r4 r5 r6 r= )rn rL r! � )r r r �atr8 r9 )r: r� r r r r � test_ufunc_at{ s z TestUFuncCompat.test_ufunc_atN)rZ r[ r\ rE r� r� r r r r r r# r' r* r- r r r r r 5 s * r c @ sj e Zd Zej�ddeg�dd� �Zej�ddeg�dd� �Zej�de j e j e j e j e jg�dd � �ZdS ) �TestObjectDtypeEquivalencer& Nc C sv |}t dddg|d�}t tjtjtjg|d�}t�||�}t�||�}tj| }t�||� |tj }t�||� d S )Nr4 r5 r6 )r& )r r r� r8 rS rU )r: r& rV rW r� r r; r r r � test_numarr_with_dtype_add_nan� s z9TestObjectDtypeEquivalence.test_numarr_with_dtype_add_nanc C sl |}t dddg|d�}t dddg|d�}t�||�}t�||�}d| }t�||� |d }t�||� d S )Nr4 r5 r6 )r& r� )r r8 rS rU )r: r& rV rW r� r r; r r r � test_numarr_with_dtype_add_int� s z9TestObjectDtypeEquivalence.test_numarr_with_dtype_add_intr� c C sL t tj�d�t�d�td�}|d|�}|d|�t��}t� |�t�|� d S )Nr! )rn r& g �?) r r rM rN rp �objectrq r� r8 r9 )r: r� rt r; r r r r � test_operators_reverse_object� s z8TestObjectDtypeEquivalence.test_operators_reverse_object)rZ r[ r\ rE r� r� r1 r/ r0 r� r r r� r� r� r2 r r r r r. � s r. c @ s� e Zd Zdd� Zdd� Zdd� Zej�de j e j e j e j e je jg�dd � �Zd d � Zej�d ejejg�ej�d dddg�dd� ��ZdS )� TestNumericArithmeticUnsortedc C s� x�|D ]�}x>t |d�D ]0\}}|||�}|t|�t|��}t�||� qW x>|D ]6} x0|D ](} || | �}|t| �| �}t�||� qZW qPW qW d S )Nr5 )r r r8 r� ) r: r �scalars�idxsr� rO rP r; r r� Zscalarr r r � check_binop� s z)TestNumericArithmeticUnsorted.check_binopc C s` t jt jt jt jt jg}dddg}tddd�tddd�tddd�tdd d�g}| �|||� d S ) Nr r4 r5 r r! r$ i����r> ���)r� r r r� r� r� r r6 )r: r r4 r5 r r r � test_binops� s z)TestNumericArithmeticUnsorted.test_binopsc C s8 t g}ddg}tddd�tddd�g}| �|||� d S )Nr4 r5 r r! r$ )r� r r6 )r: r r4 r5 r r r �test_binops_pow� s z-TestNumericArithmeticUnsorted.test_binops_powr� c C s~ t d�}ttj�d��}|t|�|�}|||�}t�||� t�tj�dd��}|t�||g�|�}|||�}t� ||� d S )Nr> r5 ) r r r rM rN r8 r9 r@ rC r� )r: r� rn rw r r; r r r �$test_arithmetic_with_frame_or_series� s zBTestNumericArithmeticUnsorted.test_arithmetic_with_frame_or_seriesc C s4 t ddd�}|d }t ddd�}tj||dd� |d }t ddd�}tj||dd� |d }t d d d�}tj||dd� |d }t dd d ��d �}tj||dd� |d }t ddd�d }tj||dd� |d }|}tj||dd� || }t|j|j �}tj||dd� t ddd�}|d }|jd }tjt|j�|dd� t ddd�dt ddd �ft ddd�dt ddd �ft ddd �dt ddd �jd ft ddd �dt ddd �jd ft d�dt d�ft ddd�dt dd d �ft ddd�dt d d d �ft d d d�dt d�fg}x(|D ] \}}}tj|| |dd� �q W d S ) Nr r! r5 r$ r� T)�exactr+ ���r� r> r4 r7 i� i� i����i7�������! �C �d g @�2 r6 r7 i����i����r i����i8���)r r8 r� rq r r� Z _int64index)r: r� r; r Z cases_exactr� r r r �test_numeric_compat2� sL z2TestNumericArithmeticUnsorted.test_numeric_compat2r& �deltar4 r r c C s� ||�}t dddg|d�}|| }t |j| |d�}t�||� || }t |j| |d�}t�||� t�|| d| � t�|| d| � || jr�t�d S )Nr! � r+ )r& r5 r )r r� r8 r� �emptyr )r: r& rC rn r; r r r r �test_addsub_arithmetic0 s z4TestNumericArithmeticUnsorted.test_addsub_arithmeticN)rZ r[ r\ r6 r8 r9 rE r� r� r� r r r� r� r� r� r: rB r rr r7 rF r r r r r3 � s ;r3 c C s~ t �dddgdd dgd��} t jddddgiddd gd �}| j|dd �}t �tjd d d gd tjd tjgd��}t�||� d S )Nr r4 r5 g�������?)r� r� r� g333333�?g�������?r6 )rn )� fill_valueg �?g )r@ rC r r r� r� r8 r� )rI rw r; r r r r �test_fill_value_inf_maskingD s "rH c C s� t jt�d�tjddddgd tjgd t�d�d�td�td�d �} t jtj�dd�td �td �d �}t� d �� | j |d d � W d Q R X d S )Nr! r4 r5 r6 r� )r� r� r� r{ Z abcdefghijr� )rn r� Z abcdefghjkZABCXr )rG ) r@ rC r rp r� r� rM rN r8 Zassert_produces_warningr� )Zpdf1Zpdf2r r r �test_dataframe_div_silencedR s rI zdata, expected_datar4 r5 r� c C s� t |dd�}| |�}|||� }||�| }t| krHtj}t|dd�} nLt|krdtj}t|dd�} n0t| |fkr�tj}t|�} ntj}tj |dd�} ||| � ||| � d S )NZInt64)r& r1 ) r r r8 r9 r r� r Zassert_numpy_array_equalr ) r Z box_1d_arrayr� Z expected_datart � containerr_ re Zassert_functionr r r r � test_integer_array_add_list_likee s" rK c C s� t jdtj�d�it jjddgdddggddgd �d �} | �� }|j� dd�|_| | }t jd gd dg| jd �}t � ||� d S ) NrO � rP r r4 r5 ZlevAZlevB)�names)rn g )r� rn ) r@ rC r rM rN Z MultiIndexZ from_productr� rn Z swaplevelr8 r� )rI Zdf2r; r r r r �"test_sub_multiindex_swapped_levels� s rN )7� __future__r � collectionsr �decimalr � itertoolsr r� �typingr �numpyr rE Zpandasr@ r r r r r r r r r Zpandas._testingZ_testingr8 Z pandas.corer Zpandas.core.computationr r� ZfixtureZto_arrayr r r �__annotations__r# �extendr3 r] r� r� r� r r. r3 rH rI r� r� rK rN r r r r �<module> sP , C A d rM/
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка