U Hv f8@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 rrr>/tmp/pip-unpacked-wheel-96ln3f52/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 szget_args..rcSsg|]\}}||qSr)r!)r@rAr2rrrrBsr)r?rndarraycopylenr(sumr;r/ziparrayZbroadcast_arraysZix_ZreshapeT)Zargspecrargsnargsmsrrrget_argss   rMc @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@argrrrrBsz'MpmathData.__init__..zr")r/osenvirongetr scipy_func mpmath_funcarg_specdpsprecrrtolatolignore_inf_signnan_okr?rrC issubdtypedtypecomplexfloating is_complexanydistinguish_nan_and_infgetattrname param_filter)rrVrWrXrfrYrZrr[r\r]rdr^rgZis_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')mpmathmpcr5r3rrrmptypesz MpmathData.check..mptypecSstt|Sr')riZmpfr rkrrrrlscSs0t|jddt|jkr"tjSt|jSdS)NgؗҜ.pytypecsjt|Sr')rWmap)r rlrprrrz"MpmathData.check..F)Z vectorizedr[r\r]rdr^rgr)rrandomseedrMrXrrimprYrZr_r`ra mpc2complex enumeraterrVr[r\r]rdr^rgAssertionErrorrEsysexc_info __traceback__with_traceback) rZargarrZold_dpsZold_precZdps_listjrYtpvaluetbrrrrchecksH         zMpmathData.checkcCs"|jrd|jfSd|jfSdS)Nzz)rbrf)rrrr__repr__s zMpmathData.__repr__) NNNNrOrPFTTN)r"r#r$rrrrrrrrNs ';rNcOst||}|dSr')rNr)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 rinstrrkrrrro+s rocCstt|jt|jSr')r5ror,r-rkrrrrx7srxcsddfdd}|S)NcSs t|tjrt|St|SdSr')r?rirjr5r rkrrrtofloat<s ztrace_args..tofloatc sbtjdtt|ftjz||}tjd|W5tjdtjX|S)Nz%r:  z-> %r)r{stderrwritetuplerqflush)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)rr{settracerr 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') Exceptionrrnrrrrrs zexception_to_nan..wraprrrrrexception_to_nans rcsfdd}|S)z1Decorate function to return nan if it returns infcs||}t|stjS|Sr')risfinitern)r rr=rrrrs  zinf_to_nan..wraprrrrr inf_to_nans rFFgsB    o  j  4