Fix test failures

Aloqeely · Jun 6, 2024 · 2570d57 · 2570d57
1 parent 02ec7b2
commit 2570d57
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pandas/tests/groupby/test_raises.py b/pandas/tests/groupby/test_raises.py
@@ -8,6 +8,8 @@
 import numpy as np
 import pytest
 
+from pandas.util._exceptions import Pandas40DeprecationWarning
+
 from pandas import (
     Categorical,
     DataFrame,
@@ -85,7 +87,7 @@ def df_with_cat_col():
 
 
 def _call_and_check(klass, msg, how, gb, groupby_func, args, warn_msg=""):
-    warn_klass = None if warn_msg == "" else FutureWarning
+    warn_klass = None if warn_msg == "" else Pandas40DeprecationWarning
     with tm.assert_produces_warning(warn_klass, match=warn_msg, check_stacklevel=False):
         if klass is None:
             if how == "method":