o &?e8@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[/home/www/facesmatcher.com/pyenv/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?)intrfloorsqrtr,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_)rrv1v2vrrrr s , z IntArg.valuesN)r7r8r*rrrrr6s  r6cCst|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 )rBrCr2rrrrDsr)rArndarraycopylenr'sumr;r/ziparrayZbroadcast_arraysZix_ZreshapeT)Zargspecrargsnargsmsrrrget_argss   rOc@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@)rBargrrrrDsz'MpmathData.__init__..zr!)r/osenvirongetr scipy_func mpmath_funcarg_specdpsprecrrtolatolignore_inf_signnan_okrArrE issubdtypedtypecomplexfloating is_complexanydistinguish_nan_and_infgetattrname param_filter)rrXrYrZrhr[r\rr]r^r_rfr`riZis_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 Wqty|t|d krt\}}} |j| ur|| |YqIwW||tj_tj_ dSW||tj_tj_ dS||tj_tj_ w) NicStt|Sr&)mpmathmpcr5r3rrrmptypez MpmathData.check..mptypecSrkr&)rlZmpfr rnrrrrorpcSs,t|jddt|jkrtjSt|jS)NgؗҜ.pytypecsjt|Sr&)rYmap)r rortrrrsz"MpmathData.check..F)Z vectorizedr]r^r_rfr`rir)rrandomseedrOrZrrlmpr[r\rarbrc mpc2complex enumeraterrXr]r^r_rfr`riAssertionErrorrGsysexc_info __traceback__with_traceback) rZargarrZold_dpsZold_precZdps_listjr[tpvaluetbrrvrchecksP         * zMpmathData.checkcCs"|jr d|jdSd|jdS)Nz >)rdrh)rrrr__repr__szMpmathData.__repr__) NNNNrQrRFTTN)r!r"r#rrrrrrrrPs ' ;rPcOst|i|}|dSr&)rPr)r kwdrrrassert_mpmath_equals rcCstjjdd|S)Nz8 Test not yet functional (too slow), needs more work.)reason)pytestmarkskipfuncrrrnonfunctional_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 rlnstrrnrrrrs+s rscCstt|jt|jSr&)r5rsr,r-rnrrrr{7sr{csddfdd}|S)NcSst|tjr t|St|Sr&)rArlrmr5r rnrrrtofloat<s ztrace_args..tofloatc sztjtt|dtjz|i|}tjd|Wtjdtj|Stjdtjw)Nz: z-> %r )r~stderrwritetupleruflush)r rrrrrrwrapBs    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)signumframerrrsigalrm_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)reventrT) start_timertracerrr~sz7time_limited..deco..wrap..trace)rr~settracerr 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&) Exceptionrrrrrrrrs   zexception_to_nan..wraprrrrrexception_to_nan rcr)z1Decorate function to return nan if it returns infcs"|i|}t|stjS|Sr&)risfiniterr)r rr?rrrrs zinf_to_nan..wraprrrrr inf_to_nanrrFFgsF     o  j  4