OwlCyberSecurity - MANAGER
Edit File: test_einsum.cpython-38.pyc
U bL�]�� � @ s� d dl mZmZmZ d dlZd dlZd dlmZm Z m Z mZmZm Z mZmZ dZe�dddddddd dddg�Zeeee��ZG d d� de�ZG dd � d e�Zdd� ZdS )� )�division�absolute_import�print_functionN)�assert_�assert_equal�assert_array_equal�assert_almost_equal� assert_raises�suppress_warnings�assert_raises_regex�assert_allcloseZ abcdefghij� � � � � c @ s@ e Zd Zdd� Zdd� ZdOdd�Zdd � Zd d� Zdd � Zdd� Z dd� Z dd� Zdd� Zdd� Z dd� Zdd� Zdd� Zdd� Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Zd4d5� Zd6d7� ZdPd9d:�Zd;d<� Zd=d>� Z d?d@� Z!dAdB� Z"dCdD� Z#dEdF� Z$dGdH� Z%dIdJ� Z&dKdL� Z'dMdN� Z(d8S )Q� TestEinsumc C sF dD �]:}t ttj|d� t ttjd|d� t ttjdd|d� t ttjddd|d� t ttjddd|d� t ttjddd |d � t ttjddd|d� t ttjddd|d � t ttjfd�d|i� t ttjddd|d� t ttjdddgdg|d� t ttjddg|d� t ttjdd|d� t ttjdddg|d� t ttjdd|d� t ttjdddg|d� t ttjdd|d� t ttjdddg|d� t ttjdddg|d� t ttjdddg|d� t ttjdddg|d� t ttjdddg|d� t ttjdddg|d� t ttjdddg|d� t ttjdddg|d� t ttjdddg|d� t ttjdddgddgg|d� t ttjd t�d!��d"d#�|d� t ttjd$t�d!��d"d#�|d� t ttjdt�d!��d"d#�|d� t ttjd%dd&gdd&ggt�d'��d"d"�|d� ttd(��( t�d)�}t�d*�}t�d+||� W 5 Q R X qd S ),N�TF��optimize� r Ztest)�outr �W)�orderr Zblah)�castingr Z bad_data_type)�dtyper )Zbad_argr )?NNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNr �,�i�ijz...izi...jzi...zij...zi..z.i...zj->..jzj->.j...zi%...z...j$zi->&zi->ijzij->jij�iir r r �ii->izi->i� r z'b')r r r r r )r r r z aabcb,abc) r � ValueError�np�einsum� TypeError�arange�reshaper �ones)�self�do_opt�a�b� r- �>/usr/lib/python3/dist-packages/numpy/core/tests/test_einsum.py�test_einsum_errors sz ������� � �� � zTestEinsum.test_einsum_errorsc s� dD �]v}t �d�� d� _t jd� |d�}t|j� k� t j� tg|d�}t|j� k� t jd� |d�}t|j� k� t|� � t j� ddg|d�}t|j� k� t|� � t jd� |d�}t|jd � d � jd <