Make ipynb tests compatible with ipython 8.3.0+ (#3008)

This commit is contained in:
Marco Edward Gorelli 2022-04-14 00:13:33 +01:00 committed by GitHub
parent 96bd428524
commit 712f8b37fb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1,3 +1,4 @@
import contextlib
from dataclasses import replace
import pathlib
import re
@ -18,6 +19,8 @@
from _pytest.monkeypatch import MonkeyPatch
from tests.util import DATA_DIR
with contextlib.suppress(ModuleNotFoundError):
import IPython
pytestmark = pytest.mark.jupyter
pytest.importorskip("IPython", reason="IPython is an optional dependency")
pytest.importorskip("tokenize_rt", reason="tokenize-rt is an optional dependency")
@ -139,10 +142,15 @@ def test_non_python_magics(src: str) -> None:
format_cell(src, fast=True, mode=JUPYTER_MODE)
@pytest.mark.skipif(
IPython.version_info < (8, 3),
reason="Change in how TransformerManager transforms this input",
)
def test_set_input() -> None:
src = "a = b??"
with pytest.raises(NothingChanged):
format_cell(src, fast=True, mode=JUPYTER_MODE)
expected = "??b"
result = format_cell(src, fast=True, mode=JUPYTER_MODE)
assert result == expected
def test_input_already_contains_transformed_magic() -> None: