add fftshift and ifftshift fft helpers (#2135)

* add fftshift and ifftshift fft helpers * address comments * axes have to be iterable * fix fp error in roll + add test --------- Co-authored-by: Aashiq Dheeraj <[email protected]>
2025-04-29 22:13:45 -07:00
parent 7bb063bcb3
commit bb6565ef14
9 changed files with 275 additions and 2 deletions
@@ -3859,6 +3859,9 @@ TEST_CASE("test roll") {
  y = roll(x, {1, 2}, {0, 1});
  CHECK(array_equal(y, array({8, 9, 5, 6, 7, 3, 4, 0, 1, 2}, {2, 5}))
            .item<bool>());
+
+  y = roll(array({}), 0, 0);
+  CHECK(array_equal(y, array({})).item<bool>());
 }

 TEST_CASE("test contiguous") {