From 3ea18ff01b688e692d80855f416e16e93e73a23a Mon Sep 17 00:00:00 2001 From: Nemo Date: Fri, 16 Jul 2021 16:57:09 +0530 Subject: [PATCH] [tests] Add tests for argument parser --- tests/test_cli.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) create mode 100644 tests/test_cli.py diff --git a/tests/test_cli.py b/tests/test_cli.py new file mode 100644 index 0000000..ac43b33 --- /dev/null +++ b/tests/test_cli.py @@ -0,0 +1,15 @@ +from pystitcher.skeleton import parse_args +import logging + +def test_default_args(): + args = parse_args(['tests/book-clean.md', 'o.pdf']) + assert args.loglevel == None + assert args.cleanup == True + +def test_loglevel(): + args = parse_args(['-v', 'tests/book-clean.md', 'o.pdf']) + assert args.loglevel == logging.INFO + +def test_cleanup(): + args = parse_args(['--cleanup', 'tests/book-clean.md', 'o.pdf']) + assert args.cleanup == True