Skip to content

Commit 898ef99

Browse files
committed
signal: type-test resample
1 parent 17afbfe commit 898ef99

File tree

1 file changed

+58
-0
lines changed

1 file changed

+58
-0
lines changed

tests/signal/test_resample.pyi

Lines changed: 58 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,58 @@
1+
# type-tests for `resample` from `signal/_signaltools.pyi`
2+
3+
from typing import assert_type
4+
5+
import numpy as np
6+
import optype.numpy as onp
7+
8+
from scipy.signal import resample
9+
10+
num: int
11+
12+
py_i_1d: list[int]
13+
py_i_2d: list[list[int]]
14+
py_f_1d: list[float]
15+
py_f_2d: list[list[float]]
16+
17+
i8_1d: onp.Array1D[np.int8]
18+
i8_2d: onp.Array2D[np.int8]
19+
20+
f16_1d: onp.Array1D[np.float16]
21+
f16_2d: onp.Array2D[np.float16]
22+
f32_1d: onp.Array1D[np.float32]
23+
f32_2d: onp.Array2D[np.float32]
24+
f64_1d: onp.Array1D[np.float64]
25+
f64_2d: onp.Array2D[np.float64]
26+
f80_1d: onp.Array1D[np.float128]
27+
f80_2d: onp.Array2D[np.float128]
28+
29+
c64_1d: onp.Array1D[np.complex64]
30+
c64_2d: onp.Array2D[np.complex64]
31+
c128_1d: onp.Array1D[np.complex128]
32+
c128_2d: onp.Array2D[np.complex128]
33+
c160_1d: onp.Array1D[np.complex256]
34+
c160_2d: onp.Array2D[np.complex256]
35+
36+
###
37+
38+
assert_type(resample(py_i_1d, num), onp.ArrayND[np.float64])
39+
assert_type(resample(py_f_1d, num), onp.ArrayND[np.float64])
40+
assert_type(resample(i8_1d, num), onp.Array1D[np.float64])
41+
assert_type(resample(f16_1d, num), onp.Array1D[np.float32])
42+
assert_type(resample(f32_1d, num), onp.Array1D[np.float32])
43+
assert_type(resample(f64_1d, num), onp.Array1D[np.float64])
44+
assert_type(resample(f80_1d, num), onp.Array1D[np.float128])
45+
assert_type(resample(c64_1d, num), onp.Array1D[np.complex64])
46+
assert_type(resample(c128_1d, num), onp.Array1D[np.complex128])
47+
assert_type(resample(c160_1d, num), onp.Array1D[np.complex256])
48+
49+
assert_type(resample(py_i_2d, num), onp.ArrayND[np.float64])
50+
assert_type(resample(py_f_2d, num), onp.ArrayND[np.float64])
51+
assert_type(resample(i8_2d, num), onp.Array2D[np.float64])
52+
assert_type(resample(f16_2d, num), onp.Array2D[np.float32])
53+
assert_type(resample(f32_2d, num), onp.Array2D[np.float32])
54+
assert_type(resample(f64_2d, num), onp.Array2D[np.float64])
55+
assert_type(resample(f80_2d, num), onp.Array2D[np.float128])
56+
assert_type(resample(c64_2d, num), onp.Array2D[np.complex64])
57+
assert_type(resample(c128_2d, num), onp.Array2D[np.complex128])
58+
assert_type(resample(c160_2d, num), onp.Array2D[np.complex256])

0 commit comments

Comments
 (0)