a ߙfbY@sddlZddlmZddlmZddlZddlmZddl m Z ddl m Z ddl mZgZd d gZdd d Ze d ddddd Ze d ddddd ZddZdS)N)inf)Number)Quantity) isiterable) deprecated)unitsinf_likevectorize_if_neededcs@durtjtdStfddt_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)nincsndd|dD}tdd|DrJ|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. cSs&g|]}t|ts|n |tjqS) isinstancerZto_valuecuredshift.0zr r 6lib/python3.9/site-packages/astropy/cosmology/utils.py 1sz>vectorize_redshift_method..wrapper..Ncss|]}t|ttjfVqdS)N)r rnpgenericrr r r 4z=vectorize_redshift_method..wrapper..)all__vectorized__)selfargskwargsZzsfuncr wrapperr rr )s  "z*vectorize_redshift_method..wrapper) functoolspartialvectorize_redshift_methodwrapsr vectorizer)rr r rrr#s  r#z5.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%)fxZvkwr r rr ?sz[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'cCst|rtStj|ttdS)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]) )Zdtype)rZisscalarrZ full_likefloat)r*r r rr \s"cCsFt|ttjfr|St|dr8t|dr4|tj>jS|St|tjjS)z Redshift as a `~numbers.Number` or `~numpy.ndarray` / |Quantity| / |Column|. Allows for any ndarray ducktype by checking for attribute "shape". shapeZunit) r rrrhasattrrrvaluer)rr r raszarrs   r/)Nr)r!ZmathrZnumbersrZnumpyrZ astropy.unitsrZ astropy.utilsrZastropy.utils.decoratorsrrr__all__Z__doctest_skip__r#r r r/r r r rs.       +