Cosmetics

test.py

@@ -291,15 +291,13 @@ def test_against_wdft(nxdirty, nydirty, nchan, nrow, fov):
     nplanes = 10000
     ws = mi + np.arange(nplanes)*(ma-mi)/(nplanes-1)
     for ii in range(len(ws)-1):
-        wkp1 = ws[ii+1]
-        if ii == nplanes-2:
-            wkp1 += abs(wkp1)
-        jj = ng.getIndices(bl, conf, flags, wmin=ws[ii], wmax=wkp1)
-        if len(jj) == 0:
+        wmax = ws[ii+1] if ii != nplanes-2 else ws[ii+1] + abs(ws[ii+1])
+        wmax = ws[ii+1] if ii != nplanes-2 else np.inf
+        iind = ng.getIndices(bl, conf, flags, wmin=ws[ii], wmax=wmax)
+        if len(iind) == 0:
             continue
-        dd = conf.grid2dirty_c(ng.vis2grid_c(bl, conf, jj, bl.ms2vis(ms, jj)))
-        wforplane = 0.5*(ws[ii+1]+ws[ii])
-        res0 += conf.apply_wscreen(dd, wforplane, adjoint=False).real
+        dd = conf.grid2dirty_c(ng.vis2grid_c(bl, conf, iind, bl.ms2vis(ms, iind)))
+        res0 += conf.apply_wscreen(dd, 0.5*(ws[ii+1]+ws[ii]), adjoint=False).real
 
     # Compute dft with w term
     x, y = np.meshgrid(*[-ss/2 + np.arange(ss) for ss in [nxdirty, nydirty]],