def convert_df_to_gwosc_df(df: pd.DataFrame) -> pd.DataFrame:
converted_df = df.copy()
# rename
converted_df['GPS'] = converted_df['tc']
converted_df['chirp_mass'] = converted_df['mchirp']
converted_df['symmetric_mass_ratio'] = converted_df['eta']
converted_df['luminosity_distance'] = converted_df['DL']
converted_df['ra'] = converted_df['RA']
converted_df['dec'] = converted_df['DEC']
# re-parameterisation
converted_df['mass_ratio']= \
symmetric_mass_ratio_to_mass_ratio(converted_df['symmetric_mass_ratio'])
converted_df['total_mass'] = \
chirp_mass_and_mass_ratio_to_total_mass(
converted_df['chirp_mass'],
converted_df['mass_ratio']
)
converted_df['mass_1'], converted_df['mass_2'] = \
total_mass_and_mass_ratio_to_component_masses(
converted_df['mass_ratio'],
converted_df['total_mass']
)
converted_df['chi_eff'] = (converted_df['s1z'] +
converted_df['s2z'] *
converted_df['mass_ratio']) / \
(1 + converted_df['mass_ratio'])
return converted_df
def test_symmetric_mass_ratio_to_mass_ratio(self):
mass_ratio = conversion.symmetric_mass_ratio_to_mass_ratio(self.symmetric_mass_ratio)
self.assertAlmostEqual(self.mass_ratio, mass_ratio)