mirror of
https://github.com/Threnklyn/dungeon-sheets.git
synced 2026-05-19 04:33:26 +02:00
matsavage
39 lines
1.3 KiB
Python
39 lines
1.3 KiB
Python
import unittest
|
|
import os
|
|
from pathlib import Path
|
|
|
|
from dungeonsheets import make_sheets, character
|
|
from dungeonsheets.fill_pdf_template import create_character_pdf_template, create_spells_pdf_template
|
|
|
|
|
|
EG_DIR = os.path.abspath(os.path.join(os.path.split(__file__)[0], "../examples/"))
|
|
CHARFILE = os.path.join(EG_DIR, "rogue1.py")
|
|
|
|
|
|
class CharacterFileTestCase(unittest.TestCase):
|
|
example_dir = Path(__file__).parent.parent / "examples"
|
|
|
|
def test_load_character_file(self):
|
|
charfile = CHARFILE
|
|
result = make_sheets.load_character_file(charfile)
|
|
self.assertEqual(result["strength"], 10)
|
|
|
|
|
|
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.")
|