def test_generate_fitsidi(): """ Generate a FITS-IDI file """ filename_uvf = 'data/test_lalc.uvfits' filename_idi = filename_uvf.rstrip('.uvfits')+'.fitsidi' uvf = InterFits(filename_uvf) uvf.exportFitsidi(filename_idi)
def test_compare_idi_generated(): """ Check that creating a fits-idi->interfits->fits-idi creates identical data. """ idi = InterFits('data/test_lalc.fitsidi') idi.exportFitsidi('data/test_lalc2.fitsidi') idi2 = InterFits('data/test_lalc2.fitsidi') ok_count = 0 # Check all header values h1("Testing header keywords") h2("Common") ok_count += compare_dicts(idi.h_common, idi2.h_common) h2("Parameters") ok_count += compare_dicts(idi.h_params, idi2.h_params) h2("Antenna") ok_count += compare_dicts(idi.h_antenna, idi2.h_antenna) h2("Array Geometry") ok_count += compare_dicts(idi.h_array_geometry, idi2.h_array_geometry) h2("Frequency") ok_count += compare_dicts(idi.h_frequency, idi2.h_frequency) h2("Source") ok_count += compare_dicts(idi.h_source, idi2.h_source) h2("UV DATA") ok_count += compare_dicts(idi.h_uv_data, idi2.h_uv_data) h1("Testing data tables") h2("Antenna") ok_count += compare_dicts(idi.d_antenna, idi2.d_antenna) h2("Array Geometry") ok_count += compare_dicts(idi.d_array_geometry, idi2.d_array_geometry) h2("Frequency") ok_count += compare_dicts(idi.d_frequency, idi2.d_frequency) h2("Source") ok_count += compare_dicts(idi.d_source, idi2.d_source) h2("UV DATA") ok_count += compare_dicts(idi.d_uv_data, idi2.d_uv_data) assert ok_count == 12 os.remove('data/test_lalc2.xml') os.remove('data/test_lalc2.fitsidi')