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.

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