Backport PR #31323: CI: Fix jedi upgrades causes deprecation warning (#31341)

Co-authored-by: Kaiqi Dong <[email protected]>

Author: 2 people

pandas/tests/arrays/categorical/test_warnings.py

@@ -14,6 +14,16 @@ async def test_tab_complete_warning(self, ip):
 
         code = "import pandas as pd; c = Categorical([])"
         await ip.run_code(code)
-        with tm.assert_produces_warning(None):
+
+        # GH 31324 newer jedi version raises Deprecation warning
+        import jedi
+
+        if jedi.__version__ < "0.16.0":
+            warning = tm.assert_produces_warning(None)
+        else:
+            warning = tm.assert_produces_warning(
+                DeprecationWarning, check_stacklevel=False
+            )
+        with warning:
             with provisionalcompleter("ignore"):
                 list(ip.Completer.completions("c.", 1))

pandas/tests/indexes/test_base.py

@@ -2408,7 +2408,17 @@ async def test_tab_complete_warning(self, ip):
 
         code = "import pandas as pd; idx = pd.Index([1, 2])"
         await ip.run_code(code)
-        with tm.assert_produces_warning(None):
+
+        # GH 31324 newer jedi version raises Deprecation warning
+        import jedi
+
+        if jedi.__version__ < "0.16.0":
+            warning = tm.assert_produces_warning(None)
+        else:
+            warning = tm.assert_produces_warning(
+                DeprecationWarning, check_stacklevel=False
+            )
+        with warning:
             with provisionalcompleter("ignore"):
                 list(ip.Completer.completions("idx.", 4))