diff --git a/tests/__init__.py b/tests/__init__.py index 056c322..d04989a 100644 --- a/tests/__init__.py +++ b/tests/__init__.py @@ -1,5 +1,20 @@ from unittest import TestCase +import os +from md_pdf.consts import TEMPLATES_DIR class BaseTestCase(TestCase): - pass + def deleteTemplates(self): + for template in [ + 'header.html', + 'footer.html', + 'cover.html', + 'toc-xml', + ]: + try: + os.remove(os.path.join(TEMPLATES_DIR, template)) + except OSError: + pass + + def tearDown(self): + self.deleteTemplates()