add N assertion

waltsims · Dec 19, 2023 · 3c41b7a · 3c41b7a
1 parent ce938f6
commit 3c41b7a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/kwave/utils/signals.py b/kwave/utils/signals.py
@@ -144,6 +144,8 @@ def cosine_series(n: int, N: int, coeffs: List[float]) -> np.ndarray:
     if N.size == 1:
         n = np.arange(0, N)
 
+        assert (N - 1) != 0, 'N must be greater than 1'
+
         if type_ == 'Bartlett':
             win = (2 / (N - 1) * ((N - 1) / 2 - abs(n - (N - 1) / 2))).T
         elif type_ == 'Bartlett-Hanning':