def test_power_law_primary_mass_ratio_zero_above_mmax(self): for ii in range(self.n_test): parameters = self.power_prior.sample() p_m = mass.power_law_primary_mass_ratio(self.dataset, **parameters) self.assertEqual( xp.max(p_m[self.dataset["mass_1"] >= parameters["mmax"]]), 0.0 )
def test_power_law_primary_mass_ratio_zero_below_mmin(self): m2s = self.dataset["mass_1"] * self.dataset["mass_ratio"] for ii in range(self.n_test): parameters = self.power_prior.sample() p_m = mass.power_law_primary_mass_ratio(self.dataset, **parameters) self.assertEqual(xp.max(p_m[m2s <= parameters["mmin"]]), 0.0)