def output_images(result):
BufferSkyModel(conf["buffer"], conf['outputs']['skymodel'], SkyModel(images=result[0])).sync()
deconvolved = image_gather_channels(result[0])
residual = image_gather_channels(remove_sumwt(result[1]))
restored = image_gather_channels(result[2])
BufferImage(conf["buffer"], conf['outputs']['deconvolved'], deconvolved).sync()
BufferImage(conf["buffer"], conf['outputs']['residual'], residual).sync()
BufferImage(conf["buffer"], conf['outputs']['restored'], restored).sync()
return result
def test_readwriteimage_assertion(self):
im = create_test_image()
with self.assertRaises(AssertionError):
config = {
"buffer": {
"directory": self.dir
},
"image": {
"name": "test_bufferskyimage.hdf",
"data_model": "Image"
}
}
bdm = BufferImage(config["buffer"], config["image"], im)
bdm.sync()
new_bdm = BufferSkyModel(config["buffer"], config["image"])
new_bdm.sync()
newim = bdm.memory_data_model
def test_readwriteimage(self):
im = create_test_image()
config = {
"buffer": {
"directory": self.dir
},
"image": {
"name": "test_bufferimage.hdf",
"data_model": "Image"
}
}
bdm = BufferImage(config["buffer"], config["image"], im)
bdm.sync()
new_bdm = BufferImage(config["buffer"], config["image"])
new_bdm.sync()
newim = bdm.memory_data_model
assert newim.data.shape == im.data.shape
assert numpy.max(numpy.abs(im.data - newim.data)) < 1e-15