From e7a11bdb96ba259d6dd6302ac053a6e4c78aea07 Mon Sep 17 00:00:00 2001 From: Kevin Van Brunt Date: Tue, 9 Oct 2018 18:50:07 -0400 Subject: Added tests for token_resembles_flag --- tests/test_acargparse.py | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) (limited to 'tests') diff --git a/tests/test_acargparse.py b/tests/test_acargparse.py index be3e8b97..617afd4f 100644 --- a/tests/test_acargparse.py +++ b/tests/test_acargparse.py @@ -5,7 +5,7 @@ Copyright 2018 Eric Lin Released under MIT license, see LICENSE file """ import pytest -from cmd2.argparse_completer import ACArgumentParser +from cmd2.argparse_completer import ACArgumentParser, token_resembles_flag def test_acarg_narg_empty_tuple(): @@ -51,3 +51,18 @@ def test_acarg_narg_tuple_zero_base(): def test_acarg_narg_tuple_zero_to_one(): parser = ACArgumentParser(prog='test') parser.add_argument('tuple', nargs=(0, 1)) + + +def test_token_resembles_flag(): + parser = ACArgumentParser() + + # Not valid flags + assert not token_resembles_flag('', parser) + assert not token_resembles_flag('non-flag', parser) + assert not token_resembles_flag('-', parser) + assert not token_resembles_flag('--has space', parser) + assert not token_resembles_flag('-2', parser) + + # Valid flags + assert token_resembles_flag('-flag', parser) + assert token_resembles_flag('--flag', parser) -- cgit v1.2.1