Current File : //lib64/python3.9/site-packages/numpy/lib/tests/__pycache__/test_index_tricks.cpython-39.pyc
a
z[yc J � @ s� d dl Z d dlZd dlmZmZmZmZmZm Z m
Z
d dl m Z m
Z
mZmZmZmZmZmZmZmZmZ G dd� d�ZG dd� d�ZG dd � d �ZG d
d � d �ZG d d
� d
�ZG dd� d�Z dd� Z G dd� d�Z dd� Z G dd� d�Z dd� Z!dS )� N)�assert_� assert_equal�assert_array_equal�assert_almost_equal�assert_array_almost_equal�
assert_raises�assert_raises_regex) �mgrid�ogrid� ndenumerate�
fill_diagonal� diag_indices�diag_indices_from� index_exp�ndindex�r_�s_�ix_c @ sf e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z e
j � dg d��dd� �Z
dd� ZdS )�TestRavelUnravelIndexc C s� t t�dd�d� t tjddd�d� tt�� tjddd� W d � n1 sR0 Y tt�� tjddd� W d � n1 s�0 Y tt�� tjddd � W d � n1 s�0 Y tt�� tjddd
� W d � n1 s�0 Y t t�dd�d� t t�dd�d � t t�d d�d� tttjd d� tttjd
d� tttjdd� tttjdd� tttjdd� tttjdd� tttjdd� tttjdd� t t�dd�g d�� t t�g d�d�d� t�g d�g d�g�}t t�|d�g d�� t tj|ddd �g d �� t tj|d d d �g d!�� t tj|d"d#d �g d$�� t t�d%d&�d'� t t�t�g d��d�g d�g d�g� t tjt�g d ��ddd �g d�g d�g� t t�d'd&�g d%�� d S )(N� )r r �� r )�indices�shape)r �hape)r � )� �^ )Zims)Zdims)r �B ���g �?� )���r )r r )r r! )r r )皙�����?g �. )r � � )r r r )r$ r% r% )r � r )� r% )� �) �% �F��order)� r) �
)r r% �clip��mode)r( � � �r r )r0 �wrap)� r/ r/ )r$ r r r )r% r' � � iU )r �np�
unravel_indexr � TypeError�ravel_multi_index�
ValueError�array)�self�arr� rB �G/usr/lib64/python3.9/site-packages/numpy/lib/tests/test_index_tricks.py�
test_basic sb ��
,
,
,
,������z TestRavelUnravelIndex.test_basicc C s d}d}t t|tjg d� t t|tjdd� t t|tjt�g �d� tt�tjg td�d�g g g g� t t|tjg g fd� t t|tjg dgfd� t t|tjt�g �t�g �fd� tt�tjg td�tjg td�fd�g � tt�tjg g gtd�d�g � d S ) Nz9indices must be integral: the provided empty sequence waszonly int indices permitted)�
r$ r& rB ��dtype�rE r$ �abc)r& r$ )r r< r: r; r? r �intr= )r@ Zmsg1Zmsg2rB rB rC �test_empty_indicesF s: ����
� ����z(TestRavelUnravelIndex.test_empty_indicesc C s� t jt jkrPddgddgddgddgdd gddgd
dgf}tt �|d �d d
g� ttt jdd� d
gd
gf}t �t j�j d }tt �||df�d
g� ttt j||d df� tt j||dfdd�d
g� ttt j||d dfdd� d S )Nr � r$ r& �u r4 r iK i r )r) r' �x �$ i�
r8 r% l �z� l
nm )l �� l r+ r, )
r: �intp�int64r r= r r> r; Ziinfo�max)r@ rA Z dummy_arrZhalf_maxrB rB rC �test_big_indicesZ s. �
� ����z&TestRavelUnravelIndex.test_big_indicesc C sr t jt jt jt jt jt jfD �]N}t jg d�g d�g|d�}d}d|d |d }tt � ||�|� t|t �
||�� |d d|d }tt j ||d d
�|� t|t j
||d d
�� t jg d�g d�g d �g|d�}d }d
d|d |d |d }tt � ||�|� t|t �
||�� |d d|d d|d }tt j ||d d
�|� t|t j
||d d
�� q d S )N)r r r r r$ r )r r% r r$ r r rF )r& r8 r8 r r r& r+ r, )r r$ r r r9 r& )r& r8 rE rE r ) r: Zint16Zuint16Zint32Zuint32rQ Zuint64r? r r= r; )r@ rG Zcoordsr ZuncoordsrB rB rC � test_dtypesr s8 �
��� �z!TestRavelUnravelIndex.test_dtypesc C sd t tjg d�ddd�t�g d�d�� t tjg d�ddd�t�g d�d�� tttjg d�d� d S )N)r& r r r )r r$ r' r7 r6 r1 )r r r% r )r6 �raiser0 rU )r r r r )r r: r= r r> �r@ rB rB rC �test_clipmodes� s ����z$TestRavelUnravelIndex.test_clipmodesc C s0 t �g d�d�\}}t|jj� t|jj� d S )N�r r r$ )r r& )r: r; r �flagsZ writeable)r@ �x�yrB rB rC �test_writeability� s z'TestRavelUnravelIndex.test_writeabilityc C sB t �dd�}t|d� ttdt jdgd� ttdt jdgd� d S )Nr rB z0d arrayz
out of boundsr )r: r; r r r> �r@ rZ rB rB rC �test_0d� s
�z TestRavelUnravelIndex.test_0dr2 )r0 r6 rU c C sr t jt jdt jd�d|d�}|jdks*J �tt��, t jt jdt jd�d|d� W d � n1 sd0 Y d S )N)r$ r rF �r r r r1 �r )r$ r )r: r= �zerosrP r r r> )r@ r2 �resrB rB rC �test_empty_array_ravel� s �
�z,TestRavelUnravelIndex.test_empty_array_ravelc C sx t �t jdt jd�d�}t|�dks(J �tdd� |D ��s>J �tt�� t �dgd� W d � n1 sj0 Y d S )Nr rF r_ r$ c s s | ]}|j d kV qdS )r` N)r )�.0�arB rB rC � <genexpr>� � zATestRavelUnravelIndex.test_empty_array_unravel.<locals>.<genexpr>r )r: r; ra rP �len�allr r> )r@ rb rB rB rC �test_empty_array_unravel� s
z.TestRavelUnravelIndex.test_empty_array_unravelN)�__name__�
__module__� __qualname__rD rK rS rT rW r\ r^ �pytest�mark� parametrizerc rj rB rB rB rC r s 7
r c @ sV e Zd Zdd� Zdd� Zdd� Zdd� Zej� d d
d g�d d
� �Z
dd� Z dd� Z dS )�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 � $@r" )rE )� r � ggfffff�?g �q �q�?)r r r r )r@ re �brB rB rC rD � s zTestGrid.test_basicc C s8 t jdddd�\}}t|d� t|tddd� d� d S )Nr rE T)Zretstepg9/����?y I@r/ )r: Zlinspacer r r )r@ r[ �strB rB rC �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 rr ���r r" g�������?)r rE rE )r rs rs r rE �drt rs )r r r r r: �onesr )r@ �cry rB rB rC �test_nd� s $&$($�$�zTestGrid.test_ndc C s\ t ddd�ddd�f }tddd�ddd�f }tj|� }t||�D ]\}}t||� qDd S )Nr r rr rx r )r r
r: Zbroadcast_arrays�zipr )r@ Z grid_fullZ grid_sparseZgrid_broadcast�fru rB rB rC � test_sparse� s
zTestGrid.test_sparsezstart, stop, step, expected)NrE rr )�� rE )i����rs N)i � c C sJ t |||�|||�f }t |||� }t|j|d � t|j|d � d S )Nr r )r r �size)r@ �start�stop�step�expectedZgridZ
grid_smallrB rB rC � test_mgrid_size_none_handling� s z&TestGrid.test_mgrid_size_none_handlingc C s� t ddd�f }t t�d�t�d�t�d��f }t|jtjk� t||� t ddd� }t t�d�t�d�t�d�� }t|jtjk� t||� d S )Nr" g ��Q� �?)r r: �float32r rG Zfloat64r )r@ Zgrid64Zgrid32rB rB rC �test_accepts_npfloating� s "
z TestGrid.test_accepts_npfloatingc C sP t tddd�f tddt�d��f � t tddd� tddt�d�� � d S )Nr" g333333�?y @)r r r: � complex64rV rB rB rC � test_accepts_npcomplexfloating s "� �z'TestGrid.test_accepts_npcomplexfloatingN)
rk rl rm rD rw r| r rn ro rp r� r� r� rB rB rB rC rq � s �
rq c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�TestConcatenatorc C sD t td t�g d��� t�d�}t|dd|f }t |g d�� d S )N)r r r$ r r& r% r& r ) r r r r r r r r r r r r )r r r: r? rz )r@ ru r{ rB rB rC �test_1d
s
zTestConcatenator.test_1dc C s" t ddd�f }t|jdk� d S )Ng333333$@r rE �f8)r r rG �r@ �grB rB rC �test_mixed_type s z TestConcatenator.test_mixed_typec C s4 t dt�dg�t�g d��df }t|jdk� d S )Ng333333$�r )r r$ r g $@r� )r r: r? r rG r� rB rB rC �test_more_mixed_type s "z%TestConcatenator.test_more_mixed_typec C sB t ddd� }t|jdk� t ddt�d�� }t|jdk� d S )Nr rO y Y@)�d )r r r r: r� r� rB rB rC �test_complex_step s z"TestConcatenator.test_complex_stepc 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& rE )rE r& )r: �random�randr r r r )r@ ru r{ ry rB rB rC �test_2d$ s zTestConcatenator.test_2dc C sb t tdt�d�df g d�� t tg d�t�d�f g d�� t tt�d�g d�f g d�� d S )Nr r r �r r r r$ )r r r r$ rX )r r r: r? rV rB rB rC r^ 0 s zTestConcatenator.test_0dN) rk rl rm r� r� r� r� r� r^ rB rB rB rC r� s r� c @ s e Zd Zdd� ZdS )�TestNdenumeratec C s0 t �ddgddgg�}ttt|��g d�� d S )Nr r r$ r )))r r r ))r r r )r r$ ))r r r )r: r? r �listr �r@ re rB rB rC rD 7 s �zTestNdenumerate.test_basicN)rk rl rm rD rB rB rB rC r� 6 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 )r: �aranger r r r� rB rB rC �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 )r: r� r� r r r r� rB rB rC �
test_simple_1D s :z!TestIndexExpression.test_simple_1N)rk rl rm r� r� rB rB rB rC r� = s r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �TestIx_c C sb t �td��\}t|jt j� t �g �\}t|jt j� t �t jg t jd��\}t|jt j� d S )Nr rF )r: r �ranger rG rP r? r� r� rB rB rC r� L s zTestIx_.test_regression_1c s� d}t tjfD ]|� tj� fdd�|D �� }tt||��D ]P\�\}}t|j� |� tt �fdd�t|j�D ��� tt�
|j tj �� q8qd S )N)r r& r$ r c s g | ] }� |��qS rB rB )rd �sz)�funcrB rC �
<listcomp>\ rg z0TestIx_.test_shape_and_dtype.<locals>.<listcomp>c 3 s" | ]\}}|� kr|d kV qdS )r NrB )rd �j�sh)�krB rC rf _ rg z/TestIx_.test_shape_and_dtype.<locals>.<genexpr>)
r� r: r� r � enumerater} r r r ri Z
issubdtyperG Zinteger)r@ ZsizesZarraysre r� rB )r� r� rC �test_shape_and_dtypeX s z TestIx_.test_shape_and_dtypec C s, g d�}t �|�\}tt �|�d |� d S )N)TFTTr )r: Znonzeror r )r@ Zbool_aZint_arB rB rC � test_boolb s zTestIx_.test_boolc C s" g d�g d�g}t ttj|� d S )NrX �r r& r% )r r> r: r )r@ Zidx2drB rB rC � test_1d_onlyg 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 )r: r� r r r )r@ Zlength_of_vectorrZ �outrB rB rC �test_repeated_inputk s
zTestIx_.test_repeated_inputN)rk rl rm r� r� r� r� r� rB rB rB rC r� K s
r� c C s> t jt �g d�g�ddt �g d�g�f } t| g d�g� d S )NrX r r� )r r r$ r r r r& r% )r: Zc_r? r )re rB rB rC �test_c_u s *r� c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�TestFillDiagonalc C s< t �dt�}t|d� t|t �g d�g d�g d�g�� d S )N)r$ r$ r& �r& r r �r r& r �r r r& �r: ra rJ r r r? r� rB rB rC rD { s
��zTestFillDiagonal.test_basicc C sf t �dt�}t|d� t|t �g d�g d�g d�g d�g d�g d�g d�g d�g d�g d�g
�� d S )NrH r& r� r� r� �r r r r� r� rB rB rC �test_tall_matrix� s
��z!TestFillDiagonal.test_tall_matrixc C sh t �dt�}t|dd� t|t �g d�g d�g d�g d�g d�g d�g d�g d�g d�g d�g
�� d S )NrH r& Tr� r� r� r� r� r� rB rB rC �test_tall_matrix_wrap� s ��z&TestFillDiagonal.test_tall_matrix_wrapc C s< t �dt�}t|d� t|t �g d�g d�g d�g�� d S )N)r$ rE r& )
r& r r r r r r r r r )
r r& r r r r r r r r )
r r r& r r r r r r r r� r� rB rB rC �test_wide_matrix� s
��z!TestFillDiagonal.test_wide_matrixc C sD t �dt�}t|d� t �g d��}tt �|dk�||||f� d S )N)r$ r$ r$ r$ r r� r )r: ra rJ r r? r �where)r@ re �irB rB rC �test_operate_4d_array� s
z&TestFillDiagonal.test_operate_4d_arrayc C sD t �dt�}ttd�� t|d� W d � n1 s60 Y d S )Nr$ � at least 2-dr& �r: ra rJ r r> r r� rB rB rC �test_low_dim_handling� s z&TestFillDiagonal.test_low_dim_handlingc C sD t �dt�}ttd�� t|d� W d � n1 s60 Y d S )N)r$ r$ r' r$ � equal lengthr r� r� rB rB rC �test_hetero_shape_handling� s z+TestFillDiagonal.test_hetero_shape_handlingN)
rk rl rm rD r� r� r� r� r� r� rB rB rB rC r� z s r� c C s� t d�} t�g d�g d�g d�g d�g�}d|| < t|t�g d�g d�g d �g d
�g�� t d d �}t�d
t�}d||< t|t�ddgddggddgddggg�� d S )Nr )r r r$ r )r& r% r' r8 )r9 rE rt r7 )r/ � � � r� )r� r r$ r )r& r� r' r8 )r9 rE r� r7 )r/ r� r� r� r r$ )r r r r r )r
r: r? r ra rJ )Zdire Zd3rB rB rC �test_diag_indices� s4
� ��
����r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestDiagIndicesFromc C s< t j�d�}t|�\}}t|t �d�� t|t �d�� d S )Nr5 r )r: r� r r r� )r@ rZ �rr{ rB rB rC �test_diag_indices_from� s z*TestDiagIndicesFrom.test_diag_indices_fromc C s@ t �d�}ttd�� t|� W d � n1 s20 Y d S )Nr' r� )r: rz r r> r r] rB rB rC �test_error_small_input� s
z*TestDiagIndicesFrom.test_error_small_inputc C sB t �dt�}ttd�� t|� W d � n1 s40 Y d S )N)r$ r$ r r$ r� )r: ra rJ r r> r r] rB rB rC �test_error_shape_mismatch� s z-TestDiagIndicesFrom.test_error_shape_mismatchN)rk rl rm r� r� r� rB rB rB rC r� � 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 rB rB )rd Zix�erB rB rC r� � rg z test_ndindex.<locals>.<listcomp>rX )r$ rB r )r� r r r: ra r r )rZ r� rB rB rC � test_ndindex� s
r� )"rn Znumpyr: Z
numpy.testingr r r r r r r Znumpy.lib.index_tricksr r
r r r
r r r r r r r rq r� r� r� r� r� r� r� r� r� rB rB rB rC �<module> s $4 .Q**G
Mini Shell