Edit File: test_scalarmath.cpython-37.pyc
B ��Fd�a � @ s� d dl mZmZmZ d dlZd dlZd dlZd dlZd dlZ d dl m Z d dl m Z mZmZmZmZmZmZmZmZmZ e 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&e j'gZ(e j)�*� Z+G dd� de �Z,G dd� de �Z-G d d � d e �Z.d d � Z/d d� Z0G dd� de �Z1G dd� de �Z2G dd� de �Z3G dd� de4�Z5e�s`G dd� de �Z6G dd� de �Z7G dd � d e �Z8G d d � d e �Z9G d d � d e �Z:e;d!k�r�e� dS )"� )�division�absolute_import�print_functionN)�_gen_alignment_data) �TestCase�run_module_suite�assert_� assert_equal� assert_raises�assert_almost_equal�assert_allclose�assert_array_equal�IS_PYPY�suppress_warningsc @ s2 e Zd Zd dd�Zd dd�Zd dd�Zdd � Zd S )� TestTypes� c C s0 x*t D ]"}|d�}t|dkd||f � qW d S )Nr zerror with %r: got %r)�typesr )�self�level�atype�a� r ��/builddir/build/BUILDROOT/alt-python37-numpy-1.13.3-8.el8.x86_64/opt/alt/python37/lib64/python3.7/site-packages/numpy/core/tests/test_scalarmath.py� test_types s zTestTypes.test_typesc C s� x�t t�D ]�\}}|d�}tjdg|d�}xht t�D ]\\}}|d�}tjdg|d�} || } || } t| j| jd|t�|�j|t�|�jf � q4W q W d S )N� )�dtyper z$error with types (%d/'%c' + %d/'%c'))� enumerater �np�arrayr r �char) r r �kr Za_scalarZa_array�lZbtypeZb_scalarZb_arrayZc_scalarZc_arrayr r r � test_type_add s zTestTypes.test_type_addc C sD x>t t�D ]2\}}t�dddg|�}|dddg�}t||� q W d S )Nr � r )r r r r r )r r r r r �br r r �test_type_create4 s zTestTypes.test_type_createc C s" x t d�D ]}t�dd� q W d S )Ni@ r )�ranger �add)r �ir r r � test_leak: s zTestTypes.test_leakN)r )r )r )�__name__� __module__� __qualname__r r" r% r) r r r r r s r c @ s e Zd Zdd� Zdd� ZdS )� TestBaseMathc C s� �x�t jdft jdft jdfgD �]�\}}�x�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�s2t jjd k �rNt j�sNtt �|�t �d|�|d� t �|�|d< t j|d|d � t||d |d� t �|�|d< t jd||d � t||d |d� q<W q W d S ) N� � �binary)r �type�max_size.)�err_msgr# r )�out)r r )r �float32�float64�int32r � ones_like� zeros_liker r' �arange�size�square�multiply�sys� version_info�major� py3kwarning� reciprocal�divide)r �dt�szr4 �inp1�inp2�msgZexp1r r r � test_blockedB s0 * " 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# )r4 )r �zeros�int8�viewr6 r r' r8 )r �d�or r r �test_lower_alignb s z TestBaseMath.test_lower_alignN)r* r+ r, rI rR r r r r r- A s r- c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) � TestPowerc C sD x>t jt jt jgD ]*}|d�}|d }t|dkd||f � qW d S )Nr rK �Q zerror with %r: got %r)r rN �int16�float16r )r �tr r$ r r r �test_small_typesp s zTestPower.test_small_typesc C sn xht jt jt jt jt jgD ]L}|d�}|d }d||f }t �|t j�rXt|dk|� qt |d|d� qW d S )N�3 rK zerror with %r: got %ri�:g )r3 ) r r7 �int64r5 r6 � longdouble� issubdtype�integerr r )r rW r r$ rH r r r �test_large_typesv s zTestPower.test_large_typesc C s` dd� dD �}dd� dD �}x^t �||�D ]N\}}|jjdkrPtttj||� q*t�||�}t|jj t j k� t |d� q*W dd� dD �}x^t �||�D ]N\}}|jjdkr�tttj||� q�t�||�}t|jj t j k� t |d � q�W d d� dD �}xbt �||�D ]R\}}|jjdk�r.tttj||� n(t�||�}t|jj t j k� t |d � �qW d S ) Nc S s g | ]}t �d |�d �qS )���r )r r )�.0rD r r r � <listcomp>� s zETestPower.test_integers_to_negative_integer_power.<locals>.<listcomp>Zbhilqc S s g | ]}t �d |�d �qS )r r )r r )r` rD r r r ra � s Z bhilqBHILQ�uint64g �?c S s g | ]}t �d |�d �qS )r_ r )r r )r` rD r r r ra � s g �c S s g | ]}t �d |�d �qS )r# r )r r )r` rD r r r ra � s g �?) � itertools�productr �namer � ValueError�operator�powr r1 r r6 r )r �exp�base�i1�i2�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 }xt|D ]l}xf|D ]^}|d�}|d�}|| }d|||df }t �t � |�t j �r�t |dk|� q8t |d|d� q8W q.W d S )Nr r# z(error with %r and %r:got %r, expected %r� )r3 ) r rN rU rV r5 r6 r7 rZ r\ r r] r r )r Ztypelist�t1�t2r r$ �resultrH r r r �test_mixed_types� s zTestPower.test_mixed_typesc C sh d}d}d}t |||�}xJtjtjtjfD ]6}tttj ||�||� tttj t�||��||� q*W d S )N� rK � ) rh r r7 r5 � complex64r � TypeErrorrg r )r r r$ �c�expectedrW r r r �test_modular_power� s z TestPower.test_modular_powerN)r* r+ r, rX r^ rn rs rz r r r r rS o s $rS c C s | | | | fS )Nr )�x�yr r r �floordiv_and_mod� s r} c C s | t jd krdS dS d S )N�UnsignedInteger)r )r r_ )r � typecodes)rD r r r �_signs� s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) � TestModulusc C s t jd t jd }x�ttgD ]�}x�t�||�D ]�\}}x�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 |� qNt | | ko�d kn |� qNW q0W q W d S ) N� AllInteger�Floatz*op: %s, dt1: %s, dt2: %s, sg1: %s, sg2: %s�G )r r � )r3 r_ r ) r r r} �divmodrc rd r� r* r r r ) r rD �op�dt1�dt2�sg1�sg2�fmtrH r r$ �div�remr r r �test_modulus_basic� s z TestModulus.test_modulus_basicc s@ 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 | �} x�t t gD ]v� xptj d D ]b} d � j | f } |�| �} |�| �}t� fd d �t| |�D �� \}}t|| | d� t|| | d� q�W q�W d S )Ni����r r � c s s | ]}t |� V qd S )N)r� )r` rW r r r � <genexpr>� s z7TestModulus.test_float_modulus_exact.<locals>.<genexpr>)r g g �r� zop: %s, dtype: %sc s g | ]\}}� ||��qS r r )r` �a_Zb_)r� r r ra � s z8TestModulus.test_float_modulus_exact.<locals>.<listcomp>)r3 )�listr& rc rd r r �int�T�float�wherer} r� r r* �astype�zipr )r ZnlstZplstZdividendZdivisor�arg�tgtr r$ ZtgtdivZtgtremrD rH �fa�fbr� r� r )r� r �test_float_modulus_exact� s$ " z$TestModulus.test_float_modulus_exactc C s t jd }x�ttgD ]�}x�t�||�D ]�\}}x�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 |� q<t | | ko�d kn |� q<W q&W qW d S ) Nr� )r r_ z*op: %s, dt1: %s, dt2: %s, sg1: %s, sg2: %s�N g+i�)+p>)r r )r3 r_ r ) r r r} r� rc rd r* r r r ) r rD r� r� r� r� r� r� rH r r$ r� r� r r r �test_float_modulus_roundoff� s z'TestModulus.test_float_modulus_roundoffc C sN xzt jd D ]l}t jd|d�}t �t jd|d�| �}t�||�}t||kd| � t�| | �}t|| kd| � q W t� ��}|�t d� x�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 W d Q R X d S )Nr� g �?)r g zdt: %sz&invalid value encountered in remainder) r r r � nextafterrg �modr r �filter�RuntimeWarning�inf�nan�isnan) r rD 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 r� � s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestComplexDivisionc C s� t jdd��� x�t jt jgD ]�}|d�}|d�}tt �|| �� |tt jt j��}tt �|| �� |tt jt j��}tt �|| �� |tt jt j��}tt �|| �� |tt jt j��}tt � || �� |d�}tt � || �� q W W d Q R X d S )N�ignore)�allg g �?) r �errstaterv � complex128r �isinf�complexr� r� r� )r rW r r$ r r r �test_zero_division( s z&TestComplexDivision.test_zero_divisionc C s� t jdd��� x�t jt jgD ]z}d}xp|D ]h}|d }|d }|d }|t|d |d ��|t|d |d �� }t|j|d � t|j|d � q*W q W W d Q R X d S )Nr� )r� )))g g �)g g �?)g �g �))g g �)g g �)g �?g �))g g �)g �g �)g �?g ))g g �)g �g �?)g �g ))g g �?)g g �)g �g ))g g �)g g �)g �?g �))g �g �)g g �)g �?g �))g �g �?)g g �)g �g �r r r# )r r� rv r� r� r �real�imag)r rW �data�cases�nrP �exrr r r r �test_signed_zeros9 s ,z%TestComplexDivision.test_signed_zerosc C s� t jdd��� x�t jt jgD ]�}t� }|�d� |�d� xp|D ]h}|d }|d }|d }|t|d |d ��|t|d |d �� }t|j|d � t|j |d � q@W q W W d Q R X d S )Nr� )r� ))g @g �?)g @g �?)g �?g ))g �?g @)g �?g @)g �?g r r r# ) r r� rv r� r� �appendr� r r� r� )r rW r� r� r� rP r� rr r r r � test_branchesR 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� Zd d � Zd d � Zd S )�TestConversionc C s� ddddddg}ddd d d d g}x8d t jt jgD ]&}t j||d �}tdd� |D �|� q0W 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�Zo)r c S s g | ] }t |��qS r )r� )r` �_mr r r ra u s z5TestConversion.test_int_from_long.<locals>.<listcomp>r c S s g | ] }t |��qS r )r� )r` r� r r r ra x s )r r6 rZ r r rb )r r! Zlir� r r r r �test_int_from_longp s z!TestConversion.test_int_from_longc C s� x>dD ]6}t jt �|�jd |d�}t �|�j}t||k� qW x@t jd D ]2}t jt �|�j|d�}t �|�j}t||k� qLW x@t jd D ]2}t j| t �|�j�}t �|�j}t||k� q�W d S )NZbBhHr )r r� )r r �iinfo�max�minr r �typeDict)r �coderm r� r r r �test_iinfo_long_valuesz s z%TestConversion.test_iinfo_long_valuesc C s& dd� }xdD ]}t t||� qW 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 s4 t �t j�}tt|j� t �t j�}tt|j� d S )N)r r[ r� r r� �__int__� clongdouble)r r{ r r r �test_longdouble_int� s z"TestConversion.test_longdouble_intc C s2 x�t jd D ]�}tdt jd|d�d kd|f � tdt jd|d�d k d|f � xxt jd D ]j}tt jd|d�d t jd|d�d kd||f � tt jd|d�d t jd|d�d k d||f � qbW q W �x dD �]}td t jd|d�d k d|f � td t jd|d�d k d|f � td t jd|d�d kd|f � x�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 � �qPW q�W �x4d t jd D �] }tdt jd |d�d kd|f � tdt jd |d�d k d|f � td t jd |d�d kd|f � x�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 � �q�W �qW d S ) Nr� r r )r r ztype %s failedztype %s and %s failed�BHILQPr_ �bhilqp�bhlqpr� )r r r r )r r� r� r r r �&test_numpy_scalar_relational_operators� s< "$$&"$" $&$"$"$&$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 d Q R X tt |�dk� tt� t�d�d �� d S ) NT)�record�always� r �test�NaTr ) �warnings�catch_warnings�filterwarnings� FutureWarningr r r5 �str_� datetime64�len�equal)r �wr r r � test_scalar_comparison_to_none� s z-TestConversion.test_scalar_comparison_to_noneN) r* r+ r, r� r� r� r� r� r� r r r r r� o s )r� c @ s e Zd Zdd� Zdd� ZdS )�TestReprc C s� t �|�}|j|j }|j}t �|�jd }x�dD ]�}t jdg| t jd�}|dkrv|d }d|d } d| > ||< n2|dkr�|d }d|d } d| > ||< ntd ��|� |�d } t | �} |t | ��} | dkr�| d k s2t | | � q2W d S ) N� )z small denormz small normr )r z small denormr/ r z small normZhmmg0��.�++) r �finfo�nexp�nmantr �itemsizer �uint8rf rO �repr�evalr ) r rW r� Zlast_fraction_bit_idxZlast_exponent_bit_idxZ storage_bytes�whichZconstr�byteZbytebit�valZval_repr�val2r r r �_test_type_repr� s( zTestRepr._test_type_reprc c s& x t jt jgD ]}| j|fV qW d S )N)r r5 r6 r� )r rW 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. x(t D ] }|d�}tt�|�|jk� qW d S )Nr )r r r> � getsizeof�nbytes)r r1 r{ r r r �test_equal_nbytes s z TestSizeOf.test_equal_nbytesc C s t �� }tt|jd� d S )Nr )r r5 r rw � __sizeof__)r rP 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� x�t tfD ]�}|dddg�}xNtjd D ]@}t�|��d�}t|| |t|� � t|| t|�| � q(W xXtjd �dd�D ]B}|tjd kr�q�t�|��� }t t t j ||� t t t j ||� q�W q W d S )Nr r# r r� �All�Vr� ) r� �tupler r r r1 r r� �replacer rw rg �mul)r �seq_type�seqZ numpy_typer( r r r �test_seq_repeat s z TestMultiply.test_seq_repeatc C s� G dd� dt �}x�|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.W d S )Nc @ s e Zd Zdd� Zdd� ZdS )zCTestMultiply.test_no_seq_repeat_basic_array_like.<locals>.ArrayLikec S s || _ d S )N)�arr)r r� r r r �__init__( s zLTestMultiply.test_no_seq_repeat_basic_array_like.<locals>.ArrayLike.__init__c S s | j S )N)r� )r r r r � __array__* s zMTestMultiply.test_no_seq_repeat_basic_array_like.<locals>.ArrayLike.__array__N)r* r+ r, r� r� r r r r � ArrayLike' s r� r g @)�objectr �ones� memoryviewr r5 �full�int_)r r� Zarr_liker r r �#test_no_seq_repeat_basic_array_like# s " z0TestMultiply.test_no_seq_repeat_basic_array_likeN)r* r+ r, r� r r r r r r� 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 rw rg �neg)r r r r r �test_exceptions6 s z TestNegative.test_exceptionsc C sh t jd t jd }t� �D}|�t� x2|D ]*}t jd|d�d }tt�|�| d� q,W W d Q R X d S )Nr� �AllFloatr )r r ) r r r r� r� r r rg r )r r r� rD r r r r � test_result: s zTestNegative.test_resultN)r* r+ r, r r r r r r r 5 s r c @ s e Zd Zdd� ZdS )� TestSubtractc C sj t jd t jd d }t� �B}|�t� x0|D ](}t jd|d�d }tt�||�d� q0W W d Q R X d S )Nr� r �?r )r r ) r r r r� r� r r rg �sub)r r r� rD r r r r r E s zTestSubtract.test_resultN)r* r+ r, r r r r r r C s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestAbsc C s\ xVt D ]N}|d�}t||�d� |d�}||�}t|d� |d�}||�}t|d� qW d S )Ng ��g �?g g �)�floating_typesr )r Zabsfunc�tpr{ rm r r r �_test_abs_funcP s zTestAbs._test_abs_funcc C s | � t� d S )N)r �abs)r r r r �test_builtin_abs\ s zTestAbs.test_builtin_absc C s | � tj� d S )N)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 N s r �__main__)<� __future__r r r r> r� rc rg �numpyr Znumpy.testing.utilsr Z numpy.testingr r r r r r r r r r r r� �ubyte�short�ushort�intc�uintcr �uint�longlong� ulonglong�single�doubler[ �csingle�cdoubler� r �floating�__subclasses__r r r- rS r} r� r� r� r� r� r� r� r� r r r r* r r r r �<module> s: 0 '.Q\Hq! &