1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58
|
import unittest
import pweave
import pickle
import os
class FormatterTest(unittest.TestCase):
"""Test formatters"""
def setUp(self):
self.doc = pweave.Pweb("tests/formats/formatters_test.pmd")
self.doc.run()
#pickle.dump(self.doc.executed, open("tests/formats/formatters_test.pkl", "wb"))
with open("tests/formats/formatters_test.pkl", "rb") as f:
e = pickle.load(f)
self.doc.executed = e
self.out_base = "tests/formats/formatters_test.%s"
self.ref_base = "tests/formats/formatters_test_REF.%s"
def testFormatters(self):
formats = sorted(list(pweave.formatters.PwebFormats.formats.keys()))
for format in formats:
if "pandoc2latex" in format or "2html" in format: #No pandoc on travis
continue
self.doc.setformat(format)
self.doc.format()
self.out_file = self.out_base % format
self.ref_file = self.ref_base % format
self.doc.output = self.out_file
self.doc.write()
if "2html" in format:
pass
#Need to ignore same amount from beginning
#End is variable lenght, anyway tested with test_publish
#self.assertSameAsReference(1000) #Ignore changing footer
else:
self.assertSameAsReference()
try:
os.remove(self.out_file)
except FileNotFoundError:
pass
def contentOf(self, filename, end_ignore):
fh = open(filename)
content = fh.read()
fh.close()
if end_ignore > 0:
return(content[:-end_ignore])
return content
def assertSameAsReference(self, end_ignore = -1):
self.assertEqual(self.contentOf(self.out_file, end_ignore),
self.contentOf(self.ref_file, end_ignore))
if __name__ == '__main__':
unittest.main()
|