def test_ReadFHD_model():
"""FHD to uvfits loopback test with model visibilities."""
fhd_uv = UVData()
uvfits_uv = UVData()
fhd_uv.read_fhd(testfiles, use_model=True)
fhd_uv.write_uvfits(os.path.join(DATA_PATH, 'test/outtest_FHD_1061316296_model.uvfits'),
spoof_nonessential=True)
uvfits_uv.read_uvfits(os.path.join(DATA_PATH, 'test/outtest_FHD_1061316296_model.uvfits'))
nt.assert_equal(fhd_uv, uvfits_uv)
del(fhd_uv)
del(uvfits_uv)
def test_ReadFHDWriteReadUVFits():
"""
FHD to uvfits loopback test.
Read in FHD files, write out as uvfits, read back in and check for object
equality.
"""
fhd_uv = UVData()
uvfits_uv = UVData()
fhd_uv.read_fhd(testfiles)
fhd_uv.write_uvfits(os.path.join(DATA_PATH, 'test/outtest_FHD_1061316296.uvfits'),
spoof_nonessential=True)
uvfits_uv.read_uvfits(os.path.join(DATA_PATH, 'test/outtest_FHD_1061316296.uvfits'))
nt.assert_equal(fhd_uv, uvfits_uv)
del(fhd_uv)
del(uvfits_uv)
def test_ReadMiriadWriteUVFits():
"""
Miriad to uvfits loopback test.
Read in Miriad files, write out as uvfits, read back in and check for
object equality.
"""
miriad_uv = UVData()
uvfits_uv = UVData()
miriad_file = os.path.join(DATA_PATH, 'zen.2456865.60537.xy.uvcRREAA')
testfile = os.path.join(DATA_PATH, 'test/outtest_miriad.uvfits')
miriad_status = uvtest.checkWarnings(miriad_uv.read_miriad, [miriad_file],
known_warning='miriad')
miriad_uv.write_uvfits(testfile, spoof_nonessential=True,
force_phase=True)
uvfits_uv.read_uvfits(testfile)
nt.assert_true(miriad_status)
nt.assert_equal(miriad_uv, uvfits_uv)
del(miriad_uv)
del(uvfits_uv)
def test_readwriteread():
"""
CASA tutorial uvfits loopback test.
Read in uvfits file, write out new uvfits file, read back in and check for
object equality.
"""
uv_in = UVData()
uv_out = UVData()
testfile = os.path.join(DATA_PATH, 'day2_TDEM0003_10s_norx_1src_1spw.uvfits')
write_file = os.path.join(DATA_PATH, 'test/outtest_casa.uvfits')
read_status = uvtest.checkWarnings(uv_in.read_uvfits, [testfile],
message='Telescope EVLA is not')
uv_in.write_uvfits(write_file)
write_status = uvtest.checkWarnings(uv_out.read_uvfits, [write_file],
message='Telescope EVLA is not')
nt.assert_true(read_status)
nt.assert_true(write_status)
nt.assert_equal(uv_in, uv_out)
del(uv_in)
del(uv_out)
obs_min = args.obsid_range[0]
obs_max = args.obsid_range[1]
except:
obs_min = min(file_dict.keys())
obs_max = max(file_dict.keys())
for k in file_dict.keys():
if k > obs_max or k < obs_min:
file_dict.pop(k)
for i, (k, v) in enumerate(file_dict.iteritems()):
if args.dirty:
print('converting dirty vis for obsid {}, ({} of {})'.format(k, i, len(file_dict)))
uvfits_file = op.join(output_folder, str(k) + '.uvfits')
this_uv = UVData()
this_uv.read_fhd(v)
this_uv.write_uvfits(uvfits_file, spoof_nonessential=True)
del(this_uv)
if args.model:
print('converting model vis for obsid {}, ({} of {})'.format(k, i, len(file_dict)))
uvfits_file = op.join(output_folder, str(k) + '_model.uvfits')
this_uv = UVData()
this_uv.read_fhd(v, use_model=True)
this_uv.write_uvfits(uvfits_file, spoof_nonessential=True)
del(this_uv)
#! /usr/bin/env python
import argparse
import os.path as op
from uvdata import UVData
parser = argparse.ArgumentParser()
parser.add_argument('uvfits_read',
help='name of a uvfits file to read in')
parser.add_argument('uvfits_write',
help='name of a uvfits file to write out')
args = parser.parse_args()
uvfits_file_in = args.uvfits_read
if not op.isfile(uvfits_file_in):
raise IOError('There is no file named {}'.format(args.uvfits_file_in))
uvfits_file_out = args.uvfits_write
this_uv = UVData()
this_uv.read_uvfits(uvfits_file_in)
this_uv.write_uvfits(uvfits_file_out)
del(this_uv)
