Edit File: test_extint128.cpython-37.pyc
B ��Fd� � @ s$ d dl mZmZmZ d dlZd dlZd dlZd dlZd dlZ d dl m m Z d dlmZ d dlmZmZmZ e �e j�jZe �e j�jZdZdd d Ze Z d Z d d � e d �D �d d � e d �D � dd � e dd �D � dd � e dd �D � dd � e dd �D � e e dd�� Z dd � e d �D �dd � e d �D � dd � e dd �D � dd � e dd �D � dd � e dd �D � e e dd�� dg Z!dd � e D �Z"ej#dd � �Z$d d � Z%d d � Z&d!d"� Z'd#d$� Z(d%d&� Z)d'd(� Z*d)d*� Z+d+d,� Z,d-d.� Z-d/d0� Z.ej/d1d2� �Z0d3d4� Z1d5d6� Z2e3d7k�r e4� dS )8� )�division�absolute_import�print_functionN)�long)� assert_raises� assert_equal�decl � � � l c C s g | ] }t | �qS � )� INT64_MIN)�.0�jr 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_extint128.py� <listcomp> s r � c C s g | ] }t | �qS r )� INT64_MAX)r r r r r r s c C s g | ] }t | �qS r )� INT64_MID)r r r r r r s i��c C s g | ]}d t | �qS )r )r )r r r r r r s c C s g | ]}t d | �qS )r )r )r r r r r r s i�����F c C s g | ] }t | �qS r )� INT128_MIN)r r r r r r " s c C s g | ] }t | �qS r )� INT128_MAX)r r r r r r # s c C s g | ] }t | �qS r )� INT128_MID)r r r r r r $ s c C s g | ]}d t | �qS )r )r )r r r r r r % s c C s g | ]}t d | �qS )r )r )r r r r r r & s Fc C s g | ]}|d kr|�qS )r r )r �xr r r r + s c ' sZ dg�� �fdd�}y |� V W n4 ddl }dt�d �|�� f }t|��Y nX dS )z} Iterate over Cartesian product of *args, and if an exception is raised, add information of the current iterate. Nc 3 s& x t j� � D ]} | �d< | V q W d S )Nr )� itertools�product)�v)�args�valuer r �iterate7 s zexc_iter.<locals>.iterater z At: %r %s)� traceback�repr� format_exc�AssertionError)r r r �msgr )r r r �exc_iter. s r% c C s� t jdft jdft jdfg} t| tt��v}xn|D ]f\}}}|\}}|||�}t| kr`tksvn tt t j |||� q0t � |||�}||kr0t ||� q0W W d Q R X d S )Nr r � ) �operator�add�sub�mulr% � INT64_VALUESr r r � OverflowError�mt�extint_safe_binopr ) Zops�itZxop�a�bZpyop�op�c�dr r r �test_safe_binopE s r5 c C sD t t��2} x*| D ]"\}t�|�}||krt||� qW W d Q R X d S )N)r% r+ r- � extint_to_128r )r/ r0 r1 r r r � test_to_128\ s r7 c C sj t t��X} xP| D ]H\}t| kr*tks<n tttj|� qt�|�}||krt||� qW W d Q R X d S )N) r% � INT128_VALUESr r r r, r- � extint_to_64r )r/ r0 r1 r r r � test_to_64d s r: c C sR t tt��>} x6| D ].\}}|| }t�||�}||krt||� qW W d Q R X d S )N)r% r+ r- �extint_mul_64_64r )r/ r0 r1 r3 r4 r r r �test_mul_64_64o s r< c C sz t tt��f} x^| D ]V\}}|| }t| kr6tksJn tttj||� qt�||�}||krt||� qW W d Q R X d S )N) r% r8 r r r r, r- �extint_add_128r )r/ r0 r1 r3 r4 r r r � test_add_128x s r> c C sz t tt��f} x^| D ]V\}}|| }t| kr6tksJn tttj||� qt�||�}||krt||� qW W d Q R X d S )N) r% r8 r r r r, r- �extint_sub_128r )r/ r0 r1 r3 r4 r r r � test_sub_128� s r@ c C sJ t t��8} x0| D ](\}| }t�|�}||krt||� qW W d Q R X d S )N)r% r8 r- �extint_neg_128r )r/ r0 r1 r3 r r r � test_neg_128� s rB c C sz t t��h} x`| D ]X\}|dk r8| d> dd d @ }n|d> dd d @ }t�|�}||krt||� qW W d Q R X d S )Nr r r r )r% r8 r- �extint_shl_128r )r/ r0 r1 r3 r r r � test_shl_128� s rD c C sb t t��P} xH| D ]@\}|dk r,| d? }n|d? }t�|�}||krt||� qW W d Q R X d S )Nr r )r% r8 r- �extint_shr_128r )r/ r0 r1 r3 r r r � test_shr_128� s rF c C sR t tt��>} x6| D ].\}}||k}t�||�}||krt||� qW W d Q R X d S )N)r% r8 r- � extint_gt_128r )r/ r0 r1 r3 r4 r r r � test_gt_128� s rH c C s� t tt���} x�| D ]�\}}|dkr2t||�\}}n t| |�\}}| }| }t�||�\}}||ks~||ks~|| | |krt||� t||� t|| | |� qW W d Q R X d S )Nr )r% r8 �INT64_POS_VALUES�divmodr- �extint_divmod_128_64r )r/ r0 r1 r3 �crr4 Zdrr r r �test_divmod_128_64� s rM c C sR t tt��>} x6| D ].\}}|| }t�||�}||krt||� qW W d Q R X d S )N)r% r8 rI r- �extint_floordiv_128_64r )r/ r0 r1 r3 r4 r r r �test_floordiv_128_64� s rO c C sZ t tt��F} x>| D ]6\}}|| d | }t�||�}||krt||� qW W d Q R X d S )Nr )r% r8 rI r- �extint_ceildiv_128_64r )r/ r0 r1 r3 r4 r r r �test_ceildiv_128_64� s rQ �__main__)5� __future__r r r �sysr � contextlibr' �numpy�np�numpy.core.multiarray_tests�core�multiarray_testsr- � numpy.compatr Z numpy.testingr r r �iinfo�int64�maxr �minr r r r r �range�listr+ r8 rI �contextmanagerr% r5 r7 r: r< r> r@ rB rD rF rH �slowrM rO rQ �__name__�run_module_suiter r r r �<module> sF ^ l