diff --git a/bilby/gw/likelihood.py b/bilby/gw/likelihood.py index 7c7dbe3db5d84f80c73558fabad1b36b71ddff19..6da8c8363c1ea56b92ff3561a1afd3eab40f6dfe 100644 --- a/bilby/gw/likelihood.py +++ b/bilby/gw/likelihood.py @@ -391,6 +391,14 @@ class GravitationalWaveTransient(likelihood.Likelihood): self.time_prior_array =\ self.priors['geocent_time'].prob(self._times) * delta_tc + @property + def interferometers(self): + return self._interferometers + + @interferometers.setter + def interferometers(self, interferometers): + self._interferometers = InterferometerList(interferometers) + class BasicGravitationalWaveTransient(likelihood.Likelihood): diff --git a/test/gw_likelihood_test.py b/test/gw_likelihood_test.py index 0dbfd147eb7b452114ed62b02f7cfe4ccca7c4c7..0811070fc161af2b8619d7c19ea9682860ccce00 100644 --- a/test/gw_likelihood_test.py +++ b/test/gw_likelihood_test.py @@ -134,6 +134,19 @@ class TestGWTransient(unittest.TestCase): 'priors={})'.format(self.interferometers, self.waveform_generator, False, False, False, self.prior) self.assertEqual(expected, repr(self.likelihood)) + def test_interferometers_setting_list(self): + ifos = [bilby.gw.detector.get_empty_interferometer(name=name) for name in ['H1', 'L1']] + self.likelihood.interferometers = ifos + self.assertListEqual(bilby.gw.detector.InterferometerList(ifos), self.likelihood.interferometers) + self.assertTrue(type(self.likelihood.interferometers) == bilby.gw.detector.InterferometerList) + + def test_interferometers_setting_interferometer_list(self): + ifos = bilby.gw.detector.InterferometerList([bilby.gw.detector.get_empty_interferometer(name=name) + for name in ['H1', 'L1']]) + self.likelihood.interferometers = ifos + self.assertListEqual(bilby.gw.detector.InterferometerList(ifos), self.likelihood.interferometers) + self.assertTrue(type(self.likelihood.interferometers) == bilby.gw.detector.InterferometerList) + class TestTimeMarginalization(unittest.TestCase):