remove message

Author: TomAugspurger

pandas/tests/util/test_assert_produces_warning.py

@@ -15,9 +15,3 @@ def f(a=FutureWarning, b=RuntimeWarning):
 def test_assert_produces_warning_honors_filter():
     with tm.assert_produces_warning(RuntimeWarning):
         f()
-
-
-@pytest.mark.filterwarnings('ignore:f1:FutureWarning')
-def test_assert_produces_warning_message():
-    with tm.assert_produces_warning(FutureWarning, message='f2'):
-        f(FutureWarning, FutureWarning)

pandas/util/testing.py

@@ -2571,8 +2571,7 @@ def exception_matches(self, exc_type, exc_value, trace_back):
 
 @contextmanager
 def assert_produces_warning(expected_warning=Warning, filter_level="always",
-                            clear=None, check_stacklevel=True,
-                            message=''):
+                            clear=None, check_stacklevel=True):
     """
     Context manager for running code expected to either raise a specific
     warning, or not raise any warnings. Verifies that the code raises the
@@ -2610,9 +2609,6 @@ class for all warnings. To check that no warning is returned,
         If True, displays the line that called the function containing
         the warning to show were the function is called. Otherwise, the
         line that implements the function is displayed.
-    message : str, default ''
-        Use in the filter with `filter_level` and `expected_warning`
-        the control which warnings the filter applies to.
 
     Examples
     --------
@@ -2652,7 +2648,8 @@ class for all warnings. To check that no warning is returned,
 
         saw_warning = False
         if expected_warning and filter_level:
-            warnings.filterwarnings(filter_level, message, expected_warning)
+            warnings.filterwarnings(filter_level, message='',
+                                    category=expected_warning)
         elif filter_level:
             # no expected warnings.
             warnings.simplefilter(filter_level)