def test_create_test_image_from_s3_low(self):
im = create_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))
def test_create_test_image_s3_spectral(self):
im = create_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,
flux_limit=2e-3)
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_create_low_test_image_s3_spectral_polarisation(self):
im = create_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_test_support_low_s3.fits' % (self.dir))
def test_create_test_image_from_s3_mid(self):
im = create_test_image_from_s3(npixel=1024,
channel_bandwidth=numpy.array([1e6]),
frequency=numpy.array([1e9]),
phasecentre=self.phasecentre,
flux_limit=2e-3)
assert im.data.shape[0] == 1
assert im.data.shape[1] == 1
assert im.data.shape[2] == 1024
assert im.data.shape[3] == 1024
if self.persist:
export_image_to_fits(
im, '%s/test_test_support_mid_s3.fits' % (self.dir))
