From b202aa341ff1427c668985754e5f87e76aab8591 Mon Sep 17 00:00:00 2001 From: rsiddharth Date: Sat, 13 Jul 2019 18:54:24 -0400 Subject: lps_gen.py: Update template_read. * lps_gen.py (template_read): Add template file extension. * tests/test_lps_gen.py (TestTemplates.test_read): Don't specify template file extension. --- lps_gen.py | 2 +- tests/test_lps_gen.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/lps_gen.py b/lps_gen.py index 49bc477..82d20df 100644 --- a/lps_gen.py +++ b/lps_gen.py @@ -97,7 +97,7 @@ def template_read(name): """Return template as `str`. """ p = 'lpschedule_generator' - r = 'data/{}'.format(name) + r = 'data/{}.jinja2'.format(name) t = None try: diff --git a/tests/test_lps_gen.py b/tests/test_lps_gen.py index c4ff1dd..81a3ab6 100644 --- a/tests/test_lps_gen.py +++ b/tests/test_lps_gen.py @@ -105,15 +105,15 @@ class TestTemplates(object): """ def test_read(self): - t = template_read('schedule.jinja2') + t = template_read('schedule') assert type(t) is str assert len(t) > 0 - t = template_read('speakers.jinja2') + t = template_read('speakers') assert type(t) is str assert len(t) > 0 - t = template_read('nonexistent.jinja2') + t = template_read('nonexistent') assert t is None -- cgit v1.2.3