Files
dungeon-sheets/tests/test_latex.py
T
Mark Wolfman 30369ce1d4 Ran flake8 and black linters, and other cleanup-related fixes.
Project now passes flake8 and black linter (also including more rst
cleanup). Moved latex related things to dedicated ``latex.py`` module,
and removed the ``makesheets -dF`` call from travis.
2021-04-16 11:10:17 -05:00

116 lines
4.7 KiB
Python

import unittest
from dungeonsheets import make_sheets, character, spells, features, latex
class MarkdownTestCase(unittest.TestCase):
"""Check that conversion of markdown formats to LaTeX code works
correctly."""
def test_rst_bold(self):
text = latex.rst_to_latex('**hello**')
self.assertEqual(text, '\n\\textbf{hello}\n')
def test_hit_dice(self):
text = latex.rst_to_latex('1d6+3')
self.assertEqual(text.strip("\n"), '\\texttt{1d6+3}')
def test_no_text(self):
text = latex.rst_to_latex(None)
self.assertEqual(text, '')
def test_verbatim(self):
text = latex.rst_to_latex('``hello, world``')
self.assertIn(r'\texttt{hello, world}', text)
def test_literal_backslash(self):
text = latex.rst_to_latex(r'\\')
self.assertEqual(r'\textbackslash{}', text.strip("\n"))
@unittest.skip("Headings are all screwed up because it treats them as the document title")
def test_headings(self):
# Simple heading by itself
text = latex.rst_to_latex('Hello, world\n------------\n\nGoodbye, world')
self.assertEqual('\\section*{Hello, world}\n', text)
# Simple heading with leading whitespace
text = latex.rst_to_latex(' Hello, world\n ============\n')
self.assertEqual('\\section*{Hello, world}\n', text)
# Heading with text after it
text = latex.rst_to_latex('Hello, world\n============\n\nThis is some text')
self.assertEqual('\\section*{Hello, world}\n\nThis is some text', text)
# Heading with text before it
text = latex.rst_to_latex('This is a paragraph\n\nHello, world\n============\n')
self.assertEqual('This is a paragraph\n\n\\section*{Hello, world}\n', text)
# Check that levels of headings are parsed appropriately
text = latex.rst_to_latex('Hello, world\n^^^^^^^^^^^^\n')
self.assertEqual('\\subsubsection*{Hello, world}\n', text)
text = latex.rst_to_latex('Hello, world\n^^^^^^^^^^^^\n', top_heading_level=3)
self.assertEqual('\\subparagraph*{Hello, world}\n', text)
# This is a bad heading missing with all the underline on one line
text = latex.rst_to_latex('Hello, world^^^^^^^^^^^^\n')
self.assertEqual('Hello, world\\^\\^\\^\\^\\^\\^\\^\\^\\^\\^\\^\\^\n', text)
def test_bullet_list(self):
tex = latex.rst_to_latex("\n- Hello\n- World\n\n")
expected_tex = "\\begin{itemize}\n\\item Hello\n\n\\item World\n\\end{itemize}"
self.assertEqual(expected_tex, tex.strip("\n"))
# Other bullet characters
tex = latex.rst_to_latex("\n* Hello\n* World\n\n")
self.assertEqual(expected_tex, tex.strip("\n"))
tex = latex.rst_to_latex("\n+ Hello\n+ World\n\n")
self.assertEqual(expected_tex, tex.strip("\n"))
# A real list taken from a docstring
real_list = """
- Secondhand (you have heard of the target) - +5
- Firsthand (you have met the target) - +0
- Familiar (you know the target well) - -5
"""
tex = latex.rst_to_latex(real_list)
self.assertIn("\\begin{itemize}", tex)
def test_multiline_bullet_list(self):
md_list = """
- Secondhand (you have heard
of the target) - +5
- Firsthand (you have met
the target) - +0
- Familiar (you know the target
well) - -5
"""
tex = latex.rst_to_latex(md_list)
self.assertIn("\\begin{itemize}", tex)
def test_simple_table(self):
table_rst = """
===== ===== =======
A B A and B
===== ===== =======
False False False
True False False
False True False
True True True
===== ===== =======
"""
tex = latex.rst_to_latex(table_rst)
# Check begin/end environment is fixed
self.assertNotIn("longtable", tex)
self.assertIn("supertabular", tex)
# Check headers and footers are fixed
self.assertNotIn("endfoot", tex)
self.assertNotIn("endhead", tex)
self.assertNotIn("endfirsthead", tex)
def test_rst_all_spells(self):
for spell in spells.all_spells():
tex = latex.rst_to_latex(spell.__doc__)
self.assertNotIn("DUadmonition", tex,
f"spell {spell} is not valid reStructured text")
def test_rst_all_features(self):
for feature in features.all_features():
tex = latex.rst_to_latex(feature.__doc__)
self.assertNotIn("DUadmonition", tex,
f"feature {feature} is not valid reStructured text")