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))