/usr/lib/python2.7/dist-packages/sphinx_gallery/tests/test_notebook.py is in python-sphinx-gallery 0.1.13-1ubuntu1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 | # -*- coding: utf-8 -*-
# Author: Óscar Nájera
# License: 3-clause BSD
r"""
Testing the Jupyter notebook parser
"""
from __future__ import division, absolute_import, print_function
import json
import tempfile
import os
import pytest
import sphinx_gallery.gen_rst as sg
from sphinx_gallery.notebook import (rst2md, jupyter_notebook, save_notebook,
python_to_jupyter_cli)
try:
FileNotFoundError
except NameError:
# Python2
FileNotFoundError = IOError
def test_latex_conversion():
"""Latex parsing from rst into Jupyter Markdown"""
double_inline_rst = r":math:`T<0` and :math:`U>0`"
double_inline_jmd = r"$T<0$ and $U>0$"
assert double_inline_jmd == rst2md(double_inline_rst)
align_eq = r"""
.. math::
\mathcal{H} &= 0 \\
\mathcal{G} &= D"""
align_eq_jmd = r"""
\begin{align}\mathcal{H} &= 0 \\
\mathcal{G} &= D\end{align}"""
assert align_eq_jmd == rst2md(align_eq)
def test_convert():
"""Test ReST conversion"""
rst = """hello
.. contents::
:local:
This is :math:`some` math :math:`stuff`.
.. note::
Interpolation is a linear operation that can be performed also on
Raw and Epochs objects.
.. warning::
Go away
For more details on interpolation see the page :ref:`channel_interpolation`.
.. _foo: bar
.. image:: foobar
:alt: me
:whatever: you
"""
markdown = """hello
This is $some$ math $stuff$.
<div class="alert alert-info"><h4>Note</h4><p>Interpolation is a linear operation that can be performed also on
Raw and Epochs objects.</p></div>
<div class="alert alert-danger"><h4>Warning</h4><p>Go away</p></div>
For more details on interpolation see the page `channel_interpolation`.
""" # noqa
assert rst2md(rst) == markdown
def test_jupyter_notebook():
"""Test that written ipython notebook file corresponds to python object"""
file_conf, blocks = sg.split_code_and_text_blocks('tutorials/plot_parse.py')
example_nb = jupyter_notebook(blocks)
with tempfile.NamedTemporaryFile('w', delete=False) as f:
save_notebook(example_nb, f.name)
try:
with open(f.name, "r") as fname:
assert json.load(fname) == example_nb
finally:
os.remove(f.name)
###############################################################################
# Notebook shell utility
def test_with_empty_args():
""" User passes no args, should fail with SystemExit """
with pytest.raises(SystemExit):
python_to_jupyter_cli([])
def test_missing_file():
""" User passes non existing file, should fail with FileNotFoundError """
with pytest.raises(FileNotFoundError) as excinfo:
python_to_jupyter_cli(['nofile.py'])
excinfo.match(r'No such file or directory.+nofile\.py')
def test_file_is_generated():
"""User passes good python file. Check notebook file is created"""
python_to_jupyter_cli(['examples/plot_quantum.py'])
assert os.path.isfile('examples/plot_quantum.ipynb')
os.remove('examples/plot_quantum.ipynb')
|