set sparse_matrices=False to avoid nuFFT warning with multidimensiona…

…l array.
MPoL-dev · Nov 9, 2023 · 38222cd · 38222cd
1 parent ae09abd
commit 38222cd
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/test/fourier_test.py b/test/fourier_test.py
@@ -314,7 +314,9 @@ def test_nufft_accuracy_batch_broadcast(coords, mock_visibility_data, tmp_path):
     nchan = uu.shape[0]
 
     # create a NuFFT layer to perform interpolations to these points
-    layer = fourier.NuFFT(coords=coords, nchan=nchan, uu=uu, vv=vv)
+    # uu and vv are multidimensional, so we should set `sparse_matrices=False`
+    # to avoid triggering a warning
+    layer = fourier.NuFFT(coords=coords, nchan=nchan, uu=uu, vv=vv, sparse_matrices=False)
 
     # a sky Gaussian
     kw = {