# -*- coding: utf-8 -*-
#
# Copyright © 2017 markdown-textwrap contributors.
#
# This file is part of markdown-textwrap.
#
# markdown-textwrap is free software: you can redistribute it
# and/or modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation, either version 3 of
# the License, or (at your option) any later version.
#
# markdown-textwrap is distributed in the hope that it will be
# useful, but WITHOUT ANY WARRANTY; without even the implied
# warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
# See the GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with markdown-textwrap (see COPYING). If not, see
# <http://www.gnu.org/licenses/>.
import textwrap
from mistune import Renderer
from nose.tools import assert_equal
from pkg_resources import resource_string, resource_filename
from md_tw import TWBlockLexer, TWInlineLexer, TWRenderer, TWMarkdown
def _get_data(f):
rs = resource_string(__name__, '/'.join(['data', f]))
return rs.decode()
def _get_data_path(f):
return resource_filename(__name__, '/'.join(['data', f]))
class TestTWBlockLexer(object):
def setup(self):
self.bl = TWBlockLexer()
def test_default_rules_contents(self):
assert_equal(self.bl.default_rules, ['paragraph', 'text'])
def teardown(self):
pass
class TestTWInlineLexer(object):
def setup(self):
renderer = Renderer()
self.il = TWInlineLexer(renderer)
def test_default_rules_contents(self):
assert_equal(self.il.default_rules, [])
def teardown(self):
pass
class TestTWRenderer(object):
def setup(self):
pass
def test_tw_obj_with_default_width(self):
renderer = TWRenderer()
# Check existence of textwrap.TexWrapper object.
assert isinstance(renderer.tw, textwrap.TextWrapper)
# Check its width
assert_equal(renderer.tw.width, 72)
def test_tw_obj_with_custom_width(self):
renderer = TWRenderer(tw_width=80)
# Check existence of textwrap.TexWrapper object.
assert isinstance(renderer.tw, textwrap.TextWrapper)
# Check its width
assert_equal(renderer.tw.width, 80)
def test_tw_set_options_with_valid_opts(self):
renderer = TWRenderer()
# Set valid options
renderer._tw_set_options(
width=80,
initial_indent='> ',
subsequent_indent=' ',
drop_whitespace=False)
# Confirm options are set.
assert_equal(renderer.tw.width, 80)
assert_equal(renderer.tw.initial_indent, '> ')
assert_equal(renderer.tw.subsequent_indent, ' ')
assert_equal(renderer.tw.drop_whitespace, False)
def test_tw_set_options_with_invalid_opts(self):
renderer = TWRenderer()
# Set invalid options
renderer._tw_set_options(
erase_bumps=True,
destroy_ampersands=False,
end_width='வருகிறேன்',
insert_between_paragraphs='time bombs')
# Confirm options are not set.
assert_equal(getattr(renderer.tw, 'erase_bumps', None), None)
assert_equal(getattr(renderer.tw, 'destroy_ampersands',
None), None)
assert_equal(getattr(renderer.tw, 'end_width', None), None)
assert_equal(getattr(renderer.tw, 'insert_between_paragraphs',
None), None)
def teardown(self):
pass
class TestTWMarkdown(object):
def setup(self):
self.md = TWMarkdown()
def test_renderer_obj(self):
assert isinstance(self.md.renderer, TWRenderer)
def test_inline_obj(self):
assert isinstance(self.md.inline, TWInlineLexer)
def test_block_obj(self):
assert isinstance(self.md.block, TWBlockLexer)
def teardown(self):
pass
class TestTextWrapParagraphs(object):
def setup(self):
self.md = TWMarkdown()
def test_tw_plain_paragraphs(self):
txt = _get_data('paragraphs.md')
expected_wrapped_txt = _get_data('paragraphs-wrapped.md')
wrapped_txt = self.md(txt)
assert_equal(wrapped_txt, expected_wrapped_txt)
def test_tw_paragraphs_with_inline(self):
txt = _get_data('paragraphs-with-inline.md')
expected_wrapped_txt = _get_data(
'paragraphs-with-inline-wrapped.md')
wrapped_txt = self.md(txt)
assert_equal(wrapped_txt, expected_wrapped_txt)
def teardown(self):
pass
