Skip to content
12 changes: 11 additions & 1 deletion pandas/tests/arrays/categorical/test_warnings.py
Original file line number Diff line number Diff line change
Expand Up @@ -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))
12 changes: 11 additions & 1 deletion pandas/tests/indexes/test_base.py
Original file line number Diff line number Diff line change
Expand Up @@ -2413,7 +2413,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))

Expand Down