xf  <dZddlmZmZmZmZddlZddlmZm Z m Z ddl m Z ddl mZmZmZddlmZddlmZdd lmZmZee gd gd gd gd gdgdgZe gdZdgZdZeeej8ej:ZdZGddZ y)z,Tests for the linalg._isolve.gcrotmk module )assert_assert_allclose assert_equalsuppress_warningsN)zerosarrayallclose)norm) csr_matrixeyerand)LinearOperator)splu)gcrotmkgmres)rrr )rrrrr)rrrrrr)rrrrrr)rrrrr)rrrrrr)rrrc6tdxxdz cc<t|zS)Nrr)countAm)vs P/usr/lib/python3/dist-packages/scipy/sparse/linalg/_isolve/tests/test_gcrotmk.pymatvecrs !HMH a4Krshapedtypec xdtd<t5}|jtdt t t ftt jddd|\}}dddtd}ttt zt ddtt |zt z ||fS#1swYQxYw)Nr.*called without specifying.*+=)x0tolg-q=)rtolatol) rrfilterDeprecationWarningrAbrr"rr r )kwsupr'flagcount_0s rdo_solver3$sE!H  H %'FG1aGE!''!*$55GBGDHAhG HQrT15u 5tAbDF|D w; HHs A B00B9c6eZdZdZdZdZdZdZdZdZ y) TestGCROTMKcFttj}t|jt j t j}t\}}t|\}}t|dt||dz ktt||dy)Nr!)Mrrr&)r)) rrtocscrsolver-r"r#r3rrr )selfpcr7r'r2x1count_1s rtest_preconditionerzTestGCROTMK.test_preconditioner/sp "((*  "((!'' Ij Gm GWa '!)#$Re,-r c xtjjdtdt dddz}tjj d}t 5}|j tdt||t|jdddd\}}t||t|jddd \}}dddtdtdtjj|j|z d kDsJt!|y#1swYdxYw) NrigMb@?)densityr%r)r'mkmaxiter)r'restartrDMbP?)nprandomseedr r rr+r,rrr"rrlinalgr dotr)r:r-r.r0r'flag0r<flag1s r test_arnoldizTestGCROTMK.test_arnoldi;s q IT46 6 IINN4   QC JJ)+J K1qwwqz):bAqQIBauQWWQZ'8"aPIB Q UAUAyy~~aeeBi!m,t333B Q Qs #A*D00D9ctjjddD]V}dt|z}t 5}|j t dtj|}t||d\}}t|dt|j||z dd t||dd \}}|dk(r t|j||z dd tjj|}t||d\}}t|dt|j||z dd t||dd \}}|dk(r t|j||z dd dddYy#1swYexYw) N)rr drr%rQ)rDrr&r*r(rD) rGrHrIr rr+r,onesrrrrKr )r:nr-r0r.xinfos rtest_cornercasezTestGCROTMK.test_cornercaseMsV t ! AA#a&A"$ A -/NOGGAJ!!Q34T1%a1 ae<!!QAr:419#AEE!HqL!%@IINN1%!!Q34T1%a1 ae<!!QAr:419#AEE!HqL!%@% A A A A As D3E==F ctdd}tj|d<tjd}t 5}|j t dt||dd\}}t|d dddy#1swYyxYw) Nrlil)format)rrr%rrQrTr) r rGnanrUrr+r,rr)r:r-r.r0rWrXs r test_nanszTestGCROTMK.test_nanskss % # GGAJ   "C JJ)+J Ka26GAt q ! " " "s 5A>>Bc tjjdtjjddtjdz}tjjd}dD]l}t 5}|j tdt||dd|dd\}}dddtd t|j|z d d ny#1swY8xYw) NrP)oldestsmallestr%rQg-C6?)rBrCtruncater(rDrrFrS) rGrHrIr r rr+r,rrrrK)r:r-r.rdr0rWrXs r test_truncatezTestGCROTMK.test_truncateus t IINN2r "RVVBZ / IINN2 . 8H"$ / -/NO!!Q"X4*-/4 / q ! AEE!HqL!$ 7  8 / /s ,C''C0 cpdD]}g}t||\}}tt|dkDtt|dk|r|D]\}}t|dut||\}}|rt|dt|zn t|dt||dz kt ||dy) N)TF)CU discard_Crrrrr&rS)r3rlenrr) r:rhrgr'r2cur<r=s rtest_CUzTestGCROTMK.test_CUs& 0IB"bI>KB CGaK CGqL !'DAqAI&'#bI>KBWa#b'k2Wa( Gwqy( ) B /% 0r cvtjddgddggt}|dtjddzz}tjddg}t 5}|j t dt||\}}ddddk(rt|j|yy#1swY+xYw) Nrrrr)r#rRrr%) rGrfloat nextafterrr+r,rrrK)r:r-r.r0xprXs rtest_denormalszTestGCROTMK.test_denormalss HHq!fq!f%U 3 S2<<1% %% HHaV    %C JJ)+J Kq!}HB % 19 AEE"Iq )   % %s &B//B8N) __name__ __module__ __qualname__r>rNrYr^rerlrqr rr5r5.s& . $A<" 80**r r5)!__doc__ numpy.testingrrrrnumpyrGrrr scipy.linalgr scipy.sparser r r scipy.sparse.linalg._interfacerscipy.sparse.linalgrscipy.sparse.linalg._isolverrrr.rrr"r#r-r3r5rur rr~s..((..9$6~%%%%% '()  -  &Aw*w*r