Edit File: test_linalg.cpython-37.pyc
B ��Fd,� � @ s~ d Z ddlmZmZmZ ddlZddlZddlZddlZddl Z ddl Z ddl m Z m Z mZmZmZmZmZ ddl mZmZmZmZmZ ddl mZ ddlmZmZm Z m Z m Z ddl m Z dd l!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+ d d � Z,d d � Z-e&Z.d�dd�Z&dd� Z/dd� Z0dd� Z1ddddd d hZ2G d d � d e3�Z4d d!� Z5e j6�7d"� g Z8e8e5de4d#e d$d%gd&d'gge d(�e d%d$ge d(��e4d)e d$d%gd&d'gged(�e d%d$ged(��e4d*e d$d%gd&d'gged(�e d%d$d'gd&d'd+gged(��e4d,e d-d.gd/d0gged(�e d1d-ged(��e4d2e d-d.gd/d0gged(�e d1d-ged(��e4d3e d-d.gd/d0gged(�e d1d-d4gd5d6d7gged(��e4d8e j9d9ed(�e j9d:ed(�d hd;�e4d<e j9d9ed(��:e j�e j9d=ed(��:e j�d hd;�e4d>e j6�;d?d?�e j6�;d?��e4d@e j6�;dAdA�e j6�;dA��e4dBdAdCgdDdEggdCdAg�e4dFe d$d%gd&d'gg�ed%d$g�j<�e4dGed$d%gd&d'gg�ed%d$g�j<�g �7 Z8e8e5de4dHe d$d%d&gd&d'd+gge d(�e d%d$ge d(��e4dIe d$d%gd&d'gdJd+gge d(�e d%d$d&ge d(��e4dKe d$d%d&gd&d'd+gged(�e d%d$ged(��e4dLe d$d%gd&d'gdJd+gged(�e d%d$d&ged(��e4dMe dNdOdPgdQdRdSgged(�e d1d-ged(��e4dTe dNdOgdPdUgdVdWgged(�e d1d-dPged(��e4dXe dNdOdPgdQdRdSgged(�e d1d-ged(��e4dYe dNdOgdPdUgdVdWgged(�e d1d-dPged(��e4dZe dNdOdPgdQdRdSgged(�e d1d-gd[d\gged(��e4d]e dNdOgdPdUgdVdWgged(�e d1d-gd[d\gd[d\gged(��e4d^e j6�;d?d_�e j6�;d?��e4d`e j6�;dAda�e j6�;dA��e4dbe j6�;dadA�e j6�;da��e4dce j6�;ddE�e j6�;d�d hd;�e4dde j6�;dEd�e j6�;dE�d hd;�g�7 Z8e8e5de4dee d$d%gd%d$gge d(�d�e4dfe d$d%gd%d$gged(�d�e4dge d$d.gdhdAgged(�d�e4die d$d.gdhdAgged(�d�e4dje j9d9ed(�dd hd;�e4dkdAdCgdCdAggd�e4dFe d$d%gd%d$gg�d�e4dled$d%gd%d$gg�d�e4dme j6�;dAdA�d�g �7 Z8dndo� Z=e8e=� 7 Z8dpdq� Z>drds� Z?e8e?� 7 Z8e@� e@� fdtdu�ZAG dvdw� dwe3�ZBG dxdy� dye3�ZCG dzd{� d{e3�ZDG d|d}� d}e3�ZEG d~d� de3�ZFG d�d�� d�e3�ZGd�d�� ZHd�d�� ZIG d�d�� d�eBeE�ZJG d�d�� d�eBeE�ZKG d�d�� d�eBeE�ZLG d�d�� d�eBeE�ZMG d�d�� d�eBeE�ZNG d�d�� d�eBeE�ZOG d�d�� d�eB�ZPG d�d�� d�e3�ZQG d�d�� d�eBeC�ZRG d�d�� d�eBeE�ZSG d�d�� d�eBeC�ZTG d�d�� d�e3�ZUG d�d�� d�e3�ZVG d�d�� d�eDeG�ZWG d�d�� d�eDeG�ZXG d�d�� d�e3�ZYG d�d�� d�e3�ZZG d�d�� d�eY�Z[G d�d�� d�eY�Z\G d�d�� d�eY�Z]G d�d�� d�e3�Z^d�d�� Z_G d�d�� d�e3�Z`G d�d�� d�e3�Zad�d�� Zbd�d�� Zcd�d�� ZdG d�d�� d�e3�Zeefd�k� rze(� dS )�z# Test functions for linalg module � )�division�absolute_import�print_functionN)�array�single�double�csingle�cdouble�dot�identity)�multiply� atleast_2d�inf�asarray�matrix)�linalg)� matrix_power�norm� matrix_rank� multi_dot� LinAlgError)� _multi_dot_matrix_chain_order) �assert_� assert_equal� assert_raises�assert_array_equal�assert_almost_equal�assert_allclose�run_module_suite�dec�SkipTest�suppress_warningsc C s | p|S )N� )�a�br" r" ��/builddir/build/BUILDROOT/alt-python37-numpy-1.13.3-8.el8.x86_64/opt/alt/python37/lib64/python3.7/site-packages/numpy/linalg/tests/test_linalg.py�ifthen s r& c C s | p|S )Nr" )r# r$ r" r" r% �imply s r' � � c K s8 t | �jjttfkr|}n|}t| |fd|i|�� d S )N�decimal)r �dtype�typer r �old_assert_almost_equal)r# r$ �single_decimal�double_decimal�kwr* r" r" r% r $ s r c C s t t tttt tti| S )N)r r r r )r+ r" r" r% �get_real_dtype, s r1 c C s t ttttttti| S )N)r r r r )r+ r" r" r% �get_complex_dtype1 s r2 c C s | t tfkrdS dS d S )Ng�h㈵��>g�dy���=)r r )r+ r" r" r% �get_rtol6 s r3 �square� nonsquare� hermitian� generalizedzsize-0�stridedc @ s* e Zd Ze� fdd�Zdd� Zdd� ZdS )� LinalgCasec C s. t t|t�� || _|| _|| _t|�| _dS )z� A bundle of arguments to be passed to a test case, with an identifying name, the operands a and b, and a set of tags to filter the tests N)r � isinstance�str�namer# r$ � frozenset�tags)�selfr< r# r$ r> r" r" r% �__init__E s zLinalgCase.__init__c C s || j | j| jd� dS )zN Run the function `do` on this test case, expanding arguments )r> N)r# r$ r> )r? �dor" r" r% �checkP s zLinalgCase.checkc C s d| j f S )Nz<LinalgCase: %s>)r< )r? r" r" r% �__repr__V s zLinalgCase.__repr__N)�__name__� __module__� __qualname__�setr@ rB rC r" r" r" r% r9 D s r9 c C s0 | t kstd��x|D ]}|j| hB |_qW |S )z^ Add the given tag (a string) to each of the cases (a list of LinalgCase objects) z Invalid tag)�all_tags�AssertionErrorr> )�tag�cases�caser" r" r% � apply_tagY s rM i� r g �?g @g @g @)r+ r Zdouble_2g @r y �? @y @ @y @ @y @ @y @ �?r Z cdouble_2y �? @y �? �y �? �y �? �Z0x0)r r )r )r> Z 0x0_matrix)r � Z8x8� Z1x1rN Znonarray� � � Z matrix_b_onlyZmatrix_a_and_bZ single_nsq_1Z single_nsq_2g @Z double_nsq_1Z double_nsq_2Z csingle_nsq_1y �? �?y @ @y @ �y @ �y @ "@y @ @Z csingle_nsq_2y @ "�y @ �y @ @Z cdouble_nsq_1Z cdouble_nsq_2Zcdouble_nsq_1_2y �? �y @ �Zcdouble_nsq_2_2Z8x11� Z1x5� Z5x1Z0x4Z4x0ZhsingleZhdoubleZhcsingley @ �ZhcdoubleZhemptyZ hnonarrayZhmatrix_a_and_bZ hmatrix_1x1c C s g } �xt D �]}t|jtj�s"q t�|jd|j d|j g�}|jd krPd }n t�|jd|j d|j g�}t|jd |||j dhB d�}| � |� t�|jgd d �� d|jj �}|jd kr�d }n$t�|jgd d �� d|jj �}t|jd |||j dhB d�}| � |� q W | S ) NrP rQ � r( Z_tile3r7 )r> )rQ rP Z_tile213) �CASESr: r# �np�ndarrayr r$ r9 r< r> �append�reshape�shape)� new_casesrL r# r$ �new_caser" r" r% �_make_generalized_cases� s& $ $r^ c c s� t | tj�s| dfV dS dg| j }d|d< | jdkr@d|d< | jdkrRd |d < �x�tjt|�� D �]�}d d � t| j|�D �}td d � |D ��}tj || j d�}|� tj �� d� || }| |d< |� | j�}tt�|| k�� |dd�dd � |D �� fV | jdk�rJ| jd dk�rJt| j�}d|d< tjjj| |d�}|dfV | jdk�r�| jd dk�r�t| j�}d|d< tjjj| |d�}|dfV | jdkrd| jdd� dkrdt| j�}d|d< d|d< tjjj| |d�}|dfV qdW dS )z< Generate cartesian product of strides for all axes �nopN)rN )rN rQ ������rN ���rP )rN r` ���c S s g | ]\}}t || ��qS r" )�abs)�.0r# r$ r" r" r% � <listcomp>( s z%_stride_comb_iter.<locals>.<listcomp>c S s g | ]}t d d |��qS )N)�slice)re �repeatr" r" r% rf ) s )r+ l �>[= .Zstride_�_c S s g | ] }d | �qS )z%+dr" )re �jr" r" r% rf 2 s r )�stridesZ stride_xxx_0Zstride_xxx_0_x)rN rN Zstride_xxx_0_0)r: rW rX �ndim� itertools�product�tuple�zipr[ �emptyr+ �view�uint32�fill� __class__r �all�join�listrk �lib� stride_tricks� as_strided)�xZ stride_set�repeats� new_shape�slices�xi�sr" r" r% �_stride_comb_iter sF r� c C sz g } xpt D ]h}xbt|j�D ]T\}}xJt|j�D ]<\}}t|jd | d | |||jdhB d�}| �|� q.W qW q W | S )Nri r8 )r> )rV r� r# r$ r9 r< r> rY )r\ rL r# Za_labelr$ Zb_labelr] r" r" r% �_make_strided_casesF s r� c C sn xht D ]`}|j|@ |krq|j|@ r&qy|�| � W q tk rd d| }|t�� 7 }t|��Y qX qW dS )zl Run func on each of the cases with all of the tags in require, and none of the tags in exclude zIn test case: %r N)rV r> rB � Exception� traceback� format_excrI )�func�require�excluderL �msgr" r" r% � _check_casesW s r� c @ s e Zd Zdd� Zdd� ZdS )�LinalgSquareTestCasec C s t | jdhddhd� d S )Nr4 r7 zsize-0)r� r� )r� rA )r? r" r" r% � test_sq_casesm s z"LinalgSquareTestCase.test_sq_casesc C s t | jddhdhd� d S )Nr4 zsize-0r7 )r� r� )r� rA )r? r" r" r% �test_empty_sq_casesp s z(LinalgSquareTestCase.test_empty_sq_casesN)rD rE rF r� r� r" r" r" r% r� k s r� c @ s e Zd Zdd� Zdd� ZdS )�LinalgNonsquareTestCasec C s t | jdhddhd� d S )Nr5 r7 zsize-0)r� r� )r� rA )r? r" r" r% �test_nonsq_casesv s z(LinalgNonsquareTestCase.test_nonsq_casesc C s t | jddhdhd� d S )Nr5 zsize-0r7 )r� r� )r� rA )r? r" r" r% �test_empty_nonsq_casesy s z.LinalgNonsquareTestCase.test_empty_nonsq_casesN)rD rE rF r� r� r" r" r" r% r� t s r� c @ s e Zd Zdd� Zdd� ZdS )�HermitianTestCasec C s t | jdhddhd� d S )Nr6 r7 zsize-0)r� r� )r� rA )r? r" r" r% �test_herm_cases~ s z!HermitianTestCase.test_herm_casesc C s t | jddhdhd� d S )Nr6 zsize-0r7 )r� r� )r� rA )r? r" r" r% �test_empty_herm_cases� s z'HermitianTestCase.test_empty_herm_casesN)rD rE rF r� r� r" r" r" r% r� | s r� c @ s( e Zd Zejdd� �Zejdd� �ZdS )� LinalgGeneralizedSquareTestCasec C s t | jddhdhd� d S )Nr7 r4 zsize-0)r� r� )r� rA )r? r" r" r% �test_generalized_sq_cases� s z9LinalgGeneralizedSquareTestCase.test_generalized_sq_casesc C s t | jdddhd� d S )Nr7 r4 zsize-0)r� )r� rA )r? r" r" r% � test_generalized_empty_sq_cases� s z?LinalgGeneralizedSquareTestCase.test_generalized_empty_sq_casesN)rD rE rF r �slowr� r� r" r" r" r% r� � s r� c @ s( e Zd Zejdd� �Zejdd� �ZdS )�"LinalgGeneralizedNonsquareTestCasec C s t | jddhdhd� d S )Nr7 r5 zsize-0)r� r� )r� rA )r? r" r" r% � test_generalized_nonsq_cases� s z?LinalgGeneralizedNonsquareTestCase.test_generalized_nonsq_casesc C s t | jdddhd� d S )Nr7 r5 zsize-0)r� )r� rA )r? r" r" r% �"test_generalized_empty_nonsq_cases� s zELinalgGeneralizedNonsquareTestCase.test_generalized_empty_nonsq_casesN)rD rE rF r r� r� r� r" r" r" r% r� � s r� c @ s( e Zd Zejdd� �Zejdd� �ZdS )� HermitianGeneralizedTestCasec C s t | jddhdhd� d S )Nr7 r6 zsize-0)r� r� )r� rA )r? r" r" r% �test_generalized_herm_cases� s z8HermitianGeneralizedTestCase.test_generalized_herm_casesc C s t | jdddhdhd� d S )Nr7 r6 zsize-0�none)r� r� )r� rA )r? r" r" r% �!test_generalized_empty_herm_cases� s z>HermitianGeneralizedTestCase.test_generalized_empty_herm_casesN)rD rE rF r r� r� r� r" r" r" r% r� � s r� c C s� t | �} | jdkr�| j|jkr<| jd d� |jdd � }n(| j|jd kr\| jd d� }ntd��tj|t�| |�d�}x8tjt t | jd d� �� D ]}t | | || �||< q�W |S t | |�S d S )NrQ ra rN zNot implemented...)r+ rb ) r rl r[ � ValueErrorrW rq � common_typerm rn �map�ranger )r# r$ r~ �r�cr" r" r% �dot_generalized� s r� c C sp t | �} | jdkr^tj| j| jd�}x4tjtt | jd d� �� D ]}t | jd �||< q@W |S t | jd �S d S )NrQ )r+ rb r ) r rl rW rq r[ r+ rm rn r� r� r )r# r� r� r" r" r% �identity_like_generalized� s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) � TestSolvec C s: t �||�}t|t||�� ttt|t�t|t��� d S )N)r �solver r� r r' r: r )r? r# r$ r> r| r" r" r% rA � s z TestSolve.doc c s, dd� }x t tttgD ]}||fV qW d S )Nc S s2 t jddgddgg| d�}tt�||�j| � d S )NrN g �?)r+ )rW r r r r� r+ )r+ r| r" r" r% rB � s z#TestSolve.test_types.<locals>.check)r r r r )r? rB r+ r" r" r% � test_types� s zTestSolve.test_typesc C s. G dd� dt j�}t �d��ddd�}t �d��ddd��|�}t�||�d d �dd�d d �f }t�|d d �dd�dd�f |d d �dd�d d �f �}t||� tt ||�� t tj tj|d d �dd�dd�f |� t t tj||d d �dd�d d �f � t �d��ddd�}t t tj||� t t tj|dd� |dd� � t �d��dd��|�}t�||�d d �dd�f }t�|d d �dd�dd�f |d d �dd�f �}t||� tt ||�� t �d��dd�}t t tj||� t t tj|dd� |dd� � t t tj|d d �dd�dd�f |� d S ) Nc @ s e Zd ZdS )z,TestSolve.test_0_size.<locals>.ArraySubclassN)rD rE rF r" r" r" r% � ArraySubclass� s r� rO rP r( rN rQ r ) rW rX �arangerZ rr r r� r r r: r r r� )r? r� r# r$ �expected�resultr" r" r% � test_0_size� s* "8 (& 2 zTestSolve.test_0_sizec C s G dd� dt j�}t �d��ddd�}t �d��ddd��|�}t�||�d d �d d �dd�f }t�||d d �d d �dd�f �}t||� tt ||�� t�||�d d �dd�dd�f }t�|d d �dd�dd�f |d d �dd�dd�f �}t||� tt ||�� d S ) Nc @ s e Zd ZdS )z.TestSolve.test_0_size_k.<locals>.ArraySubclassN)rD rE rF r" r" r" r% r� � s r� rR rN rP r( rQ r ) rW rX r� rZ rr r r� r r r: )r? r� r# r$ r� r� r" r" r% � test_0_size_k� s "" "8 zTestSolve.test_0_size_kN)rD rE rF rA r� r� r� r" r" r" r% r� � s !r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestInvc C s< t �|�}tt||�t|�� ttt|t�t|t��� d S )N) r �invr r� r� r r' r: r )r? r# r$ r> Za_invr" r" r% rA s z TestInv.doc c s, dd� }x t tttgD ]}||fV qW d S )Nc S s0 t jddgddgg| d�}tt�|�j| � d S )NrN g �?)r+ )rW r r r r� r+ )r+ r| r" r" r% rB s z!TestInv.test_types.<locals>.check)r r r r )r? rB r+ r" r" r% r� s zTestInv.test_typesc C s� G dd� dt j�}t jdt jd��|�}t�|�}t|jj t j k� t |j |j � tt ||�� t jdt jd��|�}t�|�}t|jj t jk� t |j |j � tt ||�� d S )Nc @ s e Zd ZdS )z*TestInv.test_0_size.<locals>.ArraySubclassN)rD rE rF r" r" r" r% r� s r� )r rN rN )r+ )r r )rW rX �zeros�int_rr r r� r r+ r, �float64r r[ r: � complex64)r? r� r# �resr" r" r% r� s zTestInv.test_0_sizeN)rD rE rF rA r� r� r" r" r" r% r� s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� TestEigvalsc C s& t �|�}t �|�\}}t||� d S )N)r �eigvals�eigr )r? r# r$ r> �ev�evalues�evectorsr" r" r% rA , s zTestEigvals.doc c s, dd� }x t tttgD ]}||fV qW d S )Nc S s` t jddgddgg| d�}tt�|�j| � t jddgddgg| d�}tt�|�jt| �� d S )NrN g �?)r+ ra )rW r r r r� r+ r2 )r+ r| r" r" r% rB 2 s z%TestEigvals.test_types.<locals>.check)r r r r )r? rB r+ r" r" r% r� 1 s zTestEigvals.test_typesc C s� G dd� dt j�}t jdt jd��|�}t�|�}t|jj t j k� t d|j � tt |t j�� t jdt jd��|�}t�|�}t|jj t jk� t d|j � tt |t j�� d S )Nc @ s e Zd ZdS )z.TestEigvals.test_0_size.<locals>.ArraySubclassN)rD rE rF r" r" r" r% r� < s r� )r rN rN )r+ )r rN )r r )r )rW rX r� r� rr r r� r r+ r, r� r r[ r: r� )r? r� r# r� r" r" r% r� : s zTestEigvals.test_0_sizeN)rD rE rF rA r� r� r" r" r" r% r� * s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestEigc C sd t �|�\}}tt||�t�|�t�|�dd d d �f t|j�d� tt t |t �t |t ��� d S )N.)�rtol) r r� r r� rW r r3 r+ r r' r: r )r? r# r$ r> r� r� r" r" r% rA O s z TestEig.doc c s, dd� }x t tttgD ]}||fV qW d S )Nc S s� t jddgddgg| d�}t j�|�\}}t|j| � t|j| � t jddgddgg| d�}t j�|�\}}t|jt| �� t|jt| �� d S )NrN g �?)r+ ra )rW r r r� r r+ r2 )r+ r| �w�vr" r" r% rB W s z!TestEig.test_types.<locals>.check)r r r r )r? rB r+ r" r" r% r� V s zTestEig.test_typesc C s� G dd� dt j�}t jdt jd��|�}t�|�\}}t|jj t j k� t|jj t j k� t |j |j � t d|j � tt |t j�� t jdt jd��|�}t�|�\}}t|jj t jk� t|jj t jk� t |j |j � t d|j � tt |t j�� d S )Nc @ s e Zd ZdS )z*TestEig.test_0_size.<locals>.ArraySubclassN)rD rE rF r" r" r" r% r� g s r� )r rN rN )r+ )r rN )r r )r )rW rX r� r� rr r r� r r+ r, r� r r[ r: r� )r? r� r# r� �res_vr" r" r% r� e s zTestEig.test_0_sizeN)rD rE rF rA r� r� r" r" r" r% r� M s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestSVDc C s� d|kr t ttj|d� d S t�|d�\}}}t|tt�|�t�|�dd d d �f t�|��t|j �d� t t t |t �t |t ��� t t t |t �t |t ��� d S )Nzsize-0r .)r� )r r r �svdr r� rW r r3 r+ r r' r: r )r? r# r$ r> �ur� �vtr" r" r% rA ~ s & z TestSVD.doc c s, dd� }x t tttgD ]}||fV qW d S )Nc S st t jddgddgg| d�}t�|�\}}}t|j| � t|jt| �� t|j| � tj|dd�}t|jt| �� d S )NrN g �?)r+ F)� compute_uv)rW r r r� r r+ r1 )r+ r| r� r� Zvhr" r" r% rB � s z!TestSVD.test_types.<locals>.check)r r r r )r? rB r+ r" r" r% r� � s zTestSVD.test_typesc C sd t jdt jd�}ttjtj|� t jdt jd�}ttjtj|� t jdt jd�}ttjtj|� d S )N)r r )r+ )r rN )rN r )rW r� r� r r r r� )r? r# r" r" r% r� � s zTestSVD.test_0_sizeN)rD rE rF rA r� r� r" r" r" r% r� | s r� c @ s e Zd Zdd� Zdd� ZdS )� TestCondSVDc C sZ t |�}d|kr&tttj|dd� d S tj|dd�}t|d |d t�|�ddd� d S ) Nzsize-0F)r� ).r ).ra rT rS )r. r/ )r r r r r� r �cond)r? r# r$ r> r� r� r" r" r% rA � s zTestCondSVD.doc C sD t �dddgdddgdddgg�}tt�|�t�|d �d � d S )Ng �?g @r g �g @g @)N.)rW r r r r� )r? �Ar" r" r% � test_stacked_arrays_explicitly� s "z*TestCondSVD.test_stacked_arrays_explicitlyN)rD rE rF rA r� r" r" r" r% r� � s r� c @ s e Zd Zdd� Zdd� ZdS )� TestCond2c C s\ t |�}d|kr&tttj|dd� d S tj|dd�}t|d |d t�|d�ddd � d S ) Nzsize-0F)r� ).r ).ra rP rT rS )r. r/ )r r r r r� r r� )r? r# r$ r> r� r� r" r" r% rA � s z TestCond2.doc C sH t �dddgdddgdddgg�}tt�|d�t�|d d�d � d S ) Ng �?g @r g �g @g @rP )N.)rW r r r r� )r? r� r" r" r% r� � s "z(TestCond2.test_stacked_arrays_explicitlyN)rD rE rF rA r� r" r" r" r% r� � s r� c @ s e Zd Zdd� ZdS )� TestCondInfc C s6 t dddgdddgdddgg�}tt�|t�d� d S )Ng �?r g �g @)r r r r� r )r? r� r" r" r% �test� s zTestCondInf.testN)rD rE rF r� r" r" r" r% r� � s r� c @ s e Zd Zdd� ZdS )�TestPinvc C sD t �|�}tt||��|�|ddd� ttt|t�t|t��� d S )NrT rS )r. r/ )r �pinvr r r r' r: r )r? r# r$ r> Za_ginvr" r" r% rA � s z TestPinv.doN)rD rE rF rA r" r" r" r% r� � s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestDetc C s� t �|�}t �|�\}}t|�jjttfkr<t|��t�}nt|��t �}t � |�}t |t j |dd�� t |t�|� t j |dd�� t�|�}t�|�}|dk} t t�|| �d� t|| t � d S )Nra )�axisr rN )r �det�slogdetr r+ r, r r �astyper r� r r �reducerW �exp� atleast_1drd r r ) r? r# r$ r> �dr� �ld�adr� �mr" r" r% rA � s z TestDet.doc C s t t�dgg�d� t tt�dgg��t� t t�dgg�d� t tt�dgg��t� t t�dgg�dt f� t tt�dgg�d �t� t tt�dgg�d �t� t t�dgg�dt f� t tt�dgg�d �t� t tt�dgg�d �t� d S )Ng y r rN )r r r� r, r r r� r )r? r" r" r% � test_zero� s zTestDet.test_zeroc c s, dd� }x t tttgD ]}||fV qW d S )Nc S s^ t jddgddgg| d�}tt j�|�j| � t j�|�\}}t|jt| �� t|j| � d S )NrN g �?)r+ )rW r r r r� r+ r� r1 )r+ r| �phr� r" r" r% rB � s z!TestDet.test_types.<locals>.check)r r r r )r? rB r+ r" r" r% r� � s zTestDet.test_typesc C s� t jdt jd�}t�|�}t|d� t|jjt jk� t� |�}t|d� t|d jjt jk� t|d jjt j k� t jdt j d�}t�|�}t|d� t|jjt j k� t� |�}t|d� t|d jjt j k� t|d jjt j k� d S )N)r r )r+ g �?)rN r r rN ) rW r� r� r r� r r r+ r, r� �float32r� )r? r# r� r" r" r% r� � s zTestDet.test_0_sizeN)rD rE rF rA r� r� r� r" r" r" r% r� � s r� c @ s e Zd Zdd� ZdS )� TestLstsqc C sd d|kr t ttj||� d S t�|�}|j\}}t�|d�\}}} t�||�\} } } } ||krzt|t || �� t | |� n t | |� t| | � |�� | |kr�||kr�t�t t� || �| ��d j dd�}t�|�}t�|�jdkr�d|_t | j|j� nt�g ��t| ��}t| |� tt�| jtj�� ttt|t�t| t��� ttt|t�t| t��� d S )Nzsize-0r rP )r� rN )rN )r r r �lstsqrW r r[ r� r r r �__array_wrap__rd �sumrl r rr r, r � issubdtyper+ �floatingr' r: r )r? r# r$ r> �arrr� �nr� r� r� r| Z residuals�rank�svZ expect_residsr" r" r% rA s. & z TestLstsq.doN)rD rE rF rA r" r" r" r% r� s r� c @ s� e Zd Zeddgddgg�Zeddgddgg�Zeddgddgg�Zedd gd d gg�Zed�Z e dd d �f � � Z e dd d �f e dd d �f< e e dd d �f< d d� Z dd� Z dd� Zdd� Zdd� Zdd� Zdd� Zd S )�TestMatrixPowerr rN ra rR i����rb � g�������?g������ @g333333�?gffffff�?Nc C s( t t| jdd d d d �| j� d S )NrP �d i � rN )r r �R90)r? r" r" r% �test_large_power= s z TestMatrixPower.test_large_powerc C s& t t| jdd d d �td�� d S )NrP r� i r� )r r r� r )r? r" r" r% � test_large_power_trailing_zeroA s z.TestMatrixPower.test_large_power_trailing_zeroc c s0 dd� }x"| j | j| jgD ]}||fV qW d S )Nc S s0 t | d�}t|t| jd �� t|j| j� d S )Nr )r r r r[ r+ )�M�mzr" r" r% �tzF s z'TestMatrixPower.testip_zero.<locals>.tz)�Arb22�arbfloat�large)r? r� r� r" r" r% � testip_zeroE s zTestMatrixPower.testip_zeroc c s0 dd� }x"| j | j| jgD ]}||fV qW d S )Nc S s& t | d�}t|| � t|j| j� d S )NrN )r r r+ )r� r� r" r" r% r� N s z&TestMatrixPower.testip_one.<locals>.tz)r� r� r� )r? r� r� r" r" r% � testip_oneM s zTestMatrixPower.testip_onec c s0 dd� }x"| j | j| jgD ]}||fV qW d S )Nc S s, t | d�}t|t| | �� t|j| j� d S )NrP )r r r r+ )r� r� r" r" r% r� V s z&TestMatrixPower.testip_two.<locals>.tz)r� r� r� )r? r� r� r" r" r% � testip_twoU s zTestMatrixPower.testip_twoc c s4 dd� }x&| j | j| j| jgD ]}||fV q W d S )Nc S s( t | d�}tt| jd �t|| �� d S )Nra r )r r r r[ r )r� r� r" r" r% r� ^ s z)TestMatrixPower.testip_invert.<locals>.tz)r� r� r� r� )r? r� r� r" r" r% � testip_invert] s z TestMatrixPower.testip_invertc s$ dd l }t|jjj� fdd�� d S )Nr c s t � jd�S )Nra )r �noninvr" )r? r" r% �<lambda>g � z;TestMatrixPower.test_invert_noninvertible.<locals>.<lambda>)� numpy.linalgr r r )r? �numpyr" )r? r% �test_invert_noninvertibled s z)TestMatrixPower.test_invert_noninvertible)rD rE rF r r� r� r� r� r r� �copy�tr� r� r� r� r� r� r r" r" r" r% r� 2 s r� c @ s e Zd Zdd� ZdS )� TestBoolPowerc C s( t ddgddgg�}tt|d�|� d S )NTFrP )r r r )r? r� r" r" r% � test_squarel s zTestBoolPower.test_squareN)rD rE rF r r" r" r" r% r j s r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) � TestEigvalshc C s^ t �|d�}t �|�\}}|jdd� t||t|j�d� t �|d�}t||t|j�d� d S )N�Lra )r� )r� �U)r �eigvalshr� �sortr r3 r+ )r? r# r$ r> r� r� r� �ev2r" r" r% rA s s zTestEigvalsh.doc c s, dd� }x t tttgD ]}||fV qW d S )Nc S s: t jddgddgg| d�}t j�|�}t|jt| �� d S )NrN g �?)r+ )rW r r r r r+ r1 )r+ r| r� r" r" r% rB s z&TestEigvalsh.test_types.<locals>.check)r r r r )r? rB r+ r" r" r% r� ~ s zTestEigvalsh.test_typesc C sX t jddgddggt jd�}ttt jj|dd� ttt jj|d� ttt jj|d� d S )NrN g �?)r+ �lrong)�UPLO�lower�upper)rW r r� r r� r r )r? r| r" r" r% � test_invalid� s zTestEigvalsh.test_invalidc C s� t jddgddggt jd�}t jddgddggt jd�}t jddgt jd�}tt j�}t j�|�}t|||d� t jj|dd�}t|||d� t jj|dd�}t|||d� t jj|d d�}t|||d� t jj|d d�}t|||d� d S ) Nr rN )r+ ra )r� r )r �lr r� )rW r r r3 r r r )r? �Klo�Kup�tgtr� r� r" r" r% � test_UPLO� s zTestEigvalsh.test_UPLOc C s� G dd� dt j�}t jdt jd��|�}t�|�}t|jj t j k� t d|j � tt |t j�� t jdt jd��|�}t�|�}t|jj t jk� t d|j � tt |t j�� d S )Nc @ s e Zd ZdS )z/TestEigvalsh.test_0_size.<locals>.ArraySubclassN)rD rE rF r" r" r" r% r� � s r� )r rN rN )r+ )r rN )r r )r )rW rX r� r� rr r r r r+ r, r� r r[ r: r� r� )r? r� r# r� r" r" r% r� � s zTestEigvalsh.test_0_sizeN)rD rE rF rA r� r r r� r" r" r" r% r q s r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �TestEighc C s� t �|�\}}t �|�\}}|jdd� t||� tt||�t�|�dd d d �f t�|� t |j �d� t �|d�\}} t||� tt|| �t�|�dd d d �f t�| � t |j �t |�d� d S )Nra )r� .)r� r )r� �err_msg) r �eighr� r r r r� rW r r3 r+ �repr) r? r# r$ r> r� Zevcr� r� r Zevc2r" r" r% rA � s z TestEigh.doc c s, dd� }x t tttgD ]}||fV qW d S )Nc S sJ t jddgddgg| d�}t j�|�\}}t|jt| �� t|j| � d S )NrN g �?)r+ )rW r r r r r+ r1 )r+ r| r� r� r" r" r% rB � s z"TestEigh.test_types.<locals>.check)r r r r )r? rB r+ r" r" r% r� � s zTestEigh.test_typesc C sX t jddgddggt jd�}ttt jj|dd� ttt jj|d� ttt jj|d� d S )NrN g �?)r+ r )r r r )rW r r� r r� r r )r? r| r" r" r% r � s zTestEigh.test_invalidc C s t jddgddggt jd�}t jddgddggt jd�}t jddgt jd�}tt j�}t j�|�\}}t|||d� t jj|dd�\}}t|||d� t jj|dd�\}}t|||d� t jj|d d�\}}t|||d� t jj|d d�\}}t|||d� d S ) Nr rN )r+ ra )r� r )r r r r� )rW r r r3 r r r )r? r r r r� r� r� r" r" r% r � s zTestEigh.test_UPLOc C s� G dd� dt j�}t jdt jd��|�}t�|�\}}t|jj t j k� t|jj t j k� t |j |j � t d|j � tt |t j�� t jdt jd��|�}t�|�\}}t|jj t jk� t|jj t jk� t |j |j � t d|j � tt |t j�� d S )Nc @ s e Zd ZdS )z+TestEigh.test_0_size.<locals>.ArraySubclassN)rD rE rF r" r" r" r% r� � s r� )r rN rN )r+ )r rN )r r )r )rW rX r� r� rr r r r r+ r, r� r r[ r: r� r� )r? r� r# r� r� r" r" r% r� � s zTestEigh.test_0_sizeN)rD rE rF rA r� r r r� r" r" r" r% r � s r c @ s\ e Zd ZdZdZdd� Zdd� Zdd� Zdd � Zd d � Z d d � Z dd� Z dd� Z dd� Z dS )� _TestNormNc C sF t tg �d� t ttg | jd��d� t tttg | jd���d� d S )Ng )r+ )r r r �dtr )r? r" r" r% � test_empty s z_TestNorm.test_emptyc C s� t �dddg�}t jd }t jd }|| }�x||D �]r}|�|�}t|t j �}tt|jj t j �� t |d� t � �:}|� td� t|d�}tt|jj t j �� t |d� W d Q R X t|d�}tt|jj t j �� t |d� t|d�}tt|jj t j �� t |d � t|d�}tt|jj t j �� t ||j� d �|j� d � � t|d �}tt|jj t j �� t ||j� d �|j� d � � t|t j�}tt|jj t j �� t |d � q4W d S )NrN r � AllInteger�AllFloatg zdivide by zero encounteredra rP g @g �?rR g �?g �?)rW r � typecodesr� r r r � issubclassr+ r, r� r r! �filter�RuntimeWarning) r? r# � exact_types� inexact_types� all_types�each_inexact_types�at�an�supr" r" r% �test_vector_return_type s: z!_TestNorm.test_vector_return_typec C s� t �dddgdddgg�}t jd }d}|| }�x�|D �]v}|�|�}t|t j �}tt|jj t j �� t |d� t � �:}|� td� t|d�}tt|jj t j �� t |d� W d Q R X t|d�}tt|jj t j �� t |d� t|d �}tt|jj t j �� t |d � t|d �}tt|jj t j �� t |d� t|t j�}tt|jj t j �� t |d� t|d �}tt|jj t j �� t |d� t|d �}tt|jj t j �� t|ddd� q8W d S )NrN r r ZfdFDg @zdivide by zero encounteredra g �?rP g�LX�z��?rb �fro�nucgU&,t=�@r( )r* )rW 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% �test_matrix_return_type9 s@ z!_TestNorm.test_matrix_return_typec s� ddddg}ddddg}ddddg}� fd d �}x|||fD ] }||� q<W x8t |� jd �t |� jd �t |� jd �fD ] }||� qvW d S ) NrN rP rQ rR ra rb rc r` c s� t jjt| �d� jd� t jjt| t�d� jd� t jjt| t �d� jd� t jjt| d�d� jd� t jjt| d�d� jd� t jjt| d �d� jd� t jjt| d �d � jd� t jjt| d �d � jd� d S )Ng��6ҭ�@)r* g @g �?rN g $@ra g� ��Q��?rP rb g�s�c���?r rR )rW �testingr r r r )r� )r? r" r% �_testq s z$_TestNorm.test_vector.<locals>._test)r+ )r r )r? r# r$ r� r. r� r" )r? r% � test_vectorl s z_TestNorm.test_vectorc C s� t ddgddgg| jd�}tt|�d� tt|d�d� tt|d�d � tt|t�d � tt|t �d � tt|d�d � tt|d �d � tt|d�d� tt|d�d� ttt|d� ttt|d� ttt|d� d S )NrN rQ rT rU )r+ g� ��T"@r* r+ g $@g (@g @ra g @rP g��W�?"@rb gї�ʄ�?Znofrorc r )r r r r r r r� )r? r� r" r" r% �test_matrix_2x2� s z_TestNorm.test_matrix_2x2c C s� dt jdddgdddgdddgg| jd� }tt|�d � tt|d �d � tt|d �d � tt|t�d � tt|t �d� tt|d�d� tt|d�d� tt|d�d� tt|d�d� d S )Ng�������?rN rP rQ r( r rT )r+ grj=Q0�?r* r+ g�B+pc�?g�������?g333333�?g �?ra g�������?g�&��.d�?rb gܸՠ���?)rW r r r r r )r? r� r" r" r% �test_matrix_3x3� s *z_TestNorm.test_matrix_3x3c s� t dddgdddgg| jd�� x�d dd dddtjtj gD ]h�� �fd d �t� jd �D �}tt� �d d �|� � �fd d �t� jd �D �}tt� �dd �|� q:W tjdd| jd�� ddd���j }x�d ddddtjtj dgD ]܉x�t � t| |�d�D ]�}|\}}|d k �r ||7 }|d k �r.||7 }||k�rLt tt��|d � q�t��|d �}||| �||k �r����fdd �t�j� �D �}n ���fdd �t�j� �D �}t||� q�W q�W d S )NrN rP rQ rR rT r( )r+ ra r c s$ g | ] }t � d d �|f �d��qS )N)�ord)r )re �k)r� �orderr" r% rf � s z'_TestNorm.test_axis.<locals>.<listcomp>)r2 r� c s$ g | ] }t � |d d �f �d��qS )N)r2 )r )re r3 )r� r4 r" r% rf � s � rb r* c s* g | ]"}t � d d � j|�d��d��qS )N)r� )r2 )r �take)re r3 )�B�k_indexr4 r" r% rf � s c s, g | ]$}t � d d � j|�d�j�d��qS )N)r� )r2 )r r6 �T)re r3 )r7 r8 r4 r" r% rf � s )r r rW �Infr� r[ r r r� rZ rl rm � combinationsr r� ) r? Z expected0� expected1�ndr� �row_axis�col_axisr� r� r" )r� r7 r8 r4 r% � test_axis� s2 z_TestNorm.test_axisc C s� t jdd| jd��ddd�}d}d}t|d d d �}t|d d d d �}tt �|�||�d d �d � d }t|j |k|�|j |d d �� x�d dddddt j t j gD ]�}x�t |j �D ]v}t|||d �}t|||d d �}tt �|�||�||�d � t |j �}d||< t|�}t|j |k|�|j |||�� q�W q�W x�d ddddt j t j ddg D ]�}x�t�t |j �d�D ]�}t|||d �}t|||d d �}tt �|�||�||�d � t |j �}d||d < d||d < t|�}t|j |k|�|j |||�� �q`W �qHW d S )NrN r5 )r+ rP rQ rR zorder {0}, axis = {1}z;Shape mismatch found {0}, expected {1}, order={2}, axis={3})r2 r� T)r2 r� �keepdims)r )rN rN rN ra r rb r* r+ )rW r� r rZ r r �squeeze�formatr r[ r: r� rl rx ro rm � permutations) r? r� Z allclose_errZ shape_errr� �found�expected_shaper4 r3 r" r" r% � test_keepdims� sB " z_TestNorm.test_keepdimsc C s� t dddgdddgg| jd�}tjdd| jd��ddd�}ttt|d d � ttt|d d � tttddgd d � tttddgd d � x<d D ]4}ttt||d � ttt||d � ttt||d� q�W ttjt|d d� ttjt|d d� ttt|d d� d S )NrN rP rQ rR rT r( )r+ r5 r* r r+ )r rQ )r rN )rN rP )rP rQ )r rN rP ) r r rW r� rZ r r� r � AxisError)r? r� r7 r4 r" r" r% � test_bad_args� s z_TestNorm.test_bad_args)rD rE rF r r r r) r, r/ r0 r1 r@ rG rI r" r" r" r% r s )3 %*r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestNorm_NonSystematicc C s* t jdt jd�}tt|dd�ddd� d S )Nr� )r+ rQ )r2 g�����L)@rP )r* )rW r� � longdoubler- r )r? r| r" r" r% �test_longdouble_norm s z+TestNorm_NonSystematic.test_longdouble_normc C s, t jdgt jd�}tt|dd�ddd� d S )Ni �)r+ rQ )r2 l rT )r* )rW r �int32r- r )r? r| r" r" r% � test_intmin s z"TestNorm_NonSystematic.test_intminc C s� t jdt jd�}d|d< d|d< d}tt jj|dd �|d d � |�t j�}tt jj|dd �|d d � |�t j�}tt jj|dd �|d d � d S )N)rP )r+ y @ @r y � @rN g���V;'@rQ )r2 r� )r* � rT ) rW rq � clongdoubler- r r r� � complex128r� )r? r� r� r" r" r% �test_complex_high_ord# s z,TestNorm_NonSystematic.test_complex_high_ordN)rD rE rF rL rN rR r" r" r" r% rJ s rJ c @ s e Zd ZejZdZdS )�TestNormDoubler) N)rD rE rF rW r r r r" r" r" r% rS 0 s rS c @ s e Zd ZejZdZdS )�TestNormSingler( N)rD rE rF rW r� r r r" r" r" r% rT 5 s rT c @ s e Zd ZejZdZdS )� TestNormInt64r) N)rD rE rF rW �int64r r r" r" r" r% rU : s rU c @ s e Zd Zdd� ZdS )�TestMatrixRankc c s� t dtt�d��fV t�d�}d|d< t t|�dfV t tt�d��dfV t tddddg�dfV t tt�d��dfV t tdg�dfV t�|t�d�t�d�g�}t t|�t�dddg�fV t td�dfV d S ) NrR g )ra ra rQ )rR rR r rN )rR )r r rW �eyer� r )r? �I�msr" r" r% �test_matrix_rankA s z TestMatrixRank.test_matrix_rankN)rD rE rF r[ r" r" r" r% rW ? s rW c C s� t j�d�} x�td�D ]�}| jdd�}|d d �df |d d �df |d d �df< tt|�d� |d d �d f |d d �d f |d d �d f< tt|�d � qW d S ) Ni�3r� )�( r� )�sizerN rP r rO rR rT rQ rO )rW �random� RandomStater� �normalr r )�rng�i�Xr" r" r% �test_reduced_rankV s ,,rd c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �TestQRc C s� t |�}|j}|j\}}t||�}tj|dd�\}}t|j|k� t|j|k� tt||�� tt||�� t|j||fk� t|j||fk� tt ||�|� tt |j � � |�t � |�� tt �|�|� tj|dd�\} } t| j|k� t| j|k� tt| |�� tt| |�� t| j||fk� t| j||fk� tt | | �|� tt | j � � | �t � |�� tt �| �| � tj|dd�} t| j|k� tt| |�� t| | � d S )N�complete)�mode�reducedr� )r, r+ r[ �minr �qrr r: r r r9 �conjrW rX �triu) r? r# Za_typeZa_dtyper� r� r3 �qr� Zq1�r1�r2r" r" r% �check_qre s8 zTestQR.check_qrc C s t �d�}ttjtj|� d S )N)r rP )rW r� r r r rj )r? r# r" r" r% � test_qr_empty� s zTestQR.test_qr_emptyc C s� t ddgddgddggtjd�}tj|dd �\}}t|jtjk� t|jtjk� t|jd k� t|jd k� tj|jdd �\}}t|jtjk� t|jtjk� t|jd k� t|jd k� d S ) NrN rP rQ rR rT r( )r+ �raw)rg )rP rQ )rP )rQ rP ) r rW r r rj r r+ r[ r9 )r? r# �h�taur" r" r% � test_mode_raw� s zTestQR.test_mode_rawc C s� t ddgddgg�}t ddgddgddgg�}xNdD ]F}|�|�}|�|�}| �|� | �|� | �|j� | �t|�� q4W x^dD ]V}dd|�|� }dd|�|� }| �|� | �|� | �|j� | �t|�� q�W d S ) NrN rP rQ rR rT r( �fdy �?)r r� rp r9 r )r? r# r$ r �m1�m2r" r" r% �test_mode_all_but_economic� s z!TestQR.test_mode_all_but_economicc C sR t �d�}ttjtj|� t �d�}ttjtj|� t �d�}ttjtj|� d S )N)r r )r rN )rN r )rW r� r r r rj )r? r# r" r" r% r� � s zTestQR.test_0_sizeN)rD rE rF rp rq ru ry r� r" r" r" r% re c s &re c @ s e Zd Zdd� ZdS )� TestCholeskyc C s� G dd� dt j�}t jdt jd��|�}t�|�}t|j|j� t |j j t j k� t t |t j�� t jdt jd��|�}t�|�}t|j|j� t |j j t jk� t t |t j�� d S )Nc @ s e Zd ZdS )z/TestCholesky.test_0_size.<locals>.ArraySubclassN)rD rE rF r" r" r" r% r� � s r� )r rN rN )r+ )rN r r )rW rX r� r� rr r �choleskyr r[ r r+ r, r� r: r� )r? r� r# r� r" r" r% r� � s zTestCholesky.test_0_sizeN)rD rE rF r� r" r" r" r% rz � s rz c C s� t jdkrd} nd} x�tjtjfD ]v}tjd|d�}|�| �}|�d��� }t|j jd� x<t j t j t j fD ](}||�}t|||�� t|||�� qlW q"W d S )N�little�<�>rR )r+ �S�=)�sys� byteorderrW r� r� rX � newbyteorder�byteswapr r+ r r� r� r� r )Znative�dttr� Zn_arrZsw_arrZroutiner� r" r" r% �test_byteorder_check� s r� c C sp t �ddgddgg�} t �ddgddgg�}t �ddddg�dd d� }| |d< ||d< tt jjt jj|� d S )NrN rP rQ rR .)r r )rW r r� r r r r� )Z invertibleZnon_invertibler| r" r" r% � test_generalized_raise_multiloop� s r� c C sD d} y t �� }W n ttfk r0 td��Y nX |dk�r t �d� t �d� dd l}|�|jd� yt j j � � W n* t k r� Y n t �t j� Y nX y0t �dgg�}t j j �ddd|d||dd� W n: t k �r } zdt|�kr�t �| � W d d }~X Y nX t �t j� n$t �� \}}t �|�| k�r@td��d S ) N� zNot POSIX or fork failed.r rN )r r g �?zDORGQR parameter number 5zNumpy xerbla not linked in.)�os�fork�OSError�AttributeErrorr �close�resourceZ setrlimitZ RLIMIT_CORErW r � lapack_lite�xerblar� �_exit� EX_CONFIGr �dorgqrr; �wait� WEXITSTATUS)Z XERBLA_OK�pidr� r# �e�statusr" r" r% �test_xerbla_override� s: r� c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )� TestMultiDotc C sl t j�d�}t j�d�}t j�d�}tt|||g�|�|��|�� tt|||g�t �|t �||��� d S )N)r( rP )rP r( )rW r^ r r r )r? r� r7 �Cr" r" r% �(test_basic_function_with_three_arguments+ s z5TestMultiDot.test_basic_function_with_three_argumentsc C s\ t j�d�}t j�d�}t j�d�}t j�d�}tt||||g�|�|��|��|�� d S )N)r( rP )rP r( )rP rN )rW r^ r r r )r? r� r7 r� �Dr" r" r% �8test_basic_function_with_dynamic_programing_optimization5 s zETestMultiDot.test_basic_function_with_dynamic_programing_optimizationc C sL t j�d�}t j�d�}t j�d�}t j�d�}tt||||g�jd� d S )NrP )rP r( )r( rP )rP rP )rP )rW r^ r r r[ )r? �A1dr7 r� r� r" r" r% � test_vector_as_first_argument> s z*TestMultiDot.test_vector_as_first_argumentc C sL t j�d�}t j�d�}t j�d�}t j�d�}tt||||g�jd� d S )N)r( rP )rP r( rP )r( )rW r^ r r r[ )r? r� r7 r� �D1dr" r" r% � test_vector_as_last_argumentH s z)TestMultiDot.test_vector_as_last_argumentc C sL t j�d�}t j�d�}t j�d�}t j�d�}tt||||g�jd� d S )NrP )rP r( )r( rP r" )rW r^ r r r[ )r? r� r7 r� r� r" r" r% �&test_vector_as_first_and_last_argumentR s z3TestMultiDot.test_vector_as_first_and_last_argumentc C sj t j�d�t j�d�t j�d�t j�d�t j�d�t j�d�g}t �ddd d d d gddd dddgddddddgddddddgddddddgddddddgg�}t jddddddgddddddgddddddgdddddd gdddddd gddddddggt jd �}|d8 }t|d d �\}}tt �|d d �dd �f �t �|d d �dd �f �� tt �|�t �|�� d S )!N)� �# )r� � )r� rT )rT r� )r� � )r� r5 g g ��@g þ@g �O�@g �1�@g ���@g ��@g �@g ջ@g ��@g p�@g ��@g ��@g @�@g X�@g ��@r rN rQ rP rR rT )r+ T)� return_costsra )rW r^ r �intr r rl )r? �arraysZ m_expectedZ s_expectedr� r� r" r" r% � test_dynamic_programming_logic\ s. z+TestMultiDot.test_dynamic_programming_logicc C s& t ttg � t tttj�d�g� d S )N)rQ rQ )r r� r rW r^ )r? r" r" r% �test_too_few_input_arraysz s z&TestMultiDot.test_too_few_input_arraysN) rD rE rF r� r� r� r� r� r� r� r" r" r" r% r� ) s r� �__main__)r( r) )g�__doc__� __future__r r r r� r� rm r� �warningsr� rW r r r r r r r r r r r r r r� r r r r r Znumpy.linalg.linalgr Z numpy.testingr r r r r r r r r r! r& r' r- r1 r2 r3 rH �objectr9 rM r^ �seedrV rq rr �randr9 r^ r� r� rG 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 rJ rS rT rU rW rd re rz r� r� r� r� rD r" r" r" r% �<module> s� $ 0 / B #/& > 8DQ ^1V