a aM8@sPddlZddlZddlZddlZddlmZddlZddlm Z z ddl Z Wne y^Yn0GdddZ GdddZ Gdd d ZGd d d Zd d ZGdddZddZddZddZddZddZz ddlZddlZdeevZWne ydZYn0GdddeZdejdfd 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 kr,dttj}|tjkrFdttj}|||_|_|||_|_ dS)Nz#a should be less than or equal to bg?) ValueErrornpinfZfinfofloatmaxab inclusive_a inclusive_b)selfr r r rrj/Users/vegardjervell/Documents/master/model/venv/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 rZzerosr rrr) rrr r Zn1Zn2ZposptsZnegptsrrrrvalues^s6     z Arg.valuesN) __name__ __module__ __qualname____doc__rrrrrrrrrrs 8rc@seZdZddZddZdS)FixedArgcCst||_dSN)rasarray_values)rrrrrrszFixedArg.__init__cCs|jSr%)r')rrrrrrszFixedArg.valuesNr r!r"rrrrrrr$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*rr+Zravel)rrmxyrrrrs zComplexArg.valuesN)r r!r"complexrrrrrrrrr)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 rr astyper-rZarangeuniqueZr_)rrv1v2vrrrrs , z IntArg.valuesN)r5r6r(rrrrr4s r4cCst|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@rAr0rrrrBrCr)r?rndarraycopylenr&sumr9r-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%)mpmathmpcr3r1rrrmptypesz 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|jfSd|jfSdS)Nzz)rcrg)rrrr__repr__s zMpmathData.__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%)r3rpr*r+rlrrrrx6srxcsddfdd}|S)NcSs t|tjrt|St|SdSr%)r?rjrkr3r rlrrrtofloat;s ztrace_args..tofloatc s|tjdtt|ftjz8|i|}tjd|Wtjdtjntjdtj0|S)Nz%r: z-> %r )r{stderrwritetuplerrflush)r rrrrrrwrapAs    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