summaryrefslogtreecommitdiff
path: root/numpy/fft/__init__.pyi
diff options
context:
space:
mode:
Diffstat (limited to 'numpy/fft/__init__.pyi')
-rw-r--r--numpy/fft/__init__.pyi43
1 files changed, 24 insertions, 19 deletions
diff --git a/numpy/fft/__init__.pyi b/numpy/fft/__init__.pyi
index 648b0bf79..510e576d3 100644
--- a/numpy/fft/__init__.pyi
+++ b/numpy/fft/__init__.pyi
@@ -2,25 +2,30 @@ from typing import Any, List
from numpy._pytesttester import PytestTester
+from numpy.fft._pocketfft import (
+ fft as fft,
+ ifft as ifft,
+ rfft as rfft,
+ irfft as irfft,
+ hfft as hfft,
+ ihfft as ihfft,
+ rfftn as rfftn,
+ irfftn as irfftn,
+ rfft2 as rfft2,
+ irfft2 as irfft2,
+ fft2 as fft2,
+ ifft2 as ifft2,
+ fftn as fftn,
+ ifftn as ifftn,
+)
+
+from numpy.fft.helper import (
+ fftshift as fftshift,
+ ifftshift as ifftshift,
+ fftfreq as fftfreq,
+ rfftfreq as rfftfreq,
+)
+
__all__: List[str]
__path__: List[str]
test: PytestTester
-
-def fft(a, n=..., axis=..., norm=...): ...
-def ifft(a, n=..., axis=..., norm=...): ...
-def rfft(a, n=..., axis=..., norm=...): ...
-def irfft(a, n=..., axis=..., norm=...): ...
-def hfft(a, n=..., axis=..., norm=...): ...
-def ihfft(a, n=..., axis=..., norm=...): ...
-def fftn(a, s=..., axes=..., norm=...): ...
-def ifftn(a, s=..., axes=..., norm=...): ...
-def rfftn(a, s=..., axes=..., norm=...): ...
-def irfftn(a, s=..., axes=..., norm=...): ...
-def fft2(a, s=..., axes=..., norm=...): ...
-def ifft2(a, s=..., axes=..., norm=...): ...
-def rfft2(a, s=..., axes=..., norm=...): ...
-def irfft2(a, s=..., axes=..., norm=...): ...
-def fftshift(x, axes=...): ...
-def ifftshift(x, axes=...): ...
-def fftfreq(n, d=...): ...
-def rfftfreq(n, d=...): ...