a Qe8@sTddlZddlZddlZddlmZddlZddlmZddl Z ddl m Z z ddl Z Wne yjYn0GdddZGdddZGd d d ZGd d d Zd dZGdddZddZddZddZddZddZzddlZdeevZWne ydZYn0GdddeZdejd 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 kr,dttj}|tjkrFdttj}|||_|_|||_|_ dS)Nz#a should be less than or equal to bg?) ValueErrornpinfZfinfofloatmaxab inclusive_a inclusive_b)selfr r rrrN/var/www/sunrise/env/lib/python3.9/site-packages/scipy/special/_mptestutils.py__init__s  z Arg.__init__cCs|dkrtd|ddkr*|d}|}n|d}|d}|dkr`tt|t||}nV|dkr|dkrt|||}n4|dkrtj|d|dd}tdt||}t||f}n|dkr"|dkr"td||}|jdkrt|d} nd} tjd| |dd}t||f}n|ddkr>|d} | } n|d} | d} tjdd|dd}|jdkr|t|d} nd} tjd| | dd} tdt|| } t| || f}t|S) Nrza should be positive F)Zendpointi)rrZlogspacelog10Zlinspacehstacksizesort)rr r nZnlogptsZnlinptsptsZlinptsZlogptsrightZnlogpts1Znlogpts2Zlogpts1Zlogpts2rrr_positive_values'sF   zArg._positive_valuesc Cs|j|j}}||kr t|S|js.|d7}|js<|d7}|ddkrV|d}|}n|d}|d}|dkr||||}g}nR|dkrg}|| | | }n0|d||}|d| |d }|dd}t|ddd|f}|js|dd}|js|dd}|S)z%Return an array containing n numbers.rrrN)r r rzerosrrrr) rrr r Zn1Zn2ZposptsZnegptsrrrrvalues_s6     z Arg.valuesN) __name__ __module__ __qualname____doc__rr rrr!rrrrrs 8rc@seZdZddZddZdS)FixedArgcCst||_dSN)rasarray_values)rr!rrrrszFixedArg.__init__cCs|jSr')r))rrrrrr!szFixedArg.valuesNr"r#r$rr!rrrrr&sr&c@s:eZdZeej ej eejejfddZddZdS) ComplexArgcCs$t|j|j|_t|j|j|_dSr')rrealimagrr r rrrrszComplexArg.__init__cCsXttt|}|j|}|j|d}|dddfd|dddfS)Nry?)intrfloorsqrtr,r!r-Zravel)rrmxyrrrr!s zComplexArg.valuesN)r"r#r$complexrr rr!rrrrr+s&r+c@seZdZdddZddZdS) IntArgcCs||_||_dSr')r r r.rrrrszIntArg.__init__cCsht|j|jtd|d|dt}tdd}t tj ||f}|||jk||jk@}|S)Nrr) rr r r!r astyper/rZarangeuniqueZr_)rrZv1Zv2vrrrr!s , z IntArg.valuesN)r7r8r*rrrrr6s r6cCst|tjr|}njt|}tdd|D}||t|td}ddt ||D}t tj tj | |dj}|S)NcSsg|]}t|trdndqS)g?g? isinstancer+).0specrrr zget_args..rcSsg|]\}}||qSr)r!)r@rAr2rrrrBrCr)r?rndarraycopylenr(sumr;r/ziparrayZbroadcast_arraysZix_ZreshapeT)Zargspecrargsnargsmsrrrget_argss   rNc @s&eZdZd ddZdd Zd d ZdS) MpmathDataNHz>YnFTcCs|durBzttjdd}Wnty4d}Yn0|r>dnd}||_||_||_||_||_ ||_ ||_ | |_ | |_ | |_t|jtjrt|jjtj|_ntdd|jD|_| |_ | |_|r|dkrt|d d}|r|dkrt|d d}||_| |_dS) NZ SCIPY_XSLOW0FiicSsg|]}t|tqSrr>)r@argrrrrBrCz'MpmathData.__init__..zr")r/osenvirongetr scipy_func mpmath_funcarg_specdpsprecrrtolatolignore_inf_signnan_okr?rrD issubdtypedtypecomplexfloating is_complexanydistinguish_nan_and_infgetattrname param_filter)rrWrXrYrgrZr[rr\r]r^rer_rhZis_xslowrrrrs6       zMpmathData.__init__c sftjdtjj}tjjtjj }}zjdurDjg}ndg}j dur^j tj_ t |j tj r|t ddnddddt|D]\}}|tj_zBtjfdd |d jjjjjjd Wq8Wqty4|t|d kr0t\}}} |j| ur,|| |Yq0qW||tj_tj_ n||tj_tj_ 0dS) NicSstt|Sr')mpmathmpcr5r3rrrmptypesz MpmathData.check..mptypecSstt|Sr')rjZmpfr rlrrrrmscSs0t|jddt|jkr"tjSt|jSdS)NgؗҜ.pytypecsjt|Sr')rXmap)r rmrqrrrrCz"MpmathData.check..F)Z vectorizedr\r]r^rer_rhr)rrandomseedrNrYrrjmprZr[r`rarb mpc2complex enumeraterrWr\r]r^rer_rhAssertionErrorrFsysexc_info __traceback__with_traceback) rZargarrZold_dpsZold_precZdps_listjrZtpvaluetbrrsrchecksF         zMpmathData.checkcCs&|jrd|jdSd|jdSdS)Nz >)rcrg)rrrr__repr__szMpmathData.__repr__) NNNNrPrQFTTN)r"r#r$rrrrrrrrOs ';rOcOst|i|}|dSr')rOr)r kwdrrrassert_mpmath_equalsrcCstjjdd|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 rZ min_fixedZ max_fixed)r rjnstrrlrrrrp+s rpcCstt|jt|jSr')r5rpr,r-rlrrrrx7srxcsddfdd}|S)NcSs t|tjrt|St|SdSr')r?rjrkr5r rlrrrtofloat<s ztrace_args..tofloatc s|tjtt|dtjz8|i|}tjd|Wtjdtjntjdtj0|S)Nz: z-> %r )r{stderrwritetuplerrflush)r rrrrrrwrapBs    ztrace_args..wraprrrrrr trace_args;s r setitimerFc@s eZdZdS) TimeoutErrorN)r"r#r$rrrrrVsrrTcs4tr"|r"ddfdd}nfdd}|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. cSs tdSr')r)signumframerrrsigalrm_handlerjsz%time_limited..sigalrm_handlercsfdd}|S)Nc sttj}ttjzz.|i|WWttjdttj|StyYWttjdttj|S0Wttjdttj|nttjdttj|0dS)Nr)signalSIGALRMr ITIMER_REALr)r rZ old_handler)r return_valrtimeoutrrrns  (time_limited..deco..wraprrrrrrrdecoms ztime_limited..decocsfdd}|S)Nc stfddtzXz|i|WWtdStyltdYWtdS0Wtdn td0dS)NcstkrtSr')timer)reventrS) start_timertracerrr~sz7time_limited..deco..wrap..trace)rr{settracerr r)rrr)rrrr{s     rrr)rrrrrzs)POSIX)rrZ use_sigalrmrrrr time_limitedZs  rcsfdd}|S)z9Decorate function to return nan if it raises an exceptioncs.z|i|WSty(tjYS0dSr') Exceptionrrorrrrrs zexception_to_nan..wraprrrrrexception_to_nans rcsfdd}|S)z1Decorate function to return nan if it returns infcs"|i|}t|stjS|Sr')risfinitero)r rr=rrrrs zinf_to_nan..wraprrrrr inf_to_nans rFFgs@     o  j  4