From 38222cd9ed3f5469ed18412b213b2ff590e54eb2 Mon Sep 17 00:00:00 2001
From: Ian Czekala <iancze@gmail.com>
Date: Thu, 9 Nov 2023 09:56:20 +0000
Subject: [PATCH] set sparse_matrices=False to avoid nuFFT warning with
 multidimensional array.

---
 test/fourier_test.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/test/fourier_test.py b/test/fourier_test.py
index 2d751806..1b5caed9 100644
--- 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 = {