Add unit tests for Integrator

mhostetter · Nov 12, 2023 · 13e842b · 13e842b
1 parent 3d716fa
commit 13e842b
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/tests/dsp/test_integrator.py b/tests/dsp/test_integrator.py
@@ -0,0 +1,23 @@
+import numpy as np
+
+import sdr
+
+
+def test_gaussian():
+    fir = sdr.Integrator()
+    x = sdr.gaussian(0.3, 5, 10)
+    y = fir(x)
+
+    y_truth = np.cumsum(x)
+
+    assert np.allclose(y, y_truth)
+
+
+def test_raised_cosine():
+    fir = sdr.Integrator()
+    x = sdr.root_raised_cosine(0.1, 8, 10)
+    y = fir(x)
+
+    y_truth = np.cumsum(x)
+
+    assert np.allclose(y, y_truth)