Homec4science
Diffusion lammps (lammps-icms) Browse

lammps/doc/utils/converters/tests/test_txt2rst.py
fc80281fd990lammps-icms

Last Change
Raw File

test_txt2rst.py
View Options

# LAMMPS Documentation Utilities
#
# Copyright (C) 2015 Richard Berger
#
# This program 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.
#
# This program 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 this program. If not, see <http://www.gnu.org/licenses/>.
import io
import tempfile
import unittest
from lammpsdoc import txt2rst
class TestBasicFormatting(unittest.TestCase):
def setUp(self):
self.txt2rst = txt2rst.Txt2Rst()
def test_empty_string(self):
self.assertEqual(self.txt2rst.convert(""), "")
def test_single_paragraph(self):
self.assertEqual("Hello World!\n\n", self.txt2rst.convert("Hello World!\n"))
def test_two_paragraphs(self):
s = self.txt2rst.convert("Hello World!\n\nBye World!\n")
self.assertEqual("Hello World!\n\n"
"Bye World!\n\n", s)
def test_line_concat(self):
s = self.txt2rst.convert("Hello World!\\\nBye World!\n")
self.assertEqual(s, "Hello World!Bye World!\n\n")
def test_html_pass_through(self):
s = self.txt2rst.convert("<div>Raw HTML</div>\n")
self.assertEqual(s, ".. raw:: html\n\n"
" <div>Raw HTML</div>\n\n")
def test_ignore_html_only_block(self):
s = self.txt2rst.convert("<!-- HTML_ONLY -->\n"
"content :p\n"
"<!-- END_HTML_ONLY -->\n")
self.assertEqual("", s)
def test_pass_through_raw_rst(self):
raw_rst = ".. toctree::\n" \
" :maxdepth: 2\n" \
" :numbered:\n" \
"\n" \
" Introduction\n"
s = self.txt2rst.convert("<!-- RST\n" + raw_rst + "END_RST -->\n")
self.assertEqual(raw_rst, s)
class TestMarkup(unittest.TestCase):
def setUp(self):
self.markup = txt2rst.RSTMarkup()
self.txt2rst = txt2rst.Txt2Rst()
def test_bold(self):
self.assertEqual("**bold**", self.markup.convert("[bold]"))
def test_italic(self):
self.assertEqual("*italic*", self.markup.convert("{italic}"))
def test_escape_markup(self):
s = self.markup.convert("[bold] = \\[bold\\]\n"
"{italic} = \\{italic\\}\n")
self.assertEqual("**bold** = [bold]\n"
"*italic* = {italic}\n", s)
def test_paragraph_with_italic(self):
self.assertEqual("A sentence with a *italic* word", self.markup.convert("A sentence with a {italic} word"))
def test_paragraph_with_partial_italic(self):
self.assertEqual("A sentence with a partial *italic*\ normal word",
self.markup.convert("A sentence with a partial {italic}normal word"))
def test_paragraph_with_partial_bold(self):
self.assertEqual("A sentence with a partial **bold**\ normal word",
self.markup.convert("A sentence with a partial [bold]normal word"))
def test_paragraph_with_mixed_formats(self):
self.assertEqual("A sentence with a partial normal\ **bold**\ *italic*\ normal word",
self.markup.convert("A sentence with a partial normal[bold]{italic}normal word"))
def test_link_markup(self):
self.assertEqual("`Text <link>`_", self.markup.convert('"Text"_link'))
def test_document_cross_reference_link(self):
self.assertEqual(":doc:`Text <link>`", self.markup.convert('"Text"_link.html'))
def test_user_atc_link(self):
self.assertEqual("`Text <USER/atc/link.html>`_", self.markup.convert('"Text"_USER/atc/link.html'))
def test_external_link(self):
self.assertEqual("`Text <http://site/index.html>`_", self.markup.convert('"Text"_http://site/index.html'))
def test_multiline_link_markup(self):
s = self.txt2rst.convert('"Te\n'
'xt"_link\n')
self.assertEqual("`Te xt <link>`_\n\n", s)
def test_ignore_punctuation_in_link(self):
self.assertEqual("`Text <link>`_.", self.markup.convert('"Text"_link.'))
self.assertEqual("`Text <link>`_,", self.markup.convert('"Text"_link,'))
self.assertEqual("`Text <link>`_;", self.markup.convert('"Text"_link;'))
self.assertEqual("`Text <link>`_:", self.markup.convert('"Text"_link:'))
self.assertEqual("`Text <link>`_?", self.markup.convert('"Text"_link?'))
self.assertEqual("`Text <link>`_!", self.markup.convert('"Text"_link!'))
self.assertEqual("`Text <link>`_(", self.markup.convert('"Text"_link('))
self.assertEqual("`Text <link>`_)", self.markup.convert('"Text"_link)'))
class TestFormatting(unittest.TestCase):
def setUp(self):
self.txt2rst = txt2rst.Txt2Rst()
def test_paragraph_formatting(self):
s = self.txt2rst.convert("Hello :p\n")
self.assertEqual("Hello\n\n", s)
def test_two_paragraphs_through_formatting(self):
text = "Hello :p\nBye :p\n"
p = list(self.txt2rst.paragraphs(text))
s = self.txt2rst.convert(text)
self.assertEqual(len(p), 2)
self.assertEqual(s, "Hello\n"
"\n"
"Bye\n"
"\n")
def test_break_formatting(self):
s = self.txt2rst.convert("Hello :b\n")
self.assertEqual("Hello\n", s)
def test_preformat_formatting(self):
s = self.txt2rst.convert("Hello :pre\n")
self.assertEqual(".. parsed-literal::\n\n"
" Hello\n\n", s)
def test_preformat_formatting_with_indentation(self):
s = self.txt2rst.convert(" Hello\n"
" World :pre\n")
self.assertEqual(".. parsed-literal::\n\n"
" Hello\n"
" World\n\n", s)
def test_header_formatting(self):
s = self.txt2rst.convert("Level 1 :h1\n"
"Level 2 :h2\n"
"Level 3 :h3\n"
"Level 4 :h4\n"
"Level 5 :h5\n"
"Level 6 :h6\n")
self.assertEqual("Level 1\n"
"#######\n"
"\n"
"Level 2\n"
"*******\n"
"\n"
"Level 3\n"
"=======\n"
"\n"
"Level 4\n"
"-------\n"
"\n"
"Level 5\n"
"^^^^^^^\n"
"\n"
"Level 6\n"
'"""""""\n'
'\n', s)
def test_filter_header_numbers(self):
s = self.txt2rst.convert("1.1 Level :h1\n")
self.assertEqual("Level\n"
"#####\n\n", s)
def test_all_breaks(self):
s = self.txt2rst.convert("one\n"
"two\n"
"three :all(b)\n")
self.assertEqual("| one\n"
"| two\n"
"| three \n"
"| \n"
"\n", s)
def test_links_with_all_breaks(self):
s = self.txt2rst.convert("\"one\"_link\n"
"\"two\"_link\n"
"\"three\"_link :all(b)\n")
self.assertEqual("| `one <link>`_\n"
"| `two <link>`_\n"
"| `three <link>`_ \n"
"| \n"
"\n", s)
class TestListFormatting(unittest.TestCase):
def setUp(self):
self.txt2rst = txt2rst.Txt2Rst()
def test_unordered_list(self):
s = self.txt2rst.convert("one\n"
"two\n"
"three :ul\n")
self.assertEqual("* one\n"
"* two\n"
"* three\n\n", s)
def test_elementwise_unordered_list(self):
s = self.txt2rst.convert("one :ulb,l\n"
"two :l\n"
"three :ule,l\n")
self.assertEqual("* one\n"
"* two\n"
"* three\n\n", s)
def test_multi_line_unordered_list_elements(self):
s = self.txt2rst.convert("one :ulb,l\n"
"two\n"
"words :l\n"
"three :ule,l\n")
self.assertEqual("* one\n"
"* two\n"
" words\n"
"* three\n\n", s)
def test_ordered_list(self):
s = self.txt2rst.convert("one\n"
"two\n"
"three :ol\n")
self.assertEqual("1. one\n"
"2. two\n"
"3. three\n\n", s)
def test_elementwise_ordered_list(self):
s = self.txt2rst.convert("one :olb,l\n"
"two :l\n"
"three :ole,l\n")
self.assertEqual("#. one\n"
"#. two\n"
"#. three\n\n", s)
def test_multi_line_ordered_list_elements(self):
s = self.txt2rst.convert("one :olb,l\n"
"two\n"
"words :l\n"
"three :ole,l\n")
self.assertEqual("#. one\n"
"#. two\n"
" words\n"
"#. three\n\n", s)
def test_paragraphs_ordered_list(self):
s = self.txt2rst.convert("first\n"
"paragraph :olb,l\n"
"second\n"
"paragraph :l\n"
"third\n"
"paragraph :ole,l\n")
self.assertEqual("#. first\n"
" paragraph\n"
"#. second\n"
" paragraph\n"
"#. third\n"
" paragraph\n\n", s)
def test_paragraphs_unordered_list(self):
s = self.txt2rst.convert("first\n"
"paragraph :ulb,l\n"
"second\n"
"paragraph :l\n"
"third\n"
"paragraph :ule,l\n")
self.assertEqual("* first\n"
" paragraph\n"
"* second\n"
" paragraph\n"
"* third\n"
" paragraph\n\n", s)
def test_definition_list(self):
s = self.txt2rst.convert("A\n"
"first\n"
"B\n"
"second :dl\n")
self.assertEqual("A\n"
" first\n"
"\n"
"B\n"
" second\n"
"\n\n", s)
class TestSpecialCommands(unittest.TestCase):
def setUp(self):
self.txt2rst = txt2rst.Txt2Rst()
def test_line(self):
self.txt2rst.document_filters = []
s = self.txt2rst.convert("one :line\n")
self.assertEqual("\n"
"----------\n"
"\n"
"one\n", s)
def test_image(self):
s = self.txt2rst.convert("one :image(file)\n")
self.assertEqual(".. image:: file\n"
"one\n", s)
def test_centered_image(self):
s = self.txt2rst.convert(":image(file),c\n")
self.assertEqual(".. image:: file\n"
" :align: center\n\n", s)
def test_image_with_link(self):
s = self.txt2rst.convert("one :image(file,link)\n")
self.assertEqual(s, ".. image:: file\n"
" :target: link\n"
"one\n")
def test_thumbnail_image(self):
# requires sphinxcontrib-images extension
s = self.txt2rst.convert("one :image(file,large_file.jpg)\n")
self.assertEqual(s, ".. thumbnail:: large_file.jpg\n"
"one\n")
def test_internal_reference_link(self):
s = self.txt2rst.convert("one :link(name)\n"
"a \"link\"_name to above\n")
self.assertEqual(".. _name:\n"
"\n"
"one \n\n"
"a :ref:`link <name>` to above\n\n", s)
def test_local_anchor_link(self):
s = self.txt2rst.convert("one :link(name)\n"
"a \"link\"_#name to above\n")
self.assertEqual(".. _name:\n"
"\n"
"one \n\n"
"a :ref:`link <name>` to above\n\n", s)
def test_define_link_alias(self):
s = self.txt2rst.convert("one :link(alias,value)\n"
"\"test\"_alias\n")
self.assertEqual(".. _alias: value\n\n"
"one \n"
"\n"
"`test <alias_>`_\n\n", s)
class TestTableCommand(unittest.TestCase):
def setUp(self):
self.txt2rst = txt2rst.Txt2Rst()
def test_convert_table_to_grid_table(self):
s = self.txt2rst.convert("a,b,c :tb")
table = "+---+---+---+\n" \
"| a | b | c |\n" \
"+---+---+---+\n\n"
self.assertEqual(table, s)
def test_avoid_rst_syntax_conflicts_with_table_separator(self):
s = self.txt2rst.convert("\"a\"_test.html: b: c :tb(s=:)")
table = "+-----------------+---+---+\n" \
"| :doc:`a <test>` | b | c |\n" \
"+-----------------+---+---+\n\n"
self.assertEqual(table, s)
class TestTxt2RstCLI(unittest.TestCase):
def setUp(self):
self.out = io.StringIO()
self.err = io.StringIO()
self.app = txt2rst.Txt2RstConverter()
def test_convert_single_file(self):
with tempfile.NamedTemporaryFile(mode='w+t') as f:
f.write('Hello World!\n')
f.flush()
args = [f.name]
self.app.run(args=args, out=self.out, err=self.err)
self.assertEqual("Hello World!\n\n", self.out.getvalue())
self.assertEqual("Converting " + f.name + " ...\n", self.err.getvalue())
class TestMathMarkup(unittest.TestCase):
def setUp(self):
self.markup = txt2rst.RSTMarkup()
self.txt2rst = txt2rst.Txt2Rst()
def test_detect_latex_equation(self):
s = self.txt2rst.convert("\\begin\\{equation\\} T_\\{ij\\}(r_\\{ij\\}) = 1 - \\left( 1 +\n"
"\\frac\{s_\\{ij\} r_\\{ij\} \\}\\{2\\} \\right)\n"
"\\exp \\left( - s_\\{ij\\} r_\\{ij\\} \\right) \\end\\{equation\\}\n")
self.assertEqual("\n.. math::\n\n"
" \\begin{equation} T_{ij}(r_{ij}) = 1 - \\left( 1 +\n"
" \\frac{s_{ij} r_{ij} }{2} \\right)\n"
" \\exp \\left( - s_{ij} r_{ij} \\right) \\end{equation}\n\n", s)
def test_detect_inline_latex_equation(self):
s = self.txt2rst.convert("Masses: \\begin\\{equation\\} M' = M + m \\end\\{equation\\}\n"
"\\begin\\{equation\\} m' = \\frac \\{M\\, m \\} \\{M'\\} \\end\\{equation\\}\n")
self.assertEqual("Masses:\n"
"\n.. math::\n\n"
" \\begin{equation} M' = M + m \end{equation}\n"
"\n"
"\n.. math::\n"
"\n"
" \\begin{equation} m' = \\frac {M\\, m } {M'} \\end{equation}\n"
"\n", s)
def test_detect_inline_math(self):
self.assertEqual(":math:`x^2`", self.markup.convert("\\( x^2 \\)"))
def test_detect_multiline_inline_math(self):
line = "\\(\\sqrt \\{ \\frac \\{2\, k_B \\mathtt\\{Tcom\\}\, m'\\}\n" \
"\\{\\mathrm dt\\, \\mathtt\\{damp\\_com\\} \\}\n" \
"\\} \\). :b\n" \
"\(f_r'\) is a random force proportional to\n"
expected = ":math:`\\sqrt { \\frac {2\\, k_B \\mathtt{Tcom}\, m'} " \
"{\\mathrm dt\\, \\mathtt{damp\\_com} } " \
"}`.\n" \
":math:`f_r'` is a random force proportional to\n\n"
self.assertEqual(expected, self.txt2rst.convert(line))
if __name__ == '__main__':
unittest.main()
c4science · Help