ó ʽ÷Xc@`sÖddlmZmZmZddlZddlZddlZddlm Z m Z ddl m Z ddl mZyddlZWnek r™nXdefd„ƒYZdefd „ƒYZd efd „ƒYZd efd „ƒYZd„Zdefd„ƒYZd„Zd„Zd„Zd„Zd„Zy.ddlZddlZdeeƒkZ Wnek r…e!Z nXde"fd„ƒYZ#dej$e%d„Z&d„Z'd„Z(ddd„Z)dS(i(tdivisiontprint_functiontabsolute_importN(tdectassert_(treraise(tassert_func_equaltArgcB`s3eZdZej ejeed„Zd„ZRS(s† Generate a set of numbers on the real axis, concentrating on 'interesting' regions and covering all orders of magnitude. cC`s†||_||_||_||_|jtj krTtjtƒj d|_n|jtjkr‚tjtƒjd|_ndS(Ni( tatbt inclusive_at inclusive_btnptinftfinfotfloattmax(tselfRR R R ((s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pyt__init__s    c C`s1tdtd|ƒƒ}tdtd|ƒƒ}td|||ƒ}tjdd|ƒ}tjtjddtd |d ƒƒd d d df}|jd kr!|jd kr!tjtjddd|d ƒtjdtj|jƒd|d ƒf}tjddd|dƒ}nƒ|jd ko>|jknrtjtjddd|dƒtjdtj|jƒd|dƒtjddd|dƒ tjdtj|j ƒd|dƒ f}tjtjddd|d ƒtjddd|d ƒ f}nŒ|jd kr˜tjtjddd|d ƒ tjdtj|j ƒd|d ƒ f}tjddd|dƒ }n g}g}tj||||d f} |j rß| | |jk} n| | |jk} |j r| | |jk} n| | |jk} tj | ƒS(s7Return an array containing approximatively `n` numbers.ig333333Ó?gš™™™™™É?iiÿÿÿÿiiöÿÿÿi iii÷ÿÿÿgÀg@i iâÿÿÿi( RtintR tlinspacetr_RR tlogspacetlog10R R tunique( Rtntn1tn2tn3tv1tv2tv3tv4tv((s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pytvalues&sF%- "&/%/!  (t__name__t __module__t__doc__R R tTrueRR"(((s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pyRs tFixedArgcB`seZd„Zd„ZRS(cC`stj|ƒ|_dS(N(R tasarrayt_values(RR"((s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pyRVscC`s|jS(N(R)(RR((s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pyR"Ys(R#R$RR"(((s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pyR'Us t ComplexArgcB`s@eZeej ej ƒeejejƒd„Zd„ZRS(cC`s4t|j|jƒ|_t|j|jƒ|_dS(N(Rtrealtimag(RRR ((s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pyR^scC`sttdttj|ƒƒƒ}|jj|ƒ}|jj|ƒ}|dd…dfd|ddd…fjƒS(Niyð?( RRR tsqrtR+R"R,tNonetravel(RRtmtxty((s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pyR"bs(R#R$tcomplexR R RR"(((s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pyR*]s/tIntArgcB`s eZddd„Zd„ZRS(iüÿÿiècC`s||_||_dS(N(RR (RRR ((s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pyRjs cC`sŽt|j|jƒjtd|d|dƒƒjtƒ}tjddƒ}tj tj ||fƒ}|||jk||jk@}|S(Niiiiûÿÿÿ( RRR R"RtastypeRR tarangeRR(RRRRR!((s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pyR"ns < (R#R$RR"(((s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pyR4iscC`sõt|tjƒr!|jƒ}nÐt|ƒ}tjg|D]!}t|tƒrUdnd^q:ƒ}||t|ƒjt ƒd}g}x3t ||ƒD]"\}}|j |j |ƒƒq›Wtj tjtj|ŒŒƒj|dƒj}|S(Ngø?gð?iiÿÿÿÿ(t isinstanceR tndarraytcopytlenR(R*tsumR5RtziptappendR"tarraytbroadcast_arraystix_treshapetT(targspecRtargstnargstspectmsR0((s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pytget_argsvs 7!0t MpmathDatac B`sAeZddddddeeedd„ Zd„Zd„ZRS(iˆgH¯¼šò×z>gYóøÂn¥cC`s6||_||_||_||_||_||_||_| |_| |_| |_ t |jt j ƒrt j |jjt jƒ|_n.tg|jD]}t |tƒ^qƒ|_| |_| |_| sã|dkrøt|ddƒ}n| s |dkr t|ddƒ}n||_| |_dS(NsR#(t scipy_funct mpmath_functarg_spectdpstprecRtrtoltatoltignore_inf_signtnan_okR7R R8t issubdtypetdtypetcomplexfloatingt is_complextanyR*tdistinguish_nan_and_inftgetattrR.tnamet param_filter(RRJRKRLRZRMRNRRORPRQRXRRR[targ((s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pyR‡s*          !.   c`s°tjjdƒtˆjˆjƒ}tjjtjj }}zQˆjdk r_ˆjg}n dg}ˆj dk r‰ˆj tj_ ntj |j tj ƒr³t‰d„‰nd„‰d„‰xÆt|ƒD]¸\}}|tj_yctˆj‡‡‡fd†|dtdˆjd ˆjd ˆjd ˆjd ˆjd ˆjƒPWqÒtk r‰|t|ƒdkrŠttjƒŒqŠqÒXqÒWWd||tj_tj_ XdS(NiÒicS`stjt|ƒƒS(N(tmpmathtmpcR3(R1((s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pytmptype¸scS`stjt|ƒƒS(N(R]tmpfR(R1((s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pyR_»scS`s>t|jƒddt|jƒkr-tjSt|jƒSdS(Ng¼‰Ø—²ÒœÊst vectorizedRORPRQRXRRR[i(R trandomtseedRHRLRR]tmpRMRNR.RSRTRUt mpc2complext enumerateRRJtFalseRORPRQRXRRR[tAssertionErrorR:Rtsystexc_info(Rtargarrtold_dpstold_prectdps_listtjRM((R_RdRs9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pytcheck¢s>           cC`s)|jrd|jfSd|jfSdS(Nss(RVRZ(R((s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pyt__repr__Ùs N(R#R$R.RmR&RRvRw(((s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pyRI†s  7cO`st||Ž}|jƒdS(N(RIRv(Rtkwtd((s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pytassert_mpmath_equalàscC`stjtdƒ|ƒS(Ns8 Test not yet functional (too slow), needs more work.(RtskipifR&(tfunc((s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pytnonfunctional_tooslowåscC`s"ttj|dddddƒƒS(sá 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 it min_fixedit max_fixed(RR]tnstr(R1((s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pyRcís cC`stt|jƒt|jƒƒS(N(R3RcR+R,(R1((s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pyRkùsc`sd„‰‡‡fd†}|S(NcS`s*t|tjƒrt|ƒSt|ƒSdS(N(R7R]R^R3R(R1((s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pyttofloatþs c`stjjdttˆ|ƒƒfƒtjjƒz'ˆ||Ž}tjjd|ƒWdtjjdƒtjjƒX|S(Ns%r: s-> %rs (RotstderrtwritettupleRetflush(RRxtr(R|R(s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pytwraps& ((R|R‡((R|Rs9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pyt trace_argsýs  t setitimert TimeoutErrorcB`seZRS((R#R$(((s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pyRŠsgà?c`sCtr-|r-d„‰‡‡‡fd†}n‡‡fd†}|S(s 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. cS`s tƒ‚dS(N(RŠ(tsignumtframe((s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pytsigalrm_handler,sc`s‡‡‡‡fd†}|S(Nc`s€tjtjˆƒ}tjtjˆƒz*yˆ||ŽSWntk rPˆSXWdtjtjdƒtjtj|ƒXdS(Ni(tsignaltSIGALRMR‰t ITIMER_REALRŠ(RRxt old_handler(R|t return_valRttimeout(s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pyR‡0s  ((R|R‡(R’RR“(R|s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pytdeco/s c`s‡‡‡fd†}|S(Nc`swtjƒ‰‡‡fd†}tj|ƒz7yˆ||ŽSWntk r`tjdƒˆSXWdtjdƒXdS(Nc`s&tjƒˆˆkr"tƒ‚ndS(N(ttimeRŠR.(RŒteventR\(t start_timeR“(s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pyttrace@s (R•RotsettraceRŠR.(RRxR˜(R|R’R“(R—s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pyR‡=s     ((R|R‡(R’R“(R|s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pyR”<s(tPOSIX(R“R’t use_sigalrmR”((R’RR“s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pyt time_limiteds    c`s‡fd†}|S(s9Decorate function to return nan if it raises an exceptionc`s-yˆ||ŽSWntk r(tjSXdS(N(t ExceptionR Rb(RRx(R|(s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pyR‡Rs ((R|R‡((R|s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pytexception_to_nanPsc`s‡fd†}|S(s1Decorate function to return nan if it returns infc`s)ˆ||Ž}tj|ƒs%tjS|S(N(R tisfiniteRb(RRxR!(R|(s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pyR‡\s((R|R‡((R|s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pyt inf_to_nanZsg—ÔFFõgttj||||ƒ||tj||ƒkƒWqntk rÒ|j|ƒqnXqnWt t t j |ƒƒƒ}dg}|jdj t|ƒ|ƒƒx¿|D]·}tj|||ddddƒ} tj|||ddddƒ} ||dkr„d} n5tj||||||ƒ} tj| dƒ} |jd j || | | ƒƒq!W|rûttd j|ƒƒnd S( s} Compare lists of mpmath.mpf's or mpmath.mpc's directly so that it can be done to higher precision than double. sLengths of inputs not equal.ts4Bad results ({} out of {}) for the following points:R~iRR is{}: {} != {} (rdiff {})s N(R:t TypeErrortlistRntrangeRR]tfabsR=RRaR RtformatR€Rmtjoin( treststdRPRORtfailurestktndigitstmsgtresreptstdreptrdiff((s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pytmp_assert_allcloseds6  >    "" ##(*t __future__RRRRoR•tnumpyR t numpy.testingRRtscipy._lib.sixRtscipy.special._testutilsRR]t ImportErrortobjectRR'R*R4RHRIRzR}RcRkRˆtposixRŽtdirRšRmRRŠRbR&RœRžR R±(((s9/tmp/pip-build-7oUkmx/scipy/scipy/special/_mptestutils.pyts@    ?  Z        4