1 files changed, 6 insertions, 0 deletions
diff --git a/subplot/jt.py b/subplot/jt.py
index 312a22a..6ce75b0 100644
--- a/subplot/jt.py
+++ b/subplot/jt.py
@@ -153,3 +153,9 @@ def file_contains(ctx, variable=None, pattern=None):
         data = f.read()
         logging.debug(f"file content: {data!r}")
         assert pattern in data
+
+
+def file_name_has_suffix(ctx, varname=None, suffix=None):
+    variables = ctx.get("variables", {})
+    filename = variables[varname]
+    assert filename.endswith(suffix)