def test_lalc(): """ Check that LA->interfits->LA->interfits creates identical data. """ lalc = LedaFits() lalc.filename = 'data/test_lalc.LA' lalc.readLfile(n_ant=256, n_chans=109) lalc.exportFitsidi('data/test_lalc_direct.fitsidi') uvf = LedaFits('data/test_lalc_direct.fitsidi') ok_count = 0 # Check all header values h1("Testing header keywords") h2("Common") ok_count += compare_dicts(uvf.h_common, lalc.h_common) h2("Parameters") ok_count += compare_dicts(uvf.h_params, lalc.h_params) h2("Antenna") ok_count += compare_dicts(uvf.h_antenna, lalc.h_antenna) h2("Array Geometry") ok_count += compare_dicts(uvf.h_array_geometry, lalc.h_array_geometry) h2("Frequency") ok_count += compare_dicts(uvf.h_frequency, lalc.h_frequency) h2("Source") ok_count += compare_dicts(uvf.h_source, lalc.h_source) h2("UV DATA") ok_count += compare_dicts(uvf.h_uv_data, lalc.h_uv_data) h1("Testing data tables") h2("Antenna") ok_count += compare_dicts(uvf.d_antenna, lalc.d_antenna) h2("Array Geometry") ok_count += compare_dicts(uvf.d_array_geometry, lalc.d_array_geometry) h2("Frequency") ok_count += compare_dicts(uvf.d_frequency, lalc.d_frequency) h2("Source") ok_count += compare_dicts(uvf.d_source, lalc.d_source) assert ok_count == 11 try: assert repr(lalc) == repr(uvf) except AssertionError: print "ERROR: __repr__ outputs do not match" print repr(uvf) print repr(lalc) os.remove('data/test_lalc_direct.fitsidi')
def test_lalc_json(): lalc = LedaFits() lalc.filename = 'data/test_lalc.LA' lalc.readLfile(n_ant=256, n_chans=109) lalc.readJson('data/test_lalc_json/d_antenna.json') lalc.generateUVW() uu = lalc.d_uv_data["UU"] vv = lalc.d_uv_data["VV"] ww = lalc.d_uv_data["WW"] #print uu #print len(uu) print "Checking data...", try: assert len(uu) == len(vv) == len(ww) except: print len(uu), len(vv), len(ww) raise try: assert len(uu) == len(lalc.d_uv_data["BASELINE"]) except AssertionError: print len(uu), len(lalc.d_uv_data["BASELINE"]) raise try: assert len(uu) == len(lalc.d_uv_data["FLUX"]) except: print len(uu), len(lalc.d_uv_data["BASELINE"]) raise print "OK" lalc.remove_miriad_baselines() h2("Exporting sewed data...") lalc.exportFitsidi('data/test_lalc_uvw.fitsidi', '../config/config.xml')
def remove_miriad_lalc(): uvf = LedaFits('data/test_lalc.fitsidi') uvf.remove_miriad_baselines() uvf.exportFitsidi('data/test_lalc_255.fitsidi')