• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• nara_wpe-master
  • examples
    • examples.rst
    • WPE_Numpy_online.ipynb
    • WPE_Numpy_offline.ipynb
    • WPE_Tensorflow_offline.ipynb
    • NTT_wrapper_offline.ipynb
    • WPE_Tensorflow_online.ipynb
  • LICENSE
  • README.rst
  • .pylintrc
  • setup.py
  • maintenance.org
  • data
  • nara_wpe
    • test_utils.py
    • ntt_wpe.py
    • benchmark_online_wpe.py
    • torch_wpe.py
    • __init__.py
    • tf_wpe.py
    • utils.py
    • wpe.py
  • .travis.yml
  • tests
    • test_notebooks.py
    • test_wpe.py
    • test_tf_wpe.py
  • .gitignore
  • docs
    • Makefile
    • make.bat
    • nara_wpe.rst
    • nara_wpe.gradient_overrides.rst
    • nara_wpe.utils.rst
    • nara_wpe.test_utils.rst
    • modules.rst
    • index.rst
    • conf.py
    • nara_wpe.tf_wpe.rst
    • nara_wpe.wpe.rst
    • make_apidoc.sh
    • nara_wpe.benchmark_online_wpe.rst

import sys
import os
import subprocess
import tempfile
import unittest

import nbformat

from nara_wpe import project_root


def _notebook_run(path):
    """Execute a notebook via nbconvert and collect output.
       :returns (parsed nb object, execution errors)
    """
    dirname = os.path.dirname(str(path))
    os.chdir(dirname)
    with tempfile.NamedTemporaryFile(suffix=".ipynb") as fout:
        args = ["jupyter", "nbconvert", "--to", "notebook", "--execute",
                "--ExecutePreprocessor.timeout=360",
                "--output", fout.name, str(path)]
        subprocess.check_call(args)

        fout.seek(0)
        nb = nbformat.read(fout, nbformat.current_nbformat)

    errors = [
        output for cell in nb.cells if "outputs" in cell
        for output in cell["outputs"]
        if output.output_type == "error"
    ]

    return nb, errors


@unittest.skipIf(sys.version_info < (3, 6, 0), 'Only with Python 3.6')
class TestNotebooks(unittest.TestCase):

    def setUp(self):
        self.root = project_root / 'examples'

    def test_wpe_numpy_offline(self):
        nb, errors = _notebook_run(self.root / 'WPE_Numpy_offline.ipynb')
        assert errors == []

    def test_wpe_numpy_online(self):
        nb, errors = _notebook_run(self.root / 'WPE_Numpy_online.ipynb')
        assert errors == []

    def test_wpe_tensorflow_offline(self):
        nb, errors = _notebook_run(self.root / 'WPE_Tensorflow_offline.ipynb')
        assert errors == []

    def test_wpe_tensorflow_online(self):
        nb, errors = _notebook_run(self.root / 'WPE_Tensorflow_online.ipynb')
        assert errors == []

    # def test_NTT_wrapper(self):
    #     nb, errors = _notebook_run(self.root / 'NTT_wrapper_offline.ipynb')
    #     assert errors == []