1 files changed, 11 insertions, 2 deletions

diff --git a/subplot.py b/subplot.py
index f4549ee..57b0fa5 100644
--- a/subplot.py
+++ b/subplot.py
@@ -6,15 +6,21 @@ def create_file(ctx, filename=None):
     with open(filename, 'wb') as f:
         f.write(get_file(filename))
 
-def run_docgen(ctx, md=None, output=None):
+def try_docgen(ctx, md=None, output=None):
     docgen = binary('sp-docgen')
     runcmd(ctx, [docgen, md, '-o', output])
+
+def run_docgen(ctx, md=None, output=None):
+    try_docgen(ctx, md=md, output=output)
     exit_code_zero(ctx)
 
-def run_codegen(ctx, filename=None):
+def try_codegen(ctx, filename=None):
     codegen = binary('sp-codegen')
     tmpldir = os.path.join(srcdir, 'templates')
     runcmd(ctx, [codegen, filename, '-o', 'test.py', '--run', '--templates', tmpldir])
+
+def run_codegen(ctx, filename=None):
+    try_codegen(ctx, filename=filename)
     exit_code_zero(ctx)
 
 def run_meta(ctx, filename=None):
@@ -50,6 +56,9 @@ def exit_code_zero(ctx):
         print('context:', ctx.as_dict())
     assert_eq(ctx.get('exit'), 0)
 
+def exit_code_nonzero(ctx):
+    assert_ne(ctx.get('exit'), 0)
+
 def binary(basename):
     return os.path.join(srcdir, 'target', 'debug', basename)