Datasetpublic

codekingpro/portable-devtools

sourceHugging Faceupdated 5mo agoView on Hugging Face

1likes14kdownloads

rootpythonuser_packagesPython313site-packagesnumpytypingtestsdatarevealfft.pyi

fft.pyi38 linesDownload Raw Back to reveal

1from typing import Any, assert_type
2
3import numpy as np
4import numpy.typing as npt
5
6AR_f8: npt.NDArray[np.float64]
7AR_c16: npt.NDArray[np.complex128]
8AR_LIKE_f8: list[float]
9
10assert_type(np.fft.fftshift(AR_f8), npt.NDArray[np.float64])
11assert_type(np.fft.fftshift(AR_LIKE_f8, axes=0), npt.NDArray[Any])
12
13assert_type(np.fft.ifftshift(AR_f8), npt.NDArray[np.float64])
14assert_type(np.fft.ifftshift(AR_LIKE_f8, axes=0), npt.NDArray[Any])
15
16assert_type(np.fft.fftfreq(5, AR_f8), npt.NDArray[np.floating])
17assert_type(np.fft.fftfreq(np.int64(), AR_c16), npt.NDArray[np.complexfloating])
18
19assert_type(np.fft.fftfreq(5, AR_f8), npt.NDArray[np.floating])
20assert_type(np.fft.fftfreq(np.int64(), AR_c16), npt.NDArray[np.complexfloating])
21
22assert_type(np.fft.fft(AR_f8), npt.NDArray[np.complex128])
23assert_type(np.fft.ifft(AR_f8, axis=1), npt.NDArray[np.complex128])
24assert_type(np.fft.rfft(AR_f8, n=None), npt.NDArray[np.complex128])
25assert_type(np.fft.irfft(AR_f8, norm="ortho"), npt.NDArray[np.float64])
26assert_type(np.fft.hfft(AR_f8, n=2), npt.NDArray[np.float64])
27assert_type(np.fft.ihfft(AR_f8), npt.NDArray[np.complex128])
28
29assert_type(np.fft.fftn(AR_f8), npt.NDArray[np.complex128])
30assert_type(np.fft.ifftn(AR_f8), npt.NDArray[np.complex128])
31assert_type(np.fft.rfftn(AR_f8), npt.NDArray[np.complex128])
32assert_type(np.fft.irfftn(AR_f8), npt.NDArray[np.float64])
33
34assert_type(np.fft.rfft2(AR_f8), npt.NDArray[np.complex128])
35assert_type(np.fft.ifft2(AR_f8), npt.NDArray[np.complex128])
36assert_type(np.fft.fft2(AR_f8), npt.NDArray[np.complex128])
37assert_type(np.fft.irfft2(AR_f8), npt.NDArray[np.float64])
38