1  from typing import Any, TypeVar, overload
 2  
 3  from numpy import generic, integer, floating, complexfloating
 4  from numpy._typing import (
 5      NDArray,
 6      ArrayLike,
 7      _ShapeLike,
 8      _ArrayLike,
 9      _ArrayLikeFloat_co,
10      _ArrayLikeComplex_co,
11  )
12  
13  _SCT = TypeVar("_SCT", bound=generic)
14  
15  __all__: list[str]
16  
17  @overload
18  def fftshift(x: _ArrayLike[_SCT], axes: None | _ShapeLike = ...) -> NDArray[_SCT]: ...
19  @overload
20  def fftshift(x: ArrayLike, axes: None | _ShapeLike = ...) -> NDArray[Any]: ...
21  
22  @overload
23  def ifftshift(x: _ArrayLike[_SCT], axes: None | _ShapeLike = ...) -> NDArray[_SCT]: ...
24  @overload
25  def ifftshift(x: ArrayLike, axes: None | _ShapeLike = ...) -> NDArray[Any]: ...
26  
27  @overload
28  def fftfreq(
29      n: int | integer[Any],
30      d: _ArrayLikeFloat_co,
31  ) -> NDArray[floating[Any]]: ...
32  @overload
33  def fftfreq(
34      n: int | integer[Any],
35      d: _ArrayLikeComplex_co,
36  ) -> NDArray[complexfloating[Any, Any]]: ...
37  
38  @overload
39  def rfftfreq(
40      n: int | integer[Any],
41      d: _ArrayLikeFloat_co,
42  ) -> NDArray[floating[Any]]: ...
43  @overload
44  def rfftfreq(
45      n: int | integer[Any],
46      d: _ArrayLikeComplex_co,
47  ) -> NDArray[complexfloating[Any, Any]]: ...