Added tests for token_resembles_flag

1 files changed, 16 insertions, 1 deletions

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 <anselor@gmail.com>
 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)