File: //lib/python3/dist-packages/numpy/lib/tests/__pycache__/test_index_tricks.cpython-310.pyc
o
6��aJ � @ s� d dl Z d dlZd dlmZmZmZmZmZm Z m
Z
mZ d dlm
Z
mZmZmZmZmZmZmZmZmZmZ G dd� d�ZG dd� d�ZG dd � d �ZG d
d� d�ZG dd
� d
�ZG dd� d�Zdd� ZG dd� d�Zdd� 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 @ sf e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zdd� 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 s)w Y tt�� tjddd� W d � n1 sDw Y tt�� tjddd � W d � n1 s_w Y tt�� tjddd
� W d � n1 szw Y t t�dd�d� t t�dd�d� t t�dd�d� tttjdd� 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|ddd �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 � )� �^ )�ims)�dims)r �B ���g �?� )���r )r r )r r$ )r r )皙�����?g �. )r# � � )r r r# )r'