a j.:@sTddlZddlZddlZddlmZddlZddlmZddl Z ddl m Z z ddl Z Wne yjYn0GdddZGdddZGd d d ZGd d d Zd dZGdddZddZddZddZddZddZzddlZdeevZWne ydZYn0GdddeZdejd fd!d"Zd#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@E:\soccer\ml_env\Lib\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 zget_args..rcSsg|]\}}||qSr)r )r?r@r1rrrrArBr)r>rndarraycopylenr'sumr:r.ziparrayZbroadcast_arraysZix_ZreshapeT)Zargspecrargsnargsmsrrrget_argss    rMc @s&eZdZd ddZdd Zd d ZdS) MpmathDataNHz>YnFTcCs|durBzttjdd}Wnty4d}Yn0|r>dnd}||_||_||_||_||_ ||_ ||_ | |_ | |_ | |_t|jtjrt|jjtj|_ntdd|jD|_| |_ | |_|r|dkrt|d d}|r|dkrt|d d}||_| |_dS) NZ SCIPY_XSLOW0FiicSsg|]}t|tqSrr=)r?argrrrrArBz'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_xslowrrrrs:       zMpmathData.__init__c sftjdtjj}tjjtjj }}zjdurDjg}ndg}j dur^j tj_ t |j tj r|t ddnddddt|D]\}}|tj_zBtjfdd |d jjjjjjd Wq8Wqty4|t|d kr0t\}}} |j| ur,|| |Yq0qW||tj_tj_ n||tj_tj_ 0dS) NicSstt|Sr&)mpmathmpcr4r2rrrmptypesz MpmathData.check..mptypecSstt|Sr&)riZmpfr rkrrrrlscSs0t|jddt|jkr"tjSt|jSdS)NgؗҜ.pytypecsjt|Sr&)rWmap)r rlrprrr rBz"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_listjrYtpvaluetbrrrrchecksJ         zMpmathData.checkcCs&|jrd|jdSd|jdSdS)Nz >)rbrf)rrrr__repr__szMpmathData.__repr__) NNNNrOrPFTTN)r!r"r#rrrrrrrrNs )>rNcOst|i|}|dSr&)rNr)r kwdrrrassert_mpmath_equal%srcCstjjdd|S)Nz8 Test not yet functional (too slow), needs more work.)reason)pytestmarkskipfuncrrrnonfunctional_tooslow*s rcCsttj|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 rinstrrkrrrro4s rocCstt|jt|jSr&)r4ror+r,rkrrrrw@srwcsddfdd}|S)NcSs t|tjrt|St|SdSr&)r>rirjr4r rkrrrtofloatEs ztrace_args..tofloatc s|tjtt|dtjz8|i|}tjd|Wtjdtjntjdtj0|S)N: z-> %r )rzstderrwritetuplerqflush)r rrrrrrwrapKs    ztrace_args..wraprrrrrr trace_argsDs r setitimerFc@s eZdZdS) TimeoutErrorN)r!r"r#rrrrr_srrTcs4tr"|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)Zsignumframerrrsigalrm_handlerssz%time_limited..sigalrm_handlercsfdd}|S)Nc sttj}ttjzz.|i|WWttjdttj|StyYWttjdttj|S0Wttjdttj|nttjdttj|0dS)Nr)signalZSIGALRMrZ ITIMER_REALr)r rZ old_handler)r return_valrtimeoutrrrws  (time_limited..deco..wraprrrrrrrdecovs ztime_limited..decocsfdd}|S)Nc stfddtzXz|i|WWtdStyltdYWtdS0Wtdn td0dS)NcstkrtSr&)timer)reventrR) start_timertracerrrsz7time_limited..deco..wrap..trace)rrzsettracerr r)rrr)rrrrs     rrr)rrrrrs)POSIX)rrZ use_sigalrmrrrr time_limitedcs  rcsfdd}|S)z9Decorate function to return nan if it raises an exceptioncs.z|i|WSty(tjYS0dSr&) Exceptionrrnrrrrrs zexception_to_nan..wraprrrrrexception_to_nans rcsfdd}|S)z1Decorate function to return nan if it returns infcs"|i|}t|stjS|Sr&)risfinitern)r rr<rrrrs zinf_to_nan..wraprrrrr inf_to_nans rFFgs@     o  o   4