U
|
P ±d7 ã @ s
|
d Z ddlZddlZddlZddlZddlZddlZddlZddlZ ddlm
|
Z
|
m Z m Z m Z mZmZmZmZ ddlmZ ddlmZmZmZmZ ddlmZ 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,e&Z-dóddZ&dd Z.dd Z/dd Z0ddddddhZ1G d d d Z2d d Z3e j4 5d ¡ g Z6e6e3de2d!e
|
d"d#gd$d%gge d&e
|
d#d"ge d&e2d'e
|
d"d#gd$d%gge d&e
|
d#d"ge d&e2d(e
|
d"d#gd$d%gge d&e
|
d#d"d%gd$d%d)gge d&e2d*e
|
d+d,gd-d.gge d&e
|
d/d+ge d&e2d0e
|
d+d,gd-d.gged&e
|
d/d+ged&e2d1e
|
d+d,gd-d.gged&e
|
d/d+d2gd3d4d5gged&e2d6e j7d7e d&e j7d8e d&dhd9e2d:e j4 8d;d;¡e j4 8d;¡e2d<e j4 8d=d=¡e j4 8d=¡e2d>d=d?gd@dAggd?d=gg
|
7 Z6e6e3de2dBe
|
d"d#d$gd$d%d)gge d&e
|
d#d"ge d&e2dCe
|
d"d#gd$d%gdDd)gge d&e
|
d#d"d$ge d&e2dEe
|
d"d#d$gd$d%d)gge d&e
|
d#d"ge d&e2dFe
|
d"d#gd$d%gdDd)gge d&e
|
d#d"d$ge d&e2dGe
|
dHdIdJgdKdLdMgge d&e
|
d/d+ge d&e2dNe
|
dHdIgdJdOgdPdQgge d&e
|
d/d+dJge d&e2dRe
|
dHdIdJgdKdLdMgged&e
|
d/d+ged&e2dSe
|
dHdIgdJdOgdPdQgged&e
|
d/d+dJged&e2dTe
|
dHdIdJgdKdLdMgged&e
|
d/d+gdUdVgged&e2dWe
|
dHdIgdJdOgdPdQgged&e
|
d/d+gdUdVgdUdVgged&e2dXe j4 8d;dY¡e j4 8d;¡e2dZe j4 8d=d[¡e j4 8d=¡e2d\e j4 8d[d=¡e j4 8d[¡e2d]e j4 8ddA¡e j4 8d¡dhd9e2d^e j4 8dAd¡e j4 8dA¡dhd9g7 Z6e6e3de2d_e
|
d"d#gd#d"gge d&de2d`e
|
d"d#gd#d"gge d&de2dae
|
d"d,gdbd=gge d&de2dce
|
d"d,gdbd=gged&de2dde j7d7e d&ddhd9e2ded=d?gd?d=ggde2dfe
|
d"d#gd#d"ggde2dge j4 8d=d=¡dg7 Z6dhdi Z9e6e9 7 Z6djdk Z:dldm Z;e6e; 7 Z6G dndo doZ<G dpdq dqe<Z=G drds dse<Z>G dtdu due<Z?G dvdw dwe<Z@G dxdy dye<ZAG dzd{ d{e<ZBd|d} ZCd~d ZDG dd de=e@ZEG dd deEZFG dd
d
e=e@ZGG dd deGZHG dd de=e@ZIG dd deIZJG dd de=e@ZKG dd deKZLG dd dZMG dd de=e@ZNG dd deNeMZOG dd de?eBZPG dd dePeMZQG dd de=e@ZRG dd deRZSG dd de=e>e@eAZTG d d¡ d¡eTZUG d¢d£ d£e?eBZVG d¤d¥ d¥eVZWG d¦d§ d§e=e@ZXG d¨d© d©eXZYG dªd« d«e=e>ZZG d¬d deZZ[ej\ ]d®d¯d° d±D ¡G d²d³ d³Z^G d´dµ dµe?eBZ_G d¶d· d·Z`G d¸d¹ d¹e?eBZaG dºd» d»ZbG d¼d½ d½ZcG d¾d¿ d¿ecZdG dÀdÁ dÁecZeG dÂdà dÃeeedZfG dÄdÅ dÅZgG dÆdÇ dÇecZhG dÈdÉ dÉecZiG dÊdË dËecZjG dÌdÍ dÍefehZkG dÎdÏ dÏefeiZlG dÐdÑ dÑefejZmG dÒdÓ dÓZndÔdÕ ZoG dÖd× d×ZpG dØdÙ dÙZqdÚdÛ Zrej\jse+dÜdÝdÞdß Ztdàdá Zuej\jse+dâdÝej\jvdãdä ZwG dådæ dæZxG dçdè dèZyG dédê dêZzdëdì Z{ej\j|dídÝdîdï Z}ej\j~e* dðdÝdñdò ZdS )ôz# Test functions for linalg module
|
|
é N)ÚarrayÚsingleÚdoubleÚcsingleÚcdoubleÚdotÚidentityÚmatmul)Úswapaxes)ÚmultiplyÚ
|
atleast_2dÚinfÚasarray)Ú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Úsuppress_warningsÚassert_raises_regexÚ HAS_LAPACK64ÚIS_WASMc C s" t | t|tjrt |ntjkS ©N)ÚtypeÚ
|
isinstanceÚnpÚndarray)ÚoutZin_© r& úUd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/linalg/tests/test_linalg.pyÚconsistent_subclass s ÿr( é é c K s8 t | jjttfkr|}n|}t| |fd|i| d S )NÚdecimal)r Údtyper! r r Úold_assert_almost_equal)ÚaÚbÚsingle_decimalÚdouble_decimalÚkwr+ r& r& r' r $ s r c C s t t tttt tti| S r )r r r r ©r, r& r& r' Úget_real_dtype, s ÿÿr4 c C s t ttttttti| S r )r r r r r3 r& r& r' Úget_complex_dtype1 s ÿÿr5 c C s | t tfkrdS dS d S )Ngñhãµøä>gdyáý¥=)r r r3 r& r& r' Úget_rtol6 s r6 ÚsquareÚ nonsquareÚ hermitianÚ generalizedúsize-0Ústridedc @ s* e Zd Ze fddZdd 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 r" ÚstrÚnamer. r/ Ú frozensetÚtags)Úselfr? r. r/ rA r& r& r' Ú__init__F s
|
zLinalgCase.__init__c C s || j | j| jd dS )zN
|
Run the function `do` on this test case, expanding arguments
|
©rA N)r. r/ rA )rB Údor& r& r' ÚcheckQ s zLinalgCase.checkc C s d| j dS )Nz <LinalgCase: ú>)r? ©rB r& r& r' Ú__repr__W s zLinalgCase.__repr__N)Ú__name__Ú
|
__module__Ú __qualname__ÚsetrC rF rI r& r& r& r' r= E s r= c C s, | t kstd|D ]}|j| hB |_q|S )z^
|
Add the given tag (a string) to each of the cases (a list of LinalgCase
|
objects)
|
z Invalid tag)Úall_tagsÚAssertionErrorrA )ÚtagZcasesÚcaser& r& r' Ú apply_tag[ s rR éÒ r ç ð?ç @ç @ç @r3 r Zdouble_2ç @r y ð? @y @ @y @ @y @ @y @ ð?r Z cdouble_2y ð? @y ð? Ày ð? Ày ð? ÀZ0x0©r r ©r rD Z8x8é Z1x1é Znonarrayé é é Z 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 hnonarrayZ matrix_b_onlyZ hmatrix_1x1c C s g } t D ]}t|jtjs qt |jd|j d|j g¡}|jd krLd }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| S )
|
Nr] r^ é r) Z_tile3r: rD ©r^ r] Z_tile213) ÚCASESr" r. r# r$ r r/ r= r? rA ÚappendÚreshapeÚshape)Ú new_casesrQ r. r/ Únew_caser& r& r' Ú_make_generalized_casesë s*
|
|
|
ÿ
|
$
|
$
|
ÿ rj c c sè t | tjs| dfV dS dg| j }d|d< | jdkr@d|d< | jdkrRd |d
|
< tjt| D ]}d d t| j|D }td d |D }tj || j
|
d}| tj ¡ d¡ || }| |d< | | j¡}tt || k¡ |dd dd |D ¡ fV | jdkrF| jd dkrFt| j}d|d< tjjj| |d}|dfV | jdkr| jd dkrt| j}d|d< tjjj| |d}|dfV | jdkr`| jdd
dkr`t| j}d|d< d|d< tjjj| |d}|dfV q`dS )z<
|
Generate cartesian product of strides for all axes
|
ZnopN©r\ )r\ r^ éüÿÿÿéÿÿÿÿr\ éþÿÿÿr] )r\ rl éýÿÿÿ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 r )Úslice)rq Úrepeatr& r& r' rr s r3 l ï>[= .Zstride_Ú_c S s g | ] }d | qS )z%+dr& )rq Újr& r& r' rr ( s r )ÚstridesZ stride_xxx_0Zstride_xxx_0_x©r\ r\ Zstride_xxx_0_0)r" r# r$ ÚndimÚ itertoolsÚproductÚtupleÚziprg Úemptyr, ÚviewZuint32ÚfillÚ __class__r ÚallÚjoinÚlistrw ÚlibZ stride_tricksZ
|
as_strided)ÚxZ
|
stride_setZrepeatsÚ new_shapeZslicesÚxiÚsr& r& r' Ú_stride_comb_iter sF
|
|
|
|
|
|
|
|
r c C sn g } t D ]`}t|jD ]P\}}t|jD ]<\}}t|jd | d | |||jdhB d}| |¡ q(qq| S )Nru r< rD )rd r r. r/ r= r? rA re )rh rQ r. Za_labelr/ Zb_labelri r& r& r' Ú_make_strided_cases= s
|
ÿr c @ s" e Zd ZeZe e fddZdS )ÚLinalgTestCasec C s | j D ]v}|j|@ |krq|j|@ r&qz| | j¡ W q tk
|
rz } z&d|d}|t ¡ 7 }t||W 5 d}~X Y qX qdS )zx
|
Run func on each of the cases with all of the tags in require, and none
|
of the tags in exclude
|
zIn test case: z
|
|
N)Ú
|
TEST_CASESrA rF rE Ú ExceptionÚ tracebackÚ
|
format_excrO )rB ÚrequireÚexcluderQ ÚeÚmsgr& r& r' Ú check_casesQ s
|
|
zLinalgTestCase.check_casesN)rJ rK rL rd r rM r r& r& r& r' r N s r c @ s e Zd Zdd Zdd ZdS )ÚLinalgSquareTestCasec C s | j dhddhd d S )Nr7 r: r; ©r r ©r rH r& r& r' Ú test_sq_casesg s ÿz"LinalgSquareTestCase.test_sq_casesc C s | j ddhdhd d S )Nr7 r; r: r r rH r& r& r' Útest_empty_sq_casesk s
|
ÿz(LinalgSquareTestCase.test_empty_sq_casesN)rJ rK rL r r r& r& r& r' r e s r c @ s e Zd Zdd Zdd ZdS )ÚLinalgNonsquareTestCasec C s | j dhddhd d S )Nr8 r: r; r r rH r& r& r' Útest_nonsq_casesr s ÿz(LinalgNonsquareTestCase.test_nonsq_casesc C s | j ddhdhd d S )Nr8 r; r: r r rH r& r& r' Útest_empty_nonsq_casesv s
|
ÿz.LinalgNonsquareTestCase.test_empty_nonsq_casesN)rJ rK rL r r r& r& r& r' r p s r c @ s e Zd Zdd Zdd ZdS )ÚHermitianTestCasec C s | j dhddhd d S )Nr9 r: r; r r rH r& r& r' Útest_herm_cases} s ÿz!HermitianTestCase.test_herm_casesc C s | j ddhdhd d S )Nr9 r; r: r r rH r& r& r' Útest_empty_herm_cases s
|
ÿz'HermitianTestCase.test_empty_herm_casesN)rJ rK rL r r r& r& r& r' r { s r c @ s, e Zd Zejjdd Zejjdd ZdS )Ú LinalgGeneralizedSquareTestCasec C s | j ddhdhd d S )Nr: r7 r; r r rH r& r& r' Útest_generalized_sq_cases s
|
ÿz9LinalgGeneralizedSquareTestCase.test_generalized_sq_casesc C s | j dddhd d S )Nr: r7 r; ©r r rH r& r& r' Ú test_generalized_empty_sq_cases s z?LinalgGeneralizedSquareTestCase.test_generalized_empty_sq_casesN)rJ rK rL ÚpytestÚmarkÚslowr¢ r¤ r& r& r& r' r¡ s
|
r¡ c @ s, e Zd Zejjdd Zejjdd ZdS )Ú"LinalgGeneralizedNonsquareTestCasec C s | j ddhdhd d S )Nr: r8 r; r r rH r& r& r' Ú test_generalized_nonsq_cases s
|
ÿz?LinalgGeneralizedNonsquareTestCase.test_generalized_nonsq_casesc C s | j dddhd d S )Nr: r8 r; r£ r rH r& r& r' Ú"test_generalized_empty_nonsq_cases s zELinalgGeneralizedNonsquareTestCase.test_generalized_empty_nonsq_casesN)rJ rK rL r¥ r¦ r§ r© rª r& r& r& r' r¨ s
|
r¨ c @ s, e Zd Zejjdd Zejjdd ZdS )Ú HermitianGeneralizedTestCasec C s | j ddhdhd d S )Nr: r9 r; r r rH r& r& r' Útest_generalized_herm_cases s
|
ÿz8HermitianGeneralizedTestCase.test_generalized_herm_casesc C s | j dddhdhd d S )Nr: r9 r; Únoner r rH r& r& r' Ú!test_generalized_empty_herm_cases¥ s ÿz>HermitianGeneralizedTestCase.test_generalized_empty_herm_casesN)rJ rK rL r¥ 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
}ntdtj|t | |¡d}tjt t
|
| jd d
D ]}t | | || ||< q|S t | |S d S )Nr^ rm r\ zNot implemented...r3 rn ) r ry rg Ú
|
ValueErrorr# r~ Z common_typerz r{ ÚmapÚranger )r. r/ r ÚrÚcr& r& r' Údot_generalized« s
|
r´ c C sL t | } | jdkr:tj| j| jd}t| jd |d< |S t| jd S d S )Nr^ r3 rn .r )r ry r# r~ rg r, r )r. r² r& r& r' Úidentity_like_generalized¾ s
|
rµ c @ s e Zd Zdd ZdS )Ú
|
SolveCasesc C s. t ||¡}t|t|| tt|| d S r )r Úsolver r´ r r( )rB r. r/ rA r r& r& r' rE Ê s z SolveCases.doN©rJ rK rL rE r& r& r& r' r¶ È s r¶ c @ s: e Zd Zej deeee g¡dd Z
|
dd Z dd Z dS ) Ú TestSolver, c C s2 t jddgddgg|d}tt ||¡j| d S ©Nr\ ç à?r3 )r# r r r r· r, ©rB r, 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©rJ rK rL r& r& r& r' Ú ArraySubclass× s r¿ r[ r] r) r\ r^ r ) r# r$ Úarangerf r r r· r r r" r r r¯ ©rB 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>.ArraySubclassNr¾ r& r& r& r' r¿ ù s r¿ r_ r\ r] r) r^ r )
|
r# r$ rÀ rf r r r· r r r" rÁ r& r& r' Ú test_0_size_k÷ s ""
|
"8
|
zTestSolve.test_0_size_kN) rJ rK rL r¥ r¦ Ú parametrizer r r r r½ rÄ rÅ r& r& r& r' r¹ Ð s
|
!r¹ c @ s e Zd Zdd ZdS )ÚInvCasesc C s0 t |¡}tt||t| tt|| d S r )r Úinvr r´ rµ r r( )rB r. r/ rA Za_invr& r& r' rE s
|
|
|
ÿz InvCases.doNr¸ r& r& r& r' rÇ
|
s rÇ c @ s2 e Zd Zej deeee g¡dd Z
|
dd Z dS )ÚTestInvr, c C s0 t jddgddgg|d}tt |¡j| d S rº )r# r r r rÈ r, 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>.ArraySubclassNr¾ r& r& r& r' r¿ s r¿ ©r r\ r\ r3 rY )r# r$ ÚzerosÚint_r r rÈ r r, r! Úfloat64r rg r" Ú complex64©rB r¿ r. Úresr& r& r' rÄ s
|
|
zTestInv.test_0_sizeN© rJ rK rL r¥ r¦ rÆ r r r r r½ rÄ r& r& r& r' rÉ s
|
rÉ c @ s e Zd Zdd ZdS )Ú EigvalsCasesc C s& t |¡}t |¡\}}t|| d S r )r ÚeigvalsÚeigr )rB r. r/ rA ÚevÚevaluesÚevectorsr& r& r' rE , s
|
zEigvalsCases.doNr¸ r& r& r& r' rÒ * s rÒ c @ s2 e Zd Zej deeee g¡dd Z
|
dd Z dS )Ú TestEigvalsr, c C s` t jddgddgg|d}tt |¡j| t jddgddgg|d}tt |¡jt| d S ©Nr\ r» r3 rm )r# r r r rÓ r, r5 r¼ r& r& r' r½ 3 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>.ArraySubclassNr¾ r& r& r& r' r¿ < s r¿ rÊ r3 ©r r\ rY rZ )r# r$ rË rÌ r r rÓ r r, r! rÍ r rg r" rÎ rÏ r& r& r' rÄ : s
|
|
zTestEigvals.test_0_sizeNrÑ r& r& r& r' rØ 2 s
|
rØ c @ s e Zd Zdd ZdS )ÚEigCasesc C sX t |¡\}}tt||t |¡t |¡dd d d
f t|jd tt || d S )N.©Úrtol)
|
r rÔ r r´ r# r r6 r, r r( )rB r. r/ rA rÖ r× r& r& r' rE O s
|
þz EigCases.doNr¸ r& r& r& r' rÛ M s rÛ c @ s2 e Zd Zej deeee g¡dd Z
|
dd Z dS )ÚTestEigr, c C s t jddgddgg|d}t j |¡\}}t|j| t|j| t jddgddgg|d}t j |¡\}}t|jt| t|jt| d S rÙ )r# r r rÔ r r, r5 ©rB r, r ÚwÚvr& r& r' r½ X 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>.ArraySubclassNr¾ r& r& r& r' r¿ f s r¿ rÊ r3 rÚ rY rZ )r# r$ rË rÌ r r rÔ r r, r! rÍ r rg r" rÎ ©rB r¿ r. rÐ Zres_vr& r& r' rÄ d s zTestEig.test_0_sizeNrÑ r& r& r& r' rÞ W s
|
rÞ c @ s. e Zd ZdZej deee e
|
g¡dd Z dS )Ú SVDBaseTestsFr, c C sx t jddgddgg|d}t |¡\}}}t|j| t|jt| t|j| tj|d| jd}t|jt| d S )Nr\ r» r3 F©Ú
|
compute_uvr9 )r# r r Úsvdr r, r4 r9 )rB r, r Úur Úvhr& r& r' r½ ~ s zSVDBaseTests.test_typesN) rJ rK rL r9 r¥ r¦ rÆ r r r r r½ r& r& r& r' rã { s rã c @ s e Zd Zdd ZdS )ÚSVDCasesc C sp t |d¡\}}}t|tt |¡t |¡dd d d
f t |¡t|jd tt || tt || d S )NF.rÜ )
|
r ræ r r´ r# r r6 r, r r( )rB r. r/ rA rç r Úvtr& r& r' rE s &ÿþz SVDCases.doNr¸ r& r& r& r' ré s ré c @ s e Zd Zdd ZdS )ÚTestSVDc C s t d¡}tj|d| jd\}}}t|jd t|jd t|t d¡ t d¡}tj|d| jd\}}}t|jd t|jd t|t d¡ dS ) z6 Empty input should put an identity matrix in u or vh )r_ r Trä ©r_ r_ rY r_ )r r_ N)r# r~ r ræ r9 r rg Úeye)rB r rç r rè r& r& r' Útest_empty_identity s
|
|
zTestSVD.test_empty_identityN)rJ rK rL rî r& r& r& r' rë s rë c @ s e Zd Zdd ZdS )ÚSVDHermitianCasesc C sò t j|ddd\}}}t|tt |¡t |¡dd d d
f t |¡t|jd dd }tt |||¡t
|
t |j d ¡|j ¡ tt |||¡t
|
t |j d ¡|j ¡ t t |¡dd d d
f | tt|| tt|| d S ) NFT©r9 .rÜ c S s< t t| j}|d |d |d< |d< t tj| |d¡S )Nrn rm ©Úaxes)r r± ry r# ÚconjÚ transpose)Úmatrò r& r& r' r9 « s z'SVDHermitianCases.do.<locals>.hermitianrm )r ræ r r´ r# r r6 r, r r Ú broadcast_torí rg r Úsortr r( )rB r. r/ rA rç r rê r9 r& r& r' rE ¦ s &ÿþ,, zSVDHermitianCases.doNr¸ r& r& r& r' rï ¤ s rï c @ s e Zd ZdZdS )ÚTestSVDHermitianTN)rJ rK rL r9 r& r& r& r' rø · s rø c @ s e Zd Zdd ZdS )Ú CondCasesc C sê t |}d|kr"tttj| d S tj|dd}tt |¡|d |d ddd tt |d ¡|d |d ddd tt |d
|
¡|d |d ddd tj |¡}tt |d ¡t |
|
d
|
¡ d ¡t |
|
d
|
¡ d ¡ ddd tt |d ¡t |
|
d
|
¡ d ¡t |
|
d
|
¡ d ¡ ddd tt |tj ¡t |
|
d ¡ d ¡t |
|
d ¡ d ¡ ddd tt |tj ¡t |
|
d ¡ d ¡t |
|
d ¡ d ¡ ddd tt |d ¡t t |d
|
d ¡
|
d ¡t |d
|
d ¡
|
d ¡ ¡ddd d S )Nr; F)rå ).r ).rm ra r` ©r0 r1 r] rn r\ rm Úfro)r r r r Úcondræ r r# rÈ rp ÚsumÚmaxÚminr Úsqrt)rB r. r/ rA r³ r Zcinvr& r& r' rE ¾ sp þ
|
þ
|
þ
|
& ý
|
& ý & ý& ý
|
ÿ üz CondCases.doNr¸ r& r& r& r' rù » s rù c @ s> e Zd Zdd Zdd Zejjddddd d
|
Zd d Z d S )ÚTestCondc C s t dddgdddgdddgg}tt |t¡d tt |t ¡d tt |d¡d tt |d¡d tt |d
|
¡t d ¡ d S ) NrT r r\ g ÀrV r_ gUUUUUUå?rm r» rû gUUUUU6@)r r r rü r r# r ©rB ÚAr& r& r' Útest_basic_nonsvdê s zTestCond.test_basic_nonsvdc C sx t d¡t d¡g}d dddg}ddg}t ||¡D ] \}}tt ||¡dk q4t ||¡D ]\}}t ||¡ q^d S )N©r] r] r\ r] rû rm rn ç 4&õk C)r# rË Úonesrz r{ r r rü )rB ZAsÚp_posZp_negr Úpr& r& r' Ú test_singularó s zTestCond.test_singularTFz/Platform/LAPACK-dependent failure, see gh-18914)ÚrunÚreasonc C sö d dddddg}d dddg}t d¡}t j|d< |D ].}t ||¡}tt|t j tt |¡ q4t d¡}t j|d < |D ]t}t ||¡}tt |d ¡ ||krÈt|d
|
d k t|d d k q|tt |d
|
¡ tt |d ¡ q|d S ) Nr\ rm r] rn rû r rÚ )r^ r] r] )r\ r r\ r r ) r# r Únanr rü r r" Zfloat_Úisnan)rB Zpsr r r r³ r& r& r' Útest_nan s$
|
|
|
|
zTestCond.test_nanc C s t j d¡ t j dddd¡}d|d< d|d< dD ]T}t ||¡}t|d t j t|d t j tt |d ¡ tt |d ¡ q2d S ) NrS r] r rY rx )Nr\ r] rû rm rn rÚ ©r\ r )
|
r# ÚrandomÚseedÚrandr rü r r r Úisfinite)rB r r r³ r& r& r' Útest_stacked_singular s z TestCond.test_stacked_singularN)
|
rJ rK rL r r
|
r¥ r¦ Úxfailr r r& r& r& r' r é s
|
ÿ
|
r c @ s e Zd Zdd ZdS )Ú PinvCasesc C s< t |¡}t}t||||||ddd tt|| d S )Nra r` rú ©r Úpinvr´ r r r( ©rB r. r/ rA Za_ginvr r& r& r' rE 1 s
|
z PinvCases.doNr¸ r& r& r& r' r , s r c @ s e Zd ZdS )ÚTestPinvNr¾ r& r& r& r' r 9 s r c @ s e Zd Zdd ZdS )ÚPinvHermitianCasesc C s@ t j|dd}t}t||||||ddd tt|| d S )NTrð ra r` rú r r r& r& r' rE ? s zPinvHermitianCases.doNr¸ r& r& r& r' r = s r c @ s e Zd ZdS )ÚTestPinvHermitianNr¾ r& r& r& r' r G s r c @ s e Zd Zdd ZdS )ÚDetCasesc
|
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 )Nrm ©Úaxisr r\ )r ÚdetÚslogdetr r, r! r r Úastyper rÓ r r Úreducer# ÚexpZ
|
atleast_1drp r r )
|
rB r. r/ rA Údr ZldÚadrÕ Úmr& r& r' rE M s
|
|
|
|
z DetCases.doNr¸ r& r& r& r' r K s r c @ s: e Zd Zdd Zej deee e
|
g¡dd Z dd Z dS ) ÚTestDetc 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 )Nç y r r\ )r r r! r! r r r" r rH r& r& r' Ú test_zero` s zTestDet.test_zeror, c C s^ t jddgddgg|d}tt j |¡j| t j |¡\}}t|jt| t|j| d S rº )r# r r r r! r, r" r4 )rB r, r Úphr r& r& r' r½ m 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 )NrY r3 rT r r r\ ) r# rË rÎ r r! r r r, r! r" Úfloat32rÍ )rB r. rÐ r& r& r' rÄ u s
|
|
|
|
|
|
|
|
zTestDet.test_0_sizeN) rJ rK rL r+ r¥ r¦ rÆ r r r r r½ rÄ r& r& r& r' r) _ s
|
r) c @ s e Zd Zdd ZdS )Ú
|
LstsqCasesc C sL t |¡}|j\}}t |d¡\}}} tj||dd\}
|
} } } |dkrVt|
|
dk ¡ ||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 | jt j¡ tt|
|
| tt| | d S ) NFrm ©Zrcondr r] r r\ rk )r# r rg r ræ Úlstsqr r r r r Z__array_wrap__rp rý ry r r r! Ú
|
issubdtyper, Úfloatingr( )rB r. r/ rA Úarrr( Únrç r rê r Ú residualsÚrankÚsvZ expect_residsr& r& r' rE s2
|
|
|
ÿÿ
|
|
z LstsqCases.doNr¸ r& r& r& r' r. s r. c
|
@ sF e Zd Zdd Zej dddgdddd d
|
d d g¡d d Zdd ZdS )Ú TestLstsqc C sô t ddddddgddddddgddddddgddddddgg¡j}t ddddddg¡}t }| td¡}t ||¡\}}}}t|d k tj||d
|
d \}}}}t|d k tj||d d \}}}}t|d k tt |dk W 5 Q R X d S ) Nr* rT rU rW rV r\ r z `rcond` parameter will changer_ rm r/ r^ )
|
r# r ÚTr ÚrecordÚ FutureWarningr r0 r Úlen) rB r. r/ Úsuprà r r5 r6 r r& r& r' Útest_future_rcond¨ s ý zTestLstsq.test_future_rcondr( r4 Ún_rhs)r_ r] r] )r r_ r\ )r r_ r] )r_ r r\ )r_ r r] )r_ r] r )r r r c C sÜ t || ¡ ||¡}t ||f¡}tj||d d\}}}} |dkrTt|dk ¡ t|j ||f t|j ||krx|fnd ||kr´|dkr´|t
|
||¡ }
|
t ||
|
|
|
j dd t|t || t| j t ||f d S )Nr/ r rZ rn r )r# rÀ rf r r r0 r r r rg r r rý rÿ ) rB r( r4 r? r. r/ r r5 r6 r r² r& r& r' Útest_empty_a_bº s
|
zTestLstsq.test_empty_a_bc C sj t ddddg¡}t ddddd g¡}t |t t|¡g¡j}ttd
|
tj ||d d W 5 Q R X d S ) Nr r\ r] r^ rm gÉ?gÍÌÌÌÌÌì?gÍÌÌÌÌÌ @gffffff
|
@zIncompatible dimensionsr/ )
|
r# r Zvstackr r< r9 r r r r0 )rB r Úyr r& r& r' Útest_incompatible_dimsÒ s
|
z TestLstsq.test_incompatible_dimsN) rJ rK rL r> r¥ r¦ rÆ r@ rB r& r& r& r' r8 § s ù
|
r8 Údtc C s g | ]}t |¡qS r& )r# r, )rq r³ r& r& r' rr Û s rr z?bBhHiIqQefdgFDGOc @ sè e Zd Ze d¡Zeddddg Zeddddg Zeddddg ZeeeegZ e
|
ddgddggZ e egggd ¡Z ee d¡e d¡e d¡gZd d
|
Zd d Zd d Zdd Zdd Zdd Zdd Zejjedddd Z dS ) ÚTestMatrixPowerr_ r^ r r\ r] r ÚgÚGc C s | j |¡}tt|dd d d d | j tt|dd d d d | j tt|dd d d d | j tt|dd d d d | j d S )Nr] éd i é r r\ r^ )Úrshft_1r# r r Úrshft_0Úrshft_2Úrshft_3)rB rC Zrshftr& r& r' Útest_large_powerè s" ÿ ÿ ÿ ÿz TestMatrixPower.test_large_powerc C s> dd }| j D ]*}|| |¡ |tkr|| j |¡ qd S )Nc S s* t | d}t|t| t|j| j d S )Nr )r r rµ r, )ÚMÚmzr& r& r' Útzô s
|
z.TestMatrixPower.test_power_is_zero.<locals>.tz©Ú rshft_allr# ÚobjectÚstacked©rB rC rP rõ r& r& r' Útest_power_is_zeroó s
|
|
z"TestMatrixPower.test_power_is_zeroc C s> dd }| j D ]*}|| |¡ |tkr|| j |¡ qd S )Nc S s& t | d}t|| t|j| j d S )Nr\ )r r r, )rõ rO r& r& r' rP ÿ s
|
|
z-TestMatrixPower.test_power_is_one.<locals>.tzrQ rU r& r& r' Útest_power_is_oneþ s
|
|
z!TestMatrixPower.test_power_is_onec C s> dd }| j D ]*}|| |¡ |tkr|| j |¡ qd S )Nc S s> t | d}| jtkrtnt}t||| | t|j| j d S )Nr] )r r, rS r r r )rõ rO Úmmulr& r& r' rP
|
s
|
z-TestMatrixPower.test_power_is_two.<locals>.tzrQ rU r& r& r' Útest_power_is_two s
|
|
z!TestMatrixPower.test_power_is_twoc C s0 dd }| j D ] }|| jkr|| |¡ qd S )Nc S s4 t | d}| jtkrtnt}t||| t| d S ©Nrm )r r, rS r r r rµ )rõ ZinvmatrX r& r& r' rP s
|
ÿz3TestMatrixPower.test_power_is_minus_one.<locals>.tz)rR Údtnoinvr# rU r& r& r' Útest_power_is_minus_one s
|
|
z'TestMatrixPower.test_power_is_minus_onec C s. | j |¡}ttt|d ttt|dg d S )Ng ø?r\ )rJ r# r Ú TypeErrorr ©rB rC rõ r& r& r' Útest_exceptions_bad_power s z)TestMatrixPower.test_exceptions_bad_powerc C sP t ttt dg|¡d t ttt dgdgg|¡d t ttt d|¡d d S )Nr\ r] )r_ r^ r] )r r r r# r r )rB rC r& r& r' Útest_exceptions_non_square% s z*TestMatrixPower.test_exceptions_non_squareú fp errors don't work in wasm©r c C s, || j krd S | j |¡}ttt|d d S rZ )r[ Únoninvr# r r r r^ r& r& r' Ú test_exceptions_not_invertible* s
|
z.TestMatrixPower.test_exceptions_not_invertibleN) rJ rK rL r# rí rJ rI rK rL rR r rc ÚblockrT rS r, r[ rM rV rW rY r\ r_ r` r¥ r¦ Úskipifr rd r& r& r& r' rD Û s"
|
rD c @ s e Zd Zdd ZdS )ÚTestEigvalshCasesc C s^ t |d¡}t |¡\}}|jdd t||t|jd t |d¡}t||t|jd d S )NÚLrm r rÜ ÚU)r ÚeigvalshrÔ r÷ r r6 r, )rB r. r/ rA rÕ rÖ r× Úev2r& r& r' rE 4 s zTestEigvalshCases.doNr¸ r& r& r& r' rg 2 s rg c @ sB e Zd Zej deeee g¡dd Z
|
dd Z dd Z dd Z d
|
S ) Ú TestEigvalshr, c C s: t jddgddgg|d}t j |¡}t|jt| d S rº )r# r r rj r r, r4 )rB r, r rà r& r& r' r½ A s zTestEigvalsh.test_typesc C sX t jddgddggt jd}ttt jj|dd ttt jj|d ttt jj|d d S ©Nr\ r» r3 Zlrong©ZUPLOÚlowerÚupper)r# r r- r r¯ r rj ©rB r r& r& r' Ú test_invalidG 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 r\ r3 rm rÜ rh rn Úlri rç )r# r r r6 r rj r )rB ÚKloÚKupÚtgtrÝ rà r& r& r' Ú test_UPLOM 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>.ArraySubclassNr¾ r& r& r& r' r¿ e s r¿ rÊ r3 rÚ rY rZ )r# r$ rË rÌ r r rj r r, r! rÍ r rg r" rÎ r- rÏ r& r& r' rÄ c s
|
|
zTestEigvalsh.test_0_sizeN©rJ rK rL r¥ r¦ rÆ r r r r r½ rr rx rÄ r& r& r& r' rl @ s
|
|
rl c @ s e Zd Zdd ZdS )Ú TestEighCasesc
|
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 )Nrm r .rÜ ri )rÝ Úerr_msg) r ÚeighrÔ r÷ r r r´ r# r r6 r, Úrepr)
|
rB r. r/ rA rÕ ZevcrÖ r× rk Zevc2r& r& r' rE x s
|
|
þ
|
|
þzTestEighCases.doNr¸ r& r& r& r' rz v s rz c @ sB e Zd Zej deeee g¡dd Z
|
dd Z dd Z dd Z d
|
S ) ÚTestEighr, c C sJ t jddgddgg|d}t j |¡\}}t|jt| t|j| d S rº )r# r r r| r r, r4 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 rm )r# r r- r r¯ r r| rq r& r& r' rr 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 rs )r# r r r6 r r| r )rB ru rv rw rÝ rà rá r& r& r' rx 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>.ArraySubclassNr¾ r& r& r& r' r¿ ² s r¿ rÊ r3 rÚ rY rZ )r# r$ rË rÌ r r r| r r, r! rÍ r rg r" rÎ r- râ r& r& r' rÄ ° s zTestEigh.test_0_sizeNry r& r& r& r' r~ s
|
|
r~ c @ s e Zd ZdZdZedd ZdS )Ú _TestNormBaseNc C s: t | jjtjr"t|j| jj ntt |jjtj d S r ) Ú
|
issubclassr, r! r# Zinexactr Úrealr r2 )r rÐ r& r& r' Ú check_dtypeË s z_TestNormBase.check_dtype)rJ rK rL rC ÚdecÚ staticmethodr r& r& r& r' r Ç s r c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) Ú_TestNormGeneralc C sF t tg d t ttg | jdd t tttg | jdd d S )Nr* r3 )r r r rC r rH r& r& r' Ú
|
test_emptyÖ s z_TestNormGeneral.test_emptyc
|
C sr t dddg¡}t jd }t jd }|| }|D ]:}| |¡}t|t j }| ||¡ t|d t 2}| t
|
d¡ t|d}| ||¡ t|d W 5 Q R X t|d}| ||¡ t|d t|d}| ||¡ t|d t|d}| ||¡ t||j d ¡|j d
|
¡ t|d }| ||¡ t||j d ¡|j d ¡ t|t j}| ||¡ t|d q0d S )Nr\ r Ú
|
AllIntegerZAllFloatr* údivide by zero encounteredrm r] rU r» r_ g Ð?rT ) r# r Ú typecodesr# r r r r r ÚfilterÚRuntimeWarningr, r! © rB r. Z exact_typesZ inexact_typesZ all_typesZ each_typeÚatZanr= r& r& r' Útest_vector_return_typeÛ s:
|
|
|
|
|
|
|
|
|
|
|
|
z(_TestNormGeneral.test_vector_return_typec s ddddg}ddddg}ddddg} fd d
|
}|||fD ] }|| q:t | jd t | jd t | jd fD ] }|| qpd S ) Nr\ r] r^ r_ rm rn ro rl c sÔ t jjt| d jd t jjt| td jd t jjt| t d jd t jjt| dd jd t jjt| dd jd t jjt| d d jd t jjt| d
|
d jd t jjt| d d jd d S )Ng¥6Òè@©r+ rW rT r\ ç $@rm g¸
ëQ¸Þ?r] rn gÊsÔcÙÑê?r r_ )r# Útestingr r r r )rá rH r& r' Ú_test s0 ÿÿÿÿÿÿÿÿz+_TestNormGeneral.test_vector.<locals>._testr3 )r rC )rB r. r/ r³ r rá r& rH r' Ú test_vector s
|
ÿz _TestNormGeneral.test_vectorc s¼ t dddgdddgg| jd d dd dddtjtj fD ]h fd
|
d t jd D }tt d d | fd d t jd D }tt dd | q8tjdd| jd ddd¡j
|
}d ddddtjtj dfD ]Øt t| |d¡D ]¾}|\}}|d k r||7 }|d k r&||7 }||krDt tt|d qöt|d }||| ||k rfdd tj D }n fdd tj D }t|| qöqÞd S )Nr\ r] r^ r_ ra r) r3 rm r c s$ g | ] }t d d
|f dqS ©N©Úord©r ©rq Úk©r Úorderr& r' rr ( s z._TestNormGeneral.test_axis.<locals>.<listcomp>©r r c s$ g | ] }t |d d
f dqS r r r r r& r' rr * s é rn rû c s* g | ]"}t d d
j|ddqS ©Nr r )r Útaker ©ÚBÚk_indexr r& r' rr @ s ÿc s, g | ]$}t d d
j|djdqS r )r r r9 r r r& r' rr C s ÿ)r rC r# ÚInfr± rg r r rÀ rf ry rz Ú combinationsr r¯ ) rB Z expected0Z expected1Zndr Zrow_axisZcol_axisr4 r r& )r r¡ r¢ r r' Ú test_axis" s6
|
|
|
|
ÿ ÿz_TestNormGeneral.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 ¡ d dddddt j
|
t j
|
fD ]}t |j D ]v}t|||d }t|||d
|
d }tt |¡|| ||¡d t |j }d||< t|}t|j |k| |j |||¡ q¨qd ddddt j
|
t j
|
ddf D ]¢}t t |j d¡D ]}t|||d }t|||d
|
d }tt |¡|| ||¡d t |j }d||d < d||d < t|}t|j |k| |j |||¡ qTq>d S )Nr\ r r3 r] r^ r_ zorder {0}, axis = {1}z;Shape mismatch found {0}, expected {1}, order={2}, axis={3}r T)r r Zkeepdims)r{ )r\ r\ r\ rm r rn rû Únuc)r# rÀ rC rf r r ZsqueezeÚformatr rg r£ r± ry r r| rz Ú permutations) rB r Z allclose_errZ shape_errr ÚfoundZexpected_shaper r r& r& r' Ú test_keepdimsG sN
|
ÿ
|
ÿ
|
ÿ
|
|
ÿ
|
ÿ
|
|
ÿz _TestNormGeneral.test_keepdimsN)rJ rK rL r r r r¥ rª r& r& r& r' r
Ô s
|
) %r
c @ s: e Zd ZejZdd Zdd Zdd Zdd Zd d
|
Z d S ) Ú _TestNorm2Dc C s t t| jg g| jdd d S )Nr3 r* )r r r rC rH r& r& r' Útest_matrix_emptyw s z _TestNorm2D.test_matrix_emptyc
|
C sv | dddgdddgg¡}tjd }d}|| }|D ]:}| |¡}t|tj }| ||¡ t|d t 2}| t
|
d¡ t|d}| ||¡ t|d W 5 Q R X t|d}| ||¡ t|d t|d }| ||¡ t|d
|
t|d }| ||¡ t|d t|tj}| ||¡ t|d t|d }| ||¡ t|d t|d }| ||¡ tj j|ddd q4d S )Nr\ r r ZfdFDrU r rm rT r] gªLXèz¶û?rn rû r¦ gU&,t=Û@r) r ) r r# r r# r r r r r r r r r r& r& r' Útest_matrix_return_typez s@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
z#_TestNorm2D.test_matrix_return_typec C sÚ | j ddgddgg| jd}tt|d tt|dd tt|dd tt|td
|
tt|t d tt|dd tt|d d tt|dd tt|dd ttt|d ttt|d ttt|d d S )Nr\ r^ ra rb r3 g
¹T"@rû r¦ r g (@rW rm rX r] göW³?"@rn gÑÊì?Znofroro r )r rC r r r r r¯ r r& r& r' Útest_matrix_2x2 s z_TestNorm2D.test_matrix_2x2c C sÀ d| j dddgdddgdddgg| jd }tt|d tt|d
|
d tt|d d tt|td tt|t d tt|dd tt|dd tt|dd tt|dd d S )Ng¹?r\ r] r^ r) r ra r3 grj=Q0î?rû r¦ gþB+pcõ?gñ?g333333ã?rT rm gÙ?gû&Þê.dì?rn gÜ¸Õ çÈ?)r rC r r r r r& r& r' Útest_matrix_3x3½ s &ÿz_TestNorm2D.test_matrix_3x3c C s
|
| j dddgdddgg| jd}tjdd| jd ddd¡}ttt|d d
|
ttt|d d
|
tttddgd d tttddgd d tttddgd d d D ]4}ttt||d ttt||d ttt||d qttjt|d d ttjt|d d ttt|d d d S )Nr\ r] r^ r_ ra r) r3 r rû r r¦ Útest©r r^ rÚ )r\ r] ©r] r^ )r r\ r] ) r rC r# rÀ rf r r¯ r Z AxisError)rB r r¡ r r& r& r' Ú test_bad_argsÎ s z_TestNorm2D.test_bad_argsN)
|
rJ rK rL r# r r¬ r r® r¯ r³ r& r& r& r' r« r s 3r« c @ s e Zd ZdS )Ú _TestNormNr¾ r& r& r& r' r´ ê s r´ c @ s$ e Zd Zdd Zdd Zdd ZdS )ÚTestNorm_NonSystematicc C s* t jdt jd}tt|ddddd d S )Né
|
r3 r^ r gÍÌÌÌÌL)@r] r )r# rÀ Z
|
longdoubler- r rq r& r& r' Útest_longdouble_normð s z+TestNorm_NonSystematic.test_longdouble_normc C s, t jdgt jd}tt|ddddd d S )Ni r3 r^ r l ra r )r# r Úint32r- r rq 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©r] r3 y @ @r y À @r\ g¨ËþÿV;'@r^ r r¶ r é ra ) r# r~ Z clongdoubler- r r r# Ú
|
complex128rÎ )rB r& rÐ r& r& r' Útest_complex_high_ordü s z,TestNorm_NonSystematic.test_complex_high_ordN)rJ rK rL r· r¹ r½ r& r& r& r' rµ î s rµ c @ s e Zd ZejZdZdS )Ú_TestNormDoubleBaser* N)rJ rK rL r# r rC r r& r& r& r' r¾
|
s r¾ c @ s e Zd ZejZdZdS )Ú_TestNormSingleBaser) N)rJ rK rL r# r- rC r r& r& r& r' r¿ s r¿ c @ s e Zd ZejZdZdS )Ú_TestNormInt64Baser* N)rJ rK rL r# Úint64rC r r& r& r& r' rÀ s rÀ c @ s e Zd ZdS )ÚTestNormDoubleNr¾ r& r& r& r' r s r c @ s e Zd ZdS )ÚTestNormSingleNr¾ r& r& r& r' rà s rà c @ s e Zd ZdS )Ú TestNormInt64Nr¾ r& r& r& r' rÄ ! s rÄ c @ s e Zd Zdd Zdd ZdS )ÚTestMatrixRankc C sÊ t dtt d¡ t d¡}d|d< t t|d t tt d¡d t tddddgd t tt d¡d t tdgd t |t d¡t d¡g¡}t t|t dddg¡ t tdd d S ) Nr_ r* ©rm rm r^ rì r r\ )r_ )r r r# rí rË r )rB ÚIÚmsr& r& r' Útest_matrix_rank' s
|
z TestMatrixRank.test_matrix_rankc C s t dtt d¡dd t dtt d¡dd t dtt d¡dd t d¡}d|d< t d t|dd d
|
|d< t dt|dd d t d t|dd d d S )Nr_ Trð r\ rì r r* rÆ r^ g:0âyE>g¾Ü|"BE>)r9 Ztolgµ;ä¡°E>)r r r# rí r rË )rB rÇ r& r& r' Útest_symmetric_rank; s
|
z"TestMatrixRank.test_symmetric_rankN)rJ rK rL rÉ rÊ r& r& r& r' rÅ % s rÅ c C s t j d¡} tdD ]}| 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 qd S ) Ni3rG )é( r¶ ©Úsizer\ r] r r» r_ ra r^ r[ )r# r Z RandomStater± Únormalr r )ÚrngÚiÚXr& r& r' Útest_reduced_rankI s ,,rÒ c @ s¤ e Zd ZejZdd Zej ddgdddg¡dd Z d
|
d Z
|
d d Z dd Z ej ddddddg¡ej ddddg¡ej dej ejejejg¡dd ZdS ) Ú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, rg rÿ r Úqrr r" r r r9 ró r# rí Útriu) rB r. Úa_typeÚa_dtyper( r4 r Úqr² Úq1Úr1Úr2r& r& r' Úcheck_qrZ s8
|
|
zTestQR.check_qrr( r4 )r^ r r± rY c C st t ||}t ||f¡}| |¡ tjj|dd\}}t|jtj t|jtj t|j ||f t|j |f d S )NÚrawrÕ )
|
rÿ r# r~ rà r rØ r r, r rg )rB r( r4 r r. ÚhÚtaur& r& r' Ú test_qr_empty s
|
|
zTestQR.test_qr_emptyc C sÄ | j 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 ) Nr\ r] r^ r_ ra r) r3 rá rÕ r² rº rc ) r r# r r rØ r r, rg r9 )rB r. râ rã r& r& r' Ú test_mode_raw s "zTestQR.test_mode_rawc C s | ddgddgg¡}| ddgddgddgg¡}dD ]8}| |¡}| |¡}| |¡ | |¡ | |j¡ q6dD ]H}dd| |¡ }dd| |¡ }| |¡ | |¡ | |j¡ qtd S ) Nr\ r] r^ r_ ra r) Úfdù ð?)r r# rà r9 )rB r. r/ rC Úm1Úm2r& r& r' Útest_mode_all_but_economic¨ s
|
|
|
|
|
|
z!TestQR.test_mode_all_but_economicc C sf t |}|j}|jdd
\}}t||}tj|dd\}}t|j|k t|j|k tt|| tt|| t|jdd
||fk t|jdd
||fk tt ||| t
|
|jd ¡} t
|
| |jd d
|jd fd ¡}
|
tt t |dd ¡ ||
|
tt
|
|dd d
d d
f ¡| tj|dd\} } t| j|k t| j|k tt| | tt| | t| jdd
||fk t| jdd
||fk tt | | | t
|
| jd ¡} t
|
| | jd d
| jd fd ¡}
|
tt t | dd ¡ | |
|
tt
|
| dd d
d d
f ¡| tj|dd} t| j|k tt| | t| | d S ) Nrn rÔ rÕ rm r] .r× r² )r! r, rg rÿ r rØ r r" r r r# r rö r
|
ró rÙ )rB r. rÚ rÛ r( r4 r rÜ r² ZI_matZ stack_I_matrÝ rÞ rß r& r& r' Úcheck_qr_stacked¹ sL
|
ÿ " ÿÿ"zTestQR.check_qr_stackedrÍ )r^ r_ )r_ r^ rì Ú
|
outer_sizer rº )r] r^ r_ rC c C sP t jj|| d |¡}t jj|| d |¡}| |¡ | |d| ¡ d S )NrÌ rç )r# r rÎ r# rë )rB rì rÍ rC r r¡ r& r& r' Útest_stacked_inputsæ s
|
|
zTestQR.test_stacked_inputsN)rJ rK rL r# r rà r¥ r¦ rÆ rä rå rê rë r r r r rí r& r& r& r' rÓ V s< ' ý
|
- þ ÿ þrÓ c @ sR e Zd Zej ddddddg¡ej dejejej ej
|
f¡dd Z d
|
d Z d S ) Ú TestCholeskyrg rx r ©r^ r^ )é2 rð )r^ r¶ r¶ r, c C sô t j d¡ t jj| }t |t j¡r:|dt jj| }ttt|}d|dd
< t |
|
|¡ ¡ |¡}t j ||d}t j |¡}t ||
|
|¡ ¡ ¡}t ¡ d|jd t |¡j }W 5 Q R X t|||| d| d | d | d
|
d S ) Nr\ rç )rm rn rn r3 iô r ú Ú
|
)Úatolr{ )r# r r Úrandnr1 Zcomplexfloatingr r± r< r rô ró r r ÚcholeskyZ_no_nep50_warningrg ZfinfoZepsr )rB rg r, r. Útr³ r/ ró r& r& r' Útest_basic_propertyù s
|
$z TestCholesky.test_basic_propertyc 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>.ArraySubclassNr¾ r& r& r& r' r¿ s r¿ rÊ r3 )r\ r r )r# r$ rË rÌ r r rõ r rg r r, r! rÍ r" rÎ rÏ r& r& r' rÄ s
|
|
zTestCholesky.test_0_sizeN) rJ rK rL r¥ r¦ rÆ r# r- rÍ rÎ r¼ r÷ rÄ r& r& r& r' rî ö s ÿ ÿ rî c C s t jdkrd} nd} tjtjfD ]r}tjd|d}| | ¡}| d¡ ¡ }t|j jd t
|
j t
|
j t
|
j fD ](}||}t||| t||| qhq d S )NÚlittleú<rG r_ r3 ÚSú=)ÚsysÚ byteorderr# r- rÍ rí Z newbyteorderZbyteswapr r, r rÈ r! r r )ZnativeZdttr3 Zn_arrZsw_arrZroutinerÐ r& r& r' Útest_byteorder_check$ s
|
|
rþ ra rb c C sp t ddgddgg¡} t ddgddgg¡}t ddddg¡dd d
}| |d< ||d< tt jjt jj| d S )Nr\ r] r^ r_ .rY )r# r rË r r r rÈ )Z
|
invertibleZnon_invertibler r& r& r' Ú test_generalized_raise_multiloop9 s rÿ c C sR d} z t ¡ }W n" ttfk
|
r2 t d¡ Y nX |dkr(t d¡ t d¡ dd l}| |j d¡ zt
|
j j ¡ W n2 tk
|
r Y n tk
|
rª t t j¡ Y nX z0t
|
dgg¡}t
|
j j ddd|d||dd¡ W n< tk
|
r } z dt|krt | ¡ W 5 d }~X Y nX t t j¡ n&t ¡ \}}t |¡| krNt d¡ d S ) Néÿ zNot POSIX or fork failed.r r\ rY rT zDORGQR parameter number 5zNumpy xerbla not linked in.)ÚosÚforkÚOSErrorÚAttributeErrorr¥ ÚskipÚcloseÚresourceZ setrlimitZ RLIMIT_COREr# r Ú lapack_liteZxerblar¯ r Ú_exitZ EX_CONFIGr Zdorgqrr> ÚwaitÚ WEXITSTATUS)Z XERBLA_OKÚpidr r. r Ústatusr& r& r' Útest_xerbla_overrideH sH
|
|
|
ý r zCannot start subprocessc C sd ddg} t d¡}| D ]H}|jdd|d}t tjd|g¡ |jdd|d}t tjd|g¡ qd S ) NzPyQt5.QtWidgetsZIPythonzÕ
|
import sys
|
{before}
|
try:
|
import {bad_lib}
|
except ImportError:
|
sys.exit(0)
|
{after}
|
x = np.ones(2, dtype=np.float32)
|
sys.exit(0 if np.allclose(x.dot(x), 2.0) else 1)
|
zimport numpy as npÚ )ÚbeforeÚafterÚbad_libz-c)r r r )ÚtextwrapÚdedentr§ Ú
|
subprocessÚ
|
check_callrü Ú
|
executable)Zbad_libsÚtemplater Úcoder& r& r' Útest_sdot_bug_8577y s
|
ÿÿr c @ sd e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dS )Ú TestMultiDotc C sl t j d¡}t j d¡}t j d¡}tt|||g| |¡ |¡ tt|||gt |t ||¡¡ d S ©N©r) r] ©r] r) ©r# r r r r )rB r r¡ ÚCr& r& r' Ú(test_basic_function_with_three_arguments s
|
z5TestMultiDot.test_basic_function_with_three_argumentsc C sN t j d¡}t j d¡}tt||g| |¡ tt||gt ||¡ d S r r )rB r r¡ r& r& r' Ú&test_basic_function_with_two_arguments© s z3TestMultiDot.test_basic_function_with_two_argumentsc C s\ t j d¡}t j d¡}t j d¡}t j d¡}tt||||g| |¡ |¡ |¡ d S )Nr r ©r] r\ r )rB r r¡ r ÚDr& r& r' Ú9test_basic_function_with_dynamic_programming_optimization± s
|
zFTestMultiDot.test_basic_function_with_dynamic_programming_optimizationc C sL t j d¡}t j d¡}t j d¡}t j d¡}tt||||gjd d S )Nr] r r r rº ©r# r r r rg )rB ÚA1dr¡ 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||||gjd d S )Nr r r] )r) r& )rB r r¡ r ÚD1dr& r& r' Ú test_vector_as_last_argumentÄ s
|
z)TestMultiDot.test_vector_as_last_argumentc C sL t j d¡}t j d¡}t j d¡}t j d¡}tt||||gjd d S )Nr] r r r& r& )rB r' r¡ r r) r& r& r' Ú&test_vector_as_first_and_last_argumentÎ s
|
z3TestMultiDot.test_vector_as_first_and_last_argumentc C s t j d¡}t j d¡}t j d¡}t d¡}t|||g|d}||ksLtt|| |¡ |¡ t|t |t ||¡¡ d S )Nr r ©r% ©r# r rË r rO r r )rB r r¡ r r% Úretr& r& r' Ú test_three_arguments_and_outØ s
|
z)TestMultiDot.test_three_arguments_and_outc C sd t j d¡}t j d¡}t d¡}t||g|d}||ks>tt|| |¡ t|t ||¡ d S )Nr r )r) r) r, r- )rB r r¡ r% r. r& r& r' Útest_two_arguments_and_outå s
|
z'TestMultiDot.test_two_arguments_and_outc C sz t j d¡}t j d¡}t j d¡}t j d¡}t d¡}t||||g|d}||ksZtt|| |¡ |¡ |¡ d S )Nr r r# )r) r\ r, r- )rB r r¡ r r$ r% r. r& r& r' Ú-test_dynamic_programming_optimization_and_outï s
|
z:TestMultiDot.test_dynamic_programming_optimization_and_outc C sh 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 gddddddggtd }|d8 }t|d d \}}tt |d d
dd
f ¡t |d d
dd
f ¡ tt |¡t |¡ d S )!N)é é# )r3 é )r4 ra )ra r¶ )r¶ é )r5 r r* g ÃÎ@g þ@g OÂ@g 1Ç@g Í@g ¤@g ±@g Õ»@g Ä@g p@g £@g ÿ´@g @@g X«@g ³@r r\ r^ r] r_ ra r3 T)Z return_costsrm )r# r r Úintr r rÙ )rB ZarraysZ
|
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 )Nrï )r r¯ r r# r rH r& r& r' Útest_too_few_input_arrays s z&TestMultiDot.test_too_few_input_arraysN)rJ rK rL r! r" r% r( r* r+ r/ r0 r1 r7 r8 r& r& r& r' r s
|
|
|
|
|
r c @ sv e Zd Zej de d¡dfe d¡dfg¡dd Zej dd d
|
g¡d d Z ej d ddg¡dd Z
|
dd Z dS )Ú TestTensorinvzarr, ind©r_ r) r[ r] r] ©r^ r^ r] r\ c C s& t t tj||d W 5 Q R X d S )N©Úind)r r r Ú tensorinv)rB r3 r= r& r& r' Útest_non_square_handling s
|
z&TestTensorinv.test_non_square_handlingz
|
shape, ind)©r_ r) r[ r^ r] )©é r[ r^ r\ c C sN t d¡}||_tj||d}|j|d
|jd |
}|j}t|| d S )NrB ©r. r= )r# rí rg r r> r )rB rg r= r. Úainvr Úactualr& r& r' Útest_tensorinv_shape( s
|
z"TestTensorinv.test_tensorinv_shaper= r rn c C s6 t d¡}d|_tt tj||d W 5 Q R X d S )NrB r@ rC )r# rí rg r r¯ r r> )rB r= r. r& r& r' Útest_tensorinv_ind_limit5 s
|
|
z&TestTensorinv.test_tensorinv_ind_limitc C sJ t d¡}d|_tj|dd}t d¡}tt ||d¡t j ||¡ d S )NrB rA r\ r< ) r# rí rg r r> r r Ú tensordotÚ tensorsolve)rB r. rD r/ r& r& r' Útest_tensorinv_result> s
|
|
|
z#TestTensorinv.test_tensorinv_resultN) rJ rK rL r¥ r¦ rÆ r# r r? rF rG rJ r& r& r& r' r9 s þ
|
ý
|
ÿ
|
r9 c @ sV e Zd Zej de d¡dfe d¡dfg¡dd Zej dd d
|
d g¡d d Z dS )ÚTestTensorsolveza, axesr: Nr; )r r] c C s< t t* t |jd d
¡}tj|||d W 5 Q R X d S ©Nr] rñ )r r r# r rg r rI )rB r. rò r/ r& r& r' r? I s
|
z(TestTensorsolve.test_non_square_handlingrg )r] r^ r) )r^ r_ r_ r^ )r r^ r^ r c C sN t jj| }t |jd d
¡}t j ||¡}tt j||t |jd| d S rL )
|
r# r rô r rg r rI r rH r< )rB rg r. r/ r r& r& r' Útest_tensorsolve_resultR s z'TestTensorsolve.test_tensorsolve_result)
|
rJ rK rL r¥ r¦ rÆ r# r r? rM r& r& r& r' rK G s þ
|
ÿrK c C s> t jddgddggdd} ttd t | ¡ W 5 Q R X d S )Nr\ rn r] ra Úfloat16r3 zunsupported in linalg)r# r r r] r rõ )r3 r& r& r' Útest_unsupported_commontype\ s rO z,Bad memory reports lead to OOM in ci testingc C sR d} t jd| gt jd}t jddgt jd}d|d< t ||¡}t|d d d S )Nì r\ r3 )r rm )r# rË r- r r r )r4 r. r/ r³ r& r& r' Útest_blas64_dotg s rQ z*Numpy not compiled with 64-bit BLAS/LAPACKc
|
C sº t j} t jjj}d}d}|}t jddg| d}t jdg| d}t jdg| d}|||||||dd}t|d d t|d | t|d | t| ¡ } t d | k o®d
|
k n d S ) Nl r\ r3 rm r Úinfor( r4 rP l )
|
r# rÍ r r ZdgeqrfrË r r6 Úitemr )
|
r, Zlapack_routiner( r4 Zldar. Zworkrã ÚresultsZlworkr& r& r' Ú!test_blas64_geqrf_lwork_smoketestq s
|
rU )r) r* )Ú__doc__r rü rz r r r r¥ Únumpyr# r r r r r r r r Z
|
numpy.corer
|
r r r r r Z numpy.linalgr r r r r Znumpy.linalg.linalgr Z numpy.testingr r r r r r r r r r r( r- r4 r5 r6 rN r= rR r r rd r~ r rj 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 r r r r r) r. r8 r¦ rÆ rD rg rl rz r~ r r
r« r´ rµ r¾ r¿ rÀ r rà rÄ rÅ rÒ rÓ rî rþ rf rÿ r r§ r r r9 rK rO r rQ r rU r& r& r& r' Ú<module> s$ ( 0
|
|
þ þþþþþþ ý
|
þ
|
þþã#þ þþ þ ÿý ÿý ÿý ÿý ÿý ÿ ý
|
þ
|
þ
|
þ
|
ý
|
ýÎ9þþþþ ýþþ þé
|
0
|
|
:
|
$ .C ý
|
* 4V6; x $ !.
|
1 " )
|
|
ÿ
|
