diff --git a/test/likelihood_tests.py b/test/likelihood_tests.py index a3d6da004a6b6690332c957acee07256c6b3d67b..49eea134e8f2a4f950a2b84f573667ed7db62603 100644 --- a/test/likelihood_tests.py +++ b/test/likelihood_tests.py @@ -19,6 +19,11 @@ class TestLikelihoodBase(unittest.TestCase): def tearDown(self): del self.likelihood + def test_repr(self): + self.likelihood = tupak.core.likelihood.Likelihood(parameters=['a', 'b']) + expected = 'Likelihood(parameters=[\'a\', \'b\'])' + self.assertEqual(expected, repr(self.likelihood)) + def test_base_log_likelihood(self): self.assertTrue(np.isnan(self.likelihood.log_likelihood()))