• Search by Module
• Search by Words
• Search Projects

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

• ipymd-master
  • Makefile
  • ipymd
    • utils
      • tempdir.py
      • __init__.py
      • utils.py
      • tests
        • test_utils.py
        • test_tempdir.py
        • __init__.py
    • lib
      • markdown.py
      • __init__.py
      • tests
        • test_base_lexer.py
        • test_markdown.py
        • test_opendocument.py
        • test_python.py
        • __init__.py
      • python.py
      • opendocument.py
      • base_lexer.py
    • ext
      • six.py
      • __init__.py
    • __init__.py
    • core
      • prompt.py
      • scripts.py
      • contents_manager.py
      • format_manager.py
      • __init__.py
      • tests
        • test_scripts.py
        • test_prompt.py
        • __init__.py
        • test_core.py
    • formats
      • notebook.py
      • markdown.py
      • atlas.py
      • __init__.py
      • tests
        • test_utils.py
        • test_atlas.py
        • test_markdown.py
        • test_opendocument.py
        • _utils.py
        • test_python.py
        • test_notebook.py
        • __init__.py
      • python.py
      • opendocument.py
  • .coveragerc
  • examples
    • ex1.notebook.ipynb
    • ex1.py
    • ex2.markdown.md
    • ex2.python.py
    • ex4.markdown.md
    • ex1.atlas.md
    • ex3.markdown.md
    • ex4.py
    • ex1.opendocument.odt
    • ex2.notebook.ipynb
    • ex3.notebook.ipynb
    • ex3.py
    • ex1.python.py
    • ex1.markdown.md
    • ex2.opendocument.odt
    • ex4.notebook.ipynb
    • ex2.atlas.md
    • ex2.py
  • setup.py
  • requirements-dev.txt
  • setup.cfg
  • .travis.yml
  • README.md
  • .gitignore
  • LICENSE.txt
  • MANIFEST.in

# -*- coding: utf-8 -*-

"""Testing utilities."""

#------------------------------------------------------------------------------
# Imports
#------------------------------------------------------------------------------

import os
import os.path as op
import difflib
from pprint import pprint

from ...core.format_manager import format_manager, convert
from ...core.scripts import _load_file, _save_file
from ...ext.six import exec_


#------------------------------------------------------------------------------
# Test Markdown parser
#------------------------------------------------------------------------------

def _script_dir():
    return op.dirname(op.realpath(__file__))


def _test_file_path(basename, format=None):
    """Return the full path to an example filename in the 'examples'
    directory."""
    if format is not None:
        file_extension = format_manager().file_extension(format)
        filename = basename + '.' + format + file_extension
    else:
        # format=None ==> .py test file
        filename = basename + '.py'
    return op.realpath(op.join(_script_dir(), '../../../examples', filename))


def _exec_test_file(basename):
    """Return the 'output' object defined in a Python file."""
    path = _test_file_path(basename)
    with open(path, 'r') as f:
        contents = f.read()
    ns = {}
    exec_(contents, ns)
    return ns.get('output', None)


def _read_test_file(basename, format):
    """Read a test file."""
    path = _test_file_path(basename, format)
    return _load_file(path, format)


def _test_reader(basename, format, ignore_notebook_meta=True):
    """Return converted and expected ipymd cells of a given example."""
    contents = _read_test_file(basename, format)
    converted = convert(contents, from_=format)
    expected = _exec_test_file(basename)
    converted = [cell for cell in converted
                 if (not ignore_notebook_meta) or
                 cell["cell_type"] != "notebook_metadata"]

    return converted, expected


def _test_writer(basename, format):
    """Return converted and expected ipymd cells of a given example."""
    cells = _exec_test_file(basename)
    converted = convert(cells, to=format)
    expected = _read_test_file(basename, format)
    return converted, expected