Edit File: test_index_tricks.cpython-37.pyc
B ��Fdd4 � @ s d dl mZmZmZ d dlZd dlmZmZm Z m Z m Z m Z m Z mZ d dlmZmZmZmZmZmZmZmZmZmZ G dd� de�ZG dd� de�ZG d d � d e�Z G d d � d e�Z G d d� de�Z G dd� de�Z dd� Z dd� Z!dd� Z"dd� Z#dd� Z$e%dk�r e� dS ) � )�division�absolute_import�print_functionN)�run_module_suite�TestCase�assert_� assert_equal�assert_array_equal�assert_almost_equal�assert_array_almost_equal� assert_raises) �mgrid� ndenumerate� fill_diagonal� diag_indices�diag_indices_from� index_exp�ndindex�r_�s_�ix_c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �TestRavelUnravelIndexc C s t t�dd�d� t t�dd�d� t t�dd�d� t t�dd�d� tttjdd� tttjdd� tttjd d� tttjd d� tttjd d� tttjd d� tttjd d� tttjdd� t t�dd�ddd g� t t�ddd gd�d� t�dddgd ddgg�}t t�|d�dddg� t tj|ddd�ddd g� t tj|d d d �dd d!g� t tj|d"d#d �d$d d g� t t�d%d&�d'� t t�t�dddg�d�dddgd ddgg� t tjt�ddd g�ddd�dddgd ddgg� t t�d'd&�ddd dg� d S )(N� )r r )� r � )� �^ )r �B ���g �?� )���r )r r )r r )r r )g�������?g �. )r � � r r" r# � )� r# � �) �% �F)�order� � )r r# �clip)�mode� � )r r )r- �wrap� )r" r r r )r# r% � � iU )r �np� unravel_index�ravel_multi_indexr � ValueError� TypeError�array)�self�arr� r= ��/builddir/build/BUILDROOT/alt-python37-numpy-1.13.3-8.el8.x86_64/opt/alt/python37/lib64/python3.7/site-packages/numpy/lib/tests/test_index_tricks.py� test_basic s< z TestRavelUnravelIndex.test_basicc C s� t jt jkrPddgddgddgddgdd gddgd dgf}tt �|d �d d g� d gd gf}t �t j�jd }tt �||df�d g� ttt j||d df� tt j||dfdd�d g� ttt j||d dfdd� d S )Nr � r" r$ �u r0 r iK i r )r' r% �x �$ i� r3 r# l �z� l nm r) )r* ) r5 �intp�int64r r7 �iinfo�maxr r8 )r; r< Z dummy_arrZhalf_maxr= r= r> �test_big_indices2 s z&TestRavelUnravelIndex.test_big_indicesc C s� �x�t jt jt jt jt jt jgD �]v}t jddddddgddddddgg|d�}d}d |d |d }tt � ||�|� t|t � ||�� |d d |d }tt j ||d d �|� t|t j ||d d �� t jddddddgddddddgddddd d gg|d�}d}dd |d |d |d }tt � ||�|� t|t � ||�� |d d |d d |d }tt j ||d d �|� t|t j ||d d �� q W d S )Nr r r r" r r# )�dtype)r$ r3 r3 r$ r) )r* r4 )r$ r3 � rJ ) r5 �int16�uint16�int32�uint32rE �uint64r: r r7 r6 )r; rI �coords�shapeZuncoordsr= r= r> � test_dtypesG s. &, z!TestRavelUnravelIndex.test_dtypesc C sx t tjddddgddd�t�ddddgd�� t tjddddgdd d�t�ddd dgd�� tttjddddgd� d S ) Nr$ r r r )r r" r% r2 r1 )r. r# )r1 �raiser- rS r )r r5 r7 r r8 )r; r= r= r> �test_clipmodesb s z$TestRavelUnravelIndex.test_clipmodesc C s6 t �dddgd�\}}| �|jj� | �|jj� d S )Nr r r" )r r$ )r5 r6 � assertTrue�flags� writeable)r; �x�yr= r= r> �test_writeabilityn s z'TestRavelUnravelIndex.test_writeabilityN)�__name__� __module__� __qualname__r? rH rR rT rZ r= r= r= r> r s # r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestGridc C s� t ddd� }t ddd� }t|jdk� t|jdk� t|d dk� t|d d� t|d dk� t|d |d dd� t|d |d d d� t|d |d d d� d S ) Nr r y $@g�������?)rJ )� r � ggfffff�?g �q �q�?)r r rQ r )r; �a�br= r= r> r? v s zTestGrid.test_basicc C s8 t jdddd�\}}t|d� t|tddd� d� d S )Nr rJ r )�retstepg9/����?y I@r, )r5 �linspacer r r )r; rY �str= r= r> �test_linspace_equivalence� s z"TestGrid.test_linspace_equivalencec C sV t ddd�ddd�f }t ddd�ddd�f }t|jdk� t|jd k� t|d d d d �f t�d d � � t|d d d �d f dt�d d � � t|d dd d �f t�d d �d � t|d d d �df dt�d d � d � t|d dd d �f |d d d d �f dt�dd � d � t|dd d �df |dd d �d f dt�dd � d � d S )Nr r y $@���r g�������?g�������?)r rJ rJ )r r_ r_ r rJ �dr` r_ )r r rQ r r5 �onesr )r; �crh r= r= r> �test_nd� s $&$($$zTestGrid.test_ndN)r[ r\ r] r? rf rk r= r= r= r> r^ u s r^ c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )�TestConcatenatorc C s` t td t�ddddddg�� t�d�}t|dd|f }t |ddddddddddddg � d S ) N)r r r" r r$ r# r r r" r r$ r# r )r r r5 r: ri )r; rb rj r= r= r> �test_1d� s zTestConcatenator.test_1dc C s" t ddd�f }t|jdk� d S )Ng333333$@r rJ �f8)r r rI )r; �gr= r= r> �test_mixed_type� s z TestConcatenator.test_mixed_typec C s6 t dt�dg�t�dddg�df }t|jdk� d S )Ng333333$�r r r" r g $@rn )r r5 r: r rI )r; ro r= r= r> �test_more_mixed_type� s $z%TestConcatenator.test_more_mixed_typec C s� t j�dd�}t j�dd�}td||f }t|jdk� t|d d �d d�f |� t|d d �dd �f |� t||f }t|jdk� t|d d�d d �f |� t|dd �d d �f |� d S )Nr$ �1)r$ rJ )rJ r$ )r5 �random�randr r rQ r )r; rb rj rh r= r= r> �test_2d� s zTestConcatenator.test_2dc s� ddg� ddg�t jd� �f }t jd� �f }tt|�t j� tt|�t j� tt �|�ddddgg� tt �|�dgdgdgdgg� tt� �fdd�� d S ) Nr r r" r �rrj c s t jd� �f S )N�rc)r5 r r= )ra rb r= r> �<lambda>� � z.TestConcatenator.test_matrix.<locals>.<lambda>)r5 r r �type�matrixr: r r8 )r; Zab_rZab_cr= )ra rb r> � test_matrix� s z TestConcatenator.test_matrixc C s@ t jdddgdf }tt|�t j� tt �|�dddgg� d S )Nrv r r r" )r5 r r rz r{ r: )r; rv r= r= r> �test_matrix_scalar� s z#TestConcatenator.test_matrix_scalarc C sp t �dg�}t �dg�}t �dg�}t �dg�}t jd }t �||g||gg�}t||� tt|�t|�� d S )Nr r r" r z a, b; c, d)r5 r: r �bmatr rz )r; ra rb rj rh �actual�expectedr= r= r> �test_matrix_builder� s z$TestConcatenator.test_matrix_builderN) r[ r\ r] rm rp rq ru r| r} r� r= r= r= r> rl � s rl c @ s e Zd Zdd� ZdS )�TestNdenumeratec C s4 t �ddgddgg�}ttt|��ddddg� d S ) Nr r r" r ))r r r ))r r r ))r r r" ))r r r )r5 r: r �listr )r; ra r= r= r> r? � s zTestNdenumerate.test_basicN)r[ r\ r] r? r= r= r= r> r� � s r� c @ s e Zd Zdd� Zdd� ZdS )�TestIndexExpressionc C sJ t �d�}t|d d� |td d� � t|d d� |td d� � d S )Nr r )r5 �aranger r r )r; ra r= r= r> �test_regression_1� s z%TestIndexExpression.test_regression_1c C s� t j�ddd�}t|d d �d d�ddgf |td d �d d�ddgf � t|d d �d d�ddgf |td d �d d�ddgf � d S )Nr r$ r# r" r r )r5 rs rt r r r )r; ra r= r= r> � test_simple_1� s :z!TestIndexExpression.test_simple_1N)r[ r\ r] r� r� r= r= r= r> r� � s r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �TestIx_c C s6 x0t tjfD ]"}t�|d��\}t|jtj� q W d S )Nr )�ranger5 r� r r rI rD )r; �funcra r= r= r> r� � s zTestIx_.test_regression_1c s� d}x�t tjfD ]~� tj� fdd�|D �� }x`tt||��D ]N\�\}}t|j� |� tt �fdd�t|j�D ��� tt� |j t �� q<W qW d S )N)r r$ r" r c s g | ] }� |��qS r= r= )�.0�sz)r� r= r> � <listcomp>� s z0TestIx_.test_shape_and_dtype.<locals>.<listcomp>c 3 s" | ]\}}|� kr|d kV qdS )r Nr= )r� �j�sh)�kr= r> � <genexpr>� s z/TestIx_.test_shape_and_dtype.<locals>.<genexpr>) r� r5 r� r � enumerate�zipr rQ r �all� issubdtyperI �int)r; �sizes�arraysra r� r= )r� r� r> �test_shape_and_dtype� s z TestIx_.test_shape_and_dtypec C s0 ddddg}t �|�\}tt �|�d |� d S )NTFr )r5 �nonzeror r )r; Zbool_aZint_ar= r= r> � test_bool� s zTestIx_.test_boolc C s& dddgdddgg}t ttj|� d S )Nr r r" r r$ r# )r r8 r5 r )r; Zidx2dr= r= r> � test_1d_only� s zTestIx_.test_1d_onlyc C sR d}t �|�}t||�}t|d j|df� t|d jd|f� t|j|f� d S )Nr$ r r )r5 r� r r rQ )r; Zlength_of_vectorrX �outr= r= r> �test_repeated_input s zTestIx_.test_repeated_inputN)r[ r\ r] r� r� r� r� r� r= r= r= r> r� � s r� c C sN t jt �dddgg�ddt �dddgg�f } t| ddddddddgg� d S )Nr r r" r r r$ r# )r5 �c_r: r )ra r= r= r> �test_c_ s .r� c c s� t �dt�} t| d� t| t �dddgdddgdddgg�fV t �dt�} t| d� t| t �dddgdddgdddgdddgdddgdddgdddgdddgdddgdddgg �fV t �dt�} t| dd� t| t �dddgdddgdddgdddgdddgdddgdddgdddgdddgdddgg �fV t �dt�} t| d� t| t �ddddddddddg ddddddddddg ddddddddddg g�fV t �dt�} t| d� t �dd d g�}tt �| dk�||||ffV d S ) N)r" r" r$ r )rJ r" T)r" rJ )r" r" r" r" r r r )r5 �zerosr� r r r: r �where)ra �ir= r= r> �test_fill_diagonal sT r� c c s� t d�} t�ddddgddddgd d d d gd dddgg�}d|| < t|t�ddddgddddgd d dd gd dddgg�fV t dd�}t�dt�}d||< t|t�ddgddggddgddggg�fV d S )Nr r r r" r$ r# r% r3 r4 rJ r` r2 r, � � � �d )r r r r )r r5 r: r r� r� )�dira �d3r= r= r> �test_diag_indicesE s& r� c C s< t j�d�} t| �\}}t|t �d�� t|t �d�� d S )N)r r r )r5 rs r r r� )rX rv rj r= r= r> �test_diag_indices_from` s r� c C s� t tddd��} dd� tt�d��D �}t| |� t td��} t| |� t td��} t| t td��� t t� �} t| dg� t td��} t| dg� t td g� �} t| g � d S ) Nr r r" c S s g | ] \}}|�qS r= r= )r� �ix�er= r= r> r� i s z test_ndindex.<locals>.<listcomp>)r r r" )r" r= r )r� r r r5 r� r r )rX r� r= r= r> � test_ndindexg s r� �__main__)&� __future__r r r �numpyr5 Z numpy.testingr r r r r r r r Znumpy.lib.index_tricksr r r r r r r r r r r r^ rl r� r� r� r� r� r� r� r� r[ r= r= r= r> �<module> s (0g!;%5