Add unit tests for shifted Zadoff-Chu

mhostetter · Aug 3, 2023 · 3371cb1 · 3371cb1
1 parent e2d47e8
commit 3371cb1
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/sequences/test_zadoff_chu.py b/tests/sequences/test_zadoff_chu.py
@@ -314,3 +314,13 @@ def test_25_11():
         ]
     )
     np.testing.assert_array_almost_equal(seq, seq_truth)
+
+
+def test_shift():
+    x = sdr.zadoff_chu(13, 3)
+    x_shift = sdr.zadoff_chu(13, 3, shift=1)
+
+    # The shifted version should be a cyclic shift of the original sequence multiplied by a complex value
+    y = np.roll(x, -1) / x_shift
+    y /= y[0]
+    np.testing.assert_array_almost_equal(y, 1)