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