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