Skip to content
Snippets Groups Projects

Use marginalized_parameters attribute for reconstruction

Merged Colm Talbot requested to merge flexible-marginalization-reconstruction into master
1 file
+ 6
5
Compare changes
  • Side-by-side
  • Inline
@@ -650,16 +650,16 @@ class TestMarginalizations(unittest.TestCase):
)
@parameterized.expand(
itertools.product(*itertools.repeat([True, False], 3)),
itertools.product(["regular", "roq"], *itertools.repeat([True, False], 3)),
name_func=lambda func, num, param: (
f"{func.__name__}_{num}_" + "_".join([
f"{func.__name__}_{num}__{param.args[0]}_" + "_".join([
["D", "P", "T"][ii] for ii, val
in enumerate(param.args) if val
in enumerate(param.args[1:]) if val
])
)
)
def test_marginalization_reconstruction(self, distance, phase, time):
if time and "ROQ" in self.__class__.__name__:
def test_marginalization_reconstruction(self, kind, distance, phase, time):
if time and kind == "roq":
pytest.skip("Time reconstruction not supported for ROQ likelihood")
marginalizations = dict(
geocent_time=time,
@@ -667,6 +667,7 @@ class TestMarginalizations(unittest.TestCase):
phase=phase,
)
like = self.get_likelihood(
kind=kind,
distance_marginalization=distance,
time_marginalization=time,
phase_marginalization=phase,
Loading