Test the Python command-line interface

src/python/code/gnoll/__main__.py

@@ -92,11 +92,16 @@ def main(EXPR, times, no_builtins, **kwargs):
             EXPR,
             builtins=not no_builtins,
             **kwargs)
-        if breakdown:
-            print(breakdown[0], '-->', result)
-        else:
-            print(result)
+        yield (breakdown[0], '-->', result) if breakdown else (result,)
+
+
+def main_with_args(args):
+    """Parse the commandline args and then run `main`
+    @param args - the arguments from the commandline (excluding the python3 call)
+    """
+    yield from main(**vars(parse_cmdline_args(args)))
 
 
 if __name__ == '__main__':
-    main(**vars(parse_cmdline_args(sys.argv[1:])))
+    for line in main_with_args(sys.argv[1:]):
+        print(*line)

tests/python/test_cli.py

@@ -0,0 +1,20 @@
+import gnoll.__main__
+
+
+m = lambda *x: list(gnoll.__main__.main_with_args(x))
+
+
+def test_cli():
+
+    [[r]] = m('1d4')
+    assert isinstance(r, int)
+
+    [[(die1, die2), a, r]] = m('2d4', '--breakdown')
+    assert all((isinstance(x, int) for x in [die1, die2, r]))
+    assert a == '-->'
+
+    executions = m('4d6kh3', '+', '1', '--breakdown', '--times', '6', '--no-builtins')
+    assert len(executions) == 6
+    for (die1, die2, die3, die4), a, r in executions:
+        assert all((isinstance(x, int) for x in [die1, die2, die3, die4, r]))
+        assert a == '-->'