diff options
-rw-r--r-- | lps_gen.py | 10 | ||||
-rw-r--r-- | tests/test_lps_gen.py | 14 |
2 files changed, 24 insertions, 0 deletions
@@ -10,6 +10,7 @@ import json import re import sys +import pkg_resources as pkgr import pytz from argparse import ArgumentParser @@ -92,6 +93,15 @@ def json_read(filename): object_pairs_hook=OrderedDict) +def template_read(name): + """Return template as `str`. + """ + p = 'lpschedule_generator' + r = 'data/{}'.format(name) + + return pkgr.resource_string(p, r).decode('utf-8') + + class LPiCal(object): """Used for producing iCal for LP schedule. """ diff --git a/tests/test_lps_gen.py b/tests/test_lps_gen.py index 8d57620..26c6132 100644 --- a/tests/test_lps_gen.py +++ b/tests/test_lps_gen.py @@ -100,6 +100,20 @@ class TestJSONUtils(object): os.chdir(self.old_cwd) +class TestTemplates(object): + """Petty tests for lp templates. + """ + + def test_read(self): + t = template_read('schedule.jinja2') + assert type(t) is str + assert len(t) > 0 + + t = template_read('speakers.jinja2') + assert type(t) is str + assert len(t) > 0 + + class TestLPiCal(object): """ Testing LPiCal class. |