Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/numpy/lib/tests/__pycache__/test_index_tricks.cpython-37.pyc
Назад
B �5�g J � @ s� d dl Z d dlZd dlmZmZmZmZmZm Z m Z m Z d dl m Z mZmZmZmZmZmZmZmZmZmZ G dd� d�ZG dd� d�ZG dd � d �ZG d d � d �ZG d d � d �Z G 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� assert_warns) �mgrid�ogrid� ndenumerate� fill_diagonal� diag_indices�diag_indices_from� index_exp�ndindex�r_�s_�ix_c @ sh e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z e j � ddddg�dd� �Z dd� ZdS )�TestRavelUnravelIndexc C s� t t�dd�d� t tjddd�d� tt�� tjddd� W d Q R X tt�� tjddd� W d Q R X tt�� tjddd � W d Q R X tt�� tjddd � W d Q R X 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�dddg� t t�dddgd�d� t�dddgdddgg�}t t�|d�dd d g� t tj|dd d �d d d!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�dd d g�d�dddgdddgg� t tjt�d d d!g�dd d �dddgdddgg� t t�d,d+�ddddg� d S )-N� )r r )� r )�indices�shape)r �hape)r � )� �^ )Zims)Zdims)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_indexr � TypeError�ravel_multi_index� ValueError�array)�self�arr� r> �W/home/digitalm-up/venv/lib/python3.7/site-packages/numpy/lib/tests/test_index_tricks.py� test_basic sR 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% r> )�dtype)rA r# �abc)r% r# )r r8 r6 r7 r; r �intr9 )r< Zmsg1Zmsg2r> r> r? �test_empty_indicesG 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 r1 r iK i r )r( r& �x �$ i� r4 r$ l �z� l nm )l �� l r* )r+ ) r6 �intp�int64r r9 r r: r7 Ziinfo�max)r< r= Z dummy_arrZhalf_maxr> r> r? �test_big_indices[ 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$ )rB )r% r4 r4 r% r* )r+ r5 )r% r4 rA rA ) r6 Zint16Zuint16Zint32Zuint32rK Zuint64r; r r9 r7 )r< rB Zcoordsr Zuncoordsr> r> r? � test_dtypess 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& r3 r2 )r/ r$ )r2 �raiser. rO r )r r6 r9 r r: )r< r> r> r? �test_clipmodes� s z$TestRavelUnravelIndex.test_clipmodesc C s2 t �dddgd�\}}t|jj� t|jj� d S )Nr r r# )r r% )r6 r7 r �flagsZ writeable)r< �x�yr> r> r? �test_writeability� s z'TestRavelUnravelIndex.test_writeabilityc C sB t �dd�}t|d� ttdt jdgd� ttdt jdgd� d S )Nr r> z0d arrayz out of boundsr )r6 r7 r r r: )r< rR r> r> r? �test_0d� s z TestRavelUnravelIndex.test_0dr/ r. r2 rO c C s^ t jt jdt jd�d|d�}|jdks*t�tt��" t jt jdt jd�d|d� W d Q R X d S )N)r# r )rB )r r r )r/ )r )r# r )r6 r9 �zerosrJ r �AssertionErrorr r: )r< r/ �resr> r> r? �test_empty_array_ravel� s z,TestRavelUnravelIndex.test_empty_array_ravelc C sd t �t jdt jd�d�}t|�dks(t�tdd� |D ��s>t�tt�� t �dgd� W d Q R X d S )Nr )rB )r r r r# c s s | ]}|j d kV qdS ))r N)r )�.0�ar> r> r? � <genexpr>� s zATestRavelUnravelIndex.test_empty_array_unravel.<locals>.<genexpr>r ) r6 r7 rV rJ �lenrW �allr r: )r< rX r> r> r? �test_empty_array_unravel� s z.TestRavelUnravelIndex.test_empty_array_unravelN)�__name__� __module__� __qualname__r@ rE rM rN rP rT rU �pytest�mark� parametrizerY r_ r> r> r> r? 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 y $@g�������?)rA )� r � ggfffff�?g �q �q�?)r r r r )r< r[ �br> r> r? r@ � s zTestGrid.test_basicc C s8 t jdddd�\}}t|d� t|tddd� d� d S )Nr rA T)Zretstepg9/����?y I@r- )r6 Zlinspacer r r )r< rS �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 rA rA )r rg rg r rA �drh rg )r r r r r6 �onesr )r< �crm r> r> r? �test_nd� s $&$($$zTestGrid.test_ndc C s` t ddd�ddd�f }tddd�ddd�f }tj|� }x t||�D ]\}}t||� qFW d S )Nr r y $@rl r )r r r6 Zbroadcast_arrays�zipr )r< Z grid_fullZ grid_sparseZgrid_broadcast�fri r> r> r? � test_sparse� s zTestGrid.test_sparsezstart, stop, step, expected)NrA y $@)�� rA )i����rg 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�expected�gridZ grid_smallr> r> r? � 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 )Ng�������?g ��Q� �?)r r6 �float32r rB Zfloat64r )r< Zgrid64Zgrid32r> r> r? �test_accepts_npfloating� s " z TestGrid.test_accepts_npfloatingc C sP t tddd�f tddt�d��f � t tddd� tddt�d�� � d S )Ng�������?g333333�?y @)r r r6 � complex64)r< r> r> r? � test_accepts_npcomplexfloating s &z'TestGrid.test_accepts_npcomplexfloatingN) r` ra rb r@ rk rp rs rc rd re r| r~ r� r> r> r> r? rf � s rf c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )�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 r6 r; rn )r< ri ro r> r> r? �test_1d s zTestConcatenator.test_1dc C s" t ddd�f }t|jdk� d S )Ng333333$@r rA �f8)r r rB )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 $@r� )r r6 r; r rB )r< r� r> r> r? �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 rI y Y@)�d )r r r r6 r )r< r� r> r> r? �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% rA )rA r% )r6 �random�randr r r r )r< ri ro rm r> r> r? �test_2d% s zTestConcatenator.test_2dc C sp t tdt�d�df dddg� t tdddgt�d�f ddddg� t tt�d�dddgf ddddg� d S )Nr r r r# )r r r6 r; )r< r> r> r? rU 1 s &zTestConcatenator.test_0dN) r` ra rb r� r� r� r� r� rU r> r> r> r? r� s r� 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 )r6 r; r �listr )r< r[ r> r> r? r@ 8 s zTestNdenumerate.test_basicN)r` ra rb r@ r> r> r> r? r� 7 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 )r6 �aranger r r )r< r[ 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 )r6 r� r� r r r )r< r[ r> r> r? � test_simple_1E s :z!TestIndexExpression.test_simple_1N)r` ra rb r� r� r> r> r> r? 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 )rB )r6 r �ranger rB rJ r; r} )r< r[ r> r> r? r� M s zTestIx_.test_regression_1c s� d}x�t tjfD ]�� tj� fdd�|D �� }xbtt||��D ]P\�\}}t|j� |� tt �fdd�t|j�D ��� tt� |j tj �� q<W qW d S )N)r r% r# r c s g | ] }� |��qS r> r> )rZ �sz)�funcr> r? � <listcomp>] s z0TestIx_.test_shape_and_dtype.<locals>.<listcomp>c 3 s" | ]\}}|� kr|d kV qdS )r Nr> )rZ �j�sh)�kr> r? r\ ` s z/TestIx_.test_shape_and_dtype.<locals>.<genexpr>) r� r6 r� r � enumeraterq r r r r^ Z issubdtyperB �integer)r< �sizesZarraysr[ r� r> )r� r� r? �test_shape_and_dtypeY s z TestIx_.test_shape_and_dtypec C s0 ddddg}t �|�\}tt �|�d |� d S )NTFr )r6 Znonzeror r )r< Zbool_aZint_ar> r> r? � test_boolc s zTestIx_.test_boolc C s& dddgdddgg}t ttj|� d S )Nr r r# r r% r$ )r r: r6 r )r< Zidx2dr> r> r? � test_1d_onlyh 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 )r6 r� r r r )r< Zlength_of_vectorrR �outr> r> r? �test_repeated_inputl s zTestIx_.test_repeated_inputN)r` ra rb r� r� r� r� r� r> r> r> r? r� L s r� c C sN t jt �dddgg�ddt �dddgg�f } t| ddddddddgg� d S )Nr r r# r r r% r$ )r6 Zc_r; r )r[ r> r> r? �test_c_v s .r� c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )�TestFillDiagonalc C sB t �dt�}t|d� t|t �dddgdddgdddgg�� d S )N)r# r# r% r )r6 rV rD r r r; )r< r[ r> r> r? r@ | s zTestFillDiagonal.test_basicc C sz t �dt�}t|d� t|t �dddgdddgdddgdddgdddgdddgdddgdddgdddgdddgg �� d S )N)rA r# r% r )r6 rV rD r r r; )r< r[ r> r> r? �test_tall_matrix� s z!TestFillDiagonal.test_tall_matrixc C s| t �dt�}t|dd� t|t �dddgdddgdddgdddgdddgdddgdddgdddgdddgdddgg �� d S )N)rA r# r% Tr )r6 rV rD r r r; )r< r[ r> r> r? �test_tall_matrix_wrap� s z&TestFillDiagonal.test_tall_matrix_wrapc C sl t �dt�}t|d� t|t �ddddddddddg ddddddddddg ddddddddddg g�� d S )N)r# rA r% r )r6 rV rD r r r; )r< r[ r> r> r? �test_wide_matrix� s z!TestFillDiagonal.test_wide_matrixc C sF t �dt�}t|d� t �dddg�}tt �|dk�||||f� d S )N)r# r# r# r# r r r r )r6 rV rD r r; r �where)r< r[ �ir> r> r? �test_operate_4d_array� s z&TestFillDiagonal.test_operate_4d_arrayc C s0 t �dt�}ttd�� t|d� W d Q R X d S )Nr# z at least 2-dr% )r6 rV rD r r: r )r< r[ r> r> r? �test_low_dim_handling� s z&TestFillDiagonal.test_low_dim_handlingc C s0 t �dt�}ttd�� t|d� W d Q R X d S )N)r# r# r& r# z equal lengthr )r6 rV rD r r: r )r< r[ r> r> r? �test_hetero_shape_handling� s z+TestFillDiagonal.test_hetero_shape_handlingN) r` ra rb r@ r� r� r� r� r� r� r> r> r> r? r� { s r� c C s� t d�} t�ddddgddddgd d d d gd dddgg�}d|| < t|t�ddddgddddgd d dd gd dddgg�� t dd�}t�dt�}d||< t|t�ddgddggddgddggg�� d S )Nr r r r# r% r$ r& r4 r5 rA rh r3 r- � � � r� )r r r r )r r6 r; r rV rD )Zdir[ Zd3r> r> r? �test_diag_indices� s& r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestDiagIndicesFromc C s< t j�d�}t|�\}}t|t �d�� t|t �d�� d S )N)r r r )r6 r� r r r� )r< rR �rro r> r> r? �test_diag_indices_from� s z*TestDiagIndicesFrom.test_diag_indices_fromc C s, t �d�}ttd�� t|� W d Q R X d S )Nr& z at least 2-d)r6 rn r r: r )r< rR r> r> r? �test_error_small_input� s z*TestDiagIndicesFrom.test_error_small_inputc C s. t �dt�}ttd�� t|� W d Q R X d S )N)r# r# r r# z equal length)r6 rV rD r r: r )r< rR r> r> r? �test_error_shape_mismatch� s z-TestDiagIndicesFrom.test_error_shape_mismatchN)r` ra rb r� r� r� r> r> r> r? 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 r> r> )rZ Zix�er> r> r? r� � s z test_ndindex.<locals>.<listcomp>)r r r# )r# r> r )r� r r r6 rV r r )rR rz r> r> r? � test_ndindex� s r� )#rc �numpyr6 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 rf r� r� r� r� r� r� r� r� r� r> r> r> r? �<module> s (4 .Q**G
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка