mirror of
https://github.com/Threnklyn/dungeon-sheets.git
synced 2026-05-19 12:33:27 +02:00
Mark Wolfman
3b8dbc0566
- Improved test coverage in ``make_sheets.py``. - Refactor file loading to make it more flexible. - Added a stub ``gm.py`` example.
83 lines
3.0 KiB
Python
83 lines
3.0 KiB
Python
import unittest
|
|
import os
|
|
|
|
from dungeonsheets import make_sheets, character
|
|
from dungeonsheets.classes import monk
|
|
|
|
|
|
EG_DIR = os.path.abspath(os.path.join(os.path.split(__file__)[0], "../examples/"))
|
|
CHARFILE = os.path.join(EG_DIR, "rogue1.py")
|
|
|
|
|
|
class PdfOutputTestCase(unittest.TestCase):
|
|
basename = "clara"
|
|
|
|
def tearDown(self):
|
|
temp_files = [f"{self.basename}.pdf"]
|
|
for f in temp_files:
|
|
if os.path.exists(f):
|
|
os.remove(f)
|
|
|
|
def test_file_created(self):
|
|
# Check that a file is created once the function is run
|
|
pdf_name = f"{self.basename}.pdf"
|
|
# self.assertFalse(os.path.exists(pdf_name), f'{pdf_name} already exists.')
|
|
char = character.Character(name="Clara")
|
|
char.saving_throw_proficiencies = ["strength"]
|
|
make_sheets.create_character_pdf_template(
|
|
character=char, basename=self.basename
|
|
)
|
|
self.assertTrue(os.path.exists(pdf_name), f"{pdf_name} not created.")
|
|
|
|
|
|
class TexCreatorTestCase(unittest.TestCase):
|
|
"""Test various helper functions for creating TeX from a character."""
|
|
|
|
def new_character(self):
|
|
char = character.Character(
|
|
classes=["Monk", "Druid", "Artificer"],
|
|
levels=[1, 1, 1],
|
|
subclasses=["way of the open hand", None, None],
|
|
magic_items=["cloak of protection"],
|
|
spells=["invisibility"],
|
|
wild_shapes=["crocodile"],
|
|
infusions=["boots of the winding path"]
|
|
)
|
|
return char
|
|
|
|
def test_create_subclasses_tex(self):
|
|
char = self.new_character()
|
|
tex = make_sheets.create_subclasses_tex(character=char)
|
|
self.assertIn(r"\section*{Subclasses}", tex)
|
|
self.assertIn(r"\subsection*{Way of the Open Hand}", tex)
|
|
|
|
def test_create_features_tex(self):
|
|
char = self.new_character()
|
|
tex = make_sheets.create_features_tex(character=char)
|
|
self.assertIn(r"\section*{Features}", tex)
|
|
self.assertIn(r"\subsection*{Martial Arts}", tex)
|
|
|
|
def test_create_magic_items_tex(self):
|
|
char = self.new_character()
|
|
tex = make_sheets.create_magic_items_tex(character=char)
|
|
self.assertIn(r"\section*{Magic Items}", tex)
|
|
self.assertIn(r"\subsection*{Cloak of Protection}", tex)
|
|
|
|
def test_create_spellbook_tex(self):
|
|
char = self.new_character()
|
|
tex = make_sheets.create_spellbook_tex(character=char)
|
|
self.assertIn(r"\section*{Spells}", tex)
|
|
self.assertIn(r"\section*{Invisibility}", tex)
|
|
|
|
def test_create_infusions_tex(self):
|
|
char = self.new_character()
|
|
tex = make_sheets.create_infusions_tex(character=char)
|
|
self.assertIn(r"\section*{Infusions}", tex)
|
|
self.assertIn(r"\subsection*{Boots of the Winding Path}", tex)
|
|
|
|
def test_create_druid_shapes_tex(self):
|
|
char = self.new_character()
|
|
tex = make_sheets.create_druid_shapes_tex(character=char)
|
|
self.assertIn(r"\section*{Known Beasts}", tex)
|
|
self.assertIn(r"\section*{Crocodile}", tex)
|