Skip to content

Commit 869136e

Browse files
timcornwelltimcornwell
committed
Writing test results is an option
1 parent 726d814 commit 869136e

2 files changed

Lines changed: 93 additions & 48 deletions

File tree

‎tests/processing_components/test_griddata_gridding.py‎

Lines changed: 39 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,7 @@ class TestGridDataGridding(unittest.TestCase):
4444
def setUp(self):
4545
from data_models.parameters import arl_path
4646
self.dir = arl_path('test_results')
47+
self.persist = False
4748

4849
def actualSetUp(self, zerow=True):
4950
self.doplot = False
@@ -88,11 +89,13 @@ def actualSetUp(self, zerow=True):
8889

8990
# Calculate the model convolved with a Gaussian.
9091
self.cmodel = smooth_image(self.model)
91-
export_image_to_fits(self.model, '%s/test_gridding_model.fits' % self.dir)
92-
export_image_to_fits(self.cmodel, '%s/test_gridding_cmodel.fits' % self.dir)
92+
if self.persist:
93+
export_image_to_fits(self.model, '%s/test_gridding_model.fits' % self.dir)
94+
export_image_to_fits(self.cmodel, '%s/test_gridding_cmodel.fits' % self.dir)
9395
pb = create_pb_generic(self.model, diameter=35.0, blockage=0.0)
9496
self.cmodel.data *= pb.data
95-
export_image_to_fits(self.cmodel, '%s/test_gridding_cmodel_pb.fits' % self.dir)
97+
if self.persist:
98+
export_image_to_fits(self.cmodel, '%s/test_gridding_cmodel_pb.fits' % self.dir)
9699
self.peak = numpy.unravel_index(numpy.argmax(numpy.abs(self.cmodel.data)), self.cmodel.shape)
97100

98101

@@ -106,7 +109,8 @@ def test_griddata_weight(self):
106109
griddata, sumwt = grid_weight_to_griddata(self.vis, griddata=griddata, cf=cf)
107110
im = fft_griddata_to_image(griddata, gcf)
108111
im = normalize_sumwt(im, sumwt)
109-
export_image_to_fits(im, '%s/test_gridding_weight.fits' % self.dir)
112+
if self.persist:
113+
export_image_to_fits(im, '%s/test_gridding_weight.fits' % self.dir)
110114
self.check_peaks(im, 97.00435128311616, tol=1e-7)
111115

112116
def test_griddata_invert_pswf(self):
@@ -116,7 +120,8 @@ def test_griddata_invert_pswf(self):
116120
griddata, sumwt = grid_visibility_to_griddata(self.vis, griddata=griddata, cf=cf)
117121
im = fft_griddata_to_image(griddata, gcf)
118122
im = normalize_sumwt(im, sumwt)
119-
export_image_to_fits(im, '%s/test_gridding_dirty_pswf.fits' % self.dir)
123+
if self.persist:
124+
export_image_to_fits(im, '%s/test_gridding_dirty_pswf.fits' % self.dir)
120125
self.check_peaks(im, 97.00435128311616, tol=1e-7)
121126

122127
def test_griddata_invert_pswf_w(self):
@@ -126,35 +131,40 @@ def test_griddata_invert_pswf_w(self):
126131
griddata, sumwt = grid_visibility_to_griddata(self.vis, griddata=griddata, cf=cf)
127132
im = fft_griddata_to_image(griddata, gcf)
128133
im = normalize_sumwt(im, sumwt)
129-
export_image_to_fits(im, '%s/test_gridding_dirty_pswf_w.fits' % self.dir)
134+
if self.persist:
135+
export_image_to_fits(im, '%s/test_gridding_dirty_pswf_w.fits' % self.dir)
130136
self.check_peaks(im, 97.02498821941138, tol=1e-7)
131137

132138
def test_griddata_invert_aterm(self):
133139
self.actualSetUp(zerow=True)
134140
make_pb = functools.partial(create_pb_generic, diameter=35.0, blockage=0.0)
135141
pb = make_pb(self.model)
136-
export_image_to_fits(pb, "%s/test_gridding_aterm_pb.fits" % self.dir)
142+
if self.persist:
143+
export_image_to_fits(pb, "%s/test_gridding_aterm_pb.fits" % self.dir)
137144
gcf, cf = create_awterm_convolutionfunction(self.model, make_pb=make_pb, nw=1, oversampling=16, support=16,
138145
use_aaf=False)
139146
griddata = create_griddata_from_image(self.model)
140147
griddata, sumwt = grid_visibility_to_griddata(self.vis, griddata=griddata, cf=cf)
141148
im = fft_griddata_to_image(griddata, gcf)
142149
im = normalize_sumwt(im, sumwt)
143-
export_image_to_fits(im, '%s/test_gridding_dirty_aterm.fits' % self.dir)
150+
if self.persist:
151+
export_image_to_fits(im, '%s/test_gridding_dirty_aterm.fits' % self.dir)
144152
self.check_peaks(im, 97.11809546469424, tol=1e-7)
145153

146154
def test_griddata_invert_aterm_noover(self):
147155
self.actualSetUp(zerow=True)
148156
make_pb = functools.partial(create_pb_generic, diameter=35.0, blockage=0.0)
149157
pb = make_pb(self.model)
150-
export_image_to_fits(pb, "%s/test_gridding_aterm_pb.fits" % self.dir)
158+
if self.persist:
159+
export_image_to_fits(pb, "%s/test_gridding_aterm_pb.fits" % self.dir)
151160
gcf, cf = create_awterm_convolutionfunction(self.model, make_pb=make_pb, nw=1, oversampling=1, support=16,
152161
use_aaf=True)
153162
griddata = create_griddata_from_image(self.model)
154163
griddata, sumwt = grid_visibility_to_griddata(self.vis, griddata=griddata, cf=cf)
155164
im = fft_griddata_to_image(griddata, gcf)
156165
im = normalize_sumwt(im, sumwt)
157-
export_image_to_fits(im, '%s/test_gridding_dirty_aterm_noover.fits' % self.dir)
166+
if self.persist:
167+
export_image_to_fits(im, '%s/test_gridding_dirty_aterm_noover.fits' % self.dir)
158168
self.check_peaks(im, 97.11833094589)
159169

160170
def test_griddata_invert_box(self):
@@ -164,7 +174,8 @@ def test_griddata_invert_box(self):
164174
griddata, sumwt = grid_visibility_to_griddata(self.vis, griddata=griddata, cf=cf)
165175
im = fft_griddata_to_image(griddata, gcf)
166176
im = normalize_sumwt(im, sumwt)
167-
export_image_to_fits(im, '%s/test_gridding_dirty_box.fits' % self.dir)
177+
if self.persist:
178+
export_image_to_fits(im, '%s/test_gridding_dirty_box.fits' % self.dir)
168179
self.check_peaks(im, 97.11833094588997, tol=1e-7)
169180

170181
def test_griddata_invert_fast(self):
@@ -174,7 +185,8 @@ def test_griddata_invert_fast(self):
174185
griddata, sumwt = grid_visibility_to_griddata_fast(self.vis, griddata=griddata, cf=cf, gcf=gcf)
175186
im = fft_griddata_to_image(griddata, gcf)
176187
im = normalize_sumwt(im, sumwt)
177-
export_image_to_fits(im, '%s/test_gridding_dirty_fast.fits' % self.dir)
188+
if self.persist:
189+
export_image_to_fits(im, '%s/test_gridding_dirty_fast.fits' % self.dir)
178190
self.check_peaks(im, 97.11833094588997, tol=1e-7)
179191

180192
def check_peaks(self, im, peak=100.0, tol=1e-3):
@@ -187,31 +199,36 @@ def test_griddata_invert_wterm(self):
187199

188200
cf_image = convert_convolutionfunction_to_image(cf)
189201
cf_image.data = numpy.real(cf_image.data)
190-
export_image_to_fits(cf_image, "%s/test_gridding_wterm_cf.fits" % self.dir)
202+
if self.persist:
203+
export_image_to_fits(cf_image, "%s/test_gridding_wterm_cf.fits" % self.dir)
191204

192205
griddata = create_griddata_from_image(self.model, nw=1)
193206
griddata, sumwt = grid_visibility_to_griddata(self.vis, griddata=griddata, cf=cf)
194207
im = fft_griddata_to_image(griddata, gcf)
195208
im = normalize_sumwt(im, sumwt)
196-
export_image_to_fits(im, '%s/test_gridding_dirty_wterm.fits' % self.dir)
209+
if self.persist:
210+
export_image_to_fits(im, '%s/test_gridding_dirty_wterm.fits' % self.dir)
197211
self.check_peaks(im, 97.13863141113761)
198212

199213
def test_griddata_invert_awterm(self):
200214
self.actualSetUp(zerow=False)
201215
make_pb = functools.partial(create_pb_generic, diameter=35.0, blockage=0.0)
202216
pb = make_pb(self.model)
203-
export_image_to_fits(pb, "%s/test_gridding_awterm_pb.fits" % self.dir)
217+
if self.persist:
218+
export_image_to_fits(pb, "%s/test_gridding_awterm_pb.fits" % self.dir)
204219
gcf, cf = create_awterm_convolutionfunction(self.model, make_pb=make_pb, nw=100, wstep=8.0,
205220
oversampling=16, support=32, use_aaf=True)
206221
cf_image = convert_convolutionfunction_to_image(cf)
207222
cf_image.data = numpy.real(cf_image.data)
208-
export_image_to_fits(cf_image, "%s/test_gridding_awterm_cf.fits" % self.dir)
223+
if self.persist:
224+
export_image_to_fits(cf_image, "%s/test_gridding_awterm_cf.fits" % self.dir)
209225

210226
griddata = create_griddata_from_image(self.model, nw=100, wstep=8.0)
211227
griddata, sumwt = grid_visibility_to_griddata(self.vis, griddata=griddata, cf=cf)
212228
im = fft_griddata_to_image(griddata, gcf)
213229
im = normalize_sumwt(im, sumwt)
214-
export_image_to_fits(im, '%s/test_gridding_dirty_awterm.fits' % self.dir)
230+
if self.persist:
231+
export_image_to_fits(im, '%s/test_gridding_dirty_awterm.fits' % self.dir)
215232
self.check_peaks(im, 97.13891073376104)
216233

217234
def test_griddata_predict_pswf(self):
@@ -262,7 +279,8 @@ def test_griddata_predict_awterm(self):
262279
self.actualSetUp(zerow=False)
263280
make_pb = functools.partial(create_pb_generic, diameter=35.0, blockage=0.0)
264281
pb = make_pb(self.model)
265-
export_image_to_fits(pb, "%s/test_gridding_awterm_pb.fits" % self.dir)
282+
if self.persist:
283+
export_image_to_fits(pb, "%s/test_gridding_awterm_pb.fits" % self.dir)
266284
gcf, cf = create_awterm_convolutionfunction(self.model, make_pb=make_pb, nw=100, wstep=8.0,
267285
oversampling=16, support=32, use_aaf=True)
268286
griddata = create_griddata_from_image(self.model, nw=100, wstep=8.0)
@@ -272,7 +290,7 @@ def test_griddata_predict_awterm(self):
272290
assert qa.data['rms'] < 120.0, str(qa)
273291
self.plot_vis(newvis, 'awterm')
274292

275-
def test_griddata_weight(self):
293+
def test_griddata_weight_1(self):
276294
self.actualSetUp(zerow=True)
277295
gcf, cf = create_box_convolutionfunction(self.model)
278296
gd = create_griddata_from_image(self.model)
@@ -282,7 +300,8 @@ def test_griddata_weight(self):
282300
gd, sumwt = grid_visibility_to_griddata(self.vis, griddata=gd, cf=cf)
283301
im = fft_griddata_to_image(gd, gcf)
284302
im = normalize_sumwt(im, sumwt)
285-
export_image_to_fits(im, '%s/test_gridding_dirty_2d_uniform.fits' % self.dir)
303+
if self.persist:
304+
export_image_to_fits(im, '%s/test_gridding_dirty_2d_uniform.fits' % self.dir)
286305
self.check_peaks(im, 99.42031190701735)
287306

288307

0 commit comments

Comments
 (0)