tests/test_no_ipynb.py

   1 import pytest
   2 import os
   3
   4 from tests.util import THIS_DIR
   5 from black import main, jupyter_dependencies_are_installed
   6 from click.testing import CliRunner
   7 from _pytest.tmpdir import tmpdir
   8
   9 pytestmark = pytest.mark.no_jupyter
  10
  11 runner = CliRunner()
  12
  13
  14 def test_ipynb_diff_with_no_change_single() -> None:
  15     jupyter_dependencies_are_installed.cache_clear()
  16     path = THIS_DIR / "data/notebook_trailing_newline.ipynb"
  17     result = runner.invoke(main, [str(path)])
  18     expected_output = (
  19         "Skipping .ipynb files as Jupyter dependencies are not installed.\n"
  20         "You can fix this by running ``pip install black[jupyter]``\n"
  21     )
  22     assert expected_output in result.output
  23
  24
  25 def test_ipynb_diff_with_no_change_dir(tmpdir: tmpdir) -> None:
  26     jupyter_dependencies_are_installed.cache_clear()
  27     runner = CliRunner()
  28     nb = os.path.join("tests", "data", "notebook_trailing_newline.ipynb")
  29     tmp_nb = tmpdir / "notebook.ipynb"
  30     with open(nb) as src, open(tmp_nb, "w") as dst:
  31         dst.write(src.read())
  32     result = runner.invoke(main, [str(tmpdir)])
  33     expected_output = (
  34         "Skipping .ipynb files as Jupyter dependencies are not installed.\n"
  35         "You can fix this by running ``pip install black[jupyter]``\n"
  36     )
  37     assert expected_output in result.output