OwlCyberSecurity - MANAGER
Edit File: test_scalarmath.cpython-38.pyc
U ���]wj � @ s� d dl mZmZmZ d dlZd dlZd dlZd dlZd dlZd dl Z d dl Z d dl m Z mZmZmZmZmZmZmZmZ e je je je je je je j e j e j e j e j e j!e j"e j#e j$e j%e j&gZ'e j(�)� Z*e j+�)� Z,G dd� de-�Z.G dd� de-�Z/G dd � d e-�Z0d d � Z1d d � Z2G dd� de-�Z3G dd� de-�Z4G dd� de-�Z5G dd� de-�Z6e�sjG dd� de-�Z7G dd� de-�Z8G dd� de-�Z9G d d � d e-�Z:G d d � d e-�Z;dS ) � )�division�absolute_import�print_functionN) �assert_� assert_equal� assert_raises�assert_almost_equal�assert_array_equal�IS_PYPY�suppress_warnings�_gen_alignment_data� assert_warnsc @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) � TestTypesc C s, t D ]"}|d�}t|dkd||f � qd S )N� �error with %r: got %r)�typesr )�self�atype�a� r �B/usr/lib/python3/dist-packages/numpy/core/tests/test_scalarmath.py� test_types s zTestTypes.test_typesc C s� t t�D ]�\}}|d�}tjdg|d�}t t�D ]\\}}|d�}tjdg|d�}|| } || } t| j| jd|t�|�j|t�|�jf � q0qd S )N� ��dtyper z$error with types (%d/'%c' + %d/'%c'))� enumerater �np�arrayr r �char) r �kr Za_scalarZa_array�lZbtypeZb_scalarZb_arrayZc_scalarZc_arrayr r r � test_type_add" s ��zTestTypes.test_type_addc C s@ t t�D ]2\}}t�dddg|�}|dddg�}t||� qd S )Nr � r )r r r r r )r r r r �br r r �test_type_create6 s zTestTypes.test_type_createc C s t d�D ]}t�dd� qd S )Ni@ r )�ranger �add)r �ir r r � test_leak< s zTestTypes.test_leakN)�__name__� __module__� __qualname__r r! r$ r( r r r r r s r c @ s e Zd Zdd� Zdd� ZdS )� TestBaseMathc C s� t jdft jdft jdffD �]�\}}t|d|d�D �]p\}}}}t �|�}t �|�|d< t �|�|d< tt �||�||d� tt �|d�|d |d� tt �d|�||d� t j|||d � t|||d� |d t j |j |d �d 7 < tt � |�t � ||�|d� |t jk�s*t jjd k �rFt j�sFtt �|�t �d|�|d� t �|�|d< t j|d|d � t||d |d� t �|�|d< t jd||d � t||d |d� q4q d S ) N� � Zbinary)r �typeZmax_size.��err_msgr" r ��outr r )r �float32�float64�int32r � ones_likeZ zeros_liker r& Zarange�sizeZsquareZmultiply�sys� version_info�majorZ py3kwarningZ reciprocalZdivide)r �dtZszr3 Zinp1Zinp2�msgZexp1r r r � test_blockedD s: &� �" �zTestBaseMath.test_blockedc C s� t jdt jd�dd� �t j�}t jdt jd�dd� �t j�}t|| |d � t j|||d� t jt �|�||d� t j|t �|�|d� t �t �|�|� t �|t �|�� d S )N� r � ���r" r2 )r Zzeros�int8�viewr5 r r& r7 )r �d�or r r �test_lower_alignd s z TestBaseMath.test_lower_alignN)r) r* r+ r> rF r r r r r, C s r, c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) � TestPowerc C s@ t jt jt jfD ]*}|d�}|d }t|dkd||f � qd S )Nr r@ �Q r )r rB �int16�float16r �r �tr r# r r r �test_small_typesr s zTestPower.test_small_typesc C sj t jt jt jt jt jfD ]L}|d�}|d }d||f }t �|t j�rVt|dk|� qt |d|d� qd S )N�3 r@ r i�:g r0 ) r r6 �int64r4 r5 � longdouble� issubdtype�integerr r )r rL r r# r= r r r �test_large_typesx s zTestPower.test_large_typesc C sR dd� dD �}dd� dD �}t �||�D ]N\}}|jtjkrNtttj||� q(t�||�}t |jj tj k� t |d� q(dd� dD �}t �||�D ]N\}}|jtjkr�tttj||� q�t�||�}t |jj tj k� t |d� q�d d� dD �}t �||�D ]P\}}|jtjk�r$tttj||� q�t�||�}t |jj tj k� t |d � q�d S ) Nc S s g | ]}t �d |�d �qS ����r �r r ��.0r< r r r � <listcomp>� s zETestPower.test_integers_to_negative_integer_power.<locals>.<listcomp>Zbhilqc S s g | ]}t �d |�d �qS )r r rV rW r r r rY � s Z bhilqBHILQ� �?c S s g | ]}t �d |�d �qS rT rV rW r r r rY � s � �c S s g | ]}t �d |�d �qS )r" r rV rW r r r rY � s g �?) � itertools�productr r �uint64r � ValueError�operator�powr r/ r5 r )r �exp�baseZi1Zi2�resr r r �'test_integers_to_negative_integer_power� s, z1TestPower.test_integers_to_negative_integer_powerc C s� t jt jt jt jt jt jt jt jt jg }|D ]h}|D ]^}|d�}|d�}|| }d|||df }t �t � |�t j �r�t |dk|� q4t |d|d� q4q,d S )Nr r" z(error with %r and %r:got %r, expected %r� r0 ) r rB rI rJ r4 r5 r6 rO rQ r rR r r )r Ztypelist�t1�t2r r# �resultr= r r r �test_mixed_types� s& � �zTestPower.test_mixed_typesc C sd d}d}d}t |||�}tjtjtjfD ]6}tttj ||�||� tttj t�||��||� q(d S )N� r@ � ) ra r r6 r4 � complex64r � TypeErrorr` r )r r r# �cZexpectedrL r r r �test_modular_power� s z TestPower.test_modular_powerN)r) r* r+ rM rS re rj rp r r r r rG q s $rG c C s | | | | fS �Nr )�x�yr r r �floordiv_and_mod� s rt c C s | t jd krdS dS d S )NZUnsignedInteger)r �r rU )r � typecodes)r< r r r �_signs� s rw c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) � TestModulusc C s t jd t jd }ttfD ]�}t�||�D ]�\}}t�t|�t|��D ]�\}}d}||j||||f }t j|d |d�d } t j|d |d�d } || | �\} } t | | | | |d� |d kr�t | | k o�d kn |� qHt | | ko�d kn |� qHq,q d S ) N� AllInteger�Float�*op: %s, dt1: %s, dt2: %s, sg1: %s, sg2: %s�G r r � r0 rU r ) r rv rt �divmodr\ r] rw r) r r r � r r< �op�dt1�dt2Zsg1Zsg2Zfmtr= r r# �div�remr r r �test_modulus_basic� s z TestModulus.test_modulus_basicc s8 t tdd��}t tdd��}|dg | }|| }t t�||��}t dd� |D ��}tj|td�j\}}tj|td�j\} } t� | dk|dk |dk A @ d | �} t� | dk|dk @ d | �} t t fD ]r� tj d D ]b} d � j | f } |�| �} |�| �}t� fd d �t| |�D �� \}}t|| | d� t|| | d� q�q�d S )Ni����r r � c s s | ]}t |� V qd S rq )r~ )rX rL r r r � <genexpr>� s z7TestModulus.test_float_modulus_exact.<locals>.<genexpr>r � � �rz zop: %s, dtype: %sc s g | ]\}}� ||��qS r r )rX Za_Zb_�r� r r rY � s z8TestModulus.test_float_modulus_exact.<locals>.<listcomp>r0 )�listr% r\ r] r r �int�T�float�wherert r~ rv r) Zastype�zipr )r ZnlstZplstZdividendZdivisor�arg�tgtr r# ZtgtdivZtgtremr< r= ZfaZfbr� r� r r� r �test_float_modulus_exact� s$ " z$TestModulus.test_float_modulus_exactc C s� t jd }ttfD ]�}t�||�D ]�\}}t�dd�D ]�\}}d}||j||||f }t j|d d |d�d } t j|d |d�d } || | �\} } t| | | | |d� |d kr�t | | k o�d kn |� q6t | | ko�d kn |� q6q"qd S ) Nrz ru r{ �N g+i�)+p>r r r0 rU r ) r rv rt r~ r\ r] r) r r r r r r r �test_float_modulus_roundoff� s z'TestModulus.test_float_modulus_roundoffc C sF t jd D ]l}t jd|d�}t �t jd|d�| �}t�||�}t||kd| � t�| | �}t|| kd| � q t� ��}|�t d� t jd D ]�}t jd|d�}t jd|d�}t jt j |d�}t jt j |d�} t�||�}tt � |�d| � t�|| �}tt � |�d| � t�||�}tt � |�d| � q�W 5 Q R X d S )Nrz rZ r r� zdt: %sz&invalid value encountered in remainder) r rv r Z nextafterr` �modr r �filter�RuntimeWarning�inf�nan�isnan) r r< r# r r� �supZfoneZfzerZfinfZfnanr r r � test_float_modulus_corner_cases s( z+TestModulus.test_float_modulus_corner_casesN)r) r* r+ r� r� r� r� r r r r rx � s rx c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestComplexDivisionc C s� t jdd��� t jt jfD ]�}|d�}|d�}tt �|| �� |tt jt j��}tt �|| �� |tt jt j��}tt �|| �� |tt jt j��}tt �|| �� |tt jt j��}tt � || �� |d�}tt � || �� qW 5 Q R X d S )N�ignore��allr� rZ ) r �errstaterm � complex128r Zisinf�complexr� r� r� rK r r r �test_zero_division* s z&TestComplexDivision.test_zero_divisionc C s� t jdd��� t jt jfD ]v}d}|D ]h}|d }|d }|d }|t|d |d ��|t|d |d �� }t|j|d � t|j|d � q&qW 5 Q R X d S )Nr� r� ))�r� r[ �r� rZ �r[ r� �r� r� �rZ r� )r� �r� r[ �rZ r� )r� �r� rZ �r[ r� )r� r� r� r� )r� r� r� )r� r� r� r r r" )r r� rm r� r� r �real�imag�r rL �dataZcases�nrD �exri r r r �test_signed_zeros; s ,z%TestComplexDivision.test_signed_zerosc C s� t jdd��� t jt jfD ]�}t� }|�d� |�d� |D ]h}|d }|d }|d }|t|d |d ��|t|d |d �� }t|j|d � t|j |d � q<qW 5 Q R X d S )Nr� r� )�� @rZ r� r� )�rZ r� r� r� r r r" ) r r� rm r� r� �appendr� r r� r� r� r r r � test_branchesT s ,z!TestComplexDivision.test_branchesN)r) r* r+ r� r� r� r r r r r� ) s r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zejj e d d �d d � �Z ejj e � e j�e � e j�kd d �ejj e�� �d�dd �dd� ��Zdd� Zdd� Zdd� ZdS )�TestConversionc C s� ddddddg}ddd d d d g}d t jt jfD ]&}t j||d �}tdd� |D �|� q.t j|d d� t jd �}tdd� |D �|d d� � d S )Ng ��.Ag ��mBg �Ngm��Cg ��.�g ��m�g �Ngm���i@B l J)�l �N�Zoi���l���� J)�l���� �N�Zor c S s g | ] }t |��qS r �r� �rX �_mr r r rY w s z5TestConversion.test_int_from_long.<locals>.<listcomp>r c S s g | ] }t |��qS r r� r� r r r rY z s )r r5 rO r r r^ )r r Zlir� r r r r �test_int_from_longr s z!TestConversion.test_int_from_longc C s� dD ]6}t jt �|�jd |d�}t �|�j}t||k� qt jd D ]2}t jt �|�j|d�}t �|�j}t||k� qFt jd D ]2}t j| t �|�j�}t �|�j}t||k� q�d S )NZbBhHr r ry )r r �iinfo�max�minr rv �typeDict)r �coderd r� r r r �test_iinfo_long_values| s z%TestConversion.test_iinfo_long_valuesc C s" dd� }dD ]}t t||� q d S )Nc S s t j| t �| �jd � d S �Nr )r r� r� r� r r r r �overflow_error_func� s zDTestConversion.test_int_raise_behaviour.<locals>.overflow_error_funcZlLqQ)r � OverflowError)r r� r� r r r �test_int_raise_behaviour� s z'TestConversion.test_int_raise_behaviourc C sb t �t j�}ttt|� t� �:}|�t j� t � t j�}ttt|� t t |j �d� W 5 Q R X d S r� ) r rP r� r r� r� r �record�ComplexWarning� clongdoubler �len�log�r rr r� r r r �!test_int_from_infinite_longdouble� s z0TestConversion.test_int_from_infinite_longdoublezTest is PyPy only (gh-9972))�reasonc C sb t �t j�}tt|j� t� �:}|�t j� t � t j�}tt|j� t t |j �d� W 5 Q R X d S r� ) r rP r� r r� �__int__r r� r� r� r r� r� r� r r r �)test_int_from_infinite_longdouble___int__� s z8TestConversion.test_int_from_infinite_longdouble___int__z long double is same as doubleZppczIBM double doublec C sP t �t j�jd }dt �d�| }dd| }t|t jk� tt|�|� d S )Nr i� r" ) r �finfo�doubleZmaxexprP r r� r r� )r rb Zhuge_ldZhuge_ir r r � test_int_from_huge_longdouble� s z,TestConversion.test_int_from_huge_longdoublec C s4 t �d�}tt|�d� t �d�}tt|�d� d S )N� �?r g %�i����)r rP r r� )r rr r r r �test_int_from_longdouble� s z'TestConversion.test_int_from_longdoublec C s t jd D ]�}tdt jd|d�d kd|f � tdt jd|d�d k d|f � t jd D ]j}tt jd|d�d t jd|d�d kd||f � tt jd|d�d t jd|d�d k d||f � q^q dD �]}td t jd|d�d k d|f � td t jd|d�d k d|f � td t jd|d�d kd|f � d D ]�}tt jd|d�d t jd |d�d kd||f � tt jd|d�d t jd |d�d k d||f � tt jd|d�d t jd |d�d kd||f � �qBq�d t jd D �] }tdt jd |d�d kd|f � tdt jd |d�d k d|f � td t jd |d�d kd|f � d t jd D ]�}tt jd|d�d t jd |d�d kd||f � tt jd|d�d t jd |d�d k d||f � tt jd |d�d t jd |d�d kd||f � �qn�q�d S ) Nry r r r r ztype %s failedztype %s and %s failedZBHILQPrU ZbhilqpZbhlqprz )r rv r r )r r� r� r r r �&test_numpy_scalar_relational_operators� sL "$$ �& � "$"$ �& �$ � "$"$ �& �$ �z5TestConversion.test_numpy_scalar_relational_operatorsc C s� t jdd���}t �ddt� tt�d�d k � tt�d�d k � tt�d�d k � tt�d�d k� tt�d�d k� tt�d�d k� W 5 Q R X tt |�dk� tt� t�d�d �� d S ) NT)r� �always� r ZtestZNaTr ) �warnings�catch_warnings�filterwarnings� FutureWarningr r r4 Zstr_Z datetime64r� Zequal)r �wr r r � test_scalar_comparison_to_none� s z-TestConversion.test_scalar_comparison_to_noneN)r) r* r+ r� r� r� r� �pytestZmarkZskipifr r� r r� r� rP �platform�machine� startswithr� r� r� r� r r r r r� q s �� )r� c @ s e Zd Zdd� Zdd� ZdS )�TestReprc C s� t �|�}|j|j }|j}t �|�jd }dD ]�}t jdg| t jd�}|dkrt|d }d|d } d| > ||< n2|dkr�|d }d|d } d| > ||< ntd ��|� |�d } t | �} |t | ��} | dkr�| d k s0t | | � q0d S ) N� )� small denorm� small normr r r� r. r r� Zhmmg0��.�++) r r� ZnexpZnmantr �itemsizer Zuint8r_ rC �repr�evalr ) r rL r� Zlast_fraction_bit_idxZlast_exponent_bit_idxZ storage_bytesZwhichZconstr�byteZbytebit�valZval_reprZval2r r r �_test_type_repr s( zTestRepr._test_type_reprc C s t jt jfD ]}| �|� q d S rq )r r4 r5 r� )r rL r r r �test_float_repr s zTestRepr.test_float_reprN)r) r* r+ r� r� r r r r r� s r� c @ s e Zd Zdd� Zdd� ZdS )� TestSizeOfc C s* t D ] }|d�}tt�|�|jk� qd S )Nr )r r r9 � getsizeof�nbytes)r r/ rr r r r �test_equal_nbytes' s z TestSizeOf.test_equal_nbytesc C s t �� }tt|jd� d S )Nr )r r4 r rn � __sizeof__)r rD r r r � test_error, s zTestSizeOf.test_errorN)r) r* r+ r� r� r r r r r� % s r� c @ s e Zd Zdd� Zdd� ZdS )� TestMultiplyc C s& t tjd �}dh}t tjd �| | }|dh8 }ttfD ]�}|dddg�}|D ]@}t�|��d�}t|| |t|� � t|| t|�| � qR|D ]N}t�|��� }tt t t j ||�|t|� � tt t t j ||�t|�| � q�|D ]2}t�|��� }t tt j ||� t tt j ||� q�q<d S )Nry �?ZAll�Vr r" r )�setr rv r� �tupler r/ r r� r �DeprecationWarningr` �mulr rn )r Zaccepted_typesZdeprecated_typesZforbidden_typesZseq_type�seqZ numpy_typer' r r r �test_seq_repeat2 s2 � � �z TestMultiply.test_seq_repeatc C s� G dd� dt �}|t�d��tt�d��fD ]t}t|t�d� t�dd�� tt�d�| t�dd�� t|t�d� t�dd�� tt�d�| t�dd�� q,d S )Nc @ s e Zd Zdd� Zdd� ZdS )zCTestMultiply.test_no_seq_repeat_basic_array_like.<locals>.ArrayLikec S s || _ d S rq ��arr)r r r r r �__init__W s zLTestMultiply.test_no_seq_repeat_basic_array_like.<locals>.ArrayLike.__init__c S s | j S rq r �r r r r � __array__Y s zMTestMultiply.test_no_seq_repeat_basic_array_like.<locals>.ArrayLike.__array__N)r) r* r+ r r r r r r � ArrayLikeV s r r g @)�objectr �ones� memoryviewr r4 Zfull�int_)r r Zarr_liker r r �#test_no_seq_repeat_basic_array_likeR s z0TestMultiply.test_no_seq_repeat_basic_array_likeN)r) r* r+ r� r r r r r r� 1 s r� c @ s e Zd Zdd� Zdd� ZdS )� TestNegativec C s& t jdt jd�d }tttj|� d S �Nr r )r r �bool_r rn r` �neg�r r r r r �test_exceptionse s z TestNegative.test_exceptionsc C sd t jd t jd }t� �@}|�t� |D ]*}t jd|d�d }tt�|�| d� q*W 5 Q R X d S �Nry ZAllFloatr r r ) r rv r r� r� r r r` r �r r r� r< r r r r � test_resulti s zTestNegative.test_resultN�r) r* r+ r r r r r r r d s r c @ s e Zd Zdd� Zdd� ZdS )� TestSubtractc C s( t jdt jd�d }tttj||� d S r )r r r r rn r` �subr r r r r s s z TestSubtract.test_exceptionsc C sb t jd t jd }t� �>}|�t� |D ](}t jd|d�d }tt�||�d� q*W 5 Q R X d S r ) r rv r r� r� r r r` r r r r r r w s zTestSubtract.test_resultNr r r r r r r s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestAbsc C s� t t D ]�}|d�}t||�d� |d�}||�}t|d� |d�}||�}t|d� |t�|�j�}t||�|j� |t�|�j�}t||�|j� |t�|�j�}t||�|j � qd S )Ng ��r� r� r� ) �floating_types�complex_floating_typesr r r� r� r� Ztinyr� )r Zabsfunc�tprr rd r r r �_test_abs_func� s zTestAbs._test_abs_funcc C s | � t� d S rq )r �absr r r r �test_builtin_abs� s zTestAbs.test_builtin_absc C s | � tj� d S rq )r r r r r r r �test_numpy_abs� s zTestAbs.test_numpy_absN)r) r* r+ r r r r r r r r � s r )<Z __future__r r r r9 r� r\ r` r� r� Znumpyr Z numpy.testingr r r r r r r r r r r� ZubyteZshortZushortZintcZuintcr ZuintZlonglongZ ulonglongZsingler� rP ZcsingleZcdoubler� r Zfloating�__subclasses__r Zcomplexfloatingr r r r, rG rt rw rx r� r� r� r� r� r r r r r r r �<module> sL , � '.Q\H ! 3