B 0dM8@sTddlZddlZddlZddlZddlmZddlZddlm Z y ddl Z Wne k r`YnXGdddZ GdddZ Gdd d ZGd d d Zd d ZGdddZddZddZddZddZddZy ddlZddlZdeekZWne k rdZYnXGdddeZdejdfd 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 bgg?) ValueErrornpinfZfinfofloatmaxab inclusive_a inclusive_b)selfr r r r rL/var/www/html/venv/lib/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)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 r rr) rrr r Zn1Zn2ZposptsZnegptsrrrrvalues^s6     z Arg.valuesN) __name__ __module__ __qualname____doc__rrrrrrrrrrs 8rc@seZdZddZddZdS)FixedArgcCst||_dS)N)rasarray_values)rrrrrrszFixedArg.__init__cCs|jS)N)r%)rrrrrrszFixedArg.valuesN)rr r!rrrrrrr#sr#c@s:eZdZeej ej eejejfddZddZdS) ComplexArgcCs$t|j|j|_t|j|j|_dS)N)rrealimag)rr r rrrrszComplexArg.__init__cCsXttt|}|j|}|j|d}|dddfd|dddfS)Nry?)intrfloorsqrtr'rr(Zravel)rrmxyrrrrs zComplexArg.valuesN)rr r!complexrrrrrrrrr&s&r&c@seZdZdddZddZdS) IntArgcCs||_||_dS)N)r r )rr r rrrrszIntArg.__init__cCsht|j|jtd|d|dt}tdd}t tj ||f}|||jk||jk@}|S)Nrr) rr r rr astyper)rZarangeuniqueZr_)rrZv1Zv2vrrrrs , z IntArg.valuesN)r1r2)rr r!rrrrrrr0s r0cCst|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)r9r:r,rrrr;sr)r8rndarraycopylenr$sumr5r)ziparrayZbroadcast_arraysZix_ZreshapeT)Zargspecrargsnargsmsrrrget_argss   rFc @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)r8r&)r9argrrrr;sz'MpmathData.__init__..zr)r)osenvirongetr scipy_func mpmath_funcarg_specdpsprecrrtolatolignore_inf_signnan_okr8rr< issubdtypedtypecomplexfloating is_complexanydistinguish_nan_and_infgetattrname param_filter)rrOrPrQr_rRrSrrTrUrVr]rWr`Zis_xslowrrrrs6      zMpmathData.__init__c sTtjdtjj}tjjtjj }}z jdk rDjg}ndg}j dk r^j tj_ t |j tj r|t ddnddddxt|D]\}}|tj_y.mptypecSstt|S)N)rbZmpfr)r-rrrrdscSs0t|jddt|jkr"tjSt|jSdS)NgؗҜ.pytypecsjt|S)N)rPmap)r )rdrhrrrz"MpmathData.check..F)Z vectorizedrTrUrVr]rWr`r)rrandomseedrFrQrrbmprRrSrXrYrZ mpc2complex enumeraterrOrTrUrVr]rWr`AssertionErrorr>sysexc_info __traceback__with_traceback) rZargarrZold_dpsZold_precZdps_listjrRtpvaluetbr)rdrhrrchecksD         zMpmathData.checkcCs"|jrd|jfSd|jfSdS)Nzz)r[r_)rrrr__repr__s zMpmathData.__repr__) NNNNrHrIFTTN)rr r!rrzr{rrrrrGs  $;rGcOst||}|dS)N)rGrz)r kwdrrrassert_mpmath_equals r~cCstjjdd|S)Nz8 Test not yet functional (too slow), needs more work.)reason)pytestmarkskip)funcrrrnonfunctional_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 r) min_fixed max_fixed)rrbnstr)r-rrrrg*s rgcCstt|jt|jS)N)r/rgr'r()r-rrrro6srocsddfdd}|S)NcSs t|tjrt|St|SdS)N)r8rbrcr/r)r-rrrtofloat;s ztrace_args..tofloatc sbtjdtt|ftjz||}tjd|WdtjdtjX|S)Nz%r: z-> %r )rrstderrwritetupleriflush)r r|r)rrrrwrapAs    ztrace_args..wrapr)rrr)rrr trace_args:s r setitimerFc@s eZdZdS) TimeoutErrorN)rr r!rrrrrVsrg?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)signumframerrrsigalrm_handlerjsz%time_limited..sigalrm_handlercsfdd}|S)Nc sdttj}ttjz$y ||Stk r<SXWdttjdttj|XdS)Nr)signalSIGALRMr ITIMER_REALr)r r|Z old_handler)r return_valrtimeoutrrrns  z(time_limited..deco..wrapr)rr)rrr)rrdecoms ztime_limited..decocsfdd}|S)Nc sbtfddtz.y ||Stk rLtdSXWdtdXdS)NcstkrtS)N)timer)reventrK) start_timertracerrr~sz7time_limited..deco..wrap..trace)rrrsettracer)r r|)rrr)rrrr{s    z(time_limited..deco..wrapr)rr)rr)rrrzs)POSIX)rrZ use_sigalrmrr)rrrr time_limitedZs  rcsfdd}|S)z9Decorate function to return nan if it raises an exceptioncs&y ||Stk r tjSXdS)N) Exceptionrrf)r r|)rrrrs zexception_to_nan..wrapr)rrr)rrexception_to_nans rcsfdd}|S)z1Decorate function to return nan if it returns infcs||}t|stjS|S)N)risfiniterf)r r|r7)rrrrs  zinf_to_nan..wrapr)rrr)rr inf_to_nans rFFg TypeErrorlistrqrangerrbfabsappendr)rerrformatrjoin) resZstdrUrTrZfailureskndigitsmsgZresrepZstdrepZrdiffrrrmp_assert_allcloses6  4  r)rr)"rLrrrnumpyrZ numpy.testingrrZscipy.special._testutilsrrb ImportErrorrr#r&r0rFrGr~rrgrorposixrdirrrrrfrrrrrrrrs@   o  j  4