sg6dZddlmZddlmZddlZddlZddlm Z m Z m Z m Z ddl mZddlmZddlZddlmZdd lmZmZmZmZmZmZdd lmZej<ej>hZ ejBeejDejFejHejJhZ&ejNejPejRhZ*e+e&e*z e, Z-d Z.d Z/GddZ0GddZ1ejdjgde-ejdjgde-ejdjgdddgdZ4y)z;Test functions for the sparse.linalg._expm_multiply module.)partial)productN)assert_allcloseassert_ assert_equalsuppress_warnings)SparseEfficiencyWarning)aslinearoperator)expm)_theta_compute_p_max_onenormest_matrix_power expm_multiply_expm_multiply_simple_expm_multiply_interval)np_long)keycfd}|S)zIf trace is estimated, it should warn. We warn that estimation of trace might impact performance. All result have to be correct nevertheless! cvtjtd5|i|cdddS#1swYyxYw)Nz%Trace of LinearOperator not available)match)pytestwarns UserWarning)argskwdsfuncs _/var/www/html/venv/lib/python3.12/site-packages/scipy/sparse/linalg/tests/test_expm_multiply.pywrappedzestimated..wrapped"s7 \\+ GI '&& ' ' 's/8)rrs` r estimatedr s' Nc<tj||xs||kS)N)npallclose)abs rless_than_or_closer')s ;;q!  'Q'r!cLeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z y ) TestExpmActionSimplezR These tests do not consider the case of multiple time steps in one call. cttj}t|dd|ddD]\\}}\}}t ||ky)N)sortedr itemszipr)selfpairsm_atheta_am_btheta_bs rtest_theta_monotonicityz,TestExpmActionSimple.test_theta_monotonicity2sNv||~&.1%*eABi.H ' *NS'NS' Gg% & 'r!c:d}d}t|}t||y)N7)r r)r0m_maxexpected_p_maxobserved_p_maxs rtest_p_max_defaultz'TestExpmActionSimple.test_p_max_default7s '.^^4r!ctddD]@}t|}t||dz z|dzk|dz}t||dz z|dzkDBy)Nr,8)ranger r)r0r:p_max p_too_bigs rtest_p_max_rangez%TestExpmActionSimple.test_p_max_range=s^1d^ ;E"5)E E519%2 3 I Iy1}- 9 :  ;r!c tjjdd}d}t|D]}tj j tjj||}tdD]}|stj|}ntj|}t||}tj j|d}tt||tt|d|zy)N( r,)r#randomseedr@scipylinalginvrandnidentitydotrnormrr') r0nnsamplesiApMr exacts rtest_onenormest_matrix_powerz1TestExpmActionSimple.test_onenormest_matrix_powerDs t x @A   A!67A1X @ AAq! A4Q:  q!,*9e<=*5!I+>? @ @r!c\tjjdd}d}d}t|D]}tj j tjj||}tjj||}t||}tjt||}t||ttt||}t||tj|} tt||| }t||y)NrErFrIrGtraceA)r#rJrKr@rLrMrNrOrrQsp_expmrr r trace) r0rSkrTrUrVBobservedexpectedr]s rtest_expm_multiplyz'TestExpmActionSimple.test_expm_multiplyTs t  x 0A   A!67A 1%A$Q*Hvvgaj!,H Hh //y/0@0CQGH Hh /XXa[F$%5a%8!FKH Hh / 0r!ctjjdd}d}t|D]}tj j tjj||}tjj|}t||}tjt||}t||ttt||}t||y)NrErFrG)r#rJrKr@rLrMrNrOrrQr^rr r )r0rSrTrUrVvrbrcs rtest_matrix_vector_multiplyz0TestExpmActionSimple.test_matrix_vector_multiplyes t x 0A   A!67A "A$Q*Hvvgaj!,H Hh //y/0@0CQGH Hh / 0r!cntjjdd}d}d}tt |gdD]\}}tj d5t jjtjj||}tjj||}t|||}tjt||z|} t|| ttt|||}t|| dddy#1swYxYw) NrErFrIrG)g?g?g?ignore)invalidt)r#rJrKrr@errstaterLrMrNrOrrQr^rr r ) r0rSr`rTrUrlrVrarbrcs rtest_scaled_expm_multiplyz.TestExpmActionSimple.test_scaled_expm_multiplyrs t  E(O_= 4DAqX. 4LL$$RYY__Q%:;IIOOAq)0A;66'!A#,2(3;9%:;$Q'a (3 4 4 4 4 4s CD++D4 ctjjdd}d}d}tjj||}tjj||}t |||}t ||zj |}t||ttt|||}t||y)NrE皙?rk) r#rJrKrOrr^rQrr r )r0rlrSr`rVrarbrcs r*test_scaled_expm_multiply_single_timepointz?TestExpmActionSimple.test_scaled_expm_multiply_single_timepoints t    IIOOAq ! IIOOAq !(A31Q3<##A&(+3923 Q a  (+r!cBtjjdd}d}d}t|D]}tj j ||d}tjj||}t||}t5}|jtd|jtdt|j|} dddt| ttt!||}t|| y#1swYCxYw) NrErFrIrG皙?density&splu converted its input to CSC formatCspsolve is more efficient when sparse b is in the CSC matrix format)r#rJrKr@rLsparserandrOrrfilterr r^rQrr r ) r0rSr`rTrUrVrarbsuprcs rtest_sparse_expm_multiplyz.TestExpmActionSimple.test_sparse_expm_multiplys t  x 0A !!!Q!5A 1%A$Q*H"$ - 2CE 201#1:>>!,  - Hh //y/0@0CQGH Hh / 0 - -s ADD ctjddgddggt}tjddg}t||}tjdtjdzddtj dztj dz zzdtjdzgt}t||ttt||}t||y)N?rdtyper,) r#arraycomplexrexpcossinrr r )r0rVrarbrcs r test_complexz!TestExpmActionSimple.test_complexs HH H G# % HHb"X  A&88 ObBrvvayL266!9$<= = O$+- (++9]+,++E*-B&(LCJJ6GIJJ6 89(+1gL !''!A#,2B2B62JKL LL L  LLLs A(EE c tjjddddd}tgdgdD]\}}tj j tjj||}tjj|}tjd d|i|}t||fd|i|}t||D],\}} t|t| |zj|.ttt||fd|i|} tt||fd|i|d tj |i} tt||fd|i|d tj |d zi} t| | | |D]H\} }}} t| |zj|}t| |t||t||Jy) NrErprTrrrrr,rrrqrFrr]rqr)r#rJrKrrLrMrNrOrrr/rr^rQr r r_)r0intervalrrSrVrfrrrrlXguessXgivenXwrong sol_guess sol_given sol_wrongcorrects r"test_expm_multiply_interval_vectorz9TestExpmActionInterval.test_expm_multiply_interval_vectors t #4@k+<= 4FC   A!67A "Akk6c6X6Ga88x8A"1g ? !'!A#,*:*:1*=> ?.Y}-.>q.A1C25C9ACF##3A#67s7h7*,((1+7F##3A#69s9h9*,((1+a-9F69&&:@'7K 42 9i!!A#,**1- 73 73 73  4! 4r!c tjjddddd}tgdgddd gD]"\}}}tj j tjj||}tjj||}tjd d |i|}t||fd |i|}t||D],\} } t| t| |zj|.ttt||fd |i|}t||D],\} } t| t| |zj|.%y) NrErprTrrrr,rrrr)r#rJrKrrLrMrNrOrrr/rr^rQr r ) r0rrrSr`rVrarrrrls r"test_expm_multiply_interval_matrixz9TestExpmActionInterval.test_expm_multiply_interval_matrixs: t #4@ .?!QH ?ICA   A!67A 1%Akk6c6X6Ga88x8A"1g ? !'!A#,*:*:1*=> ?( -()9!) ? ?r!ctjjtjddt }tj dt }tjjtjtjdd}tt||ddd|j|tjjd tjdzdt}tj dt }tjjtjd tjdzd}tt||ddd|j|tjjtjddt }tjdd t}tjjtjtjdd}tt||ddd|j|y) Nrqcsr)formatrr)rrr,r+yr) rLrzdiagsr#arangeintonesrrrrQrfull)r0rVraAexpms r)test_sparse_expm_multiply_interval_dtypesz@TestExpmActionInterval.test_sparse_expm_multiply_interval_dtypess LL  ryy|5  D GGAS ! ""266"))A,#7u"E a!A.r2EIIaLA LL  s299Q</uG  L GGAS ! ""266#biil*:#;5"I a!A.r2EIIaLA LL  ryy|5  D GGAr ) ""266"))A,#7u"E a!A.r2EIIaLAr!c&|jdy)Nr(_help_test_specific_expm_interval_statusr0s r$test_expm_multiply_interval_status_0z;TestExpmActionInterval.test_expm_multiply_interval_status_0 55a8r!c&|jdy)Nr,rrs r$test_expm_multiply_interval_status_1z;TestExpmActionInterval.test_expm_multiply_interval_status_1rr!c&|jdy)Nrrrrs r$test_expm_multiply_interval_status_2z;TestExpmActionInterval.test_expm_multiply_interval_status_2rr!c ttjjdd}d}d}d}d}d}d}d } gd |zD]}tjj||} tjj||} t | | ||||d } | |k(s[t | | ||||d \} } t | j |||ftj|||| }t| |D],\}}t|t|| zj| .| dz } | sdt|zdz}t|y)NrErprrTrqrrrGrr)rrrr status_onlyFrr,zfailed to find a status-z interval)r#rJrKrOrrshaperr/rr^rQstr Exception)r0 target_statusrrrrrSr`nrepeats nsuccessesrVrastatusrrrrlmsgs rrz?TestExpmActionInterval._help_test_specific_expm_interval_statussN t   ) C 1%A 1%A,Qdh $&F&3Aq#$C($)+ 6QWWsAqk2++E(4#&q'?CKHa#Hgacl.>.>q.ABCa  ,s=/AAKOCC. r!N)rrrrmark fail_slowrrrrrrrrrr!rrrs [[1LL2 [[1444 [[1 ? ?B&999!r!rdtype_adtype_b b_is_matrixFTc||htzrttddnt}tjj d}d}|r|dfn|f}|t vr@tjj|j ||gj|}nVtjj|j ||gd|j ||gzzj|}|t vr$d|j |zj|}n5|j |d|j |zzj|}t||} ttt||} tjt||} || | || | tt||tj | } || | d d d d d} tj"di| } t||fi| }ttt||fi| }t%||| D]5\} } }t||zj|} || | || | 7y)zAMake sure `expm_multiply` handles all numerical dtypes correctly.ga2U0*S?gh㈵>)rtolatolrErIrrrr\rprrTrNr) IMPRECISErrr#rJ default_rng REAL_DTYPESrLrMrNastyperr r rQr^r_rr/)rrrassert_allclose_rngrSb_shaperVrasol_matsol_op direct_solrrX_matX_oprls rtest_expm_multiply_dtyper6s. $W- 9 f4H?N ))   %C A#q!f!G+ LL  SZZA/ 0 7 7 @ LL   JJ1v CJJ1v$6!6 6 &/ + szz'" " * *7 3 ZZ 2cjj&9#9 9 A A' JAq!G %Y} %&6q&91 =F A&JWj)VZ( +A."((1+ FFVZ(c"$GHkk%H%G !Q +( +E #9] #$4Q$7 Gh GD!%w7-QqS\%%a( *-,-r!)5r functoolsr itertoolsrnumpyr#r numpy.testingrrrr scipy.sparser scipy.sparse.linalgr scipy.linalgrLr r^"scipy.sparse.linalg._expm_multiplyr r rrrrscipy._lib._utilrsinglecsinglerintclonglongfloat32float64 longdoubler complex64 complex128 clongdoubleCOMPLEX_DTYPESr-rDTYPESr r'r)rr parametrizerrr!rrsA ..00/!!%YY # wwzz2::r}}6 ,, r~~>  n,# 6 (D,D,N!!DF+F+ 6$-7,,$-r!