Use np.linspace() where appropriate

Fixes #46
mhostetter · Jul 27, 2023 · f063e7f · f063e7f
1 parent bf65fb7
commit f063e7f
Show file tree

Hide file tree

Showing 5 changed files with 100 additions and 47 deletions.
diff --git a/docs/examples/farrow-resampler.ipynb b/docs/examples/farrow-resampler.ipynb
diff --git a/docs/examples/phase-locked-loop.ipynb b/docs/examples/phase-locked-loop.ipynb
diff --git a/docs/examples/psk.ipynb b/docs/examples/psk.ipynb
diff --git a/src/sdr/_modulation/_symbol_mapping.py b/src/sdr/_modulation/_symbol_mapping.py
@@ -33,7 +33,7 @@ def binary_code(n: int) -> np.ndarray:
     if not n >= 1:
         raise ValueError(f"Argument 'n' must be greater than or equal to 1, not {n}.")
 
-    return np.arange(0, 2**n)
+    return np.arange(2**n)
 
 
 @export

diff --git a/tests/resamplers/test_farrow.py b/tests/resamplers/test_farrow.py
@@ -11,9 +11,9 @@
 def generate_signal():
     N = 1_000  # samples
     omega = 2 * np.pi / 40  # radians
-    x = np.exp(1j * omega * np.arange(0, N))  # Complex exponential input signal
+    x = np.exp(1j * omega * np.arange(N))  # Complex exponential input signal
     x *= np.exp(-np.arange(N) / 300)  # Exponential decay
-    tx = np.arange(0, N)  # Time axis for the input signal
+    tx = np.arange(N)  # Time axis for the input signal
     return tx, x