Update groupby corrwith's warning

Author: Aloqeely

pandas/core/groupby/generic.py

@@ -33,7 +33,10 @@
     Substitution,
     doc,
 )
-from pandas.util._exceptions import find_stack_level
+from pandas.util._exceptions import (
+    Pandas40DeprecationWarning,
+    find_stack_level,
+)
 
 from pandas.core.dtypes.common import (
     ensure_int64,
@@ -2791,7 +2794,7 @@ def corrwith(
         """
         warnings.warn(
             "DataFrameGroupBy.corrwith is deprecated",
-            FutureWarning,
+            Pandas40DeprecationWarning,
             stacklevel=find_stack_level(),
         )
         result = self._op_via_apply(

pandas/tests/groupby/test_all_methods.py

@@ -13,6 +13,8 @@
 
 import pytest
 
+from pandas.util._exceptions import Pandas40DeprecationWarning
+
 import pandas as pd
 from pandas import DataFrame
 import pandas._testing as tm
@@ -26,7 +28,7 @@ def test_multiindex_group_all_columns_when_empty(groupby_func):
     method = getattr(gb, groupby_func)
     args = get_groupby_method_args(groupby_func, df)
     if groupby_func == "corrwith":
-        warn = FutureWarning
+        warn = Pandas40DeprecationWarning
         warn_msg = "DataFrameGroupBy.corrwith is deprecated"
     else:
         warn = None
@@ -72,7 +74,7 @@ def test_dup_labels_output_shape(groupby_func, idx):
 
     args = get_groupby_method_args(groupby_func, df)
     if groupby_func == "corrwith":
-        warn = FutureWarning
+        warn = Pandas40DeprecationWarning
         warn_msg = "DataFrameGroupBy.corrwith is deprecated"
     else:
         warn = None

pandas/tests/groupby/test_apply.py

@@ -6,6 +6,8 @@
 import numpy as np
 import pytest
 
+from pandas.util._exceptions import Pandas40DeprecationWarning
+
 import pandas as pd
 from pandas import (
     DataFrame,
@@ -1198,7 +1200,7 @@ def test_apply_is_unchanged_when_other_methods_are_called_first(reduction_func):
     grp = df.groupby(by="a")
     args = get_groupby_method_args(reduction_func, df)
     if reduction_func == "corrwith":
-        warn = FutureWarning
+        warn = Pandas40DeprecationWarning
         msg = "DataFrameGroupBy.corrwith is deprecated"
     else:
         warn = None

pandas/tests/groupby/test_categorical.py

@@ -3,6 +3,8 @@
 import numpy as np
 import pytest
 
+from pandas.util._exceptions import Pandas40DeprecationWarning
+
 import pandas as pd
 from pandas import (
     Categorical,
@@ -1474,7 +1476,7 @@ def test_dataframe_groupby_on_2_categoricals_when_observed_is_true(reduction_fun
 
     args = get_groupby_method_args(reduction_func, df)
     if reduction_func == "corrwith":
-        warn = FutureWarning
+        warn = Pandas40DeprecationWarning
         warn_msg = "DataFrameGroupBy.corrwith is deprecated"
     else:
         warn = None
@@ -1520,7 +1522,7 @@ def test_dataframe_groupby_on_2_categoricals_when_observed_is_false(
         return
 
     if reduction_func == "corrwith":
-        warn = FutureWarning
+        warn = Pandas40DeprecationWarning
         warn_msg = "DataFrameGroupBy.corrwith is deprecated"
     else:
         warn = None
@@ -1919,7 +1921,7 @@ def test_category_order_reducer(
             getattr(gb, reduction_func)(*args)
         return
     if reduction_func == "corrwith":
-        warn = FutureWarning
+        warn = Pandas40DeprecationWarning
         warn_msg = "DataFrameGroupBy.corrwith is deprecated"
     else:
         warn = None

pandas/tests/groupby/test_groupby_dropna.py

@@ -2,6 +2,7 @@
 import pytest
 
 from pandas.compat.pyarrow import pa_version_under10p1
+from pandas.util._exceptions import Pandas40DeprecationWarning
 
 from pandas.core.dtypes.missing import na_value_for_dtype
 
@@ -544,7 +545,7 @@ def test_categorical_reducers(reduction_func, observed, sort, as_index, index_ki
 
     gb_filled = df_filled.groupby(keys, observed=observed, sort=sort, as_index=True)
     if reduction_func == "corrwith":
-        warn = FutureWarning
+        warn = Pandas40DeprecationWarning
         msg = "DataFrameGroupBy.corrwith is deprecated"
     else:
         warn = None
@@ -575,7 +576,7 @@ def test_categorical_reducers(reduction_func, observed, sort, as_index, index_ki
             expected = expected["size"].rename(None)
 
     if reduction_func == "corrwith":
-        warn = FutureWarning
+        warn = Pandas40DeprecationWarning
         msg = "DataFrameGroupBy.corrwith is deprecated"
     else:
         warn = None

pandas/tests/groupby/test_numeric_only.py

@@ -3,6 +3,7 @@
 import pytest
 
 from pandas._libs import lib
+from pandas.util._exceptions import Pandas40DeprecationWarning
 
 import pandas as pd
 from pandas import (
@@ -257,7 +258,7 @@ def test_numeric_only(kernel, has_arg, numeric_only, keys):
     if has_arg and numeric_only is True:
         # Cases where b does not appear in the result
         if kernel == "corrwith":
-            warn = FutureWarning
+            warn = Pandas40DeprecationWarning
             msg = "DataFrameGroupBy.corrwith is deprecated"
         else:
             warn = None
@@ -304,7 +305,7 @@ def test_numeric_only(kernel, has_arg, numeric_only, keys):
             msg = "'>' not supported between instances of 'type' and 'type'"
         with pytest.raises(exception, match=msg):
             if kernel == "corrwith":
-                warn = FutureWarning
+                warn = Pandas40DeprecationWarning
                 msg = "DataFrameGroupBy.corrwith is deprecated"
             else:
                 warn = None

pandas/tests/groupby/transform/test_transform.py

@@ -4,6 +4,7 @@
 import pytest
 
 from pandas._libs import lib
+from pandas.util._exceptions import Pandas40DeprecationWarning
 
 from pandas.core.dtypes.common import ensure_platform_int
 
@@ -1105,7 +1106,7 @@ def test_transform_agg_by_name(request, reduction_func, frame_or_series):
 
     args = get_groupby_method_args(reduction_func, obj)
     if func == "corrwith":
-        warn = FutureWarning
+        warn = Pandas40DeprecationWarning
         msg = "DataFrameGroupBy.corrwith is deprecated"
     else:
         warn = None
@@ -1476,7 +1477,7 @@ def test_as_index_no_change(keys, df, groupby_func):
     gb_as_index_true = df.groupby(keys, as_index=True)
     gb_as_index_false = df.groupby(keys, as_index=False)
     if groupby_func == "corrwith":
-        warn = FutureWarning
+        warn = Pandas40DeprecationWarning
         msg = "DataFrameGroupBy.corrwith is deprecated"
     else:
         warn = None

pandas/tests/io/json/test_pandas.py

@@ -151,7 +151,6 @@ def test_frame_non_unique_columns(self, orient, data, request):
         if df.iloc[:, 0].dtype == "datetime64[s]":
             expected_warning = Pandas40DeprecationWarning
 
-
         with tm.assert_produces_warning(expected_warning, match=msg):
             result = read_json(
                 StringIO(df.to_json(orient=orient)), orient=orient, convert_dates=["x"]