Add unit tests for power()

mhostetter · Jul 11, 2024 · a1913ce · a1913ce
1 parent 8bb922b
commit a1913ce
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/tests/measurements/test_power.py b/tests/measurements/test_power.py
@@ -4,6 +4,14 @@
 import sdr
 
 
+def test_power():
+    rng = np.random.default_rng()
+    A = rng.uniform(2, 10)  # Signal amplitude
+    N = 50
+    x = A * np.exp(1j * np.linspace(0, 2 * np.pi, N, endpoint=False))
+    assert np.allclose(sdr.power(x), A**2)
+
+
 def test_peak_power():
     rng = np.random.default_rng()
     A = rng.uniform(2, 10)  # Signal amplitude