From 5a85d8e0da96e003e9b7b84480909c3c85b64f53 Mon Sep 17 00:00:00 2001
From: MoritzThomasHuebner <email@moritz-huebner.de>
Date: Wed, 5 Sep 2018 18:14:58 +1000
Subject: [PATCH] Moved tests for repr in here
---
test/gw_likelihood_tests.py | 11 +++++++++++
1 file changed, 11 insertions(+)
diff --git a/test/gw_likelihood_tests.py b/test/gw_likelihood_tests.py
index c603a59e0..fa8e53b6e 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):
--
GitLab