g08dZddlmZddlmZddlZddlZddlm Z m Z m Z m Z ddl mZddlmZddlZddlmZdd lmZmZmZmZmZmZdd lmZej<ej>hZ ejBeejDejFejHejJfZ&ejNejPejRfZ*e&e*zZ+d Z,d Z-Gd dZ.GddZ/ej`jbej`jede+ej`jede+ej`jedddgdZ3y)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_longcfd}|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 estimatedrs' Nc<tj||xs||kS)N)npallclose)abs rless_than_or_closer&(s ;;q!  'Q'r cxeZdZdZdZdZdZdZejjdZ ejjdZ ejjdZ ejjd Zejjd Zejjd Zy ) 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_monotonicity1sNv||~&.1%*eABi.H ' *NS'NS' Gg% & 'r c:d}d}t|}t||y)N7)r r)r/m_maxexpected_p_maxobserved_p_maxs rtest_p_max_defaultz'TestExpmActionSimple.test_p_max_default6s '.^^4r ctddD]@}t|}t||dz z|dzk|dz}t||dz z|dzkDBy)Nr+8)ranger r)r/r9p_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 tjjd}d}d}t|D]}tj j |j||}tdD]}|stj|}ntj|}t||}tj j|d} tt|| tt| d|zy)N( r+)r"random RandomStater?scipylinalginvrandnidentitydotrnormrr&) r/rngnnsamplesiApMrexacts rtest_onenormest_matrix_powerz1TestExpmActionSimple.test_onenormest_matrix_powerCsii##D) x @A   1a1A1X @ 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)NrDrErHrFtraceA)r"rIseedr?rKrLrMrNrrPsp_expmrrr trace) r/rSkrTrUrVBobservedexpectedr]s rtest_expm_multiplyz'TestExpmActionSimple.test_expm_multiplySs 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)NrDrErF)r"rIr^r?rKrLrMrNrrPr_rrr )r/rSrTrUrVvrcrds 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) NrDrErHrF)g?g?g?ignore)invalidt)r"rIr^rr?errstaterKrLrMrNrrPr_rrr ) r/rSrarTrUrmrVrbrcrds rtest_scaled_expm_multiplyz.TestExpmActionSimple.test_scaled_expm_multiplyss 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)NrD皙?rl) r"rIr^rNrr_rPrrr )r/rmrSrarVrbrcrds r*test_scaled_expm_multiply_single_timepointz?TestExpmActionSimple.test_scaled_expm_multiply_single_timepoints t    IIOOAq ! IIOOAq !(A31Q3<##A&(+3923 Q a  (+r c,tjjd}d}d}d}t|D]}tj j ||fd|}|j||f}t||}t5} | jtd| jtdt|j|} dddt| ttt!||}t|| y#1swYCxYw) NrDrErHrF皙?densityrR&splu converted its input to CSC formatCspsolve is more efficient when sparse b is in the CSC matrix format)r"rI default_rngr?rKsparse random_arraystandard_normalrrfilterr r_rPrrr ) r/rRrSrarTrUrVrbrcsuprds rtest_sparse_expm_multiplyz.TestExpmActionSimple.test_sparse_expm_multiplysii##D)  x 0A ))1a&$C)HA##QF+A$Q*H"$ - 2CE 201#1:>>!,  - Hh //y/0@0CQGH Hh / 0 - -s AD  D ctjddgddggt}tjddg}t||}tjdtjdzddtj dztj dz zzdtjdzgt}t||ttt||}t||y)N?rdtyper+) r"arraycomplexrexpcossinrrr )r/rVrbrcrds 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 3A(D))D2 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) NrDrqrTrrrrr+rsrrrrErr]rrr)r"rIr^rrKrLrMrNrrr.rr_rPrr r`)r/intervalrrSrVrgrrrrmXguessXgivenXwrong 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) NrDrqrTrrrr+rsrr)r"rIr^rrKrLrMrNrrr.rr_rPrr ) r/rrrSrarVrbrrrrms 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) Nrrcsr)formatrr)rrr+r*yr) rKr| diags_arrayr"arangeintonesrrrrPrfull)r/rVrbAexpms r)test_sparse_expm_multiply_interval_dtypesz@TestExpmActionInterval.test_sparse_expm_multiply_interval_dtypess LL $ $RYYq\c $ J GGAS ! (( ! )=U(K a!A.r2EIIaLA LL $ $S1%5U' $ R GGAS ! ((BIIaL0@)A(O a!A.r2EIIaLA LL $ $RYYq\c $ J GGAr ) (( ! )=U(K a!A.r2EIIaLAr c&|jdy)Nr(_help_test_specific_expm_interval_statusr/s 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)Nrsrrs r$test_expm_multiply_interval_status_2z;TestExpmActionInterval.test_expm_multiply_interval_status_2rr c <tjjd}d}d}d}d}d}d}d} d } gd | zD]}|j||} |j||} t | | ||||d } | |k(s?t | | ||||d \}} t |j |||ftj|||| }t||D],\}}t|t|| zj| .| dz } | sdt|zdz}t|y)NrDrqrrTrrrsrFrr)rrrr status_onlyFrr+zfailed to find a status-z interval)r"rIrJrNrrshaperr.rr_rPstr Exception)r/ target_statusrRrrrrrSranrepeats nsuccessesrVrbstatusrrrrmmsgs rrz?TestExpmActionInterval._help_test_specific_expm_interval_statussHii##D)   ) C !QA !QA,Qdh $&F&3Aq#$C($)+ 6QWWsAqk2++E(4#&q'?CKHa#Hgacl.>.>q.ABCa  ,s=/AAKOCC. r N)rrrrr fail_slowrrrrrrrrrrr rrrs [[2LL2 [[ [[2444 [[ [[2 ? ?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.gH}]?gh㈵>)rtolatolrDrHrrsr\rqrrTrNr) IMPRECISErrr"rIr{ REAL_DTYPESrKrLrMastyperrr rPr_r`rr.)rrrassert_allclose_rRrSb_shaperVrbsol_matsol_op direct_solrrX_matX_oprms rtest_expm_multiply_dtyper=s.$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 )4r functoolsr itertoolsrnumpyr"r numpy.testingrrrr scipy.sparser scipy.sparse.linalgr scipy.linalgrKr r_"scipy.sparse.linalg._expm_multiplyr r rrrrscipy._lib._utilrsinglecsinglerintclonglongfloat32float64 longdoubler complex64 complex128 clongdoubleCOMPLEX_DTYPESDTYPESrr&r(rrr parametrizerrr rrs)A ..00/!!%YY # wwzz2::r}}6 ,, r~~> ~ % (J,J,ZA!A!HF+F+ 6$-7,,$-r