Fix: update Fourier shift test to accommodate 'ij' indexing

wheelerMT · Mar 23, 2024 · 853adf1 · 853adf1
1 parent 59bd604
commit 853adf1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/shared/test_grids.py b/tests/shared/test_grids.py
@@ -59,9 +59,9 @@ def test_correct_fourier_shift_2d():
 
     grid2d = grid.Grid((64, 64), (0.5, 0.5))
 
-    for element in grid2d.fourier_x_mesh[:, 0]:
+    for element in grid2d.fourier_x_mesh[0, :]:
         assert element == 0
-    for element in grid2d.fourier_y_mesh[0, :]:
+    for element in grid2d.fourier_y_mesh[:, 0]:
         assert element == 0