o EbM8@sNddlZddlZddlZddlZddlmZddlZddlm Z zddl Z Wn e y/YnwGdddZ GdddZ Gdd d ZGd d d Zd d ZGdddZddZddZddZddZddZzddlZddlZdeevZWn e ydZYnwGdddeZdejdfd 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 krdttj}|tjkr#dttj}|||_|_|||_|_ dS)Nz#a should be less than or equal to bg?) ValueErrornpinfZfinfofloatmaxab inclusive_a inclusive_b)selfr r r rrszget_args..rcSsg|] \}}||qSr)r)r=r>r/rrrr?sr)r<rndarraycopylenr&sumr8r.zipZarrayZbroadcast_arraysZix_ZreshapeT)Zargspecrargsnargsmsrrrget_argss   rIc@s.eZdZ    d ddZdd Zd d ZdS) MpmathDataNHz>YnFTcCs|dur!z ttjdd}Wn tyd}Ynw|rdnd}||_||_||_||_||_ ||_ ||_ | |_ | |_ | |_t|jtjrQt|jjtj|_n tdd|jD|_| |_ | |_|rh|dkrnt|d d}|rt|dkrzt|d d}||_| |_dS) NZ SCIPY_XSLOW0FiicSsg|]}t|tqSrr;)r=argrrrr?sz'MpmathData.__init__..zr )r.osenvirongetr scipy_func mpmath_funcarg_specdpsprecrrtolatolignore_inf_signnan_okr<rr@ issubdtypedtypecomplexfloating is_complexanydistinguish_nan_and_infgetattrname param_filter)rrRrSrTrbrUrVrrWrXrYr`rZrcZis_xslowrrrrs8       zMpmathData.__init__c sltjdtjj}tjjtjj }}zjdur!jg}ndg}j dur.j tj_ t |j tj r=t ddnddddt|D]J\}}|tj_ztjfdd |d jjjjjjd Wn0ty|t|d krt\}}} |j| ur|| |YqIwW||tj_tj_ dSW||tj_tj_ dS||tj_tj_ w) NicStt|Sr%)mpmathmpcr2r0rrrmptypez MpmathData.check..mptypecSrer%)rfZmpfr rhrrrrirjcSs,t|jddt|jkrtjSt|jS)NgؗҜ.pytypecsjt|Sr%)rSmap)r rirnrrrsz"MpmathData.check..F)Z vectorizedrWrXrYr`rZrcr)rZrandomZseedrIrTrrfZmprUrVr[r\r] mpc2complex enumeraterrRrWrXrYr`rZrcAssertionErrorrBsysexc_info __traceback__with_traceback) rZargarrZold_dpsZold_precZdps_listjrUtpvaluetbrrprchecksP         * zMpmathData.checkcCs|jr d|jfSd|jfS)Nzz)r^rb)rrrr__repr__s  zMpmathData.__repr__) NNNNrKrLFTTN)r r!r"rr}r~rrrrrJs ' ;rJcOst|i|}|dSr%)rJr})r kwdrrrassert_mpmath_equals rcCstjjdd|S)Nz8 Test not yet functional (too slow), needs more work.)reason)pytestZmarkskipfuncrrrnonfunctional_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 rfnstrrhrrrrm*s rmcCstt|jt|jSr%)r2rmr+r,rhrrrrr6srrcsddfdd}|S)NcSst|tjr t|St|Sr%)r<rfrgr2r rhrrrtofloat;s ztrace_args..tofloatc sztjdtt|ftjz|i|}tjd|Wtjdtj|Stjdtjw)Nz%r: z-> %r )rustderrwritetupleroflush)r rrrrrrwrapAs    ztrace_args..wraprrrrrr trace_args:s r setitimerFc@s eZdZdS) TimeoutErrorN)r r!r"rrrrrVsrrTcs6tr|rddfdd}|Sfdd}|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. cSstr%)r)Zsignumframerrrsigalrm_handlerjr(z%time_limited..sigalrm_handlercsfdd}|S)Nc sttj}ttjz2z|i|WWttjdttj|Sty@YWttjdttj|Swttjdttj|w)Nr)signalSIGALRMr ITIMER_REALr)r rZ old_handler)r return_valrtimeoutrrrns (time_limited..deco..wraprrrrrrrdecoms ztime_limited..decocsfdd}|S)Nc sztfddtz%z|i|WWtdSty6tdYWtdSwtdw)Ncstkr tSr%)timer)rZeventrN) start_timertracerrr~sz7time_limited..deco..wrap..trace)rrusettracerr r)rrr)rrrr{s     rrr)rrrrrzs)POSIX)rrZ use_sigalrmrrrr time_limitedZs rcfdd}|S)z9Decorate function to return nan if it raises an exceptioncs*z|i|WStytjYSwr%) Exceptionrrlrrrrrs   zexception_to_nan..wraprrrrrexception_to_nan rcr)z1Decorate function to return nan if it returns infcs"|i|}t|stjS|Sr%)rZisfiniterl)r rr:rrrrs zinf_to_nan..wraprrrrr inf_to_nanrrFFg     4    r)rr)"rOrurZnumpyrZ numpy.testingrrZscipy.special._testutilsrrf ImportErrorrr$r*r3rIrJrrrmrrrposixrdirrrrrlrrrrrrrrsF    o  j  4