B &]\.8@s~ddlmZmZmZddlZddlZddlZddlZddl m Z ddl Z ddl m Z ddlmZy ddlZWnek rYnXGdddeZGdd d eZGd d d eZGd d d eZddZGdddeZddZddZddZddZddZy ddlZddlZde ekZ!Wnek r<dZ!YnXGddde"Z#d ej$d!fd"d#Z%d$d%Z&d&d'Z'd+d)d*Z(dS),)divisionprint_functionabsolute_importN)assert_)reraise)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 bgg?) ValueErrornpinfZfinfofloatmaxab inclusive_a inclusive_b)selfrrrrr9lib/python3.7/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)r r Zlogspacelog10Zlinspacehstacksizesort)rrrnZnlogptsZnlinptsptsZlinptsZlogptsrightZnlogpts1Znlogpts2Zlogpts1Zlogpts2rrr_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)rrr Zzerosrrr r) rrrrZn1Zn2ZposptsZnegptsrrrrvaluesas6     z Arg.valuesN) __name__ __module__ __qualname____doc__r r rr r"rrrrrs 8rc@seZdZddZddZdS)FixedArgcCst||_dS)N)r asarray_values)rr"rrrrszFixedArg.__init__cCs|jS)N)r))rrrrrr"szFixedArg.valuesN)r#r$r%rr"rrrrr'sr'c@s:eZdZeej ej eejejfddZddZdS) ComplexArgcCs$t|j|j|_t|j|j|_dS)N)rrealimag)rrrrrrrszComplexArg.__init__cCsXttt|}|j|}|j|d}|dddfd|dddfS)Nry?)intr ZfloorZsqrtr+r"r,Zravel)rrmxyrrrr"s zComplexArg.valuesN)r#r$r%complexr r rr"rrrrr*s&r*c@seZdZdddZddZdS) IntArgcCs||_||_dS)N)rr)rrrrrrrszIntArg.__init__cCsht|j|jtd|d|dt}tdd}t tj ||f}|||jk||jk@}|S)Nrr) rrrr"r astyper-r ZarangeuniqueZr_)rrZv1Zv2vrrrr"s , z IntArg.valuesN)r3r4)r#r$r%rr"rrrrr2s r2cCst|tjr|}nt|}tdd|D}||t|td}g}x&t ||D]\}}| | |q\Wt tj tj||dj}|S)NcSsg|]}t|trdndqS)g?g?) isinstancer*).0specrrr szget_args..rr!)r:r ndarraycopylenr(sumr7r-zipappendr"ZarrayZbroadcast_arraysZix_ZreshapeT)ZargspecrargsnargsZmsr<r.rrrget_argss   rGc @s&eZdZd ddZdd Zd d ZdS) MpmathDataNHz>YnFTcCs|dkrDyttjdd}Wntk r6d}YnX|r@dnd}||_||_||_||_||_ ||_ ||_ | |_ | |_ | |_t|jtjrt|jjtj|_ntdd|jD|_| |_ | |_|r|dkrt|d d}|r|dkrt|d d}||_| |_dS) NZ SCIPY_XSLOW0FiicSsg|]}t|tqSr)r:r*)r;argrrrr=sz'MpmathData.__init__..zr#)r-osenvirongetr scipy_func mpmath_funcarg_specdpsprecrrtolatolignore_inf_signnan_okr:r r> issubdtypedtypecomplexfloating is_complexanydistinguish_nan_and_infgetattrname param_filter)rrPrQrRr`rSrTrrUrVrWr^rXraZis_xslowrrrrs6      zMpmathData.__init__c s6tjdtjj}tjjtjj }}zjdk rBjg}ndg}j dk r\j tj_ t |j tj rzt ddnddddxt|D]\}}|tj_y.mptypecSstt|S)N)rcZmpfr )r/rrrrescSs0t|jddt|jkr"tjSt|jSdS)NgؗҜ.pytypecsjt|S)N)rQmap)r)rerirrrsz"MpmathData.check..F)Z vectorizedrUrVrWr^rXrar)r ZrandomZseedrGrRrrcZmprSrTrYrZr[ mpc2complex enumeraterrPrUrVrWr^rXraAssertionErrorr@rsysexc_info)rZargarrZold_dpsZold_precZdps_listjrSr)rerirrchecks>       zMpmathData.checkcCs"|jrd|jfSd|jfSdS)Nzz)r\r`)rrrr__repr__s zMpmathData.__repr__) NNNNrIrJFTTN)r#r$r%rrrrsrrrrrHs  $7rHcOst||}|dS)N)rHrr)rkwdrrrassert_mpmath_equals rvcCstjjdd|S)Nz8 Test not yet functional (too slow), needs more work.)reason)pytestZmarkskip)funcrrrnonfunctional_tooslow#sr{cCsttj|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 rcnstr)r/rrrrh+s rhcCstt|jt|jS)N)r1rhr+r,)r/rrrrl7srlcsddfdd}|S)NcSs t|tjrt|St|SdS)N)r:rcrdr1r )r/rrrtofloat<s ztrace_args..tofloatc sbtjdtt|ftjz||}tjd|WdtjdtjX|S)Nz%r: z-> %r )rostderrwritetuplerjflush)rrtr)rzrrrwrapBs    ztrace_args..wrapr)rzrr)rzrr trace_args;s r setitimerFc@s eZdZdS) TimeoutErrorN)r#r$r%rrrrrWsrg?Tcs4tr"|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 tdS)N)r)Zsignumframerrrsigalrm_handlerksz%time_limited..sigalrm_handlercsfdd}|S)Nc sdttj}ttjz$y ||Stk r<SXWdttjdttj|XdS)Nr)signalSIGALRMr ITIMER_REALr)rrtZ old_handler)rz return_valrtimeoutrrros  z(time_limited..deco..wrapr)rzr)rrr)rzrdecons ztime_limited..decocsfdd}|S)Nc sbtfddtz.y ||Stk rLtdSXWdtdXdS)NcstkrtS)N)timer)rZeventrL) start_timertracerrrsz7time_limited..deco..wrap..trace)rrosettracer)rrt)rzrr)rrrr|s    z(time_limited..deco..wrapr)rzr)rr)rzrr{s)POSIX)rrZ use_sigalrmrr)rrrr time_limited[s  rcsfdd}|S)z9Decorate function to return nan if it raises an exceptioncs&y ||Stk r tjSXdS)N) Exceptionr rg)rrt)rzrrrs zexception_to_nan..wrapr)rzrr)rzrexception_to_nans rcsfdd}|S)z1Decorate function to return nan if it returns infcs||}t|stjS|S)N)r Zisfiniterg)rrtr9)rzrrrs  zinf_to_nan..wrapr)rzrr)rzr inf_to_nans rFFgsD    o  f  4