pytest.raises accept cutom message only when used as context manager

Author: palaviv

_pytest/python.py

@@ -1412,12 +1412,11 @@ def raises(expected_exception, *args, **kwargs):
     elif not isclass(expected_exception):
         raise TypeError(msg % type(expected_exception))
 
-    if "message" in kwargs:
-        message = kwargs.pop("message")
-    else:
-        message = "DID NOT RAISE {0}".format(expected_exception)
+    message = "DID NOT RAISE {0}".format(expected_exception)
 
     if not args:
+        if "message" in kwargs:
+            message = kwargs.pop("message")
         return RaisesContext(expected_exception, message)
     elif isinstance(args[0], str):
         code, = args

testing/python/raises.py

@@ -76,20 +76,23 @@ def test_no_raise_message(self):
             pytest.raises(ValueError, int, '0')
         except pytest.raises.Exception as e:
             assert e.msg == "DID NOT RAISE {0}".format(repr(ValueError))
+        else:
+            assert False, "Expected pytest.raises.Exception"
+
         try:
             with pytest.raises(ValueError):
                 pass
         except pytest.raises.Exception as e:
             assert e.msg == "DID NOT RAISE {0}".format(repr(ValueError))
+        else:
+            assert False, "Expected pytest.raises.Exception"
 
     def test_costum_raise_message(self):
         message = "TEST_MESSAGE"
-        try:
-            pytest.raises(ValueError, int, '0', message=message)
-        except pytest.raises.Exception as e:
-            assert e.msg == message
         try:
             with pytest.raises(ValueError, message=message):
                 pass
         except pytest.raises.Exception as e:
             assert e.msg == message
+        else:
+            assert False, "Expected pytest.raises.Exception"