o `^hi8@sRddlZddlZddlZddlmZddlZddlmZddl Z ddl m Z zddl Z Wn e y5YnwGdddZGdddZGd d d ZGd d d Zd dZGdddZddZddZddZddZddZz ddlZdeevZWn e ydZYnwGdddeZdejd fd!d"Zd#d$Z d%d&Z!d*d(d)Z"dS)+N) zip_longest)assert_)assert_func_equalc@s8eZdZdZej ejddfddZddZddZd S) ArgzGenerate a set of numbers on the real axis, concentrating on 'interesting' regions and covering all orders of magnitude. TcCsf||krtd|tj krdttj}|tjkr#dttj}|||_|_|||_|_ dS)Nz#a should be less than or equal to bg?) ValueErrornpinffinfofloatmaxab inclusive_a inclusive_b)selfr rrrrX/home/air/shanriGPT/back/venv/lib/python3.10/site-packages/scipy/special/_mptestutils.py__init__s  z Arg.__init__cCs|dkrtd|ddkr|d}|}n|d}|d}|dkr/tt|t||}n|dkr?|dkr?t|||}n|dkr^tj|d|dd}tdt||}t||f}nw|dkr|dkrtd||}|jdkrzt|d} nd} tjd| |dd}t||f}nH|ddkr|d} | } n|d} | d} tjdd|dd}|jdkrt|d} nd} tjd| | dd} tdt|| } t| || f}t|S) Nrza should be positive F)endpointi)rrlogspacelog10linspacehstacksizesort)rr rnnlogptsnlinptsptslinptslogptsrightnlogpts1nlogpts2logpts1logpts2rrr_positive_values'sF     zArg._positive_valuesc Cs|j|j}}||krt|S|js|d7}|js|d7}|ddkr+|d}|}n|d}|d}|dkrA||||}g}n)|dkrRg}|| | | }n|d||}|d| |d }|dd}t|ddd|f}|js|dd}|js|dd}|S)z%Return an array containing n numbers.rrrN)r rrzerosrrr*r) rrr rn1n2posptsnegptsr"rrrvalues_s6     z Arg.valuesN) __name__ __module__ __qualname____doc__rr rr*r1rrrrrs  8rc@seZdZddZddZdS)FixedArgcCst||_dSN)rasarray_values)rr1rrrrszFixedArg.__init__cCs|jSr7)r9)rrrrrr1zFixedArg.valuesNr2r3r4rr1rrrrr6s r6c@s:eZdZeej ej eejejfddZddZdS) ComplexArgcCs$t|j|j|_t|j|j|_dSr7)rrealimagrr rrrrrszComplexArg.__init__cCsXttt|}|j|}|j|d}|dddfd|dddfS)Nry?)intrfloorsqrtr=r1r>ravel)rrmxyrrrr1s (zComplexArg.valuesN)r2r3r4complexrr rr1rrrrr<s& r<c@seZdZdddZddZdS) IntArgcCs||_||_dSr7)r rr?rrrrs zIntArg.__init__cCsht|j|jtd|d|dt}tdd}t tj ||f}|||jk||jk@}|S)Nrr) rr rr1r astyper@rarangeuniquer_)rrv1v2vrrrr1s , z IntArg.valuesN)rIrJr;rrrrrHs  rHcCst|tjr |}|St|}tdd|D}||t|td}ddt ||D}t tj tj | |dj}|S)NcSsg|] }t|tr dndqS)g?g? isinstancer<).0specrrr szget_args..rcSsg|] \}}||qSr)r1)rVrWrDrrrrXsr+)rUrndarraycopylenr8sumrMr@ziparraybroadcast_arraysix_reshapeT)argspecrargsnargsmsrrrget_argss    rgc@s.eZdZ    d ddZdd Zd d ZdS) MpmathDataNHz>YnFTcCs|dur!z ttjdd}Wn tyd}Ynw|rdnd}||_||_||_||_||_ ||_ ||_ | |_ | |_ | |_t|jtjrQt|jjtj|_n tdd|jD|_| |_ | |_|rh|dkrnt|d d}|rt|dkrzt|d d}||_| |_dS) N SCIPY_XSLOW0FiicSsg|]}t|tqSrrT)rVargrrrrXsz'MpmathData.__init__..zr2)r@osenvirongetr scipy_func mpmath_funcarg_specdpsprecrrtolatolignore_inf_signnan_okrUrrY issubdtypedtypecomplexfloating is_complexanydistinguish_nan_and_infgetattrname param_filter)rrqrrrsrrtrurrvrwrxrryris_xslowrrrrs<       zMpmathData.__init__c sltjdtjj}tjjtjj }}zjdur!jg}ndg}j dur.j tj_ t |j tj r=t ddnddddt|D]J\}}|tj_ztjfdd |d jjjjjjd Wn0ty|t|d krt\}}} |j| ur|| |YqIwW||tj_tj_ dSW||tj_tj_ dS||tj_tj_ w) NicStt|Sr7)mpmathmpcrGrErrrmptypez MpmathData.check..mptypecSrr7)rmpfr rrrrrrcSs,t|jddt|jkrtjSt|jS)NgؗҜr=rnan mpf2floatrrrrpytypes z MpmathData.check..pytypecsjt|Sr7)rrmap)r rrrrr sz"MpmathData.check..F) vectorizedrvrwrxrryrr)rrandomseedrgrsrrmprtrurzr{r| mpc2complex enumeraterrqrvrwrxrryrAssertionErrorr[sysexc_info __traceback__with_traceback) rargarrold_dpsold_precdps_listjrttpvaluetbrrrchecksT         * zMpmathData.checkcCs"|jr d|jdSd|jdS)Nz >)r}r)rrrr__repr__szMpmathData.__repr__) NNNNrirjFTTN)r2r3r4rrrrrrrrhs ) >rhcOst|i|}|dSr7)rhr)r kwdrrrassert_mpmath_equal%s rcCstjjdd|S)Nz8 Test not yet functional (too slow), needs more work.)reason)pytestmarkskipfuncrrrnonfunctional_tooslow*s rcCsttj|ddddS)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)r rnstrrrrrr4s rcCstt|jt|jSr7)rGrr=r>rrrrr@srcsddfdd}|S)NcSst|tjr t|St|Sr7)rUrrrGr rrrrtofloatEs ztrace_args..tofloatc s|tjtt|dtjz|i|}tjd|Wtjdtj|Stjdtjw)N: z->  )rstderrwritetuplerflush)r rrrrrrwrapKs    ztrace_args..wraprrrrrr trace_argsDs r setitimerFc@s eZdZdS) TimeoutErrorN)r2r3r4rrrrr_srrTcs6tr|rddfdd}|Sfdd}|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. cSstr7)r)signumframerrrsigalrm_handlersr:z%time_limited..sigalrm_handlercsfdd}|S)Nc sttj}ttjz2z|i|WWttjdttj|Sty@YWttjdttj|Swttjdttj|w)Nr)signalSIGALRMr ITIMER_REALr)r r old_handler)r return_valrtimeoutrrrws (time_limited..deco..wraprrrrrrrdecovs ztime_limited..decocsfdd}|S)Nc sztfddtz%z|i|WWtdSty6tdYWtdSwtdw)Ncstkr tSr7)timer)reventrm) start_timertracerrrsz7time_limited..deco..wrap..trace)rrsettracerr r)rrr)rrrrs     rrr)rrrrrs)POSIX)rr use_sigalrmrrrr time_limitedcs rcfdd}|S)z9Decorate function to return nan if it raises an exceptioncs*z|i|WStytjYSwr7) Exceptionrrrrrrrs   zexception_to_nan..wraprrrrrexception_to_nan rcr)z1Decorate function to return nan if it returns infcs"|i|}t|stjS|Sr7)risfiniter)r rrSrrrrs zinf_to_nan..wraprrrrr inf_to_nanrrFFg\}}}tj || ddd } tj || ddd } |dkrvd } nt|||} t | d } | |d | d | d| dqXt dd | dSdS)z| Compare lists of mpmath.mpf's or mpmath.mpc's directly so that it can be done to higher precision than double. Nz,Lengths of inputs res and std are not equal.rz Bad results (z out of rz) for the following points:rr rz != z (rdiff )Fr)rrrrfabsappendr[r@rrrrrjoin)resstdrwrvfailureskresvalstdvalnfailndigitsmsgresrepstdreprdiffrrrmp_assert_allcloses,  $r)rr)#rnrr itertoolsrnumpyr numpy.testingrrscipy.special._testutilsrr ImportErrorrr6r<rHrgrhrrrrrrdirrrrrrrrrrrrrsF     o  o   4