U md8@s`ddlZddlZddlZddlmZddlZddlmZddl Z ddl m Z z ddl Z Wne k rlYnXGdddZGdddZGd d d ZGd d d Zd dZGdddZddZddZddZddZddZz ddlZddlZdeekZWne k rdZYnXGdddeZdejd fd!d"Z d#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 rrrS/home/sam/Atlas/atlas_env/lib/python3.8/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 rZzerosrrrr) 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)r6r7r)rrrrr5s r5cCst|tjr|}njt|}tdd|D}||t|td}ddt ||D}t tj tj | |dj}|S)NcSsg|]}t|trdndqS)g?g? isinstancer*).0specrrr szget_args..rcSsg|]\}}||qSr)r )r?r@r1rrrrAsr)r>rndarraycopylenr'sumr:r.ziparrayZbroadcast_arraysZix_ZreshapeT)Zargspecrargsnargsmsrrrget_argss   rLc @s&eZdZd ddZdd Zd d ZdS) MpmathDataNHz>YnFTcCs|dkrDzttjdd}Wntk r6d}YnX|r@dnd}||_||_||_||_||_ ||_ ||_ | |_ | |_ | |_t|jtjrt|jjtj|_ntdd|jD|_| |_ | |_|r|dkrt|d d}|r|dkrt|d d}||_| |_dS) NZ SCIPY_XSLOW0FiicSsg|]}t|tqSrr=)r?argrrrrAsz'MpmathData.__init__..zr!)r.osenvirongetr scipy_func mpmath_funcarg_specdpsprecrrtolatolignore_inf_signnan_okr>rrB issubdtypedtypecomplexfloating is_complexanydistinguish_nan_and_infgetattrname param_filter)rrUrVrWrerXrYrrZr[r\rcr]rfZis_xslowrrrrs6      zMpmathData.__init__c sVtjdtjj}tjjtjj }}zjdk rDjg}ndg}j dk r^j tj_ t |j tj r|t ddnddddt|D]\}}|tj_zBtjfdd |d jjjjjjd Wq:Wqtk r6|t|d kr2t\}}} |j| k r.|| |YqXqW5||tj_tj_ XdS) NicSstt|Sr&)mpmathmpcr4r2rrrmptypesz MpmathData.check..mptypecSstt|Sr&)rhZmpfr rjrrrrkscSs0t|jddt|jkr"tjSt|jSdS)NgؗҜ.pytypecsjt|Sr&)rVmap)r rkrorrrz"MpmathData.check..F)Z vectorizedrZr[r\rcr]rfr)rrandomseedrLrWrrhmprXrYr^r_r` mpc2complex enumeraterrUrZr[r\rcr]rfAssertionErrorrDsysexc_info __traceback__with_traceback) rZargarrZold_dpsZold_precZdps_listjrXtpvaluetbrrqrchecksH         zMpmathData.checkcCs"|jrd|jfSd|jfSdS)Nzz)rare)rrrr__repr__s zMpmathData.__repr__) NNNNrNrOFTTN)r!r"r#rrrrrrrrMs ';rMcOst||}|dSr&)rMr)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 rZ min_fixedZ max_fixed)r rhnstrrjrrrrn+s rncCstt|jt|jSr&)r4rnr+r,rjrrrrw7srwcsddfdd}|S)NcSs t|tjrt|St|SdSr&)r>rhrir4r rjrrrtofloat<s ztrace_args..tofloatc sbtjdtt|ftjz||}tjd|W5tjdtjX|S)Nz%r:  z-> %r)rzstderrwritetuplerpflush)r rrrrrrwrapBs    ztrace_args..wraprrrrrr trace_args;s r setitimerFc@s eZdZdS) TimeoutErrorN)r!r"r#rrrrrWsrrTcs4tr"|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_handlerksz%time_limited..sigalrm_handlercsfdd}|S)Nc srttj}ttjz2z||WW"Stk rJYWSXW5ttjdttj|XdS)Nr)signalSIGALRMr ITIMER_REALr)r rZ old_handler)r return_valrtimeoutrrros(time_limited..deco..wraprrrrrrrdecons ztime_limited..decocsfdd}|S)Nc sptfddtz.deco..wrap..trace)rrzsettracerr r)rrr)rrrr|s  rrr)rrrrr{s)POSIX)rrZ use_sigalrmrrrr time_limited[s  rcsfdd}|S)z9Decorate function to return nan if it raises an exceptioncs,z ||WStk r&tjYSXdSr&) Exceptionrrmrrrrrs zexception_to_nan..wraprrrrrexception_to_nans rcsfdd}|S)z1Decorate function to return nan if it returns infcs||}t|stjS|Sr&)risfiniterm)r rr<rrrrs  zinf_to_nan..wraprrrrr inf_to_nans rFFgsB    o  j  4