def test_create_low_test_image_spectral(self):
im = create_low_test_image_from_s3(npixel=1024, channel_bandwidth=numpy.array([1e6, 1e6, 1e6]),
frequency=numpy.array([1e8 - 1e6, 1e8, 1e8 + 1e6]),
phasecentre=self.phasecentre, fov=10)
assert im.data.shape[0] == 3
assert im.data.shape[1] == 1
assert im.data.shape[2] == 1024
assert im.data.shape[3] == 1024
def test_get_frequency_map_s3(self):
self.model = create_low_test_image_from_s3(
npixel=64,
cellsize=0.001,
frequency=self.frequency,
channel_bandwidth=self.channel_bandwidth)
spectral_mode, vfrequency_map = get_frequency_map(self.vis, self.model)
assert numpy.max(vfrequency_map) == self.model.nchan - 1
assert spectral_mode == 'channel'
def test_create_low_test_image_spectral_polarisation(self):
im = create_low_test_image_from_s3(npixel=1024, channel_bandwidth=numpy.array([1e6, 1e6, 1e6]),
polarisation_frame=PolarisationFrame("stokesIQUV"),
frequency=numpy.array([1e8 - 1e6, 1e8, 1e8 + 1e6]), fov=10)
assert im.data.shape[0] == 3
assert im.data.shape[1] == 4
assert im.data.shape[2] == 1024
assert im.data.shape[3] == 1024
export_image_to_fits(im, '%s/test_low_s3.fits' % (self.dir))
def test_create_low_test_image_from_s3(self):
im = create_low_test_image_from_s3(npixel=1024,
channel_bandwidth=numpy.array([1e6
]),
frequency=numpy.array([1e8]),
phasecentre=self.phasecentre,
fov=10)
assert im.data.shape[0] == 1
assert im.data.shape[1] == 1
assert im.data.shape[2] == 1024
assert im.data.shape[3] == 1024
export_image_to_fits(im,
'%s/test_test_support_low_s3.fits' % (self.dir))