set nthreads=1 in test suite

ratt-ru · Sep 4, 2024 · 446df17 · 446df17
1 parent 1ff7673
commit 446df17
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 13 deletions.
diff --git a/pfb/parser/grid.yaml b/pfb/parser/grid.yaml
@@ -117,11 +117,7 @@ inputs:
     - (.)out.yml
 
 outputs:
-  cds:
-    dtype: Directory
-    info: output cube dataset
-    mkdir: false
-
+  {}
 
 policies:
   pass_missing_as_none: true
diff --git a/tests/test_klean.py b/tests/test_klean.py
@@ -104,7 +104,7 @@ def test_klean(do_gains, ms_name):
                                            flip_v=flip_v,
                                            flip_w=flip_w,
                                            do_wgridding=True,
-                                           nthreads=8)
+                                           nthreads=1)
         model_vis[:, c, -1] = model_vis[:, c, 0]
 
 
@@ -219,7 +219,7 @@ def test_klean(do_gains, ms_name):
     grid_args["fits_mfs"] = True
     grid_args["psf"] = True
     grid_args["residual"] = False
-    grid_args["nthreads"] = 8
+    grid_args["nthreads"] = 1
     grid_args["overwrite"] = True
     grid_args["robustness"] = 0.0
     grid_args["do_wgridding"] = True
@@ -240,7 +240,7 @@ def test_klean(do_gains, ms_name):
     klean_args["gamma"] = 0.1
     klean_args["peak_factor"] = 0.75
     klean_args["sub_peak_factor"] = 0.75
-    klean_args["nthreads"] = 8
+    klean_args["nthreads"] = 1
     klean_args["do_wgridding"] = True
     klean_args["epsilon"] = epsilon
     klean_args["mop_flux"] = True

diff --git a/tests/test_polproducts.py b/tests/test_polproducts.py
@@ -107,7 +107,7 @@ def test_polproducts(do_gains, ms_name):
                                         flip_w=flip_w,
                                         epsilon=epsilon,
                                         do_wgridding=True,
-                                        nthreads=8)
+                                        nthreads=1)
     model_vis_Q = np.zeros((nrow, nchan), dtype=np.complex128)
     for c in range(nchan):
         model_vis_Q[:, c:c+1] = dirty2vis(uvw=uvw,
@@ -122,7 +122,7 @@ def test_polproducts(do_gains, ms_name):
                                         flip_w=flip_w,
                                         epsilon=epsilon,
                                         do_wgridding=True,
-                                        nthreads=8)
+                                        nthreads=1)
     model_vis_U = np.zeros((nrow, nchan), dtype=np.complex128)
     for c in range(nchan):
         model_vis_U[:, c:c+1] = dirty2vis(uvw=uvw,
@@ -137,7 +137,7 @@ def test_polproducts(do_gains, ms_name):
                                         flip_w=flip_w,
                                         epsilon=epsilon,
                                         do_wgridding=True,
-                                        nthreads=8)
+                                        nthreads=1)
     model_vis_V = np.zeros((nrow, nchan), dtype=np.complex128)
     for c in range(nchan):
         model_vis_V[:, c:c+1] = dirty2vis(uvw=uvw,
@@ -152,7 +152,7 @@ def test_polproducts(do_gains, ms_name):
                                         flip_w=flip_w,
                                         epsilon=epsilon,
                                         do_wgridding=True,
-                                        nthreads=8)
+                                        nthreads=1)
 
     model_vis = np.zeros((nrow, nchan, ncorr), dtype=np.complex128)
     model_vis[:, :, 0] = model_vis_I + model_vis_Q
@@ -281,7 +281,7 @@ def test_polproducts(do_gains, ms_name):
         grid_args["fits_mfs"] = True
         grid_args["psf"] = True
         grid_args["residual"] = False
-        grid_args["nthreads"] = 8
+        grid_args["nthreads"] = 1
         grid_args["overwrite"] = True
         grid_args["robustness"] = 0.0
         grid_args["do_wgridding"] = True