o tBhM8@sNddlZddlZddlZddlZddlmZddlZddlm Z zddl Z Wn e y/YnwGdddZ GdddZ Gdd d ZGd d d Zd d ZGdddZddZddZddZddZddZzddlZddlZdeevZWn e ydZYnwGdddeZdejdfd d!Zd"d#Zd$d%Z d)d'd(Z!dS)*N)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 r rrrq/var/www/html/riverr-enterprise-integrations-main/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 r nnlogptsnlinptsptslinptslogptsrightnlogpts1nlogpts2logpts1logpts2rrr_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 r rzerosrrr)r) rrr r n1n2posptsnegptsr!rrrvalues^s6     z Arg.valuesN) __name__ __module__ __qualname____doc__rrrr)r0rrrrrs  8rc@seZdZddZddZdS)FixedArgcCst||_dSN)rasarray_values)rr0rrrrszFixedArg.__init__cCs|jSr6)r8)rrrrrr0zFixedArg.valuesNr1r2r3rr0rrrrr5s r5c@s:eZdZeej ej eejejfddZddZdS) ComplexArgcCs$t|j|j|_t|j|j|_dSr6)rrealimagrr r rrrrszComplexArg.__init__cCsXttt|}|j|}|j|d}|dddfd|dddfS)Nry?)intrfloorsqrtr<r0r=ravel)rrmxyrrrr0s (zComplexArg.valuesN)r1r2r3complexrrrr0rrrrr;s& r;c@seZdZdddZddZdS) IntArgcCs||_||_dSr6)r r r>rrrrs zIntArg.__init__cCsht|j|jtd|d|dt}tdd}t tj ||f}|||jk||jk@}|S)Nrr) rr r r0r astyper?rarangeuniquer_)rrv1v2vrrrr0s , z IntArg.valuesN)rHrIr:rrrrrGs  rGcCst|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)r0)rUrVrCrrrrWsr*)rTrndarraycopylenr7sumrLr?ziparraybroadcast_arraysix_reshapeT)argspecrargsnargsmsrrrget_argss   rfc@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|tqSrrS)rUargrrrrWsz'MpmathData.__init__..zr1)r?osenvirongetr scipy_func mpmath_funcarg_specdpsprecrrtolatolignore_inf_signnan_okrTrrX issubdtypedtypecomplexfloating is_complexanydistinguish_nan_and_infgetattrname param_filter)rrprqrrrrsrtrrurvrwr~rxris_xslowrrrrs8       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|Sr6)mpmathmpcrFrDrrrmptypez MpmathData.check..mptypecSrr6)rmpfr rrrrrrcSs,t|jddt|jkrtjSt|jS)NgؗҜ.pytypecsjt|Sr6)rqmap)r rrrrrsz"MpmathData.check..F) vectorizedrurvrwr~rxrr)rrandomseedrfrrrrmprsrtryrzr{ mpc2complex enumeraterrprurvrwr~rxrAssertionErrorrZsysexc_info __traceback__with_traceback) rargarrold_dpsold_precdps_listjrstpvaluetbrrrchecksP         * zMpmathData.checkcCs|jr d|jfSd|jfS)Nzz)r|r)rrrr__repr__s  zMpmathData.__repr__) NNNNrhriFTTN)r1r2r3rrrrrrrrgs ' ;rgcOst|i|}|dSr6)rgr)r kwdrrrassert_mpmath_equals rcCstjjdd|S)Nz8 Test not yet functional (too slow), needs more work.)reason)pytestmarkskipfuncrrrnonfunctional_tooslow"srcCsttj|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 rnstrrrrrr*s rcCstt|jt|jSr6)rFrr<r=rrrrr6srcsddfdd}|S)NcSst|tjr t|St|Sr6)rTrrrFr rrrrtofloat;s ztrace_args..tofloatc sztjdtt|ftjz|i|}tjd|Wtjdtj|Stjdtjw)Nz%r: z-> %r )rstderrwritetuplerflush)r rrrrrrwrapAs    ztrace_args..wraprrrrrr trace_args:s r setitimerFc@s eZdZdS) TimeoutErrorN)r1r2r3rrrrrVsrrTcs6tr|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. cSstr6)r)signumframerrrsigalrm_handlerjr9z%time_limited..sigalrm_handlercsfdd}|S)Nc sttj}ttjz2z|i|WWttjdttj|Sty@YWttjdttj|Swttjdttj|w)Nr)signalSIGALRMr ITIMER_REALr)r r old_handler)r return_valrtimeoutrrrns (time_limited..deco..wraprrrrrrrdecoms ztime_limited..decocsfdd}|S)Nc sztfddtz%z|i|WWtdSty6tdYWtdSwtdw)Ncstkr tSr6)timer)reventrl) start_timertracerrr~sz7time_limited..deco..wrap..trace)rrsettracerr r)rrr)rrrr{s     rrr)rrrrrzs)POSIX)rr use_sigalrmrrrr time_limitedZs rcfdd}|S)z9Decorate function to return nan if it raises an exceptioncs*z|i|WStytjYSwr6) Exceptionrrrrrrrs   zexception_to_nan..wraprrrrrexception_to_nan rcr)z1Decorate function to return nan if it returns infcs"|i|}t|stjS|Sr6)risfiniter)r rrRrrrrs zinf_to_nan..wraprrrrr inf_to_nanrrFFg     4    r)rr)"rmrrnumpyr numpy.testingrrscipy.special._testutilsrr ImportErrorrr5r;rGrfrgrrrrrposixrdirrrrrrrrrrrrrsF    o  j  4