o EOfi8@sRddlZddlZddlZddlmZddlZddlmZddl Z ddl m Z zddl Z Wn e y5YnwGdddZGdddZGd d d ZGd d d Zd dZGdddZddZddZddZddZddZz ddlZdeevZWn e ydZYnwGdddeZdejd 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 krdttj}|tjkr#dttj}|||_|_|||_|_ dS)Nz#a should be less than or equal to bg?) ValueErrornpinfZfinfofloatmaxab inclusive_a inclusive_b)selfr r rrr:lib/python3.10/site-packages/scipy/special/_mptestutils.py__init__s  z Arg.__init__cCs|dkrtd|ddkr|d}|}n|d}|d}|dkr/tt|t||}n|dkr?|dkr?t|||}n|dkr^tj|d|dd}tdt||}t||f}nw|dkr|dkrtd||}|jdkrzt|d} nd} tjd| |dd}t||f}nH|ddkr|d} | } n|d} | d} tjdd|dd}|jdkrt|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}}||krt|S|js|d7}|js|d7}|ddkr+|d}|}n|d}|d}|dkrA||||}g}n)|dkrRg}|| | | }n|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 zFixedArg.valuesNr!r"r#rr rrrrr%s r%c@s:eZdZeej ej eejejfddZddZdS) ComplexArgcCs$t|j|j|_t|j|j|_dSr&)rrealimagrr r rrrrszComplexArg.__init__cCsXttt|}|j|}|j|d}|dddfd|dddfS)Nry?)intrZfloorZsqrtr,r r-Zravel)rrmxyrrrr s (zComplexArg.valuesN)r!r"r#complexrr rr rrrrr+s& r+c@seZdZdddZddZdS) IntArgcCs||_||_dSr&)r r r.rrrrs zIntArg.__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)r5r6r*rrrrr4s  r4cCst|tjr |}|St|}tdd|D}||t|td}ddt ||D}t tj tj | |dj}|S)NcSsg|] }t|tr dndqS)g?g? isinstancer+).0specrrr szget_args..rcSsg|] \}}||qSr)r )r>r?r0rrrr@sr)r=rndarraycopylenr'sumr9r/zipZarrayZbroadcast_arraysZix_ZreshapeT)Zargspecrargsnargsmsrrrget_argss    rJc@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=rrA issubdtypedtypecomplexfloating is_complexanydistinguish_nan_and_infgetattrname param_filter)rrSrTrUrcrVrWrrXrYrZrar[rdZis_xslowrrrrs<       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&)mpmathmpcr3r1rrrmptypez MpmathData.check..mptypecSrfr&)rgZmpfr rirrrrjrkcSs,t|jddt|jkrtjSt|jS)NgؗҜ.pytypecsjt|Sr&)rTmap)r rjrorrr sz"MpmathData.check..F)Z vectorizedrXrYrZrar[rdr)rZrandomZseedrJrUrrgZmprVrWr\r]r^ mpc2complex enumeraterrSrXrYrZrar[rdAssertionErrorrCsysexc_info __traceback__with_traceback) rZargarrZold_dpsZold_precZdps_listjrVtpvaluetbrrqrchecksT         * zMpmathData.checkcCs"|jr d|jdSd|jdS)Nz >)r_rc)rrrr__repr__szMpmathData.__repr__) NNNNrLrMFTTN)r!r"r#rr~rrrrrrKs ) >rKcOst|i|}|dSr&)rKr~)r kwdrrrassert_mpmath_equal%s rcCstjjdd|S)Nz8 Test not yet functional (too slow), needs more work.)reason)pytestZmarkskipfuncrrrnonfunctional_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 rgnstrrirrrrn4s rncCstt|jt|jSr&)r3rnr,r-rirrrrs@srscsddfdd}|S)NcSst|tjr t|St|Sr&)r=rgrhr3r rirrrtofloatEs ztrace_args..tofloatc sztjtt|dtjz|i|}tjd|Wtjdtj|Stjdtjw)N: z-> %r )rvstderrwritetuplerpflush)r rrrrrrwrapKs    ztrace_args..wraprrrrrr trace_argsDs r setitimerFc@s eZdZdS) TimeoutErrorN)r!r"r#rrrrr_srrTcs6tr|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_handlersr)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_valrtimeoutrrrws (time_limited..deco..wraprrrrrrrdecovs ztime_limited..decocsfdd}|S)Nc sztfddtz%z|i|WWtdSty6tdYWtdSwtdw)Ncstkr tSr&)timer)rZeventrO) start_timertracerrrsz7time_limited..deco..wrap..trace)rrvsettracerr r)rrr)rrrrs     rrr)rrrrrs)POSIX)rrZ use_sigalrmrrrr time_limitedcs rcfdd}|S)z9Decorate function to return nan if it raises an exceptioncs*z|i|WStytjYSwr&) Exceptionrrmrrrrrs   zexception_to_nan..wraprrrrrexception_to_nan rcr)z1Decorate function to return nan if it returns infcs"|i|}t|stjS|Sr&)rZisfiniterm)r rr;rrrrs zinf_to_nan..wraprrrrr inf_to_nanrrFFg\}}}tj || ddd } tj || ddd } |dkrvd } nt|||} t | d } | |d | d | d| dqXt dd | dSdS)z| Compare lists of mpmath.mpf's or mpmath.mpc's directly so that it can be done to higher precision than double. Nz,Lengths of inputs res and std are not equal.rz Bad results (z out of rz) for the following points:rr rz != z (rdiff )Fr)rtrrrgZfabsappendrCr/rlrrrrjoin)resZstdrYrXZfailureskZresvalZstdvalZnfailZndigitsmsgZresrepZstdrepZrdiffrrrmp_assert_allcloses,  $r)rr)#rPrvr itertoolsrZnumpyrZ numpy.testingrrZscipy.special._testutilsrrg ImportErrorrr%r+r4rJrKrrrnrsrrdirrrrrmrrrrrrrrsF     o  o   4