From 43870a452303e848d9fd9431ab82b628cea98d50 Mon Sep 17 00:00:00 2001
From: mhostetter <matthostetter@gmail.com>
Date: Fri, 19 Jul 2024 17:13:02 -0400
Subject: [PATCH] Use `scipy.signal.windows.get_window()` for FIR design

---
 src/sdr/_filter/_design_fir.py     | 164 +++++++----------------------
 tests/dsp/fir/test_bandpass_fir.py |   8 +-
 tests/dsp/fir/test_bandstop_fir.py |  20 ++--
 tests/dsp/fir/test_highpass_fir.py |  34 +++---
 tests/dsp/fir/test_lowpass_fir.py  |  20 ++--
 5 files changed, 78 insertions(+), 168 deletions(-)

diff --git a/src/sdr/_filter/_design_fir.py b/src/sdr/_filter/_design_fir.py
index e8ae9c6ef..acfdc0b86 100644
--- a/src/sdr/_filter/_design_fir.py
+++ b/src/sdr/_filter/_design_fir.py
@@ -71,50 +71,11 @@ def _ideal_bandstop(order: int, center_freq: float, bandwidth: float) -> npt.NDA
     return h_ideal
 
 
-# TODO: Replace with scipy.signal.windows.get_window()
-def _window(
-    order: int,
-    window: Literal["hamming", "hann", "blackman", "blackman-harris", "chebyshev", "kaiser"]
-    | npt.ArrayLike
-    | None = None,
-    atten: float = 60,
-) -> npt.NDArray[np.float64]:
-    if window is None:
-        h_window = np.ones(order + 1)
-    elif isinstance(window, str):
-        if window == "hamming":
-            h_window = scipy.signal.windows.hamming(order + 1)
-        elif window == "hann":
-            h_window = scipy.signal.windows.hann(order + 1)
-        elif window == "blackman":
-            h_window = scipy.signal.windows.blackman(order + 1)
-        elif window == "blackman-harris":
-            h_window = scipy.signal.windows.blackmanharris(order + 1)
-        elif window == "chebyshev":
-            h_window = scipy.signal.windows.chebwin(order + 1, at=atten)
-        elif window == "kaiser":
-            beta = scipy.signal.kaiser_beta(atten)
-            h_window = scipy.signal.windows.kaiser(order + 1, beta=beta)
-        else:
-            raise ValueError(
-                f"Argument 'window' must be in ['hamming', 'hann', 'blackman', 'blackman-harris', 'chebyshev', 'kaiser'], not {window!r}."
-            )
-    else:
-        h_window = np.asarray(window)
-        if not h_window.shape == (order + 1,):
-            raise ValueError(f"Argument 'window' must be a length-{order + 1} vector, not {h_window.shape}.")
-
-    return h_window
-
-
 @export
 def lowpass_fir(
     order: int,
     cutoff_freq: float,
-    window: None
-    | Literal["hamming", "hann", "blackman", "blackman-harris", "chebyshev", "kaiser"]
-    | npt.ArrayLike = "hamming",
-    atten: float = 60,
+    window: str | float | tuple | None = "hamming",
 ) -> npt.NDArray[np.float64]:
     r"""
     Designs a lowpass FIR filter impulse response $h[n]$ using the window method.
@@ -122,18 +83,8 @@ def lowpass_fir(
     Arguments:
         order: The filter order $N$. Must be even.
         cutoff_freq: The cutoff frequency $f_c$, normalized to the Nyquist frequency $f_s / 2$.
-        window: The time-domain window to use.
-
-            - `None`: No windowing. Equivalently, a length-$N + 1$ vector of ones.
-            - `"hamming"`: Hamming window, see :func:`scipy.signal.windows.hamming`.
-            - `"hann"`: Hann window, see :func:`scipy.signal.windows.hann`.
-            - `"blackman"`: Blackman window, see :func:`scipy.signal.windows.blackman`.
-            - `"blackman-harris"`: Blackman-Harris window, see :func:`scipy.signal.windows.blackmanharris`.
-            - `"chebyshev"`: Chebyshev window, see :func:`scipy.signal.windows.chebwin`.
-            - `"kaiser"`: Kaiser window, see :func:`scipy.signal.windows.kaiser`.
-            - `npt.ArrayLike`: A custom window. Must be a length-$N + 1$ vector.
-
-        atten: The sidelobe attenuation in dB. Only used if `window` is `"chebyshev"` or `"kaiser"`.
+        window: The SciPy window definition. See :func:`scipy.signal.windows.get_window` for details.
+            If `None`, no window is applied.
 
     Returns:
         The filter impulse response $h[n]$ with length $N + 1$. The center of the passband has 0 dB gain.
@@ -162,9 +113,9 @@ def lowpass_fir(
 
             h_hann = sdr.lowpass_fir(100, 0.2, window="hann"); \
             h_blackman = sdr.lowpass_fir(100, 0.2, window="blackman"); \
-            h_blackman_harris = sdr.lowpass_fir(100, 0.2, window="blackman-harris"); \
-            h_chebyshev = sdr.lowpass_fir(100, 0.2, window="chebyshev"); \
-            h_kaiser = sdr.lowpass_fir(100, 0.2, window="kaiser")
+            h_blackman_harris = sdr.lowpass_fir(100, 0.2, window="blackmanharris"); \
+            h_chebyshev = sdr.lowpass_fir(100, 0.2, window=("chebwin", 60)); \
+            h_kaiser = sdr.lowpass_fir(100, 0.2, window=("kaiser", 0.5))
 
             @savefig sdr_lowpass_fir_3.png
             plt.figure(); \
@@ -181,11 +132,10 @@ def lowpass_fir(
     """
     verify_scalar(order, int=True, even=True)
     verify_scalar(cutoff_freq, float=True, inclusive_min=0, inclusive_max=1)
-    verify_scalar(atten, float=True, non_negative=True)
 
-    h_ideal = _ideal_lowpass(order, cutoff_freq)
-    h_window = _window(order, window, atten=atten)
-    h = h_ideal * h_window
+    h = _ideal_lowpass(order, cutoff_freq)
+    if window is not None:
+        h *= scipy.signal.windows.get_window(window, h.size, fftbins=False)
     h = _normalize_passband(h, 0)
 
     return h
@@ -195,10 +145,7 @@ def lowpass_fir(
 def highpass_fir(
     order: int,
     cutoff_freq: float,
-    window: None
-    | Literal["hamming", "hann", "blackman", "blackman-harris", "chebyshev", "kaiser"]
-    | npt.ArrayLike = "hamming",
-    atten: float = 60,
+    window: str | float | tuple | None = "hamming",
 ) -> npt.NDArray[np.float64]:
     r"""
     Designs a highpass FIR filter impulse response $h[n]$ using the window method.
@@ -206,18 +153,8 @@ def highpass_fir(
     Arguments:
         order: The filter order $N$. Must be even.
         cutoff_freq: The cutoff frequency $f_c$, normalized to the Nyquist frequency $f_s / 2$.
-        window: The time-domain window to use.
-
-            - `None`: No windowing. Equivalently, a length-$N + 1$ vector of ones.
-            - `"hamming"`: Hamming window, see :func:`scipy.signal.windows.hamming`.
-            - `"hann"`: Hann window, see :func:`scipy.signal.windows.hann`.
-            - `"blackman"`: Blackman window, see :func:`scipy.signal.windows.blackman`.
-            - `"blackman-harris"`: Blackman-Harris window, see :func:`scipy.signal.windows.blackmanharris`.
-            - `"chebyshev"`: Chebyshev window, see :func:`scipy.signal.windows.chebwin`.
-            - `"kaiser"`: Kaiser window, see :func:`scipy.signal.windows.kaiser`.
-            - `npt.ArrayLike`: A custom window. Must be a length-$N + 1$ vector.
-
-        atten: The sidelobe attenuation in dB. Only used if `window` is `"chebyshev"` or `"kaiser"`.
+        window: The SciPy window definition. See :func:`scipy.signal.windows.get_window` for details.
+            If `None`, no window is applied.
 
     Returns:
         The filter impulse response $h[n]$ with length $N + 1$. The center of the passband has 0 dB gain.
@@ -246,9 +183,9 @@ def highpass_fir(
 
             h_hann = sdr.highpass_fir(100, 0.7, window="hann"); \
             h_blackman = sdr.highpass_fir(100, 0.7, window="blackman"); \
-            h_blackman_harris = sdr.highpass_fir(100, 0.7, window="blackman-harris"); \
-            h_chebyshev = sdr.highpass_fir(100, 0.7, window="chebyshev"); \
-            h_kaiser = sdr.highpass_fir(100, 0.7, window="kaiser")
+            h_blackman_harris = sdr.highpass_fir(100, 0.7, window="blackmanharris"); \
+            h_chebyshev = sdr.highpass_fir(100, 0.7, window=("chebwin", 60)); \
+            h_kaiser = sdr.highpass_fir(100, 0.7, window=("kaiser", 0.5))
 
             @savefig sdr_highpass_fir_3.png
             plt.figure(); \
@@ -265,11 +202,10 @@ def highpass_fir(
     """
     verify_scalar(order, int=True, even=True)
     verify_scalar(cutoff_freq, float=True, inclusive_min=0, inclusive_max=1)
-    verify_scalar(atten, float=True, non_negative=True)
 
-    h_ideal = _ideal_highpass(order, cutoff_freq)
-    h_window = _window(order, window, atten=atten)
-    h = h_ideal * h_window
+    h = _ideal_highpass(order, cutoff_freq)
+    if window is not None:
+        h *= scipy.signal.windows.get_window(window, h.size, fftbins=False)
     h = _normalize_passband(h, 1)
 
     return h
@@ -280,10 +216,7 @@ def bandpass_fir(
     order: int,
     center_freq: float,
     bandwidth: float,
-    window: None
-    | Literal["hamming", "hann", "blackman", "blackman-harris", "chebyshev", "kaiser"]
-    | npt.ArrayLike = "hamming",
-    atten: float = 60,
+    window: str | float | tuple | None = "hamming",
 ) -> npt.NDArray[np.float64]:
     r"""
     Designs a bandpass FIR filter impulse response $h[n]$ using the window method.
@@ -292,18 +225,8 @@ def bandpass_fir(
         order: The filter order $N$. Must be even.
         center_freq: The center frequency $f_{center}$, normalized to the Nyquist frequency $f_s / 2$.
         bandwidth: The two-sided bandwidth about $f_{center}$, normalized to the Nyquist frequency $f_s / 2$.
-        window: The time-domain window to use.
-
-            - `None`: No windowing. Equivalently, a length-$N + 1$ vector of ones.
-            - `"hamming"`: Hamming window, see :func:`scipy.signal.windows.hamming`.
-            - `"hann"`: Hann window, see :func:`scipy.signal.windows.hann`.
-            - `"blackman"`: Blackman window, see :func:`scipy.signal.windows.blackman`.
-            - `"blackman-harris"`: Blackman-Harris window, see :func:`scipy.signal.windows.blackmanharris`.
-            - `"chebyshev"`: Chebyshev window, see :func:`scipy.signal.windows.chebwin`.
-            - `"kaiser"`: Kaiser window, see :func:`scipy.signal.windows.kaiser`.
-            - `npt.ArrayLike`: A custom window. Must be a length-$N + 1$ vector.
-
-        atten: The sidelobe attenuation in dB. Only used if `window` is `"chebyshev"` or `"kaiser"`.
+        window: The SciPy window definition. See :func:`scipy.signal.windows.get_window` for details.
+            If `None`, no window is applied.
 
     Returns:
         The filter impulse response $h[n]$ with length $N + 1$. The center of the passband has 0 dB gain.
@@ -333,9 +256,9 @@ def bandpass_fir(
 
             h_hann = sdr.bandpass_fir(100, 0.4, 0.1, window="hann"); \
             h_blackman = sdr.bandpass_fir(100, 0.4, 0.1, window="blackman"); \
-            h_blackman_harris = sdr.bandpass_fir(100, 0.4, 0.1, window="blackman-harris"); \
-            h_chebyshev = sdr.bandpass_fir(100, 0.4, 0.1, window="chebyshev"); \
-            h_kaiser = sdr.bandpass_fir(100, 0.4, 0.1, window="kaiser")
+            h_blackman_harris = sdr.bandpass_fir(100, 0.4, 0.1, window="blackmanharris"); \
+            h_chebyshev = sdr.bandpass_fir(100, 0.4, 0.1, window=("chebwin", 60)); \
+            h_kaiser = sdr.bandpass_fir(100, 0.4, 0.1, window=("kaiser", 0.5))
 
             @savefig sdr_bandpass_fir_3.png
             plt.figure(); \
@@ -353,11 +276,10 @@ def bandpass_fir(
     verify_scalar(order, int=True, even=True)
     verify_scalar(center_freq, float=True, inclusive_min=0, inclusive_max=1)
     verify_scalar(bandwidth, float=True, inclusive_min=0, inclusive_max=2 * min(center_freq, 1 - center_freq))
-    verify_scalar(atten, float=True, non_negative=True)
 
-    h_ideal = _ideal_bandpass(order, center_freq, bandwidth)
-    h_window = _window(order, window, atten=atten)
-    h = h_ideal * h_window
+    h = _ideal_bandpass(order, center_freq, bandwidth)
+    if window is not None:
+        h *= scipy.signal.windows.get_window(window, h.size, fftbins=False)
     h = _normalize_passband(h, center_freq)
 
     return h
@@ -368,10 +290,7 @@ def bandstop_fir(
     order: int,
     center_freq: float,
     bandwidth: float,
-    window: None
-    | Literal["hamming", "hann", "blackman", "blackman-harris", "chebyshev", "kaiser"]
-    | npt.ArrayLike = "hamming",
-    atten: float = 60,
+    window: str | float | tuple | None = "hamming",
 ) -> npt.NDArray[np.float64]:
     r"""
     Designs a bandstop FIR filter impulse response $h[n]$ using the window method.
@@ -380,18 +299,8 @@ def bandstop_fir(
         order: The filter order $N$. Must be even.
         center_freq: The center frequency $f_{center}$, normalized to the Nyquist frequency $f_s / 2$.
         bandwidth: The two-sided bandwidth about $f_{center}$, normalized to the Nyquist frequency $f_s / 2$.
-        window: The time-domain window to use.
-
-            - `None`: No windowing. Equivalently, a length-$N + 1$ vector of ones.
-            - `"hamming"`: Hamming window, see :func:`scipy.signal.windows.hamming`.
-            - `"hann"`: Hann window, see :func:`scipy.signal.windows.hann`.
-            - `"blackman"`: Blackman window, see :func:`scipy.signal.windows.blackman`.
-            - `"blackman-harris"`: Blackman-Harris window, see :func:`scipy.signal.windows.blackmanharris`.
-            - `"chebyshev"`: Chebyshev window, see :func:`scipy.signal.windows.chebwin`.
-            - `"kaiser"`: Kaiser window, see :func:`scipy.signal.windows.kaiser`.
-            - `npt.ArrayLike`: A custom window. Must be a length-$N + 1$ vector.
-
-        atten: The sidelobe attenuation in dB. Only used if `window` is `"chebyshev"` or `"kaiser"`.
+        window: The SciPy window definition. See :func:`scipy.signal.windows.get_window` for details.
+            If `None`, no window is applied.
 
     Returns:
         The filter impulse response $h[n]$ with length $N + 1$. The center of the larger passband has 0 dB gain.
@@ -421,9 +330,9 @@ def bandstop_fir(
 
             h_hann = sdr.bandstop_fir(100, 0.4, 0.75, window="hann"); \
             h_blackman = sdr.bandstop_fir(100, 0.4, 0.75, window="blackman"); \
-            h_blackman_harris = sdr.bandstop_fir(100, 0.4, 0.75, window="blackman-harris"); \
-            h_chebyshev = sdr.bandstop_fir(100, 0.4, 0.75, window="chebyshev"); \
-            h_kaiser = sdr.bandstop_fir(100, 0.4, 0.75, window="kaiser")
+            h_blackman_harris = sdr.bandstop_fir(100, 0.4, 0.75, window="blackmanharris"); \
+            h_chebyshev = sdr.bandstop_fir(100, 0.4, 0.75, window=("chebwin", 60)); \
+            h_kaiser = sdr.bandstop_fir(100, 0.4, 0.75, window=("kaiser", 0.5))
 
             @savefig sdr_bandstop_fir_3.png
             plt.figure(); \
@@ -441,11 +350,10 @@ def bandstop_fir(
     verify_scalar(order, int=True, even=True)
     verify_scalar(center_freq, float=True, inclusive_min=0, inclusive_max=1)
     verify_scalar(bandwidth, float=True, inclusive_min=0, inclusive_max=2 * min(center_freq, 1 - center_freq))
-    verify_scalar(atten, float=True, non_negative=True)
 
-    h_ideal = _ideal_bandstop(order, center_freq, bandwidth)
-    h_window = _window(order, window, atten=atten)
-    h = h_ideal * h_window
+    h = _ideal_bandstop(order, center_freq, bandwidth)
+    if window is not None:
+        h *= scipy.signal.windows.get_window(window, h.size, fftbins=False)
     if center_freq > 0.5:
         h = _normalize_passband(h, 0)
     else:
diff --git a/tests/dsp/fir/test_bandpass_fir.py b/tests/dsp/fir/test_bandpass_fir.py
index 7ab196218..514714750 100644
--- a/tests/dsp/fir/test_bandpass_fir.py
+++ b/tests/dsp/fir/test_bandpass_fir.py
@@ -191,7 +191,7 @@ def test_blackman_harris():
         >> h = designBandpassFIR(FilterOrder=30, CenterFrequency=0.4, Bandwidth=0.25, Window="blackman-harris");
         >> transpose(h)
     """
-    h = sdr.bandpass_fir(30, 0.4, 0.25, window="blackman-harris")
+    h = sdr.bandpass_fir(30, 0.4, 0.25, window="blackmanharris")
     h_truth = np.array(
         [
             -0.000001060796132,
@@ -236,7 +236,7 @@ def test_chebyshev():
         >> h = designBandpassFIR(FilterOrder=30, CenterFrequency=0.4, Bandwidth=0.25, Window="chebyshev");
         >> transpose(h)
     """
-    h = sdr.bandpass_fir(30, 0.4, 0.25, window="chebyshev")
+    h = sdr.bandpass_fir(30, 0.4, 0.25, window=("chebwin", 60))
     h_truth = np.array(
         [
             -0.000309113441909,
@@ -281,8 +281,8 @@ def test_kaiser():
         >> h = designBandpassFIR(FilterOrder=30, CenterFrequency=0.4, Bandwidth=0.25, Window="kaiser");
         >> transpose(h)
     """
-    # MATLAB uses beta=0.5 for Kaiser window. Attenuation of 21.542 dB was reverse engineered.
-    h = sdr.bandpass_fir(30, 0.4, 0.25, window="kaiser", atten=21.542)
+    # MATLAB uses beta=0.5 for Kaiser window
+    h = sdr.bandpass_fir(30, 0.4, 0.25, window=("kaiser", 0.5))
     h_truth = np.array(
         [
             -0.016765450319470,
diff --git a/tests/dsp/fir/test_bandstop_fir.py b/tests/dsp/fir/test_bandstop_fir.py
index f4c25f77c..e3b689997 100644
--- a/tests/dsp/fir/test_bandstop_fir.py
+++ b/tests/dsp/fir/test_bandstop_fir.py
@@ -47,7 +47,7 @@ def test_custom():
             0.017963811098946,
         ]
     )
-    verify_impulse_response(h, h_truth, atol=1e-1)
+    verify_impulse_response(h, h_truth, atol=1e-1)  # TODO: These aren't exactly identical
 
 
 def test_hamming():
@@ -92,7 +92,7 @@ def test_hamming():
             0.001295920763505,
         ]
     )
-    verify_impulse_response(h, h_truth, atol=1e-3)
+    verify_impulse_response(h, h_truth, atol=1e-3)  # TODO: These aren't exactly identical
 
 
 def test_hann():
@@ -137,7 +137,7 @@ def test_hann():
             0,
         ]
     )
-    verify_impulse_response(h, h_truth, atol=1e-2)
+    verify_impulse_response(h, h_truth, atol=1e-2)  # TODO: These aren't exactly identical
 
 
 def test_blackman():
@@ -182,7 +182,7 @@ def test_blackman():
             -0.000000000000000,
         ]
     )
-    verify_impulse_response(h, h_truth, atol=1e-2)
+    verify_impulse_response(h, h_truth, atol=1e-2)  # TODO: These aren't exactly identical
 
 
 def test_blackman_harris():
@@ -191,7 +191,7 @@ def test_blackman_harris():
         >> h = designBandstopFIR(FilterOrder=30, CenterFrequency=0.4, Bandwidth=0.25, Window="blackman-harris");
         >> transpose(h)
     """
-    h = sdr.bandstop_fir(30, 0.4, 0.25, window="blackman-harris")
+    h = sdr.bandstop_fir(30, 0.4, 0.25, window="blackmanharris")
     h_truth = np.array(
         [
             0.000001060796132,
@@ -227,7 +227,7 @@ def test_blackman_harris():
             0.000001060796132,
         ]
     )
-    verify_impulse_response(h, h_truth, atol=1e-1)
+    verify_impulse_response(h, h_truth, atol=1e-1)  # TODO: These aren't exactly identical
 
 
 def test_chebyshev():
@@ -236,7 +236,7 @@ def test_chebyshev():
         >> h = designBandstopFIR(FilterOrder=30, CenterFrequency=0.4, Bandwidth=0.25, Window="chebyshev");
         >> transpose(h)
     """
-    h = sdr.bandstop_fir(30, 0.4, 0.25, window="chebyshev")
+    h = sdr.bandstop_fir(30, 0.4, 0.25, window=("chebwin", 60))
     h_truth = np.array(
         [
             0.000309113441909,
@@ -272,7 +272,7 @@ def test_chebyshev():
             0.000309113441909,
         ]
     )
-    verify_impulse_response(h, h_truth, atol=1e-2)
+    verify_impulse_response(h, h_truth, atol=1e-2)  # TODO: These aren't exactly identical
 
 
 def test_kaiser():
@@ -281,7 +281,7 @@ def test_kaiser():
         >> h = designBandstopFIR(FilterOrder=30, CenterFrequency=0.4, Bandwidth=0.25, Window="kaiser");
         >> transpose(h)
     """
-    h = sdr.bandstop_fir(30, 0.4, 0.25, window="kaiser", atten=20)
+    h = sdr.bandstop_fir(30, 0.4, 0.25, window=("kaiser", 0.5))
     h_truth = np.array(
         [
             0.016765450319470,
@@ -317,4 +317,4 @@ def test_kaiser():
             0.016765450319470,
         ]
     )
-    verify_impulse_response(h, h_truth, atol=1e-1)
+    verify_impulse_response(h, h_truth, atol=1e-1)  # TODO: These aren't exactly identical
diff --git a/tests/dsp/fir/test_highpass_fir.py b/tests/dsp/fir/test_highpass_fir.py
index 5658ca293..f4626b7cf 100644
--- a/tests/dsp/fir/test_highpass_fir.py
+++ b/tests/dsp/fir/test_highpass_fir.py
@@ -1,4 +1,5 @@
 import numpy as np
+import scipy.signal
 
 import sdr
 
@@ -47,7 +48,7 @@ def test_custom():
             -0.000000000000000,
         ]
     )
-    verify_impulse_response(h, h_truth, atol=1e-2)
+    verify_impulse_response(h, h_truth, atol=1e-2)  # TODO: These aren't exactly identical
 
 
 def test_hamming():
@@ -92,7 +93,7 @@ def test_hamming():
             -0.000000000000000,
         ]
     )
-    verify_impulse_response(h, h_truth, atol=1e-2)
+    verify_impulse_response(h, h_truth, atol=1e-2)  # TODO: These aren't exactly identical
 
 
 def test_hann():
@@ -137,7 +138,7 @@ def test_hann():
             0,
         ]
     )
-    verify_impulse_response(h, h_truth, atol=1e-2)
+    verify_impulse_response(h, h_truth, atol=1e-2)  # TODO: These aren't exactly identical
 
 
 def test_blackman():
@@ -182,7 +183,7 @@ def test_blackman():
             0.000000000000000,
         ]
     )
-    verify_impulse_response(h, h_truth, atol=1e-2)
+    verify_impulse_response(h, h_truth, atol=1e-2)  # TODO: These aren't exactly identical
 
 
 def test_blackman_harris():
@@ -191,7 +192,7 @@ def test_blackman_harris():
         >> h = designHighpassFIR(FilterOrder=30, CutoffFrequency=0.6, Window="blackman-harris");
         >> transpose(h)
     """
-    h = sdr.highpass_fir(30, 0.6, window="blackman-harris")
+    h = sdr.highpass_fir(30, 0.6, window="blackmanharris")
     h_truth = np.array(
         [
             -0.000000000000000,
@@ -227,7 +228,7 @@ def test_blackman_harris():
             -0.000000000000000,
         ]
     )
-    verify_impulse_response(h, h_truth, atol=1e-2)
+    verify_impulse_response(h, h_truth, atol=1e-2)  # TODO: These aren't exactly identical
 
 
 def test_chebyshev():
@@ -236,7 +237,7 @@ def test_chebyshev():
         >> h = designHighpassFIR(FilterOrder=30, CutoffFrequency=0.6, Window="chebyshev");
         >> transpose(h)
     """
-    h = sdr.highpass_fir(30, 0.6, window="chebyshev")
+    h = sdr.highpass_fir(30, 0.6, window=("chebwin", 60))
     h_truth = np.array(
         [
             -0.000000000000000,
@@ -272,7 +273,7 @@ def test_chebyshev():
             -0.000000000000000,
         ]
     )
-    verify_impulse_response(h, h_truth, atol=1e-2)
+    verify_impulse_response(h, h_truth, atol=1e-2)  # TODO: These aren't exactly identical
 
 
 # NOTE: Added extra Kaiser window tests to reverse engineer MATLAB's beta parameter
@@ -284,7 +285,7 @@ def test_kaiser_0p2():
         >> h = designHighpassFIR(FilterOrder=30, CutoffFrequency=0.2, Window="kaiser");
         >> transpose(h)
     """
-    h = sdr.highpass_fir(30, 0.2, window="kaiser", atten=30)
+    h = sdr.highpass_fir(30, 0.2, window=("kaiser", 0.5))
     h_truth = np.array(
         [
             -0.000000000000000,
@@ -320,7 +321,7 @@ def test_kaiser_0p2():
             -0.000000000000000,
         ]
     )
-    verify_impulse_response(h, h_truth, atol=1e-1)
+    verify_impulse_response(h, h_truth, atol=1e-1)  # TODO: These aren't exactly identical
 
 
 def test_kaiser_0p4():
@@ -329,7 +330,7 @@ def test_kaiser_0p4():
         >> h = designHighpassFIR(FilterOrder=30, CutoffFrequency=0.4, Window="kaiser");
         >> transpose(h)
     """
-    h = sdr.highpass_fir(30, 0.4, window="kaiser", atten=21.542)
+    h = sdr.highpass_fir(30, 0.4, window=("kaiser", 0.5))
     h_truth = np.array(
         [
             0.000000000000000,
@@ -365,7 +366,7 @@ def test_kaiser_0p4():
             0.000000000000000,
         ]
     )
-    verify_impulse_response(h, h_truth, atol=1e-1)
+    verify_impulse_response(h, h_truth, atol=1e-1)  # TODO: These aren't exactly identical
 
 
 def test_kaiser_0p6():
@@ -374,7 +375,8 @@ def test_kaiser_0p6():
         >> h = designHighpassFIR(FilterOrder=30, CutoffFrequency=0.6, Window="kaiser");
         >> transpose(h)
     """
-    h = sdr.highpass_fir(30, 0.6, window="kaiser", atten=60)
+    beta = scipy.signal.kaiser_beta(60)
+    h = sdr.highpass_fir(30, 0.6, window=("kaiser", beta))
     h_truth = np.array(
         [
             -0.000000000000000,
@@ -410,7 +412,7 @@ def test_kaiser_0p6():
             -0.000000000000000,
         ]
     )
-    verify_impulse_response(h, h_truth, atol=1e-2)
+    verify_impulse_response(h, h_truth, atol=1e-2)  # TODO: These aren't exactly identical
 
 
 def test_kaiser_0p8():
@@ -419,7 +421,7 @@ def test_kaiser_0p8():
         >> h = designHighpassFIR(FilterOrder=30, CutoffFrequency=0.8, Window="kaiser");
         >> transpose(h)
     """
-    h = sdr.highpass_fir(30, 0.8, window="kaiser", atten=20)
+    h = sdr.highpass_fir(30, 0.8, window=("kaiser", 0.5))
     h_truth = np.array(
         [
             0.000000000000000,
@@ -455,4 +457,4 @@ def test_kaiser_0p8():
             0.000000000000000,
         ]
     )
-    verify_impulse_response(h, h_truth, atol=1e-2)
+    verify_impulse_response(h, h_truth, atol=1e-2)  # TODO: These aren't exactly identical
diff --git a/tests/dsp/fir/test_lowpass_fir.py b/tests/dsp/fir/test_lowpass_fir.py
index 0c3381404..f539ff1c0 100644
--- a/tests/dsp/fir/test_lowpass_fir.py
+++ b/tests/dsp/fir/test_lowpass_fir.py
@@ -191,7 +191,7 @@ def test_blackman_harris():
         >> h = designLowpassFIR(FilterOrder=30, CutoffFrequency=0.2, Window="blackman-harris");
         >> transpose(h)
     """
-    h = sdr.lowpass_fir(30, 0.2, window="blackman-harris")
+    h = sdr.lowpass_fir(30, 0.2, window="blackmanharris")
     h_truth = np.array(
         [
             0.000000000000000,
@@ -236,7 +236,7 @@ def test_chebyshev():
         >> h = designLowpassFIR(FilterOrder=30, CutoffFrequency=0.2, Window="chebyshev");
         >> transpose(h)
     """
-    h = sdr.lowpass_fir(30, 0.2, window="chebyshev")
+    h = sdr.lowpass_fir(30, 0.2, window=("chebwin", 60))
     h_truth = np.array(
         [
             0.000000000000000,
@@ -284,8 +284,8 @@ def test_kaiser30_0p1():
         >> h = designLowpassFIR(FilterOrder=30, CutoffFrequency=0.1, Window="kaiser");
         >> transpose(h)
     """
-    # MATLAB uses beta=0.5 for Kaiser window. Attenuation of 21.542 dB was reverse engineered.
-    h = sdr.lowpass_fir(30, 0.1, "kaiser", atten=21.542)
+    # MATLAB uses beta=0.5 for Kaiser window
+    h = sdr.lowpass_fir(30, 0.1, ("kaiser", 0.5))
     h_truth = np.array(
         [
             -0.019837202437853,
@@ -330,8 +330,8 @@ def test_kaiser_30_0p2():
         >> h = designLowpassFIR(FilterOrder=30, CutoffFrequency=0.2, Window="kaiser");
         >> transpose(h)
     """
-    # MATLAB uses beta=0.5 for Kaiser window. Attenuation of 21.542 dB was reverse engineered.
-    h = sdr.lowpass_fir(30, 0.2, window="kaiser", atten=21.542)
+    # MATLAB uses beta=0.5 for Kaiser window
+    h = sdr.lowpass_fir(30, 0.2, window=("kaiser", 0.5))
     h_truth = np.array(
         [
             0.000000000000000,
@@ -376,8 +376,8 @@ def test_kaiser30_0p3():
         >> h = designLowpassFIR(FilterOrder=30, CutoffFrequency=0.3, Window="kaiser");
         >> transpose(h)
     """
-    # MATLAB uses beta=0.5 for Kaiser window. Attenuation of 21.542 dB was reverse engineered.
-    h = sdr.lowpass_fir(30, 0.3, "kaiser", atten=21.542)
+    # MATLAB uses beta=0.5 for Kaiser window
+    h = sdr.lowpass_fir(30, 0.3, ("kaiser", 0.5))
     h_truth = np.array(
         [
             0.019631739746596,
@@ -422,8 +422,8 @@ def test_kaiser_60_0p2():
         >> h = designLowpassFIR(FilterOrder=60, CutoffFrequency=0.2, Window="kaiser");
         >> transpose(h)
     """
-    # MATLAB uses beta=0.5 for Kaiser window. Attenuation of 21.542 dB was reverse engineered.
-    h = sdr.lowpass_fir(60, 0.2, "kaiser", atten=21.542)
+    # MATLAB uses beta=0.5 for Kaiser window
+    h = sdr.lowpass_fir(60, 0.2, ("kaiser", 0.5))
     h_truth = np.array(
         [
             -0.000000000000000,