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)