-i9SrSSKJr SSKJr SSKrSSKrSSKJ r J r J r J r SSK Jr SSK rSSKJr SSKrSSKJr SS KJrJrJrJrJrJr SS KJr \R<\R>1r \RB\\RD\RF\RH\RJ4r&\RN\RP\RR4r*\&\*-r+S r,S r-"S S5r."SS5r/\R`Rb\R`ReS\+5\R`ReS\+5\R`ReSSS/5S5555r3g)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_longc^U4SjnU$)zIf trace is estimated, it should warn. We warn that estimation of trace might impact performance. All result have to be correct nevertheless! c|>[R"[SS9 T"U0UD6sSSS5 $!,(df  g=f)Nz%Trace of LinearOperator not available)match)pytestwarns UserWarning)argskwdsfuncs _/var/www/html/venv/lib/python3.13/site-packages/scipy/sparse/linalg/tests/test_expm_multiply.pywrappedestimated..wrapped"s5 \\+ GI&&IIIs- ;)rrs` r estimatedr s' NcD[R"X5=(d X:$)N)npallclose)abs rless_than_or_closer')s ;;q  ''r!c\rSrSrSrSrSrSrSr\ RRS5r \ RRS5r \ RRS 5r\ RRS 5r\ RRS 5r\ RRS 5rS rg)TestExpmActionSimple-zJ These tests do not consider the case of multiple time steps in one call. c[[R"55n[USSUSS5Huup#upE[ X5:5 M g)N)sortedr itemszipr)selfpairsm_atheta_am_btheta_bs rtest_theta_monotonicity,TestExpmActionSimple.test_theta_monotonicity2sDv||~&.1%*eABi.H *NSNS G% &/Ir!c8SnSn[U5n[X25 g)N7)r r)r1m_maxexpected_p_maxobserved_p_maxs rtest_p_max_default'TestExpmActionSimple.test_p_max_default7s'.^4r!c[SS5H?n[U5n[X"S- -US-:*5 US-n[X3S- -US-:5 MA g)Nr-8)ranger r)r1r<p_max p_too_bigs rtest_p_max_range%TestExpmActionSimple.test_p_max_range=sU1d^E"5)E E19%2 3 I I1}- 9 : $r!c [RRS5nSnSn[U5Hn[R R URX"55n[S5HnU(d[R"U5nO[R"WU5n[XV5n[R RUS5n [[X55 [[U SU-55 M M g)N( r-)r#random RandomStaterCscipylinalginvrandnidentitydotrnormrr') r1rngnnsamplesiApMr exacts rtest_onenormest_matrix_power1TestExpmActionSimple.test_onenormest_matrix_powerDsii##D) xA   11A1X AAq! A4Q:  q!,*9<=*5!I+>?!r!cP[RRS5 SnSnSn[U5Hn[R R [RRX55n[RRX5n[XV5n[R"[U5U5n[Xx5 [[5"[U5U5n[Xx5 [R"U5n [[U5XiS9n[Xx5 M g)NrIrJrMrKtraceA)r#rNseedrCrPrQrRrSrrUsp_expmrr r trace) r1rXkrYrZr[Bobservedexpectedrcs rtest_expm_multiply'TestExpmActionSimple.test_expm_multiplyTs t  xA   !67A %A$Q*Hvvgaj!,H H / /0@0CQGH H /XXa[F$%5a%8!KH H /!r!c[RRS5 SnSn[U5Hn[R R [RRX55n[RRU5n[XE5n[R"[U5U5n[Xg5 [[5"[U5U5n[Xg5 M g)NrIrJrK)r#rNrdrCrPrQrRrSrrUrerr r )r1rXrYrZr[vrirjs rtest_matrix_vector_multiply0TestExpmActionSimple.test_matrix_vector_multiplyfs t xA   !67A "A$Q*Hvvgaj!,H H / /0@0CQGH H /!r!c`[RRS5 SnSnSn[[ U5/SQ5HupE[R "SS9 [ RR[RRX55n[RRX5n[XgUS9n[R"[XV-5U5n [X5 [[5"[U5XuS9n[X5 SSS5 M g!,(df  M=f) NrIrJrMrK)g?g?g?ignore)invalidt)r#rNrdrrCerrstaterPrQrRrSrrUrerr r ) r1rXrgrYrZrur[rhrirjs rtest_scaled_expm_multiply.TestExpmActionSimple.test_scaled_expm_multiplyts t  E(O_=DAX.LL$$RYY__Q%:;IIOOA)0;66'!#,23$%:;$Q' 3/.>..s B;D D- c|[RRS5 SnSnSn[RRX"5n[RRX#5n[ XEUS9n[ X-5R U5n[Xg5 [[5"[U5XQS9n[Xg5 g)NrI皙?rt) r#rNrdrSrrerUrr r )r1rurXrgr[rhrirjs r*test_scaled_expm_multiply_single_timepoint?TestExpmActionSimple.test_scaled_expm_multiply_single_timepoints t    IIOOA ! IIOOA !(313<##A&+23 Q   +r!c,[RRS5nSnSnSn[U5Hn[R R X"4SUS9nURX#45n[Xg5n[5n U R[S5 U R[S5 [U5RU5n SSS5 [UW 5 [[5"[!U5U5n[X5 M g!,(df  NI=f) NrIrJrMrK皙?densityrW&splu converted its input to CSC formatCspsolve is more efficient when sparse b is in the CSC matrix format)r#rN default_rngrCrPsparse random_arraystandard_normalrrfilterr rerUrr r ) r1rWrXrgrYrZr[rhrisuprjs rtest_sparse_expm_multiply.TestExpmActionSimple.test_sparse_expm_multiplysii##D)  xA ))1&$C)HA##QF+A$Q*H"$ 2CE 201#1:>>!, % Hh / /0@0CQGH H /!%$s ;AD D c[R"SS/SS//[S9n[R"SS/5n[X5n[R"S[R"S5-SS[R "S5-[R "S5- --S[R"S5-/[S9n[X45 [[5"[U5U5n[X45 g)N?rdtyper-) r#arraycomplexrexpcossinrr r )r1r[rhrirjs r test_complex!TestExpmActionSimple.test_complexs HH H G# % HHb"X  &88 ObBrvvayL266!9$<= = O$+- +]+,++E*-B&(CJJ6GIJJ6 89(+1 !''!A#,2B2B62JK(7 )( ! )(s #A)D D, c [RRS5 SSSS.n[/SQ/SQ5GHup#[R R [RRX355n[RRU5n[R"S SU0UD6n[XE4SU0UD6n[Xv5H+up[U[X-5RU55 M- [[5"[U5U4SU0UD6n [[U5U4SU0UDS [R "U50D6n [[U5U4SU0UDS [R "U5S -0D6n [XX5HFupp[X-5RU5n[U U5 [UU5 [UU5 MH GM g) NrIrzrTrrrrr-r|r{rrJrrcr{r)r#rNrdrrPrQrRrSrrr0rrerUr r rf)r1intervalrrXr[rnrrrruXguessXgivenXwrong sol_guess sol_given sol_wrongcorrects r"test_expm_multiply_interval_vector9TestExpmActionInterval.test_expm_multiply_interval_vectors t #4@k+<=FC   !67A "Akk6c6X6Ga88x8A"1 '!#,*:*:1*=> /}-.>q.A1C25C9ACF##3A#67s7h7*,((1+7F##3A#69s9h9*,((1+a-9F69&:@7K2 i!!#,**1- 73 73 73 7K!>r!c [RRS5 SSSS.n[/SQ/SQSS /5GHup#n[R R [RRX355n[RRX45n[R"S S U0UD6n[XV4S U0UD6n[X5H+up[U [X-5RU55 M- [[5"[U5U4S U0UD6n[X5H+up[U [X-5RU55 M- GM! g) NrIrzrTrrrr-r|rr)r#rNrdrrPrQrRrSrrr0rrerUr r ) r1rrrXrgr[rhrrrrus r"test_expm_multiply_interval_matrix9TestExpmActionInterval.test_expm_multiply_interval_matrixs t #4@ .?!QHICA   !67A %Akk6c6X6Ga88x8A"1 '!#,*:*:1*=> /-()9!) /Ir!c [RR[R"S5S[ S9n[R "S[ S9n[RR[R"[R"S55SS9n[R"S[ S9n[RRU5n[[XSS5SURU55 [[R"[XSS5S5URU55 [RRS [R"S5-S[S9n[R "S[ S9n[RR[R"S [R"S5-5SS9n[[XSS5SURU55 [[R"[XSS5S5URU55 [RR[R"S5S[ S9n[R"SS [S9n[RR[R"[R"S55SS9n[[XSS5SURU55 [R"S[S9S -n[[R"[U[RRU5SS5S5URU55 g) Nr{csr)formatrr)rrr-r,yr)rPr diags_arrayr#arangeintonesrrT csr_arrayrrrUdiagrfull)r1r[rhAexpmBI BI_sparses r)test_sparse_expm_multiply_interval_dtypes@TestExpmActionInterval.test_sparse_expm_multiply_interval_dtypessT LL $ $RYYq\c $ J GGAS ! (( ! )=U(K [[# &LL**2.  a!A.r2EIIaLA aAq A" EF RS U LL $ $S1%5U' $ R GGAS ! ((BIIaL0@)A(O a!A.r2EIIaLA aAq A" EF RS U LL $ $RYYq\c $ J GGAr ) (( ! )=U(K a!A.r2EIIaLA [[' *2 - GGM!U\\%;%;B%?AFrJ K IIaL r!c&URS5 g)Nr(_help_test_specific_expm_interval_statusr1s r$test_expm_multiply_interval_status_0;TestExpmActionInterval.test_expm_multiply_interval_status_0 55a8r!c&URS5 g)Nr-rrs r$test_expm_multiply_interval_status_1;TestExpmActionInterval.test_expm_multiply_interval_status_1"rr!c&URS5 g)Nr|rrs r$test_expm_multiply_interval_status_2;TestExpmActionInterval.test_expm_multiply_interval_status_2%rr!c &[RRS5nSnSnSnSnSnSnSn S n /S QU -HnURXw5n URXx5n [ XX4XVSS 9n X:XdM8[ XX4XVS S 9up[ UR XWU45 [R"X4XVS 9n[X5H-unn[U[UU -5RU 55 M/ U S- n M U (dS[U5-S-n[U5eg)NrIrzrrTr{r|rKrr)rrrr status_onlyFrr-zfailed to find a status-z interval)r#rNrOrSrrshaperr0rrerUstr Exception)r1 target_statusrWrrrrrXrgnrepeats nsuccessesr[rhstatusrrrrumsgs rr?TestExpmActionInterval._help_test_specific_expm_interval_status(s ii##D)   )C !A !A,Q $&F&3A#C$)+ QWWsqk2++E4#&q?KHa#Hgacl.>.>q.AB$3a * ,s=/AAKOCC. r!rN)rrrrrr fail_slowrrrrrrrrrrrr!rrrs [[2LL2 [[ [[2444 [[ [[2 ? ? 8999!r!rdtype_adtype_b b_is_matrixFTcX1[-(a[[SSS9O[n[RR S5nSnU(aUS4OU4nU[ ;a?[RRUR XU/55RU5nOT[RRUR XU/5SUR XU/5--5RU5nU[ ;a$SUR U5-RU5nO5UR U5SUR U5--RU5n[Xx5n [[5"[U5U5n [R"[U5U5n U"X5 U"X5 [[U5U[R "U5S 9n U"X5 S S S S S.n [R""S0U D6n [Xx40U D6n[[5"[U5U40U D6n[%XU 5H3upn[UU-5RU5n U"X5 U"X5 M5 g)zAMake sure `expm_multiply` handles all numerical dtypes correctly.gH}]?gh㈵>)rtolatolrIrMrr|rbrzrrTrNr) IMPRECISErrr#rNr REAL_DTYPESrPrQrRastyperr r rUrerfrr0)rrrassert_allclose_rWrXb_shaper[rhsol_matsol_op direct_solrrX_matX_oprus rtest_expm_multiply_dtyperGs $- 9 f4H?N ))   %C A#q!f!G+ LL  SZZ/ 0 7 7 @ LL   JJv CJJv$6!6 6 &/ + szz'" " * *7 3 ZZ 2cjj&9#9 9 A A' JA!G } %&6q&91 =F A&JW)V( +A."((1+ FFV(c"$GHkk%H%G ! +( +E ] #$4Q$7 Gh GD!%w7QqS\%%a( -,8r!)4r functoolsr itertoolsrnumpyr#r numpy.testingrrrr scipy.sparser rPscipy.sparse.linalgr scipy.linalgr re"scipy.sparse.linalg._expm_multiplyr r rrrrscipy._lib._utilrsinglecsinglerintclonglongfloat32float64 longdoubler complex64 complex128 clongdoubleCOMPLEX_DTYPESDTYPESr r'r)rrr parametrizerrr!rrs,A ..00/!!%YY # wwzz2::r}}6 ,, r~~> ~ % (J,J,ZJ!J!ZF+F+ 6$-7,,$-r!