def test_update_file_values_correct(backup_file): append_hjd_correction_column(backup_file, column_name='hjd_correction') with fitsio.FITS(backup_file) as infile: catalogue = infile[1] ra, dec = [catalogue[key].read() for key in ['ra', 'dec']] hjd_column_data = catalogue['hjd_correction'].read() header = catalogue.read_header() sun_ra, sun_dec = [header[key] for key in ['sun_ra', 'sun_dec']] mjd = header['mjd'] hjd_values = compute_hjd_correction(mjd, ra, dec, sun_ra, sun_dec) assert allclose(hjd_column_data, hjd_values)
def test_hjd_computation(ref_jd, ref_ra, ref_dec, sun_ra, sun_dec, expected): result = compute_hjd_correction(ref_jd, ref_ra, ref_dec, sun_ra, sun_dec) assert isclose(result, expected, rtol=1E-1, atol=0)