def test_array_geometry(self):
"""Test the 'AIPS AN' table, part 1."""
testTime = time.time() - 2*86400.0
testFile = os.path.join(self.testPath, 'uv-test-AG.fits')
# Get some data
data = self.__initData()
# Start the file
fits = uvfits.Uv(testFile, ref_time=testTime)
fits.set_stokes(['xx'])
fits.set_frequency(data['freq'])
fits.set_geometry(data['site'], data['antennas'])
fits.add_data_set(testTime, 6.0, data['bl'], data['vis'])
fits.write()
# Open the file and examine
hdulist = astrofits.open(testFile)
ag = hdulist['AIPS AN'].data
# Correct number of stands
self.assertEqual(len(data['antennas']), len(ag.field('NOSTA')))
# Correct stand names
names = ['LWA%03i' % ant.stand.id for ant in data['antennas']]
for name, anname in zip(names, ag.field('ANNAME')):
self.assertEqual(name, anname)
hdulist.close()
def test_write_tables(self):
"""Test if the UVFITS writer writes all of the tables."""
testTime = time.time() - 2*86400.0
testFile = os.path.join(self.testPath, 'uv-test-W.fits')
# Get some data
data = self.__initData()
# Start the file
fits = uvfits.Uv(testFile, ref_time=testTime)
fits.set_stokes(['xx'])
fits.set_frequency(data['freq'])
fits.set_geometry(data['site'], data['antennas'])
fits.add_comment('This is a comment')
fits.add_history('This is history')
fits.add_data_set(testTime, 6.0, data['bl'], data['vis'])
fits.write()
# Open the file and examine
hdulist = astrofits.open(testFile)
# Check that all of the extensions are there
extNames = [hdu.name for hdu in hdulist]
for ext in ['AIPS AN', 'AIPS FQ', 'AIPS SU']:
self.assertTrue(ext in extNames)
# Check the comments and history
self.assertTrue('This is a comment' in str(hdulist[0].header['COMMENT']).split('\n'))
self.assertTrue('This is history' in str(hdulist[0].header['HISTORY']).split('\n'))
hdulist.close()
def test_frequency(self):
"""Test the 'AIPS FQ' table."""
testTime = time.time() - 2*86400.0
testFile = os.path.join(self.testPath, 'uv-test-FQ.fits')
# Get some data
data = self.__initData()
# Start the file
fits = uvfits.Uv(testFile, ref_time=testTime)
fits.set_stokes(['xx'])
fits.set_frequency(data['freq'])
fits.set_geometry(data['site'], data['antennas'])
fits.add_data_set(testTime, 6.0, data['bl'], data['vis'])
fits.write()
# Open the file and examine
hdulist = astrofits.open(testFile)
fq = hdulist['AIPS FQ'].data
# Correct number of IFs
self.assertEqual(len(fq.field('FRQSEL')), 1)
# Correct channel width
self.assertEqual(fq.field('CH WIDTH')[0], data['freq'][1]-data['freq'][0])
# Correct bandwidth
self.assertEqual(fq.field('TOTAL BANDWIDTH')[0], numpy.abs(data['freq'][-1]-data['freq'][0]).astype(numpy.float32), 4)
hdulist.close()
def test_antenna(self):
"""Test the 'AIPS AN' table, part 2."""
testTime = time.time() - 2 * 86400.0
testFile = os.path.join(self.testPath, 'uv-test-AN.fits')
# Get some data
data = self._init_data()
# Start the file
fits = uvfits.Uv(testFile, ref_time=testTime)
fits.set_stokes(['xx'])
fits.set_frequency(data['freq'])
fits.set_geometry(data['site'], data['antennas'])
fits.add_data_set(unix_to_taimjd(testTime), 6.0, data['bl'],
data['vis'])
fits.write()
fits.close()
# Open the file and examine
hdulist = astrofits.open(testFile)
an = hdulist['AIPS AN'].data
# Correct number of stands
self.assertEqual(len(data['antennas']), len(an.field('NOSTA')))
hdulist.close()
def test_bandpass(self):
"""Test the 'AIPS BP' table."""
testTime = time.time() - 2*86400.0
testFile = os.path.join(self.testPath, 'uv-test-BP.fits')
# Get some data
data = self.__initData()
# Start the file
fits = uvfits.Uv(testFile, ref_time=testTime)
fits.set_stokes(['xx'])
fits.set_frequency(data['freq'])
fits.set_geometry(data['site'], data['antennas'])
fits.add_data_set(testTime, 6.0, data['bl'], data['vis'])
fits.write()
# Open the file and examine
hdulist = astrofits.open(testFile)
su = hdulist['AIPS BP'].data
hdulist.close()
def test_uvdata(self):
"""Test the primary data table."""
testTime = time.time() - 2*86400.0
testFile = os.path.join(self.testPath, 'uv-test-UV.fits')
# Get some data
data = self.__initData()
# Start the file
fits = uvfits.Uv(testFile, ref_time=testTime)
fits.set_stokes(['xx'])
fits.set_frequency(data['freq'])
fits.set_geometry(data['site'], data['antennas'])
fits.add_data_set(testTime, 6.0, data['bl'], data['vis'])
fits.write()
# Open the file and examine
hdulist = astrofits.open(testFile)
uv = hdulist[0]
# Load the mapper
try:
mp = hdulist['NOSTA_MAPPER'].data
nosta = mp.field('NOSTA')
noact = mp.field('NOACT')
except KeyError:
ag = hdulist['AIPS AN'].data
nosta = ag.field('NOSTA')
noact = ag.field('NOSTA')
mapper = {}
for s,a in zip(nosta, noact):
mapper[s] = a
# Correct number of visibilities
self.assertEqual(len(uv.data), data['vis'].shape[0])
# Correct number of frequencies
for row in uv.data:
vis = row['DATA'][0,0,:,:,:]
self.assertEqual(len(vis), len(data['freq']))
# Correct values
for row in uv.data:
bl = int(row['BASELINE'])
vis = row['DATA'][0,0,:,:,:]
# Unpack the baseline
if bl >= 65536:
a1 = int((bl - 65536) / 2048)
a2 = int((bl - 65536) % 2048)
else:
a1 = int(bl / 256)
a2 = int(bl % 256)
# Convert mapped stands to real stands
stand1 = mapper[a1]
stand2 = mapper[a2]
# Find out which visibility set in the random data corresponds to the
# current visibility
i = 0
for ant1,ant2 in data['bl']:
if ant1.stand.id == stand1 and ant2.stand.id == stand2:
break
else:
i = i + 1
# Extract the data and run the comparison
visData = numpy.zeros(len(data['freq']), dtype=numpy.complex64)
visData.real = vis[:,0,0]
visData.imag = vis[:,0,1]
for vd, sd in zip(visData, data['vis'][i,:]):
self.assertAlmostEqual(vd, sd, 8)
i = i + 1
hdulist.close()