Pynorms

src/wxflow/jinja.py

@@ -1,11 +1,11 @@
 import os
 import sys
+from functools import reduce
 from pathlib import Path
 from typing import Dict, List, Union
 
 import jinja2
 from markupsafe import Markup
-from functools import reduce
 
 from .timetools import (add_to_datetime, strftime, to_fv3time, to_isotime,
                         to_julian, to_timedelta, to_YMD, to_YMDH)

tests/test_jinja.py

@@ -31,9 +31,11 @@ def test_render_stream():
     assert j.render == f"Hello Jane! How are you? It is: {to_isotime(current_date)}"
 
     tmpl_dict = {"{{ name }}": "Jane", "{{ greeting }}": "How are you?", "{{ current_date | to_isotime }}": to_isotime(current_date)}
-    env = Jinja.get_set_env(jinja2.BaseLoader())
+    j = Jinja(j2tmpl, data, allow_missing=False)
+    loader = jinja2.BaseLoader()
+    env = j.get_set_env(loader)
     assert env.filters['replace_tmpl'](j2tmpl, tmpl_dict) == f"Hello Jane! How are you? It is: {to_isotime(current_date)}"
-                            
+
 
 def test_render_file(tmp_path, create_template):