HR-eddlZddlmZddlmZddlZddlmZddl m Z ddl m Z ddl mZgZd d gZdd Ze d d ddZe d dddZdZdS)N)inf)Number)Quantity) isiterable) deprecated)unitsinf_likevectorize_if_neededctjtStjfdt j_S)aVectorize a method of redshift(s). Parameters ---------- func : callable or None method to wrap. If `None` returns a :func:`functools.partial` with ``nin`` loaded. nin : int Number of positional redshift arguments. Returns ------- wrapper : callable :func:`functools.wraps` of ``func`` where the first ``nin`` arguments are converted from |Quantity| to :class:`numpy.ndarray`. N)nincd|dD}td|Dr|g||dRi|Sj|g||dRi|S)z :func:`functools.wraps` of ``func`` where the first ``nin`` arguments are converted from |Quantity| to `numpy.ndarray` or scalar. czg|]8}t|ts|n|tj9S) isinstancerto_valuecuredshift.0zs 7lib/python3.11/site-packages/astropy/cosmology/utils.py z>vectorize_redshift_method..wrapper..1sI    8,, IAA!**R[2I2I   Nc3XK|]%}t|ttjfV&dS)N)rrnpgenericrs r z=vectorize_redshift_method..wrapper..6s3??qz!fbj122??????r)all__vectorized__)selfargskwargszsfuncr wrappers rr&z*vectorize_redshift_method..wrapper)s  $3$Z    ??B??? ? ? :49r9DJ999&99 9%w%dGRG$stt*GGGGGGr) functoolspartialvectorize_redshift_methodwrapsr vectorizer )r%r r&s``@rr)r)s}$ | !:DDDD_THHHHHHH" \$//G Nrz5.0zSvectorize_if_needed has been removed because it constructs a new ufunc on each callz>> func = lambda x: x ** 2 >>> vectorize_if_needed(func, 2) 4 >>> vectorize_if_needed(func, [2, 3]) array([4, 9]) )anymaprrr+)fxvkws rr r AsF:*-SQ-?-?)@)@ K !2< ! !S ! !1 % %aaeKrz[inf_like has been removed because it duplicates functionality provided by numpy.full_like()z@Use numpy.full_like(z, numpy.inf) instead for a target array 'z'cztj|rtn tj|ttS)aLReturn the shape of x with value infinity and dtype='float'. Preserves 'shape' for both array and scalar inputs. But always returns a float array, even if x is of integer type. Parameters ---------- x : scalar or array-like Must work with functions `numpy.isscalar` and `numpy.full_like` (if `x` is not a scalar` Returns ------- `math.inf` or ndarray[float] thereof Returns a scalar `~math.inf` if `x` is a scalar, an array of floats otherwise. Examples -------- >>> inf_like(0.) # float scalar inf >>> inf_like(1) # integer scalar should give float output inf >>> inf_like([0., 1., 2., 3.]) # float list array([inf, inf, inf, inf]) >>> inf_like([0, 1, 2, 3]) # integer list should give float output array([inf, inf, inf, inf]) )dtype)risscalarr full_likefloat)r3s rr r as-J+a.. G33bl1c&G&G&GGrct|ttjfr|St |dr&t |dr|t jzjS|St|t jjS)z Redshift as a `~numbers.Number` or `~numpy.ndarray` / |Quantity| / |Column|. Allows for any ndarray ducktype by checking for attribute "shape". shapeunit) rrrrhasattrrrvaluer)rs raszarrr?sp !fbj)** G   1f   ,$+ + Ar{ # # ))r)Nr)r'mathrnumbersrnumpyr astropy.unitsr astropy.utilsrastropy.utils.decoratorsrr r__all____doctest_skip__r)r r r?rrrrIsK""""""$$$$$$//////  56****Z   O LLL0   6S HHH@ * * * * *r