ó î&]\c@`sèddlmZmZmZddlZddlZddlZddlZddl m Z ddl Z ddl m Z ddlmZyddlZWnek r«nXdefd„ƒYZdefd „ƒYZd efd „ƒYZd efd „ƒYZd„Zdefd„ƒYZd„Zd„Zd„Zd„Zd„Zy.ddlZddlZde eƒkZ!Wnek r—e"Z!nXde#fd„ƒYZ$dej%e&d„Z'd„Z(d„Z)ddd„Z*dS(i(tdivisiontprint_functiontabsolute_importN(tassert_(treraise(tassert_func_equaltArgcB`s<eZdZej ejeed„Zd„Zd„ZRS(s‚Generate a set of numbers on the real axis, concentrating on 'interesting' regions and covering all orders of magnitude. cC`s–||krtdƒ‚n|tj krDdtjtƒj}n|tjkrldtjtƒj}n|||_|_|||_|_ dS(Ns#a should be less than or equal to bgà¿gà?( t ValueErrortnptinftfinfotfloattmaxtatbt inclusive_at inclusive_b(tselfR RRR((s9lib/python2.7/site-packages/scipy/special/_mptestutils.pyt__init__s cC`s{|dkrtdƒ‚n|ddkr>|d}|}n|d}|d}|dkrˆtjtj|ƒtj|ƒ|ƒ}næ|dkr¸|dkr¸tj|||ƒ}n¶|dkrtj|d|dtƒ}tjdtj|ƒ|ƒ}tj||fƒ}nY|dkr |dkr tjd||ƒ}|jdkrgtj|dƒ} nd} tjd| |dtƒ}tj||fƒ}nÎ|ddkrÃ|d} | } n|d} | d} tjdd|dtƒ}|jdkrtj|dƒ} nd} tjd| | dtƒ} tjdtj|ƒ| ƒ} tj| || fƒ}tj|ƒS(Nisa should be positiveiii tendpointiâÿÿÿ( RRtlogspacetlog10tlinspacetFalsethstacktsizetsort(RR Rtntnlogptstnlinptstptstlinptstlogptstrighttnlogpts1tnlogpts2tlogpts1tlogpts2((s9lib/python2.7/site-packages/scipy/special/_mptestutils.pyt_positive_values)sF      *     c C`sr|j|j}}||kr,tj|ƒS|jsB|d7}n|jsX|d7}n|ddkr{|d}|}n|d}|d}|dkr¹|j|||ƒ}g}ng|dkræg}|j| | |ƒ }n:|jd||ƒ}|jd| |dƒ }|d}tj|ddd…|fƒ}|jsX|d}n|jsn|d }n|S(s%Return an array containing n numbers.iiiNiÿÿÿÿ(R RRtzerosRRR&R( RRR Rtn1tn2tposptstnegptsR((s9lib/python2.7/site-packages/scipy/special/_mptestutils.pytvaluesas6              "    ( t__name__t __module__t__doc__RR tTrueRR&R,(((s9lib/python2.7/site-packages/scipy/special/_mptestutils.pyRs 8tFixedArgcB`seZd„Zd„ZRS(cC`stj|ƒ|_dS(N(Rtasarrayt_values(RR,((s9lib/python2.7/site-packages/scipy/special/_mptestutils.pyRˆscC`s|jS(N(R3(RR((s9lib/python2.7/site-packages/scipy/special/_mptestutils.pyR,‹s(R-R.RR,(((s9lib/python2.7/site-packages/scipy/special/_mptestutils.pyR1‡s t ComplexArgcB`s@eZeej ej ƒeejejƒd„Zd„ZRS(cC`s4t|j|jƒ|_t|j|jƒ|_dS(N(Rtrealtimag(RR R((s9lib/python2.7/site-packages/scipy/special/_mptestutils.pyRscC`sxttjtj|ƒƒƒ}|jj|ƒ}|jj|dƒ}|dd…dfd|ddd…fjƒS(Niyð?( tintRtfloortsqrtR5R,R6tNonetravel(RRtmtxty((s9lib/python2.7/site-packages/scipy/special/_mptestutils.pyR,”s(R-R.tcomplexRR RR,(((s9lib/python2.7/site-packages/scipy/special/_mptestutils.pyR4s/tIntArgcB`s eZddd„Zd„ZRS(iüÿÿiècC`s||_||_dS(N(R R(RR R((s9lib/python2.7/site-packages/scipy/special/_mptestutils.pyRœs cC`sŽt|j|jƒjtd|d|dƒƒjtƒ}tjddƒ}tj tj ||fƒ}|||jk||jk@}|S(Niiiiûÿÿÿ( RR RR,R tastypeR7Rtarangetuniquetr_(RRtv1tv2tv((s9lib/python2.7/site-packages/scipy/special/_mptestutils.pyR, s < (R-R.RR,(((s9lib/python2.7/site-packages/scipy/special/_mptestutils.pyR@›scC`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 isinstanceRtndarraytcopytlenR2R4tsumRAR7tziptappendR,tarraytbroadcast_arraystix_treshapetT(targspecRtargstnargstspectmsR<((s9lib/python2.7/site-packages/scipy/special/_mptestutils.pytget_args¨s 7!0t MpmathDatac B`sAeZddddddeeedd„ Zd„Zd„ZRS(gH¯¼šò×z>gYóøÂn¥cC`s|dkrZyttjjddƒƒ}Wntk rDt}nX|rQdnd}n||_||_||_ ||_ ||_ ||_ ||_ | |_| |_| |_t|j tjƒrêtj|j jtjƒ|_n.tg|j D]}t|tƒ^q÷ƒ|_| |_| |_| s=|dkrRt|ddƒ}n| se|dkrzt|ddƒ}n||_| |_dS(Nt SCIPY_XSLOWt0iˆiôsR-(R:R7tostenvirontgetRRt scipy_funct mpmath_functarg_spectdpstprecRtrtoltatoltignore_inf_signtnan_okRHRRIt issubdtypetdtypetcomplexfloatingt is_complextanyR4tdistinguish_nan_and_inftgetattrtnamet param_filter(RR`RaRbRpRcRdRReRfRgRnRhRqtis_xslowtarg((s9lib/python2.7/site-packages/scipy/special/_mptestutils.pyR¹s6             !.   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(tmpmathtmpcR?(R=((s9lib/python2.7/site-packages/scipy/special/_mptestutils.pytmptypeöscS`stjt|ƒƒS(N(RttmpfR (R=((s9lib/python2.7/site-packages/scipy/special/_mptestutils.pyRvùscS`s>t|jƒddt|jƒkr-tjSt|jƒSdS(Ng¼‰Ø—²Òœst vectorizedReRfRgRnRhRqi(RtrandomtseedRYRbRRttmpRcRdR:RiRjRkt mpc2complext enumerateRR`RReRfRgRnRhRqtAssertionErrorRKRtsystexc_info(Rtargarrtold_dpstold_prectdps_listtjRc((RvR{Rs9lib/python2.7/site-packages/scipy/special/_mptestutils.pytcheckàs>           cC`s)|jrd|jfSd|jfSdS(Nss(RlRp(R((s9lib/python2.7/site-packages/scipy/special/_mptestutils.pyt__repr__s N(R-R.R:RR0RRŒR(((s9lib/python2.7/site-packages/scipy/special/_mptestutils.pyRZ¸s $ 7cO`st||Ž}|jƒdS(N(RZRŒ(R tkwtd((s9lib/python2.7/site-packages/scipy/special/_mptestutils.pytassert_mpmath_equalscC`stjjddƒ|ƒS(Ntreasons8 Test not yet functional (too slow), needs more work.(tpytesttmarktskip(tfunc((s9lib/python2.7/site-packages/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(R Rttnstr(R=((s9lib/python2.7/site-packages/scipy/special/_mptestutils.pyRz+s cC`stt|jƒt|jƒƒS(N(R?RzR5R6(R=((s9lib/python2.7/site-packages/scipy/special/_mptestutils.pyR‚7sc`sd„‰‡‡fd†}|S(NcS`s*t|tjƒrt|ƒSt|ƒSdS(N(RHRtRuR?R (R=((s9lib/python2.7/site-packages/scipy/special/_mptestutils.pyttofloat<s c`stjjdttˆ|ƒƒfƒtjjƒz'ˆ||Ž}tjjd|ƒWdtjjdƒtjjƒX|S(Ns%r: s-> %rs (R…tstderrtwritettupleR|tflush(R RŽtr(R•Rš(s9lib/python2.7/site-packages/scipy/special/_mptestutils.pytwrapBs& ((R•R ((R•Ršs9lib/python2.7/site-packages/scipy/special/_mptestutils.pyt trace_args;s  t setitimert TimeoutErrorcB`seZRS((R-R.(((s9lib/python2.7/site-packages/scipy/special/_mptestutils.pyR£Wsgà?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((s9lib/python2.7/site-packages/scipy/special/_mptestutils.pytsigalrm_handlerksc`s‡‡‡‡fd†}|S(Nc`s€tjtjˆƒ}tjtjˆƒz*yˆ||ŽSWntk rPˆSXWdtjtjdƒtjtj|ƒXdS(Ni(tsignaltSIGALRMR¢t ITIMER_REALR£(R RŽt old_handler(R•t return_valR¦ttimeout(s9lib/python2.7/site-packages/scipy/special/_mptestutils.pyR os  ((R•R (R«R¦R¬(R•s9lib/python2.7/site-packages/scipy/special/_mptestutils.pytdecons c`s‡‡‡fd†}|S(Nc`sztjƒ‰‡‡‡fd†‰tjˆƒz7yˆ||ŽSWntk rctjdƒˆSXWdtjdƒXdS(Nc`s&tjƒˆˆkr"tƒ‚nˆS(N(ttimeR£(R¥teventRs(t start_timeR¬ttrace(s9lib/python2.7/site-packages/scipy/special/_mptestutils.pyR±s (R®R…tsettraceR£R:(R RŽ(R•R«R¬(R°R±s9lib/python2.7/site-packages/scipy/special/_mptestutils.pyR |s     ((R•R (R«R¬(R•s9lib/python2.7/site-packages/scipy/special/_mptestutils.pyR­{s(tPOSIX(R¬R«t use_sigalrmR­((R«R¦R¬s9lib/python2.7/site-packages/scipy/special/_mptestutils.pyt time_limited[s    c`s‡fd†}|S(s9Decorate function to return nan if it raises an exceptionc`s-yˆ||ŽSWntk r(tjSXdS(N(t ExceptionRRy(R RŽ(R•(s9lib/python2.7/site-packages/scipy/special/_mptestutils.pyR ‘s ((R•R ((R•s9lib/python2.7/site-packages/scipy/special/_mptestutils.pytexception_to_nansc`s‡fd†}|S(s1Decorate function to return nan if it returns infc`s)ˆ||Ž}tj|ƒs%tjS|S(N(RtisfiniteRy(R RŽRG(R•(s9lib/python2.7/site-packages/scipy/special/_mptestutils.pyR ›s((R•R ((R•s9lib/python2.7/site-packages/scipy/special/_mptestutils.pyt inf_to_nan™sg—Ô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—iR˜R is{}: {} != {} (rdiff {})s N(RKt TypeErrortlistR„trangeRRttfabsRNR7RxRRtformatR™Rtjoin( treststdRfReRtfailurestktndigitstmsgtresreptstdreptrdiff((s9lib/python2.7/site-packages/scipy/special/_mptestutils.pytmp_assert_allclose£s6  >    "" ##(+t __future__RRRR]R…R®tnumpyRt numpy.testingRR’tscipy._lib.sixRtscipy.special._testutilsRRtt ImportErrortobjectRR1R4R@RYRZRR–RzR‚R¡tposixR§tdirR³RR¶R£RyR0RµR·R¹RÊ(((s9lib/python2.7/site-packages/scipy/special/_mptestutils.pytsD      o  f        4