diff --git a/test/gw_likelihood_tests.py b/test/gw_likelihood_tests.py index c603a59e033e653b9595d6392872361411cc2c12..fa8e53b6ec255b5defd1ad98ccd7da0f9f39b46b 100644 --- a/test/gw_likelihood_tests.py +++ b/test/gw_likelihood_tests.py @@ -61,6 +61,11 @@ class TestBasicGWTransient(unittest.TestCase): np.nan_to_num(-np.inf)) self.likelihood.waveform_generator.parameters['mass_2'] = 29 + def test_repr(self): + expected = 'BasicGravitationalWaveTransient(interferometers={},\n\twaveform_generator={})'.format( + self.interferometers, self.waveform_generator) + self.assertEqual(expected, repr(self.likelihood)) + class TestGWTransient(unittest.TestCase): @@ -133,6 +138,12 @@ class TestGWTransient(unittest.TestCase): np.nan_to_num(-np.inf)) self.likelihood.waveform_generator.parameters['mass_2'] = 29 + def test_repr(self): + expected = 'GravitationalWaveTransient(interferometers={},\n\twaveform_generator={},\n\t' \ + 'time_marginalization={}, distance_marginalization={}, phase_marginalization={}, ' \ + 'prior={})'.format(self.interferometers, self.waveform_generator, False, False, False, self.prior) + self.assertEqual(expected, repr(self.likelihood)) + class TestTimeMarginalization(unittest.TestCase):