^Mhi8ddlZddlZddlZddlmZddlZddlmZddl Z ddl m Z  ddl Z n #e $rYnwxYwGddZGddZGd d ZGd d Zd ZGddZdZdZdZdZdZ ddlZdeevZn #e $rdZYnwxYwGddeZdejdfdZdZ dZ!d dZ"dS)!N) zip_longest)assert_)assert_func_equalcDeZdZdZej ejddfdZdZdZdS)ArgzGenerate a set of numbers on the real axis, concentrating on 'interesting' regions and covering all orders of magnitude. Tc2||krtd|tj kr!dtjtjz}|tjkr!dtjtjz}||c|_|_||c|_|_ dS)Nz#a should be less than or equal to bg?) ValueErrornpinffinfofloatmaxab inclusive_a inclusive_b)selfrrrrs Z/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/scipy/special/_mptestutils.py__init__z Arg.__init__s q55BCC C <<RXe__((A ;;BHUOO''AA-8+*$***ch|dkrtd|dzdkr|dz}|}n |dz}|dz}|dkr|j|j}}||krtj|S|js|dz }|js|dz }|dzdkr|dz}|}n |dz}|dz}|dkr||||}g}n`|dkrg}|| | | }n=|d||}|d| |dz }|dd}tj|ddd|f}|js |dd}|js |dd}|S)z%Return an array containing n numbers.rrrN)rrr zerosrrr.r ) rr#rrn1n2posptsnegptsr&s rvaluesz Arg.values_sivtv1 668A;;   FA  FA q5A::ABBBABaB 66**1a33FFF !VVF++QBA666FF**1a44F++Ar26:::FABBZFi"v.// abb'C crc(C rN) __name__ __module__ __qualname____doc__r r rr.r6rrrrsf &BF$ F F F F666p#####rrceZdZdZdZdS)FixedArgc8tj||_dSN)r asarray_values)rr6s rrzFixedArg.__init__sz&)) rc|jSr?)rA)rr#s rr6zFixedArg.valuess |rNr7r8r9rr6r;rrr=r=s2***rr=cteZdZeej ej eejejfdZdZdS) ComplexArgct|j|j|_t|j|j|_dSr?)rrealimagrrrs rrzComplexArg.__init__s0'' '' rc2ttjtj|}|j|}|j|dz}|dddfd|dddfzzS)Nry?)intr floorsqrtrGr6rHravel)rr#mxys rr6zComplexArg.valuess $$ % % I  Q   I  QU # #!!!D& BqaaayL(//111rN)r7r8r9complexr r rr6r;rrrErEs` "&26'22ggbfbf6M6M((((22222rrEceZdZddZdZdS)IntArgc"||_||_dSr?)rrrIs rrzIntArg.__init__srcnt|j|jt d|dzz|dz t }tjdd}tj tj ||f}|||jk||jkz}|S)Nrr) rrrr6rastyperKr arangeuniquer_)rr#v1v2vs rr6z IntArg.valuess  ' 'A1Hac(:(: ; ; B B3 G G Yr1   IbeBFm $ $ qDF{q46z* +rN)rUrVrCr;rrrTrTs7rrTct|tjr|}nt |}tjd|D}||t |z ztdz}dt||D}tj tj tj | |dj}|S)Nc@g|]}t|trdndS)g?g? isinstancerE).0specs r zget_args..s+ N N NdJtZ00 9SSc N N Nrrc>g|]\}}||Sr;)r6)rfrgrOs rrhzget_args..s&???74 A???rr0)rer ndarraycopylenr@sumr[rKziparraybroadcast_arraysix_reshapeT)argspecr#argsnargsmss rget_argsrxs'2:&& R||~~G  Z N Ng N N N  "SWW*o % %c * *Q .??c'2.>.>???x+RVT];<<DDUBOOQ Krc*eZdZ d dZdZdZdS) MpmathDataNHz>YnFTc|G ttjdd}n#t$rd}YnwxYw|rdnd}||_||_||_||_||_ ||_ ||_ | |_ | |_ | |_t|jt jr/t!j|jjt j|_n#t-d|jD|_| |_ | |_|r|dkrt1|dd}|r|dkrt1|dd}||_| |_dS) N SCIPY_XSLOW0Fiic8g|]}t|tSr;rd)rfargs rrhz'MpmathData.__init__..s"FFFC,,FFFrzr7)rKosenvirongetr scipy_func mpmath_funcarg_specdpsprecr#rtolatolignore_inf_signnan_okrer rj issubdtypedtypecomplexfloating is_complexanydistinguish_nan_and_infgetattrname param_filter)rrrrrrrr#rrrrrris_xslows rrzMpmathData.__init__sm 9 !rz~~mSAABB ! ! !  !!)cA$&     . dmRZ 0 0  mDM,?ASTTDOO!FF FFFDO /'>$ 9tz)):z488D :tz)); D99D (s-2 AAc  tjdtjj}t jjt jj }} j jg}ndg}j j t j_ tj |j tj r t d nd d t|D]\}}|t j_ tj fd|djjjjjj n]#t.$rP|t1|d z kr7t3j\}}} |j| ur|| |YwxYw||ct j_t j_ dS#||ct j_t j_ wxYw) NicDtjt|Sr?)mpmathmpcrRrPs rmptypez MpmathData.check..mptypes!:gajj111rcDtjt|Sr?)rmpfrrs rrz MpmathData.check..mptypes!:eAhh///rct|jddt|jzzkr tjSt |jS)NgؗҜ.pytypes=16{{UAAF O%<<<!v (000rcFjt|Sr?)rmap)rrrrs rz"MpmathData.check.. s$66*:$*:CNN*K#L#LrF) vectorizedrrrrrrr)r randomseedrxrr#rmprrrrr mpc2complex enumeraterrrrrrrrAssertionErrorrlsysexc_info __traceback__with_traceback) rargarrold_dpsold_precdps_listjrtpvaluetbrrs ` @@rcheckzMpmathData.checks t$-00#IM69>4 >x# H:4y$!%  }V\2+=>> 1$2222000111$H-- $ $3 # $%LLLLLL#(!Y!Y(,(<040L#{%)%6    E%$$$CMM!O++(+  E2 .b88"'"6"6r":"::# ,+$-4X )FIM69>>>GX )FIM69> = = = =s3 B F0+AD.,F0.AFF0FF00%Gc<|jr d|jdSd|jdS)Nz >)rr)rs r__repr__zMpmathData.__repr__s2 ? 0949999 9/49/// /r) NNNNr{r|FTTN)r7r8r9rrrr;rrrzrzsT?C>D@D+/')')')')R<><><>|00000rrzcFt|i|}|dSr?)rzr)rkwds rassert_mpmath_equalr%s%AAGGIIIIIrcRtjd|S)Nz8 Test not yet functional (too slow), needs more work.)reason)pytestmarkskip)funcs rnonfunctional_tooslowr*s1 ;  I       rcLttj|dddS)z Convert an mpf to the nearest floating point number. Just using float directly doesn't work because of results like this: with mp.workdps(50): float(mpf("0.99999999999999999")) = 0.9999999999999999 r min_fixed max_fixed)rrnstrrs rrr4s% Qa1=== > >>rcjtt|jt|jSr?)rRrrGrHrs rrr@s& 9QV$$i&7&7 8 88rcdfd}|S)Ncrt|tjrt|St |Sr?)rerrrRrrs rtofloatztrace_args..tofloatEs- a $ $ 1:: 88Orctjtt |dtj |i|}tjd|tjdtjnB#tjdtjwxYw|S)N: z->  )rstderrwritetuplerflush)rrrrrs rwrapztrace_args..wrapKs E#gq//22888999  a2A J  [1[[ ) ) ) J  T " " " J       J  T " " " J      s *C?Dr;)rrrs` @r trace_argsrDs8       Kr setitimerFceZdZdS) TimeoutErrorN)r7r8r9r;rrrr_sDrrr Tc@tr |r dfd}nfd}|S)a Decorator for setting a timeout for pure-Python functions. If the function does not return within `timeout` seconds, the value `return_val` is returned instead. On POSIX this uses SIGALRM by default. On non-POSIX, settrace is used. Do not use this with threads: the SIGALRM implementation does probably not work well. The settrace implementation only traces the current thread. The settrace implementation slows down execution speed. Slowdown by a factor around 10 is probably typical. ctr?)r)signumframes rsigalrm_handlerz%time_limited..sigalrm_handlerss .. rcfd}|S)Nc4tjtj}tjtj |i|tjtjdtjtj|S#t$rCcYtjtjdtjtj|SwxYw#tjtjdtjtj|wxYw)Nr)signalSIGALRMr ITIMER_REALr)rr old_handlerr return_valrtimeouts rrz(time_limited..deco..wrapws$mFNOLL  !3W===?4>b>>$V%7;;;M&.+>>>> $&&&%%%$V%7;;;M&.+>>>> &$V%7;;;M&.+>>>>s%B CCCCADr;)rrrrrs` rdecoztime_limited..decovs3 ? ? ? ? ? ? ? ?Krcfd}|S)NcDtjfdtj |i|tjdS#t$r-tjdcYtjdSwxYw#tjdwxYw)NcXtjz krtSr?)timer)reventr start_timertraces rrz7time_limited..deco..wrap..traces(y{{Z/'99*nn, Lr)rrsettracer)rrrrrrrs @@rrz(time_limited..deco..wraps!Y[[ !!!!!!! U###'4>b>> L&&&& $&&&L&&&%%%L&&&& &L&&&&s#A B/B BB Br;)rrrrs` rrztime_limited..decos. ' ' ' ' ' ' 'Kr)POSIX)rr use_sigalrmrrs`` @r time_limitedrcsr !! ! ! !              " Krcfd}|S)z9Decorate function to return nan if it raises an exceptioncN |i|S#t$rtjcYSwxYwr?) Exceptionr r)rrrs rrzexception_to_nan..wraps@ 4>b>> !   6MMM s  $$r;rrs` rexception_to_nanrs# Krcfd}|S)z1Decorate function to return nan if it returns infcX|i|}tj|s tjS|Sr?)r isfiniter)rrrars rrzinf_to_nan..wraps0 D!NrNN{1~~ 6Mrr;rs` r inf_to_nanrs# KrFFg)>??11 FF >V^KLL L ;v ' '$fk&6I6I1I*I I I OOQ/ 0 0 0 MME qyyc"(4..))**d T5TT!a%TTTUUU!) E E Avv[AKKKF[AKKKF{{ Vf_f$<== E1-- JJ!CCvCC6CC5CCC D D D Dtyy~~&&&&&yr)rr)#rrr itertoolsrnumpyr numpy.testingrrscipy.special._testutilsrr ImportErrorrr=rErTrxrzrrrrrrdirrrrrrrrrr;rrrs !!!!!!!!!!!! 666666 MMMM   D llllllll^ 2 2 2 2 2 2 2 2            l0l0l0l0l0l0l0l0^     ? ? ?999(MMM CCKK 'EE EEE     9   T1111h''''''s-55 BB('B(