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_get_frequency_map_mfs(self):
self.model = create_image_from_visibility(
self.vis,
npixel=256,
cellsize=0.001,
nchan=1,
frequency=self.startfrequency)
spectral_mode, vfrequency_map = get_frequency_map(self.vis, self.model)
assert numpy.max(vfrequency_map) == 0
assert spectral_mode == 'mfs'
def test_get_frequency_map_different_channel(self):
self.model = create_image_from_visibility(
self.vis,
npixel=256,
cellsize=0.001,
frequency=self.startfrequency,
nchan=3,
channel_bandwidth=2e7)
spectral_mode, vfrequency_map = get_frequency_map(self.vis, self.model)
assert numpy.max(vfrequency_map) == self.model.nchan - 1
assert spectral_mode == 'channel'