Skip to content

Commit 7b975ec

Browse files
committed
tests: Update test_waveforms
1 parent 7524442 commit 7b975ec

File tree

1 file changed

+15
-15
lines changed

1 file changed

+15
-15
lines changed

tests/signal/test_waveforms.pyi

+15-15
Original file line numberDiff line numberDiff line change
@@ -9,56 +9,56 @@ _Array_f8: TypeAlias = onp.ArrayND[np.float64]
99
_Falsy: TypeAlias = Literal[0, False]
1010
_Truthy: TypeAlias = Literal[1, True]
1111

12-
_time: onp.ToFloat | onp.ToFloatND
12+
_gauss_pulse_time: onp.ToFloat | onp.ToFloatND | Literal["cutoff"]
1313
_float: onp.ToFloat
1414
_truthy: _Truthy
1515
_falsy: _Falsy
1616

1717
# test gausspulse function overloads
18-
assert_type(gausspulse(_time), _Array_f8)
18+
assert_type(gausspulse(_gauss_pulse_time), _Array_f8)
1919
# Full positional arguments
20-
assert_type(gausspulse(_time, _float, _float, _float, _float, _falsy, _falsy), _Array_f8)
20+
assert_type(gausspulse(_gauss_pulse_time, _float, _float, _float, _float, _falsy, _falsy), _Array_f8)
2121
assert_type(
22-
gausspulse(_time, _float, _float, _float, _float, _truthy, _falsy),
22+
gausspulse(_gauss_pulse_time, _float, _float, _float, _float, _truthy, _falsy),
2323
tuple[_Array_f8, _Array_f8],
2424
)
2525
assert_type(
26-
gausspulse(_time, _float, _float, _float, _float, _falsy, _truthy),
26+
gausspulse(_gauss_pulse_time, _float, _float, _float, _float, _falsy, _truthy),
2727
tuple[_Array_f8, _Array_f8],
2828
)
2929
assert_type(
30-
gausspulse(_time, _float, _float, _float, _float, _truthy, _truthy),
30+
gausspulse(_gauss_pulse_time, _float, _float, _float, _float, _truthy, _truthy),
3131
tuple[_Array_f8, _Array_f8, _Array_f8],
3232
)
3333
# Full keyword arguments
34-
assert_type(gausspulse(t=_time), _Array_f8)
34+
assert_type(gausspulse(t=_gauss_pulse_time), _Array_f8)
3535
assert_type(
3636
gausspulse(
37-
t=_time,
37+
t=_gauss_pulse_time,
3838
retquad=_falsy,
3939
retenv=_falsy,
4040
),
4141
_Array_f8,
4242
)
4343
assert_type(
4444
gausspulse(
45-
t=_time,
45+
t=_gauss_pulse_time,
4646
retquad=_truthy,
4747
retenv=_falsy,
4848
),
4949
tuple[_Array_f8, _Array_f8],
5050
)
5151
assert_type(
5252
gausspulse(
53-
t=_time,
53+
t=_gauss_pulse_time,
5454
retquad=_falsy,
5555
retenv=_truthy,
5656
),
5757
tuple[_Array_f8, _Array_f8],
5858
)
5959
assert_type(
6060
gausspulse(
61-
t=_time,
61+
t=_gauss_pulse_time,
6262
retquad=_truthy,
6363
retenv=_truthy,
6464
),
@@ -67,18 +67,18 @@ assert_type(
6767

6868
# Mixed positional and keyword arguments
6969
assert_type(
70-
gausspulse(_time, _float, _float, _float, _float, retquad=_falsy, retenv=_falsy),
70+
gausspulse(_gauss_pulse_time, _float, _float, _float, _float, retquad=_falsy, retenv=_falsy),
7171
_Array_f8,
7272
)
7373
assert_type(
74-
gausspulse(_time, _float, _float, _float, _float, retquad=_truthy, retenv=_falsy),
74+
gausspulse(_gauss_pulse_time, _float, _float, _float, _float, retquad=_truthy, retenv=_falsy),
7575
tuple[_Array_f8, _Array_f8],
7676
)
7777
assert_type(
78-
gausspulse(_time, _float, _float, _float, _float, retquad=_falsy, retenv=_truthy),
78+
gausspulse(_gauss_pulse_time, _float, _float, _float, _float, retquad=_falsy, retenv=_truthy),
7979
tuple[_Array_f8, _Array_f8],
8080
)
8181
assert_type(
82-
gausspulse(_time, _float, _float, _float, _float, retquad=_truthy, retenv=_truthy),
82+
gausspulse(_gauss_pulse_time, _float, _float, _float, _float, retquad=_truthy, retenv=_truthy),
8383
tuple[_Array_f8, _Array_f8, _Array_f8],
8484
)

0 commit comments

Comments
 (0)