Changed test according to probability code change

djgroen · Feb 23, 2024 · 0afab67 · 0afab67
1 parent cb4f6fd
commit 0afab67
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -24,12 +24,12 @@ def test_probability_with_mock(mock_random):
     assert utils.probability(0.4) is False
 
 
-def test_probability_raises_error():
+def test_probability_raises_warning():
     """Test the probability function raises an error."""
 
-    with pytest.raises(ValueError):
+    with pytest.warns(UserWarning):
         utils.probability(-0.1)
-    with pytest.raises(ValueError):
+    with pytest.warns(UserWarning):
         utils.probability(1.1)