add tests

davidbailey · Dec 14, 2019 · b4ac793 · b4ac793
1 parent ca2f0e0
commit b4ac793
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/test_likelihood.py b/tests/test_likelihood.py
@@ -5,21 +5,22 @@
 
 from rail import Likelihood
 
+LAM = 0.5
 
 class TestLikelihood(unittest.TestCase):
     """
     Class to test a likelihood
     """
 
     def setUp(self):
-        self.likelihood = Likelihood(0.5)
+        self.likelihood = Likelihood(LAM)
 
     def test_likelihood(self):
         """
         Test a likelihood
         """
-        self.assertEqual(self.likelihood["name"], "0.5")
-        self.assertEqual(self.likelihood["lam"], 0.5)
+        self.assertEqual(self.likelihood["name"], str(LAM))
+        self.assertEqual(self.likelihood["lam"], LAM)
 
 
 if __name__ == "__main__":