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)