← Back
Editing: test_einsum.cpython-311.pyc
� ���c � � �� � d dl Z d dlZd dlZd dlmZmZmZmZm Z m Z mZmZ dZ ej g d�� � Z e ee e� � � � Z G d� d� � Z G d� d� � Zd � ZdS ) � N)�assert_�assert_equal�assert_array_equal�assert_almost_equal� assert_raises�suppress_warnings�assert_raises_regex�assert_allclose� abcdefghij)� � � � r r r � r r r c � � e Zd Zd� Zd� Z ej � � d.d�� � Zd� Zd� Z d� Z d� Zd � Zd � Z d� Zd� Zd � Zd� Zd� Zd� Zd� Zd� Zd� Zd� Zd� Zd� Zd� Zd� Zd� Zd� Zej � dej! d ej! d z � � d� � � Z"d� Z#d � Z$d/d"�Z%d#� Z&d$� Z'd%� Z(d&� Z)d'� Z*d(� Z+d)� Z,d*� Z-d+� Z.d,� Z/d-� Z0d!S )0� TestEinsumc �x � dD �]�}t t t j |�� � t t t j d|�� � t t t j dd|�� � t t t j ddd|�� � t t t j ddd|�� � t t t j ddd |� � � t t t j ddd|�� � t t t j ddd|� � � t t t j gd�R d|i� t t t j ddd|�� � t t t j dddgdg|�� � t t t j ddg|�� � t t t j dd|�� � t t t j dddg|�� � t t t j dd|�� � t t t j dddg|�� � t t t j dd|�� � t t t j dddg|�� � t t t j dddg|�� � t t t j dddg|�� � t t t j dddg|�� � t t t j dddg|�� � t t t j dddg|�� � t t t j dddg|�� � t t t j dddg|�� � t t t j dddg|�� � t t t j dddgddgg|�� � t t t j d t j d!� � � d"d#� � |�� � t t t j d$t j d!� � � d"d#� � |�� � t t t j dt j d!� � � d"d#� � |�� � t t t j d%dd&gdd&ggt j d'� � � d"d"� � |�� � t t d(� � 5 t j d)� � }t j d*� � }t j d+||� � d d d � � n# 1 swxY w Y t t t j d%t j d!� � � d,d&� � |d-�.� � ���d S )/N�TF��optimize� r �test)�outr �W��orderr �blah)�castingr � bad_data_type)�dtyper )�bad_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����d)r r ) r � ValueError�np�einsum� TypeError�arange�reshaper �ones)�self�do_opt�a�bs �>/usr/lib/python3/dist-packages/numpy/core/tests/test_einsum.py�test_einsum_errorszTestEinsum.test_einsum_errors s� � �#� R 6� R 6�F��*�b�i�&�A�A�A�A��*�b�i��f�E�E�E�E� �)�R�Y��1�v�F�F�F�F� �)�R�Y��A�6�#)� +� +� +� +� �*�b�i��Q�c�#)� +� +� +� +� �*�b�i��Q��#)� +� +� +� +� �)�R�Y��A�_�#)� +� +� +� +� �)�R�Y��A�q�#)� +� +� +� +� �)�R�Y�M��M�M�M�f�M�M�M� �*�b�i��Q��F�K�K�K�K��*�b�i��a�!��q�c�#)� +� +� +� +��*�b�i��q�c�F�K�K�K�K� �*�b�i��a�&�I�I�I�I��*�b�i���1�v��O�O�O�O��*�b�i���V�L�L�L�L��*�b�i��1�a�&�6�R�R�R�R��*�b�i���V�L�L�L�L��*�b�i��1�a�&�6�R�R�R�R� �*�b�i���A���P�P�P�P��*�b�i��1�a�&�6�R�R�R�R��*�b�i��A�q�6�F�S�S�S�S��*�b�i��a��V�f�U�U�U�U� �*�b�i��1�a�&�6�R�R�R�R��*�b�i��1�a�&�6�R�R�R�R��*�b�i��!�Q��&�Q�Q�Q�Q� �*�b�i��1�a�&�6�R�R�R�R� �*�b�i��a��V�a��V�<L�#)� +� +� +� +� �*�b�i���)�A�,�,�.�.�q�!�4�4�v� G� G� G� G��*�b�i���)�A�,�,�.�.�q�!�4�4�v� G� G� G� G� �*�b�i��b�i��l�l�6J�6J�1�a�6P�6P�#)� +� +� +� +��*�b�i��1�a�&�1�a�&�9I� �i��l�l�2�2�1�a�8�8�6� K� K� K� K�$�Z��7�7� -� -��G�O�,�,���G�I�&�&��� �+�q�!�,�,�,� -� -� -� -� -� -� -� -� -� -� -���� -� -� -� -� �*�b�i����1���9M�9M�b�RS�9T�9T�#)�� 6� 6� 6� 6� 6�cR 6� R 6s �?U#�#U' �*U' c �, �� dD � ]�}t j d� � �d�_ t j d�|�� � }t |j �u � � t j �t g|�� � }t |j �u � � t j d�|�� � }t |j �u � � t |�� � t j �ddg|�� � }t |j �u � � t |�� � t j d�|�� � }t |j d � � d �j d <