fix test

davidbailey · Dec 14, 2019 · 8dc4f76 · 8dc4f76
1 parent 4eff5b8
commit 8dc4f76
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_likelihood.py b/tests/test_likelihood.py
@@ -19,7 +19,8 @@ def test_likelihood(self):
         """
         Test a likelihood
         """
-        self.assertRaises(ValueError, Likelihood(-1))
+        with self.assertRaises(ValueError):
+            Likelihood(-1)
         self.assertEqual(self.likelihood["name"], str(LAM))
         self.assertEqual(self.likelihood["lam"], LAM)
         self.assertIsNotNone(self.likelihood.plot())